diff options
Diffstat (limited to 'lib/urlfilter/filtertree.h')
-rw-r--r-- | lib/urlfilter/filtertree.h | 46 |
1 files changed, 46 insertions, 0 deletions
diff --git a/lib/urlfilter/filtertree.h b/lib/urlfilter/filtertree.h new file mode 100644 index 0000000..8cecf50 --- /dev/null +++ b/lib/urlfilter/filtertree.h @@ -0,0 +1,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 |