New upstream version 3.8

[minitube] / src / yt / searchvideosource.cpp

diff --git a/src/yt/searchvideosource.cpp b/src/yt/searchvideosource.cpp
new file mode 100644 (file)
index 0000000..bec7d0a
--- /dev/null
+++ b/src/yt/searchvideosource.cpp
@@ -0,0 +1,89 @@
+#include "searchvideosource.h"
+
+#include "searchparams.h"
+#include "video.h"
+#include "videoapi.h"
+
+#include "ytsearch.h"
+
+#include "ivchannelsource.h"
+#include "ivsearch.h"
+#include "ivsinglevideosource.h"
+
+#include "ytjschannelsource.h"
+#include "ytjssearch.h"
+#include "ytjssinglevideosource.h"
+
+SearchVideoSource::SearchVideoSource(SearchParams *searchParams, QObject *parent)
+    : VideoSource(parent), searchParams(searchParams) {}
+
+void SearchVideoSource::loadVideos(int max, int startIndex) {
+    if (!source) {
+        aborted = false;
+        if (VideoAPI::impl() == VideoAPI::YT3) {
+            YTSearch *ytSearch = new YTSearch(searchParams);
+            ytSearch->setAsyncDetails(true);
+            // --- connect(ytSearch, SIGNAL(gotDetails()), playlistModel, SLOT(emitDataChanged()));
+            source = ytSearch;
+        } else if (VideoAPI::impl() == VideoAPI::IV) {
+            if (searchParams->channelId().isEmpty()) {
+                source = new IVSearch(searchParams);
+            } else {
+                source = new IVChannelSource(searchParams);
+            }
+        } else if (VideoAPI::impl() == VideoAPI::JS) {
+            if (searchParams->channelId().isEmpty()) {
+                source = new YTJSSearch(searchParams);
+            } else {
+                source = new YTJSChannelSource(searchParams);
+            }
+        }
+        connectSource(max, startIndex);
+    }
+    source->loadVideos(max, startIndex);
+}
+
+bool SearchVideoSource::hasMoreVideos() {
+    if (source) return source->hasMoreVideos();
+    return VideoSource::hasMoreVideos();
+}
+
+QString SearchVideoSource::getName() {
+    if (source) return source->getName();
+    return QString();
+}
+
+const QList<QAction *> &SearchVideoSource::getActions() {
+    if (source) return source->getActions();
+    return VideoSource::getActions();
+}
+
+int SearchVideoSource::maxResults() {
+    if (source) return source->maxResults();
+    return VideoSource::maxResults();
+}
+
+void SearchVideoSource::connectSource(int max, int startIndex) {
+    connect(source, &VideoSource::finished, this, &VideoSource::finished);
+    connect(source, &VideoSource::gotVideos, this, [this](auto &videos) {
+        if (aborted) return;
+        emit gotVideos(videos);
+    });
+    connect(source, &VideoSource::error, this, [this, max, startIndex](auto msg) {
+        qDebug() << source << msg;
+        if (aborted) return;
+        if (QLatin1String(source->metaObject()->className()).startsWith(QLatin1String("YTJS"))) {
+            qDebug() << "Falling back to IV";
+            source->deleteLater();
+            if (searchParams->channelId().isEmpty()) {
+                source = new IVSearch(searchParams);
+            } else {
+                source = new IVChannelSource(searchParams);
+            }
+            connectSource(max, startIndex);
+            source->loadVideos(max, startIndex);
+        } else {
+            emit error(msg);
+        }
+    });
+}