]> git.sur5r.net Git - minitube/blobdiff - src/networkaccess.cpp
Imported Debian patch 1.6-1
[minitube] / src / networkaccess.cpp
index 378cb1a80fd51e9573e27b2e778ef6dc46141d1e..57f5637dccff1ddfccd10f571e27251a0a9d45ec 100644 (file)
@@ -6,15 +6,30 @@ namespace The {
     NetworkAccess* http();
 }
 
-const QString USER_AGENT = QString(Constants::APP_NAME)
+/*
+const QString USER_AGENT = QString(Constants::NAME)
                            + " " + Constants::VERSION
                            + " (" + Constants::WEBSITE + ")";
+*/
+
+const QString USER_AGENT = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_7_2) AppleWebKit/535.1 (KHTML, like Gecko) Chrome/14.0.835.202 Safari/535.1";
 
 NetworkReply::NetworkReply(QNetworkReply *networkReply) : QObject(networkReply) {
     this->networkReply = networkReply;
+
+    // monitor downloadProgress to impl timeout
+    connect(networkReply, SIGNAL(downloadProgress(qint64,qint64)),
+            SLOT(downloadProgress(qint64,qint64)), Qt::UniqueConnection);
+
+    readTimeoutTimer = new QTimer(this);
+    readTimeoutTimer->setInterval(5000);
+    readTimeoutTimer->setSingleShot(true);
+    connect(readTimeoutTimer, SIGNAL(timeout()), SLOT(readTimeout()));
+    readTimeoutTimer->start();
 }
 
 void NetworkReply::finished() {
+    // qDebug() << "Finished" << networkReply->url();
 
     QUrl redirection = networkReply->attribute(QNetworkRequest::RedirectionTargetAttribute).toUrl();
     if (redirection.isValid()) {
@@ -28,50 +43,95 @@ void NetworkReply::finished() {
         networkReply = redirectReply;
 
         // when the request is finished we'll invoke the target method
-        connect(networkReply, SIGNAL(finished()), this, SLOT(finished()), Qt::AutoConnection);
+        connect(networkReply, SIGNAL(finished()), this, SLOT(finished()), Qt::UniqueConnection);
+
+        // monitor downloadProgress to impl timeout
+        connect(networkReply, SIGNAL(downloadProgress(qint64,qint64)),
+                SLOT(downloadProgress(qint64,qint64)), Qt::UniqueConnection);
+        readTimeoutTimer->start();
+
+        // error signal
+        connect(networkReply, SIGNAL(error(QNetworkReply::NetworkError)),
+                SLOT(requestError(QNetworkReply::NetworkError)), Qt::UniqueConnection);
+
+        connect(readTimeoutTimer, SIGNAL(timeout()), SLOT(readTimeout()), Qt::UniqueConnection);
+        readTimeoutTimer->start();
 
         return;
     }
 
-
+    emit data(networkReply->readAll());
     emit finished(networkReply);
 
-    // get the HTTP response body
-    QByteArray bytes = networkReply->readAll();
-
-    emit data(bytes);
+#ifndef QT_NO_DEBUG_OUTPUT
+    if (!networkReply->attribute(QNetworkRequest::SourceIsFromCacheAttribute).toBool()) {
+        qDebug() << networkReply->url().toEncoded();
+    }
+#endif
 
     // bye bye my reply
     // this will also delete this NetworkReply as the QNetworkReply is its parent
     networkReply->deleteLater();
 }
 
-void NetworkReply::requestError(QNetworkReply::NetworkError code) {
+void NetworkReply::requestError(QNetworkReply::NetworkError /* code */) {
     emit error(networkReply);
 }
 
+void NetworkReply::downloadProgress(qint64 bytesReceived, qint64 bytesTotal) {
+    // qDebug() << "Downloading" << bytesReceived << bytesTotal << networkReply->url();
+    if (bytesReceived > 0) {
+        readTimeoutTimer->stop();
+        disconnect(networkReply, SIGNAL(downloadProgress(qint64,qint64)),
+                   this, SLOT(downloadProgress(qint64,qint64)));
+    }
+}
+
+void NetworkReply::readTimeout() {
+    // qDebug() << "HTTP read timeout" << networkReply->url();
+    networkReply->disconnect();
+    networkReply->abort();
+
+    QNetworkReply *retryReply = The::http()->simpleGet(networkReply->url(), networkReply->operation());
+
+    setParent(retryReply);
+    networkReply->deleteLater();
+    networkReply = retryReply;
+
+    // when the request is finished we'll invoke the target method
+    connect(networkReply, SIGNAL(finished()), this, SLOT(finished()), Qt::UniqueConnection);
+
+    // monitor downloadProgress to impl timeout
+    connect(networkReply, SIGNAL(downloadProgress(qint64,qint64)),
+            SLOT(downloadProgress(qint64,qint64)), Qt::UniqueConnection);
+    readTimeoutTimer->start();
+
+    // error signal
+    connect(networkReply, SIGNAL(error(QNetworkReply::NetworkError)),
+            SLOT(requestError(QNetworkReply::NetworkError)));
+
+    // emit error(networkReply);
+}
+
 /* --- NetworkAccess --- */
 
 NetworkAccess::NetworkAccess( QObject* parent) : QObject( parent ) {}
 
-QNetworkReply* NetworkAccess::simpleGet(QUrl url, int operation) {
+QNetworkReply* NetworkAccess::manualGet(QNetworkRequest request, int operation) {
 
     QNetworkAccessManager *manager = The::networkAccessManager();
-
-    QNetworkRequest request(url);
-    request.setRawHeader("User-Agent", USER_AGENT.toUtf8());
-    request.setRawHeader("Connection", "Keep-Alive");
+    // manager->setCookieJar(new QNetworkCookieJar());
 
     QNetworkReply *networkReply;
     switch (operation) {
 
     case QNetworkAccessManager::GetOperation:
-        qDebug() << "GET" << url.toString();
+        // qDebug() << "GET" << request.url().toEncoded();
         networkReply = manager->get(request);
         break;
 
     case QNetworkAccessManager::HeadOperation:
-        qDebug() << "HEAD" << url.toString();
+        // qDebug() << "HEAD" << request.url().toEncoded();
         networkReply = manager->head(request);
         break;
 
@@ -86,7 +146,20 @@ QNetworkReply* NetworkAccess::simpleGet(QUrl url, int operation) {
             this, SLOT(error(QNetworkReply::NetworkError)));
 
     return networkReply;
+}
 
+QNetworkRequest NetworkAccess::buildRequest(QUrl url) {
+    QNetworkRequest request(url);
+    request.setRawHeader("User-Agent", USER_AGENT.toUtf8());
+    request.setRawHeader("Accept-Charset", "ISO-8859-1,utf-8;q=0.7,*;q=0.7");
+    request.setRawHeader("Accept", "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8");
+    request.setRawHeader("Accept-Language", "en-us,en;q=0.5");
+    request.setRawHeader("Connection", "Keep-Alive");
+    return request;
+}
+
+QNetworkReply* NetworkAccess::simpleGet(QUrl url, int operation) {
+    return manualGet(buildRequest(url), operation);
 }
 
 NetworkReply* NetworkAccess::get(const QUrl url) {
@@ -99,7 +172,7 @@ NetworkReply* NetworkAccess::get(const QUrl url) {
             reply, SLOT(requestError(QNetworkReply::NetworkError)));
 
     // when the request is finished we'll invoke the target method
-    connect(networkReply, SIGNAL(finished()), reply, SLOT(finished()), Qt::AutoConnection);
+    connect(networkReply, SIGNAL(finished()), reply, SLOT(finished()), Qt::UniqueConnection);
 
     return reply;
 
@@ -115,65 +188,12 @@ NetworkReply* NetworkAccess::head(const QUrl url) {
             reply, SLOT(requestError(QNetworkReply::NetworkError)));
 
     // when the request is finished we'll invoke the target method
-    connect(networkReply, SIGNAL(finished()), reply, SLOT(finished()), Qt::AutoConnection);
+    connect(networkReply, SIGNAL(finished()), reply, SLOT(finished()), Qt::UniqueConnection);
 
     return reply;
 
 }
 
-/*** sync ***/
-
-
-QNetworkReply* NetworkAccess::syncGet(QUrl url) {
-
-    working = true;
-
-    networkReply = simpleGet(url);
-    connect(networkReply, SIGNAL(metaDataChanged()),
-            this, SLOT(syncMetaDataChanged()), Qt::AutoConnection);
-    connect(networkReply, SIGNAL(finished()),
-            this, SLOT(syncFinished()), Qt::AutoConnection);
-    connect(networkReply, SIGNAL(error(QNetworkReply::NetworkError)),
-            this, SLOT(error(QNetworkReply::NetworkError)));
-
-    // A little trick to make this function blocking
-    while (working) {
-        // Do something else, maybe even network processing events
-        qApp->processEvents();
-    }
-
-    networkReply->deleteLater();
-    return networkReply;
-
-}
-
-void NetworkAccess::syncMetaDataChanged() {
-
-    QUrl redirection = networkReply->attribute(QNetworkRequest::RedirectionTargetAttribute).toUrl();
-    if (redirection.isValid()) {
-
-        qDebug() << "Redirect" << redirection;
-        networkReply->deleteLater();
-        syncGet(redirection);
-
-        /*
-        QNetworkAccessManager *manager = The::networkAccessManager();
-        networkReply->deleteLater();
-        networkReply = manager->get(QNetworkRequest(redirection));
-        connect(networkReply, SIGNAL(metaDataChanged()),
-                this, SLOT(metaDataChanged()), Qt::AutoConnection);
-        connect(networkReply, SIGNAL(finished()),
-                this, SLOT(finished()), Qt::AutoConnection);
-        */
-    }
-
-}
-
-void NetworkAccess::syncFinished() {
-    // got it!
-    working = false;
-}
-
 void NetworkAccess::error(QNetworkReply::NetworkError code) {
     // get the QNetworkReply that sent the signal
     QNetworkReply *networkReply = static_cast<QNetworkReply *>(sender());
@@ -182,6 +202,8 @@ void NetworkAccess::error(QNetworkReply::NetworkError code) {
         return;
     }
 
+    networkReply->deleteLater();
+
     // Ignore HEADs
     if (networkReply->operation() == QNetworkAccessManager::HeadOperation)
         return;
@@ -192,14 +214,4 @@ void NetworkAccess::error(QNetworkReply::NetworkError code) {
             tr("Network error: %1").arg(networkReply->errorString()));
 
     qDebug() << "Network error:" << networkReply->errorString() << code;
-
-    networkReply->deleteLater();
-}
-
-QByteArray NetworkAccess::syncGetBytes(QUrl url) {
-    return syncGet(url)->readAll();
-}
-
-QString NetworkAccess::syncGetString(QUrl url) {
-    return QString::fromUtf8(syncGetBytes(url));
 }