File indexing completed on 2024-06-23 04:53:41

0001 /* ============================================================
0002 * Falkon - Qt web browser
0003 * Copyright (C) 2018 David Rosca <nowrep@gmail.com>
0004 *
0005 * This program is free software: you can redistribute it and/or modify
0006 * it under the terms of the GNU General Public License as published by
0007 * the Free Software Foundation, either version 3 of the License, or
0008 * (at your option) any later version.
0009 *
0010 * This program is distributed in the hope that it will be useful,
0011 * but WITHOUT ANY WARRANTY; without even the implied warranty of
0012 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
0013 * GNU General Public License for more details.
0014 *
0015 * You should have received a copy of the GNU General Public License
0016 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
0017 * ============================================================ */
0018 #pragma once
0019 
0020 #include <QWebEngineUrlSchemeHandler>
0021 
0022 #include "qzcommon.h"
0023 
0024 class FALKON_EXPORT ExtensionSchemeHandler : public QObject
0025 {
0026 public:
0027     explicit ExtensionSchemeHandler(QObject *parent = nullptr) : QObject(parent) { }
0028 
0029     virtual void requestStarted(QWebEngineUrlRequestJob *job) = 0;
0030 
0031 protected:
0032     void setReply(QWebEngineUrlRequestJob *job, const QByteArray &contentType, const QByteArray &content);
0033 };
0034 
0035 class FALKON_EXPORT ExtensionSchemeManager : public QWebEngineUrlSchemeHandler
0036 {
0037 public:
0038     explicit ExtensionSchemeManager(QObject *parent = nullptr);
0039 
0040     void requestStarted(QWebEngineUrlRequestJob *job) override;
0041 
0042     void registerHandler(const QString &name, ExtensionSchemeHandler *handler);
0043     void unregisterHandler(ExtensionSchemeHandler *handler);
0044 
0045 private:
0046     QHash<QString, ExtensionSchemeHandler*> m_handlers;
0047 };