/*
 * This file is part of smolbote. It's copyrighted by the contributors recorded
 * in the version control history of the file, available from its original
 * location: https://neueland.iserlohn-fortress.net/gitea/aqua/smolbote
 *
 * SPDX-License-Identifier: GPL-3.0
 */

#include "filterrule.h"

inline bool isMatchingDomain(const QString &domain, const QString &filter)
{
    // domain and filter are the same
    if(domain == filter) {
        return true;
    }

    // domain can't be matched by filter if it doesn't end with filter
    // ex. example2.com isn't matched by example.com
    if(!domain.endsWith(filter)) {
        return false;
    }

    // match with subdomains
    // ex. subdomain.example.com is matched by example.com
    int index = domain.indexOf(filter);

    // match if (domain ends with filter) && (filter has been found) and (character before filter is '.')
    return index > 0 && domain[index - 1] == QLatin1Char('.');
}

bool FilterRule::isEnabled() const
{
    return m_isEnabled;
}

bool FilterRule::isBlocking() const
{
    return m_isBlocking;
}

bool FilterRule::matchesDomain(uint domainHash) const
{
    // no domains have been allowed or blocked -> allow on all domains
    if(allowedDomains_hashes.isEmpty() && blockedDomains_hashes.isEmpty()) {
        return true;
    }

    // blockedDomains prevents the rules from being matched on those domains
    if(blockedDomains_hashes.contains(domainHash)) {
        return false;
    }

    // allowedDomains means the rule should only be matched on those domains
    return allowedDomains_hashes.contains(domainHash);
}

bool FilterRule::matchesType(QWebEngineUrlRequestInfo::ResourceType type) const
{
    // no options have been specified -> match all resource types
    if(m_resourceTypeOptions.isEmpty())
        return true;

    // this resource type has not been specified -> reject it
    if(!m_resourceTypeOptions.contains(type))
        return false;

    // resource type has been specified; true to match, false to exception
    return m_resourceTypeOptions.value(type);
}

bool FilterRule::matchesUrl(const QUrl &url) const
{
    switch(urlMatchType) {
    case InvalidMatch:
        return false;

    case RegularExpressionMatch:
        return regexp.match(url.toString()).hasMatch();

    case StringContains:
        return url.toString().contains(match);

    case StringStartsWith:
        return url.toString().startsWith(match);

    case StringEndsWith:
        return url.toString().endsWith(match);

    case StringEquals:
        return url.toString() == match;

    case DomainMatch:
        return isMatchingDomain(url.host(), match);
    }
}