aboutsummaryrefslogtreecommitdiff
path: root/lib/urlfilter/filterleaf.h
blob: 64f465d3b6d2271dd63ee3fe137ff73251f10232 (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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
/*
 * 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
 */

#ifndef SMOLBOTE_FILTERLEAF_H
#define SMOLBOTE_FILTERLEAF_H

#include <QHash>
#include <QObject>
#include <QString>
#include <QWebEngineUrlRequestInfo>
#include <optional>
#include <utility>
#include <QVariant>

class FilterLeaf
{
public:
    enum Action {
        NotMatched,
        Allow,
        Block,
        Redirect
    };

    enum UrlMatchType {
        InvalidMatch,
        RegularExpressionMatch,
        StringContains,
        StringStartsWith,
        StringEndsWith,
        StringEquals,
        DomainMatch
    };

    virtual ~FilterLeaf() = default;

    virtual bool match(const QUrl &requestUrl) const = 0;
    virtual std::pair<Action, QVariant> action() const = 0;

    const QString request() const;
    std::optional<bool> option(QWebEngineUrlRequestInfo::ResourceType opt) const;

protected:
    // rule matching
    UrlMatchType matchType = InvalidMatch;
    QHash<QWebEngineUrlRequestInfo::ResourceType, bool> resourceTypeOptions;
    QString m_request;

    // rule action
    bool m_isBlocking;
};

Q_DECLARE_METATYPE(FilterLeaf::Action)

#endif // SMOLBOTE_FILTERLEAF_H