#include "webview.qt.impl.hpp"

#include <QFile>

#include <QBuffer>

#include <QWebEngineUrlRequestJob>

namespace saucer

{

const std::string webview::impl::ready_script = "window.saucer.on_message('dom_loaded')";

const std::string webview::impl::inject_script = []()

{

QFile web_channel_api(":/qtwebchannel/qwebchannel.js");

if (!web_channel_api.open(QIODevice::ReadOnly))

{

throw std::runtime_error("Failed to open required qwebchannel.js");

}

auto content = web_channel_api.readAll().toStdString();

web_channel_api.close();

return content +

R"js(

)js";

}();

webview::impl::web_class::web_class(webview *parent) : QObject(parent->m_impl->web_view), m_parent(parent) {}

void webview::impl::web_class::on_message(const QString &message)

{

m_parent->on_message(message.toStdString());

}

webview::impl::url_scheme_handler::url_scheme_handler(webview *parent)

: QWebEngineUrlSchemeHandler(parent->m_impl->web_view), m_parent(parent)

{

}

void webview::impl::url_scheme_handler::requestStarted(QWebEngineUrlRequestJob *request)

{

auto url = request->requestUrl().toString(QUrl::RemoveQuery).toStdString();

if (!url.starts_with(scheme_prefix))

{

request->fail(QWebEngineUrlRequestJob::UrlInvalid);

return;

}

url = url.substr(scheme_prefix.size());

if (!m_parent->m_embedded_files.contains(url))

{

request->fail(QWebEngineUrlRequestJob::UrlNotFound);

return;

}

const auto &file = m_parent->m_embedded_files.at(url);

auto *buffer = new QBuffer;

buffer->open(QIODevice::WriteOnly);

auto size = file.content.size();

const auto *content = file.content.data();

buffer->write(reinterpret_cast<const char *>(content), static_cast<std::int64_t>(size));

buffer->close();

connect(request, &QObject::destroyed, buffer, &QObject::deleteLater);

request->reply(QString::fromStdString(file.mime).toUtf8(), buffer);

}

template <>

void webview::impl::setup<web_event::load_finished>(webview *self)

{

if (load_finished)

{

return;

}

auto handler = [self](bool success)

{

if (!success)

{

return;

}

self->m_events.at<web_event::load_finished>().fire();

};

load_finished = web_view->connect(web_view, &QWebEngineView::loadFinished, handler);

}

template <>

void webview::impl::setup<web_event::url_changed>(webview *self)

{

if (url_changed)

{

return;

}

auto handler = [self](const QUrl &url)

{

self->m_events.at<web_event::url_changed>().fire(url.toString().toStdString());

};

url_changed = web_view->connect(web_view, &QWebEngineView::urlChanged, handler);

}

template <>

void webview::impl::setup<web_event::load_started>(webview *)

{

}

template <>

void webview::impl::setup<web_event::dom_ready>(webview *)

{

}

} // namespace saucer