1 files changed, 107 insertions, 0 deletions
diff --git a/src/adblock/adblockmanager.cpp b/src/adblock/adblockmanager.cpp
new file mode 100644
index 00000000..70960b44
--- /dev/null
+++ b/src/adblock/adblockmanager.cpp
@@ -0,0 +1,107 @@
+/* ============================================================
+*
+* This file is a part of the rekonq project
+*
+* Copyright (C) 2009 by Andrea Diamantini <adjam7 at gmail dot com>
+*
+*
+* 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 2 of
+* the License or (at your option) version 3 or any later version
+* accepted by the membership of KDE e.V. (or its successor approved
+* by the membership of KDE e.V.), which shall act as a proxy 
+* defined in Section 14 of version 3 of the license.
+* 
+* 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/>.
+*
+* ============================================================ */
+
+
+#include "adblockmanager.h"
+#include "adblockmanager.moc"
+
+
+// Local Includes
+#include "adblockrule.h"
+
+// KDE Includes
+#include <KSharedConfig>
+#include <KConfigGroup>
+#include <KDebug>
+
+// Qt Includes
+#include <QUrl>
+
+
+AdBlockManager::AdBlockManager(QObject *parent)
+    : QObject(parent)
+    , _isAdblockEnabled(false)
+{
+    loadSettings();
+}
+
+
+AdBlockManager::~AdBlockManager()
+{
+}
+
+
+void AdBlockManager::loadSettings()
+{
+    _blockList.clear();
+    
+    KSharedConfig::Ptr config = KSharedConfig::openConfig("webkitrc", KConfig::NoGlobals);
+    KConfigGroup cg(config, "adblock");
+    
+    _isAdblockEnabled = cg.readEntry("Enabled", false);
+
+    int num = cg.readEntry("Count", 0);
+    for (int i = 0; i < num; ++i)
+    {
+        QString key = "Filter-" + QString::number(i);
+        QString filter = cg.readEntry( key, QString() );
+        if(!filter.isEmpty())
+        {
+            if (filter.startsWith(QLatin1String("@@")))
+                _whiteList << filter;
+            else
+                _blockList << filter;
+        }
+    }
+}
+
+
+bool AdBlockManager::isUrlAllowed(const QUrl &url)
+{
+    kDebug() << "matching rule...";
+    if (!_isAdblockEnabled)
+        return true;
+
+    QString urlString = QString::fromUtf8(url.toEncoded());
+    
+    foreach(const QString &str, _whiteList)
+    {
+        AdBlockRule rule(str);
+        kDebug() << "checking white list rule...";
+        if(rule.match(urlString))
+            return true;
+    }
+    
+    foreach(const QString &str, _blockList)
+    {
+        AdBlockRule rule(str);
+        kDebug() << "checking block list rule...";
+        if(rule.match(urlString))
+            return false;
+    }
+    
+    kDebug() << "done";
+    return true;
+}