goldendict-ng/resourceschemehandler.cc

#include "resourceschemehandler.hh"

ResourceSchemeHandler::ResourceSchemeHandler(ArticleNetworkAccessManager& articleNetMgr, QObject *parent):
    QWebEngineUrlSchemeHandler(parent),
    mManager(articleNetMgr){
}
void ResourceSchemeHandler::requestStarted(QWebEngineUrlRequestJob *requestJob)
{
    QUrl url = requestJob->requestUrl();

    QNetworkRequest request;
    request.setUrl(url);
    request.setAttribute(QNetworkRequest::CacheLoadControlAttribute, QNetworkRequest::PreferCache);
    QNetworkReply *reply = this->mManager.getArticleReply( request);
    connect(reply, &QNetworkReply::finished, requestJob, [=]() {
      if (reply->error() == QNetworkReply::ContentNotFoundError) {
        requestJob->fail(QWebEngineUrlRequestJob::UrlNotFound);
        return;
      }
      if (reply->error() != QNetworkReply::NoError) {
        qDebug() << "resource handler failed:" << reply->error() << ":" << reply->request().url();
        requestJob->fail(QWebEngineUrlRequestJob::RequestFailed);
        return;
      }
      QMimeType mineType = db.mimeTypeForUrl(url);
      QString contentType = mineType.name();
      // Reply segment
      requestJob->reply(contentType.toLatin1(), reply);
    });
    connect(requestJob, &QObject::destroyed, reply, &QObject::deleteLater);
}