aboutsummaryrefslogtreecommitdiff
path: root/lib/urlfilter/filtertree.h
blob: 8cecf5088ddd7959490cb63f3b932da78eb2b0b2 (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
/*
 * 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 <QObject>
#include <QVector>
#include <vector>
#include <QIODevice>
#include "filterleaf.h"

/** FilterTree: B+ tree of filter rules
 *  The root of the tree contains branches that represent domains, on which their rules are to be applied.
 *  Each branch contains leaves - rules
 */
class FilterTree : public QObject
{
    Q_OBJECT

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

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

private:
    struct Branch {
        ~Branch() { qDeleteAll(leaves); }

        // TODO: replace domain type with domain-matching class
        std::string domain;
        std::vector<FilterLeaf *> leaves;
    };

    std::vector<Branch> m_branches;
};

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

#endif // SMOLBOTE_FILTERTREE_H