From c74367d82c1c7bec393548d2e5014c794333822f Mon Sep 17 00:00:00 2001
From: Aqua-sama <aqua@iserlohn-fortress.net>
Date: Tue, 16 Oct 2018 17:25:40 +0200
Subject: urlfilter: Add FilterTree class

FilterTree is a class that holds filter rules, sorted by the domain they
are to be applied on. The rules are to follow FilterLeaf as interface.
- Add a hostlist rule format to FilterTree.
- Add a test for hostlist format.
---
 lib/urlfilter/filtertree.h | 46 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 lib/urlfilter/filtertree.h

(limited to 'lib/urlfilter/filtertree.h')

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
-- 
cgit v1.2.1