#include "filterdomain.h"
#include <QVector>

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('.');
}

FilterDomain::FilterDomain(MatchType type, QObject *parent)
    : QObject(parent)
{
    setType(type);
}

void FilterDomain::setType(MatchType type)
{
    m_type = type;
}

void FilterDomain::addDomain(const QString &domain)
{
    if(!domain.isEmpty())
        m_domains.append(domain);
}

bool FilterDomain::hasMatch(const QString &host) const
{
    // match all domains -> this rule applies to all domains
    if(m_type == WhitelistAll)
        return true;

    // match no domains -> this rule applies to no domains
    if(m_type == BlacklistAll)
        return false;

    // is this a whitelist or blacklist domain rule?
    // should it apply to added domains or not
    const bool whitelist = (m_type == Whitelist);

    for(const auto &domain : qAsConst(m_domains)) {
        if(isMatchingDomain(host, domain))
            return whitelist;
    }

    return !whitelist;
}