Improved filter rules

1 files changed, 89 insertions, 0 deletions

diff --git a/src/blocker/filterrule.h b/src/blocker/filterrule.h
new file mode 100644
index 0000000..44340fc
--- /dev/null
+++ b/src/blocker/filterrule.h
@@ -0,0 +1,89 @@
+/*******************************************************************************
+ **
+ ** smolbote: yet another qute browser
+ ** Copyright (C) 2017  Xian Nox
+ **
+ ** This program is free software: you can redistribute it and/or modify
+ ** it under the terms of the GNU General Public License as published by
+ ** the Free Software Foundation, either version 3 of the License, or
+ ** (at your option) any later version.
+ **
+ ** This program is distributed in the hope that it will be useful,
+ ** but WITHOUT ANY WARRANTY; without even the implied warranty of
+ ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ ** GNU General Public License for more details.
+ **
+ ** You should have received a copy of the GNU General Public License
+ ** along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ **
+ ******************************************************************************/
+
+#ifndef ADBLOCKRULE_H
+#define ADBLOCKRULE_H
+
+#include <QObject>
+#include <QUrl>
+#include "regexp.h"
+#include <QWebEngineUrlRequestInfo>
+
+class FilterRule : public QObject
+{
+    Q_OBJECT
+public:
+
+    enum ResourceType {
+        NoType = 0,
+        MainFrame = 1,
+        SubFrame = 2,
+        Stylesheet = 4,
+        Script = 8,
+        Image = 16,
+        FontResource = 32,
+        SubResource = 64,
+        Object = 128,
+        Media = 256,
+        Worker = 512,
+        SharedWorker = 1024,
+        Prefetch = 2048,
+        Favicon = 4096,
+        Xhr = 8192,
+        Ping = 16384,
+        ServiceWorker = 32768,
+        CspReport = 65536,
+        PluginResource = 131072,
+        Unknown = 262144
+    };
+    Q_DECLARE_FLAGS(Resources, ResourceType)
+
+    struct ResourceRules {
+        Resources allowed;
+        Resources blocked;
+    };
+
+    explicit FilterRule(const QString firstPartyUrl, const QString requestUrl, ResourceRules rules, bool shouldBlock, QObject *parent = 0);
+
+    bool hasMatch(const QWebEngineUrlRequestInfo &info);
+    bool shouldBlock(const QWebEngineUrlRequestInfo &info);
+
+    bool isValid();
+
+    QString filter() const;
+
+signals:
+
+public slots:
+
+private:
+    bool testFlag(ResourceType flag);
+
+    RegExp m_firstPartyUrl;
+    RegExp m_requestUrl;
+
+    ResourceRules m_rules;
+
+    bool m_valid = false;
+    bool m_shouldBlock;
+
+};
+
+#endif // ADBLOCKRULE_H