aboutsummaryrefslogtreecommitdiff
path: root/lib/urlfilter/adblock/adblocklist.h
blob: ee41e11a5d55cab7a6b66f35fc89c35dfafe292d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
/*
 * 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 "urlfilter.h"
#include "matcher.h"
#include <QHash>
#include <QWebEngineUrlRequestInfo>

class QIODevice;
class AdBlockList : public UrlFilter
{
public:
    // TODO: check if all keys are listed
    const QStringList keys = { "Version", "Title", "Last modified", "Expires", "Homepage", "Licence", "Redirect" };

    AdBlockList(QIODevice *device);
    ~AdBlockList();

    QString metadata(const QString &key) const override;
    int ruleCount() const;
    std::pair<MatchResult, QString> match(const QUrl &firstParty, const QUrl &requestUrl, QWebEngineUrlRequestInfo::ResourceType type) const override;

protected:
    void parseLine(const QString &line);

private:
    QHash<QString, QString> m_metadata;

    struct Rule {
        UrlFilter::MatchResult action = UrlFilter::NotMatched;
        Matcher *matcher;
        QStringList enabledOn, disabledOn;
        QHash<QWebEngineUrlRequestInfo::ResourceType, bool> options;
    };

    std::vector<Rule> rules;
};