mirror of
https://github.com/xiaoyifang/goldendict-ng.git
synced 2024-12-18 03:14:06 +00:00
102 lines
3.4 KiB
C++
102 lines
3.4 KiB
C++
#include "iframeschemehandler.h"
|
|
|
|
#include <QTextCodec>
|
|
|
|
IframeSchemeHandler::IframeSchemeHandler(QObject * parent):QWebEngineUrlSchemeHandler(parent){
|
|
|
|
}
|
|
void IframeSchemeHandler::requestStarted(QWebEngineUrlRequestJob *requestJob)
|
|
{
|
|
QUrl url = requestJob->requestUrl();
|
|
|
|
// website dictionary iframe url
|
|
url = QUrl( Utils::Url::queryItemValue( url, "url" ) );
|
|
QNetworkRequest request;
|
|
request.setUrl( url );
|
|
request.setAttribute(QNetworkRequest::RedirectPolicyAttribute,QNetworkRequest::RedirectPolicy::NoLessSafeRedirectPolicy);
|
|
|
|
QNetworkReply * reply = mgr.get( request );
|
|
|
|
auto finishAction = [ = ]() -> void
|
|
{
|
|
// Handle reply data
|
|
|
|
QByteArray replyData = reply->readAll();
|
|
QString articleString;
|
|
|
|
QTextCodec * codec = QTextCodec::codecForHtml( replyData, QTextCodec::codecForName( "UTF-8" ) );
|
|
articleString = codec->toUnicode( replyData );
|
|
|
|
// Change links from relative to absolute
|
|
|
|
QString root = reply->url().scheme() + "://" + reply->url().host();
|
|
QString base = root + reply->url().path();
|
|
while( !base.isEmpty() && !base.endsWith( "/" ) )
|
|
base.chop( 1 );
|
|
|
|
QRegularExpression tags( "<\\s*(a|link|img|script)\\s+[^>]*(src|href)\\s*=\\s*['\"][^>]+>",
|
|
QRegularExpression::CaseInsensitiveOption );
|
|
QRegularExpression links( "\\b(src|href)\\s*=\\s*(['\"])([^'\"]+['\"])",
|
|
QRegularExpression::CaseInsensitiveOption );
|
|
int pos = 0;
|
|
QString articleNewString;
|
|
QRegularExpressionMatchIterator it = tags.globalMatch( articleString );
|
|
while( it.hasNext() )
|
|
{
|
|
QRegularExpressionMatch match = it.next();
|
|
articleNewString += articleString.mid( pos, match.capturedStart() - pos );
|
|
pos = match.capturedEnd();
|
|
|
|
QString tag = match.captured();
|
|
|
|
QRegularExpressionMatch match_links = links.match( tag );
|
|
if( !match_links.hasMatch() )
|
|
{
|
|
articleNewString += tag;
|
|
continue;
|
|
}
|
|
|
|
QString url = match_links.captured( 3 );
|
|
|
|
if( url.indexOf( ":/" ) >= 0 || url.indexOf( "data:" ) >= 0 || url.indexOf( "mailto:" ) >= 0 ||
|
|
url.startsWith( "#" ) || url.startsWith( "javascript:" ) )
|
|
{
|
|
// External link, anchor or base64-encoded data
|
|
articleNewString += tag;
|
|
continue;
|
|
}
|
|
|
|
QString newUrl = match_links.captured( 1 ) + "=" + match_links.captured( 2 );
|
|
if( url.startsWith( "//" ) )
|
|
newUrl += reply->url().scheme() + ":";
|
|
else if( url.startsWith( "/" ) )
|
|
newUrl += root;
|
|
else
|
|
newUrl += base;
|
|
newUrl += match_links.captured( 3 );
|
|
|
|
tag.replace( match_links.capturedStart(), match_links.capturedLength(), newUrl );
|
|
articleNewString += tag;
|
|
}
|
|
if( pos )
|
|
{
|
|
articleNewString += articleString.mid( pos );
|
|
articleString = articleNewString;
|
|
articleNewString.clear();
|
|
}
|
|
|
|
QBuffer * buffer = new QBuffer(requestJob);
|
|
buffer->setData(codec->fromUnicode(articleString));
|
|
|
|
QByteArray contentType="text/html;charset=UTF-8";
|
|
auto contentTypeHeader=reply->header(QNetworkRequest::ContentTypeHeader);
|
|
if(contentTypeHeader.isValid())
|
|
contentType= contentTypeHeader.toByteArray();
|
|
|
|
requestJob->reply(contentType , buffer );
|
|
};
|
|
connect( reply, &QNetworkReply::finished, requestJob, finishAction );
|
|
|
|
connect( requestJob, &QObject::destroyed, reply, &QObject::deleteLater );
|
|
}
|