2022-05-13 16:00:23 +00:00
# include "iframeschemehandler.h"
# include <QTextCodec>
IframeSchemeHandler : : IframeSchemeHandler ( QObject * parent ) : QWebEngineUrlSchemeHandler ( parent ) {
}
void IframeSchemeHandler : : requestStarted ( QWebEngineUrlRequestJob * requestJob )
{
QUrl url = requestJob - > requestUrl ( ) ;
// website dictionary iframe url
2022-09-14 12:01:57 +00:00
url = QUrl ( Utils : : Url : : queryItemValue ( url , " url " ) ) ;
2022-05-13 16:00:23 +00:00
QNetworkRequest request ;
request . setUrl ( url ) ;
2022-05-16 14:10:12 +00:00
request . setAttribute ( QNetworkRequest : : RedirectPolicyAttribute , QNetworkRequest : : RedirectPolicy : : NoLessSafeRedirectPolicy ) ;
2022-05-13 16:00:23 +00:00
QNetworkReply * reply = mgr . get ( request ) ;
2022-05-16 14:10:12 +00:00
2023-03-31 06:05:17 +00:00
auto finishAction = [ = ] ( ) {
2022-05-13 16:00:23 +00:00
2023-03-31 06:05:17 +00:00
QByteArray contentType = " text/html " ;
QString codecName ;
const auto ctHeader = reply - > header ( QNetworkRequest : : ContentTypeHeader ) ;
if ( ctHeader . isValid ( ) ) {
contentType = ctHeader . toByteArray ( ) ;
const auto ct = ctHeader . toString ( ) ;
const auto index = ct . indexOf ( " charset= " ) ;
if ( index > - 1 ) {
codecName = ct . mid ( index + 8 ) ;
}
}
2022-07-09 02:37:40 +00:00
QBuffer * buffer = new QBuffer ( requestJob ) ;
2022-09-14 12:01:01 +00:00
2022-05-13 16:00:23 +00:00
QByteArray replyData = reply - > readAll ( ) ;
QString articleString ;
2022-12-18 13:30:35 +00:00
QTextCodec * codec = QTextCodec : : codecForUtfText ( replyData , QTextCodec : : codecForName ( codecName . toUtf8 ( ) ) ) ;
if ( codec )
articleString = codec - > toUnicode ( replyData ) ;
else
articleString = QString : : fromUtf8 ( replyData ) ;
2022-09-14 12:01:01 +00:00
// Handle reply data
// 404 response may have response body.
if ( reply - > error ( ) ! = QNetworkReply : : NoError & & articleString . isEmpty ( ) )
{
2022-09-15 23:54:09 +00:00
if ( reply - > error ( ) = = QNetworkReply : : ContentNotFoundError ) {
2022-12-18 13:30:35 +00:00
buffer - > deleteLater ( ) ;
2022-09-15 23:54:09 +00:00
//work around to fix QTBUG-106573
requestJob - > redirect ( url ) ;
return ;
}
2022-09-14 12:01:01 +00:00
QString emptyHtml = QString ( " <html><body>%1</body></html> " ) . arg ( reply - > errorString ( ) ) ;
buffer - > setData ( emptyHtml . toUtf8 ( ) ) ;
requestJob - > reply ( contentType , buffer ) ;
return ;
}
2022-05-13 16:00:23 +00:00
// Change links from relative to absolute
QString root = reply - > url ( ) . scheme ( ) + " :// " + reply - > url ( ) . host ( ) ;
QString base = root + reply - > url ( ) . path ( ) ;
2022-07-16 01:13:33 +00:00
QRegularExpression baseTag ( " <base \\ s+.*?> " ,
2022-07-08 14:14:15 +00:00
QRegularExpression : : CaseInsensitiveOption | QRegularExpression : : DotMatchesEverythingOption ) ;
QString baseTagHtml = " <base href= \" " + base + " \" > " ;
2022-09-02 13:27:25 +00:00
2022-09-04 11:03:27 +00:00
QString depressionFocus = " <script type= \" application/javascript \" > HTMLElement.prototype.focus=function(){console.log( \" focus() has been disabled. \" );}</script> "
" <script type= \" text/javascript \" src= \" qrc:///scripts/iframeResizer.contentWindow.min.js \" ></script> "
" <script type= \" text/javascript \" src= \" qrc:///scripts/iframe-defer.js \" ></script> " ;
2022-07-08 14:14:15 +00:00
// remove existed base tag
2022-07-16 01:13:33 +00:00
articleString . remove ( baseTag ) ;
2022-07-17 10:22:39 +00:00
QRegularExpression headTag ( " <head \\ b.*?> " ,
2022-08-10 11:49:31 +00:00
QRegularExpression : : CaseInsensitiveOption
| QRegularExpression : : DotMatchesEverythingOption ) ;
2022-07-16 01:13:33 +00:00
auto match = headTag . match ( articleString , 0 ) ;
if ( match . hasMatch ( ) )
{
articleString . insert ( match . capturedEnd ( ) , baseTagHtml ) ;
2022-09-02 13:27:25 +00:00
articleString . insert ( match . capturedEnd ( ) , depressionFocus ) ;
2022-07-16 01:13:33 +00:00
}
2022-08-10 11:49:31 +00:00
else
{
// the html contain no head element
// just insert at the beginning of the html ,and leave it at the mercy of browser(chrome webengine)
articleString . insert ( 0 , baseTagHtml ) ;
2022-09-02 13:27:25 +00:00
articleString . insert ( 0 , depressionFocus ) ;
2022-08-10 11:49:31 +00:00
}
2022-05-13 16:00:23 +00:00
2023-03-31 06:05:17 +00:00
buffer - > setData ( articleString . toUtf8 ( ) ) ;
//"text/html;charset=utf-8" will be treated as text/plain in Linux .
# if defined( Q_OS_WIN32 ) || defined( Q_OS_MAC )
requestJob - > reply ( contentType , buffer ) ;
# else
requestJob - > reply ( " text/html " , buffer ) ;
# endif
2022-05-13 16:00:23 +00:00
} ;
connect ( reply , & QNetworkReply : : finished , requestJob , finishAction ) ;
connect ( requestJob , & QObject : : destroyed , reply , & QObject : : deleteLater ) ;
}
