File indexing completed on 2024-05-12 16:23:45

 // SPDX-FileCopyrightText: 2020 Jonah Brüchert <jbb@kaidan.im>
 //
 // SPDX-License-Identifier: GPL-2.0-or-later
 
 #include "adblockfilterlistsmanager.h"
 
 #include <QDir>
 #include <QFile>
 #include <QNetworkReply>
 #include <QStandardPaths>
 
 #include "angelfishsettings.h"
 
 void copyStream(QIODevice &input, QIODevice &output)
 {
     constexpr auto BUFFER_SIZE = 1024;
 
     QByteArray buffer;
     buffer.reserve(BUFFER_SIZE);
 
     while (true) {
         int64_t read = input.read(buffer.data(), BUFFER_SIZE);
 
         if (read > 0) {
             output.write(buffer.data(), read);
         } else {
             break;
         }
     }
 }
 
 const QString filterListIdFromUrl(const QString &url) {
     QCryptographicHash fileNameHash(QCryptographicHash::Sha256);
     fileNameHash.addData(url.toUtf8());
     return QString::fromUtf8(fileNameHash.result().toHex());
 }
 
 AdblockFilterListsManager::AdblockFilterListsManager(QObject *parent)
     : QObject(parent)
     , m_filterLists(loadFromConfig())
 {
     connect(&m_networkManager, &QNetworkAccessManager::finished, this, &AdblockFilterListsManager::handleListFetched);
     m_networkManager.setRedirectPolicy(QNetworkRequest::SameOriginRedirectPolicy);
 }
 
 void AdblockFilterListsManager::refreshLists()
 {
     // Delete old lists, in case the names change.
     // Otherwise we might not be overwriting all of them.
     const QDir dir(filterListPath());
     const auto entries = dir.entryList();
     for (const auto &entry : entries) {
         QFile::remove(dir.path() + QDir::separator() + entry);
     }
 
     for (const auto &list : std::as_const(m_filterLists)) {
         m_runningRequests++;
         m_networkManager.get(QNetworkRequest(list.url));
     }
 }
 
 QString AdblockFilterListsManager::filterListPath()
 {
     static const auto path = []() -> QString {
         QString path = QStandardPaths::writableLocation(QStandardPaths::AppDataLocation) + QStringLiteral("/filterlists/");
         QDir(path).mkpath(QStringLiteral("."));
         return path;
     }();
     return path;
 }
 
 void AdblockFilterListsManager::handleListFetched(QNetworkReply *reply)
 {
     Q_ASSERT(reply);
 
     m_runningRequests--;
 
     if (m_runningRequests < 1) {
         Q_EMIT refreshFinished();
     }
 
     const auto id = filterListIdFromUrl(reply->url().toString());
 
     QFile file(filterListPath() + id);
     if (!file.open(QIODevice::WriteOnly)) {
         qDebug() << "Failed to open" << file.fileName() << "for writing."
                  << "Filter list not updated";
         return;
     }
 
     copyStream(*reply, file);
 }
 
 QVector<AdblockFilterListsManager::FilterList> AdblockFilterListsManager::loadFromConfig()
 {
     const auto &filterNames = AngelfishSettings::adblockFilterNames();
     const auto &filterUrls = AngelfishSettings::adblockFilterUrls();
 
     auto namesIt = filterNames.begin();
     auto urlsIt = filterUrls.begin();
 
     QVector<AdblockFilterListsManager::FilterList> out;
 
     // Otherwise list is corrupted, but we will still not crash in release mode
     Q_ASSERT(filterNames.size() == filterUrls.size());
 
     while (namesIt != filterNames.end() && urlsIt != filterUrls.end()) {
         out.push_back(FilterList{*namesIt, *urlsIt});
 
         namesIt++;
         urlsIt++;
     }
 
     return out;
 }
 
 void AdblockFilterListsManager::writeToConfig(const QVector<AdblockFilterListsManager::FilterList> &filters)
 {
     QStringList filterNames;
     QList<QUrl> filterUrls;
 
     for (const auto &filterList : filters) {
         filterNames.push_back(filterList.name);
         filterUrls.push_back(filterList.url);
     }
 
     AngelfishSettings::setAdblockFilterNames(filterNames);
     AngelfishSettings::setAdblockFilterUrls(filterUrls);
 }
 
 const QVector<AdblockFilterListsManager::FilterList> &AdblockFilterListsManager::filterLists() const
 {
     return m_filterLists;
 }
 
 void AdblockFilterListsManager::addFilterList(const QString &name, const QUrl &url)
 {
     m_filterLists.push_back(FilterList{name, url});
     writeToConfig(m_filterLists);
 }
 
 void AdblockFilterListsManager::removeFilterList(const int index)
 {
     m_filterLists.removeAt(index);
     writeToConfig(m_filterLists);
 }
 
 #include "moc_adblockfilterlistsmanager.cpp"