aboutsummaryrefslogtreecommitdiff
path: root/lib/web/urlfilter/filterrule.h
blob: 5b9a6cfd99ec1b3051ab4f3453440b4daedcea5e (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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
/*
 * 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/smolbote.hg
 *
 * SPDX-License-Identifier: GPL-3.0
 */

#ifndef SMOLBOTE_FILTERRULE_H
#define SMOLBOTE_FILTERRULE_H

#include <QRegularExpression>
#include <QStringList>
#include <QStringMatcher>
#include <QUrl>
#include <QWebEngineUrlRequestInfo>
#include <memory>

class FilterRule
{
public:
    enum UrlMatchType {
        InvalidMatch,
        RegularExpressionMatch,
        StringContains,
        StringStartsWith,
        StringEndsWith,
        StringEquals,
        DomainMatch
    };

    FilterRule() = default;

    bool isEnabled() const;
    bool isBlocking() const;

    /**
     * @brief matchesDomain
     * @param domain
     * @return
     */
    bool matchesDomain(const QString &domain) const;

    /**
     * @brief matchesType
     * @param type
     * @return true if type matches, false otherwise
     */
    bool matchesType(QWebEngineUrlRequestInfo::ResourceType type) const;

    /**
     * @brief matchesUrl
     * @param url
     * @return
     */
    bool matchesUrl(const QUrl &url) const;

    QString toString() const;

protected:
    bool m_isEnabled = false;
    bool m_isBlocking = true;

    QString originalFilter;

    UrlMatchType urlMatchType = InvalidMatch;
    QHash<QWebEngineUrlRequestInfo::ResourceType, bool> m_resourceTypeOptions;
    QStringList allowedDomains, blockedDomains;

    QString match;
    QRegularExpression regexp;

};

#endif // SMOLBOTE_FILTERRULE_H