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