goldendict-ng/webmultimediadownload.cc
2022-12-15 15:11:09 +08:00

91 lines
2.1 KiB
C++

#include "webmultimediadownload.hh"
#include "filetype.hh"
namespace Dictionary {
#define MAX_REDIRECTS 10
WebMultimediaDownload::WebMultimediaDownload( QUrl const & url,
QNetworkAccessManager & _mgr ) :
mgr( _mgr ),
redirectCount( 0 )
{
connect( &mgr, SIGNAL(finished(QNetworkReply*)),
this, SLOT(replyFinished(QNetworkReply*)), Qt::QueuedConnection );
reply = mgr.get( QNetworkRequest( url ) );
#ifndef QT_NO_SSL
connect( reply, SIGNAL( sslErrors( QList< QSslError > ) ),
reply, SLOT( ignoreSslErrors() ) );
#endif
}
void WebMultimediaDownload::cancel()
{
reply = NULL;
finish();
}
void WebMultimediaDownload::replyFinished( QNetworkReply * r )
{
if ( !r || r != reply )
return; // Not our reply
if ( r->error() == QNetworkReply::NoError )
{
// Check for redirect reply
QVariant possibleRedirectUrl = reply->attribute( QNetworkRequest::RedirectionTargetAttribute );
QUrl redirectUrl = possibleRedirectUrl.toUrl();
if( !redirectUrl.isEmpty() )
{
disconnect( reply, 0, 0, 0 );
reply->deleteLater();
if( ++redirectCount > MAX_REDIRECTS )
{
setErrorString( "Too many redirects detected" );
finish();
return;
}
reply = mgr.get( QNetworkRequest( redirectUrl ) );
#ifndef QT_NO_SSL
connect( reply, SIGNAL( sslErrors( QList< QSslError > ) ),
reply, SLOT( ignoreSslErrors() ) );
#endif
return;
}
// Handle reply data
Mutex::Lock _( dataMutex );
data.resize( r->bytesAvailable() );
r->read( data.data(), data.size() );
hasAnyData = true;
}
else
setErrorString( r->errorString() );
disconnect( r, 0, 0, 0 );
r->deleteLater();
reply = NULL;
finish();
}
bool WebMultimediaDownload::isAudioUrl( QUrl const & url )
{
// Note: we check for forvo sound links explicitly, as they don't have extensions
return ( url.scheme() == "http" || url.scheme() == "https" ) && (
Filetype::isNameOfSound( url.path().toUtf8().data() ) || url.host() == "apifree.forvo.com" );
}
}