aboutsummaryrefslogtreecommitdiff
path: root/lib/urlfilter/filtertree.h
blob: f453a3d6e1aa61524645f0cdca36ab55b30f9260 (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
/*
 * 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_FILTERTREE_H
#define SMOLBOTE_FILTERTREE_H

#include "domain.h"
#include "filterleaf.h"
#include <QIODevice>
#include <QObject>
#include <QVector>
#include <vector>
#include <QMutex>

/** FilterTree: B+ tree of filter rules
 * The tree contains branches that represent domains
 * Each domain-branch contains leaves (rules) that are to be applied to it.
 * Rules may be applied to multiple branches.
 */
class FilterTree : public QObject
{
    Q_OBJECT

public:
    ~FilterTree();

    const QStringList branches() const;
    QVector<const FilterLeaf *> match(const QUrl &domain, const QUrl &requestUrl) const;

    bool addRule(FilterLeaf *rule, const QString &domain);

private:
    struct Branch {
        explicit Branch(const QString &host)
            : domain(host)
        {
        }
        explicit Branch(Branch &&other)
            : domain(std::move(other.domain))
            , leaves(std::move(other.leaves))
        {
        }

        Domain domain;
        std::vector<FilterLeaf *> leaves;
    };

    Branch& branch(const QString &domain);

    QMutex branchLock;
    std::vector<Branch> m_branches;
};

bool loadHostlist(QIODevice &from, FilterTree *tree);

#endif // SMOLBOTE_FILTERTREE_H