aboutsummaryrefslogtreecommitdiff
path: root/src/webengine/urlinterceptor.h
blob: ff47ffe1235ca79320d1682d0f2cc72d80e89181 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
/*
 * 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/smolbote.hg
 *
 * SPDX-License-Identifier: GPL-3.0
 */

#ifndef SMOLBOTE_URLREQUESTINTERCEPTOR_H
#define SMOLBOTE_URLREQUESTINTERCEPTOR_H

#include "web/urlfilter/filterrule.h"
#include <QByteArray>
#include <QVector>
#include <QWebEngineUrlRequestInterceptor>
#include <memory>

typedef std::pair<std::string, std::string> Header;

class Configuration;
class UrlRequestInterceptor : public QWebEngineUrlRequestInterceptor
{
    Q_OBJECT
public:
    struct HostRule {
        bool isBlocking;
    };

    explicit UrlRequestInterceptor(const std::unique_ptr<Configuration> &config, QObject *parent = nullptr);
    ~UrlRequestInterceptor() = default;

    void interceptRequest(QWebEngineUrlRequestInfo &info) override;

private:
    QHash<QString, HostRule> rules;
    std::vector<FilterRule> filters;
    std::vector<Header> m_headers;
};

QHash<QString, UrlRequestInterceptor::HostRule> parse(const QString &filename);
inline std::vector<FilterRule> parseAdBlockList(const QString &filename);

#endif // SMOLBOTE_URLREQUESTINTERCEPTOR_H