aboutsummaryrefslogtreecommitdiff
path: root/lib/urlfilter/filtertree.h
blob: 004cf5a5df7b73678b6bba14283b576c038da325 (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
/*
 * 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"
#include "domain.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 {
        explicit Branch(const QString &host) : domain(host) {}
        explicit Branch(Branch &&other)
            : domain(std::move(other.domain))
            , leaves(std::move(other.leaves))
            {}
        ~Branch() { qDeleteAll(leaves); }

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

    std::vector<Branch> m_branches;
};

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

#endif // SMOLBOTE_FILTERTREE_H