Imported Upstream version 2.4

[minitube] / src / paginatedvideosource.cpp

diff --git a/src/paginatedvideosource.cpp b/src/paginatedvideosource.cpp
new file mode 100644 (file)
index 0000000..396b836
--- /dev/null
+++ b/src/paginatedvideosource.cpp
@@ -0,0 +1,170 @@
+/* $BEGIN_LICENSE
+
+This file is part of Minitube.
+Copyright 2009, Flavio Tordini <flavio.tordini@gmail.com>
+
+Minitube is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+Minitube is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with Minitube.  If not, see <http://www.gnu.org/licenses/>.
+
+$END_LICENSE */
+
+#include "paginatedvideosource.h"
+
+#include "yt3.h"
+#include "yt3listparser.h"
+#include "datautils.h"
+
+#include "video.h"
+#include "networkaccess.h"
+
+namespace The {
+NetworkAccess* http();
+QHash<QString, QAction*>* globalActions();
+}
+
+PaginatedVideoSource::PaginatedVideoSource(QObject *parent) : VideoSource(parent)
+  , tokenTimestamp(0)
+  , reloadingToken(false)
+  , currentMax(0)
+  , currentStartIndex(0)
+  , asyncDetails(false) { }
+
+bool PaginatedVideoSource::hasMoreVideos() {
+    qDebug() << __PRETTY_FUNCTION__ << nextPageToken;
+    return !nextPageToken.isEmpty();
+}
+
+bool PaginatedVideoSource::maybeReloadToken(int max, int startIndex) {
+    // kind of hackish. Thank the genius who came up with this stateful stuff
+    // in a supposedly RESTful (aka stateless) API.
+
+    if (nextPageToken.isEmpty()) {
+        // previous request did not return a page token. Game over.
+        // emit gotVideos(QList<Video*>());
+        emit finished(0);
+        return true;
+    }
+
+    if (isPageTokenExpired()) {
+        reloadingToken = true;
+        currentMax = max;
+        currentStartIndex = startIndex;
+        reloadToken();
+        return true;
+    }
+    return false;
+}
+
+bool PaginatedVideoSource::setPageToken(const QString &value) {
+    tokenTimestamp = QDateTime::currentDateTime().toTime_t();
+    nextPageToken = value;
+
+    if (reloadingToken) {
+        reloadingToken = false;
+        loadVideos(currentMax, currentStartIndex);
+        currentMax = currentStartIndex = 0;
+        return true;
+    }
+
+    return false;
+}
+
+bool PaginatedVideoSource::isPageTokenExpired() {
+    uint now = QDateTime::currentDateTime().toTime_t();
+    return now - tokenTimestamp > 1800;
+}
+
+void PaginatedVideoSource::reloadToken() {
+    qDebug() << "Reloading pageToken";
+    QObject *reply = The::http()->get(lastUrl);
+    connect(reply, SIGNAL(data(QByteArray)), SLOT(parseResults(QByteArray)));
+    connect(reply, SIGNAL(error(QNetworkReply*)), SLOT(requestError(QNetworkReply*)));
+}
+
+void PaginatedVideoSource::loadVideoDetails(const QList<Video*> &videos) {
+    QString videoIds;
+    foreach (Video *video, videos) {
+        // TODO get video details from cache
+        if (!videoIds.isEmpty()) videoIds += ",";
+        videoIds += video->id();
+        videoMap.insert(video->id(), video);
+    }
+
+    if (videoIds.isEmpty()) {
+        if (!asyncDetails) {
+            emit gotVideos(videos);
+            emit finished(videos.size());
+        }
+        return;
+    }
+
+    QUrl url = YT3::instance().method("videos");
+
+#if QT_VERSION >= 0x050000
+    {
+        QUrl &u = url;
+        QUrlQuery url;
+#endif
+
+        url.addQueryItem("part", "contentDetails,statistics");
+        url.addQueryItem("id", videoIds);
+
+#if QT_VERSION >= 0x050000
+        u.setQuery(url);
+    }
+#endif
+
+    QObject *reply = The::http()->get(url);
+    connect(reply, SIGNAL(data(QByteArray)), SLOT(parseVideoDetails(QByteArray)));
+    connect(reply, SIGNAL(error(QNetworkReply*)), SLOT(requestError(QNetworkReply*)));
+}
+
+void PaginatedVideoSource::parseVideoDetails(const QByteArray &bytes) {
+
+    QScriptEngine engine;
+    QScriptValue json = engine.evaluate("(" + QString::fromUtf8(bytes) + ")");
+
+    QScriptValue items = json.property("items");
+    if (items.isArray()) {
+        QScriptValueIterator it(items);
+        while (it.hasNext()) {
+            it.next();
+            QScriptValue item = it.value();
+            if (!item.isObject()) continue;
+
+            // qDebug() << item.toString();
+
+            QString id = item.property("id").toString();
+            Video *video = videoMap.value(id);
+            if (!video) {
+                qWarning() << "No video for id" << id;
+                continue;
+            }
+
+            QString isoPeriod = item.property("contentDetails").property("duration").toString();
+            int duration = DataUtils::parseIsoPeriod(isoPeriod);
+            video->setDuration(duration);
+
+            uint viewCount = item.property("statistics").property("viewCount").toUInt32();
+            video->setViewCount(viewCount);
+
+            // TODO cache by etag?
+        }
+    }
+    if (!asyncDetails) {
+        emit gotVideos(videoMap.values());
+        emit finished(videoMap.size());
+    } else {
+        emit gotDetails();
+    }
+}