Add soundcloud

4 files changed, 298 insertions, 11 deletions

diff --git a/src/plugins/Pornhub.cpp b/src/plugins/Pornhub.cpp
index b4e908a..74b66b6 100644
--- a/src/plugins/Pornhub.cpp
+++ b/src/plugins/Pornhub.cpp
@@ -140,13 +140,13 @@ namespace QuickMedia {
         return search_result_to_plugin_result(get_videos_in_page(url, is_tor_enabled(), result_items));
     }
 
-    PluginResult PornhubSearchPage::submit(const std::string&, const std::string&, std::vector<Tab> &result_tabs) {
-        result_tabs.push_back(Tab{nullptr, std::make_unique<PornhubVideoPage>(program), nullptr});
+    PluginResult PornhubSearchPage::submit(const std::string&, const std::string &url, std::vector<Tab> &result_tabs) {
+        result_tabs.push_back(Tab{nullptr, std::make_unique<PornhubVideoPage>(program, url), nullptr});
         return PluginResult::OK;
     }
 
-    PluginResult PornhubRelatedVideosPage::submit(const std::string&, const std::string&, std::vector<Tab> &result_tabs) {
-        result_tabs.push_back(Tab{nullptr, std::make_unique<PornhubVideoPage>(program), nullptr});
+    PluginResult PornhubRelatedVideosPage::submit(const std::string&, const std::string &url, std::vector<Tab> &result_tabs) {
+        result_tabs.push_back(Tab{nullptr, std::make_unique<PornhubVideoPage>(program, url), nullptr});
         return PluginResult::OK;
     }
 
diff --git a/src/plugins/Soundcloud.cpp b/src/plugins/Soundcloud.cpp
new file mode 100644
index 0000000..cfb6011
--- /dev/null
+++ b/src/plugins/Soundcloud.cpp
@@ -0,0 +1,287 @@
+#include "../../plugins/Soundcloud.hpp"
+#include "../../include/NetUtils.hpp"
+#include "../../include/StringUtils.hpp"
+#include "../../include/Scale.hpp"
+
+namespace QuickMedia {
+    static std::string client_id = "Na04L87fnpWDMVCCW2ngWldN4JMoLTAc";
+
+    class SoundcloudPlaylist : public BodyItemExtra {
+    public:
+        BodyItems tracks;
+    };
+
+    // Return empty string if transcoding files are not found
+    static std::string get_best_transcoding_audio_url(const Json::Value &media_json) {
+        const Json::Value &transcodings_json = media_json["transcodings"];
+        if(transcodings_json.isArray() && !transcodings_json.empty() && transcodings_json[0].isObject()) {
+            const Json::Value &transcoding_url = transcodings_json[0]["url"];
+            if(transcoding_url.isString())
+                return transcoding_url.asString();
+        }
+        return "";
+    }
+
+    static std::string duration_to_descriptive_string(int64_t seconds) {
+        seconds /= 1000;
+        time_t minutes = seconds / 60;
+        time_t hours = minutes / 60;
+
+        std::string str;
+        if(hours >= 1) {
+            str = std::to_string(hours) + " hour" + (hours == 1 ? "" : "s");
+            seconds -= (hours * 60 * 60);
+        }
+        
+        if(minutes >= 1) {
+            if(!str.empty())
+                str += ", ";
+            str += std::to_string(minutes) + " minute" + (minutes == 1 ? "" : "s");
+            seconds -= (minutes * 60);
+        }
+        
+        if(!str.empty() || seconds > 0) {
+            if(!str.empty())
+                str += ", ";
+            str += std::to_string(seconds) + " second" + (seconds == 1 ? "" : "s");
+        }
+
+        return str;
+    }
+
+    static std::string collection_item_get_duration(const Json::Value &item_json) {
+        const Json::Value &full_duration_json = item_json["full_duration"];
+        if(full_duration_json.isInt64())
+            return duration_to_descriptive_string(full_duration_json.asInt64());
+
+        const Json::Value &duration_json = item_json["duration"];
+        if(duration_json.isInt64())
+            return duration_to_descriptive_string(duration_json.asInt64());
+
+        return "";
+    }
+
+    static std::shared_ptr<BodyItem> parse_collection_item(const Json::Value &item_json) {
+        std::string title;
+
+        const Json::Value &title_json = item_json["title"];
+        const Json::Value &username_json = item_json["username"];
+        if(title_json.isString())
+            title = title_json.asString();
+        else if(username_json.isString())
+            title = username_json.asString();
+        else
+            return nullptr;
+
+        auto body_item = BodyItem::create(std::move(title));
+        std::string description;
+
+        const Json::Value &media_json = item_json["media"];
+        if(media_json.isObject())
+            body_item->url = get_best_transcoding_audio_url(media_json);
+
+        if(body_item->url.empty()) {
+            const Json::Value &tracks_json = item_json["tracks"];
+            if(tracks_json.isArray()) {
+                auto playlist = std::make_shared<SoundcloudPlaylist>();
+                for(const Json::Value &track_json : tracks_json) {
+                    if(!track_json.isObject())
+                        continue;
+
+                    auto track = parse_collection_item(track_json);
+                    if(track)
+                        playlist->tracks.push_back(std::move(track));
+                }
+
+                description = "Playlist with " + std::to_string(playlist->tracks.size()) + " track" + (playlist->tracks.size() == 1 ? "" : "s");
+                body_item->extra = std::move(playlist);
+                body_item->url = "track";
+            }
+        }
+
+        if(body_item->url.empty()) {
+            const Json::Value &id_json = item_json["id"];
+            if(id_json.isInt64())
+                body_item->url = "https://api-v2.soundcloud.com/stream/users/" + std::to_string(id_json.asInt64());
+        }
+
+        const Json::Value &artwork_url_json = item_json["artwork_url"];
+        const Json::Value &avatar_url_json = item_json["avatar_url"];
+        if(artwork_url_json.isString()) {
+            // For larger thumbnails
+            /*
+            if(strstr(artwork_url_json.asCString(), "-large") != 0) {
+                std::string artwork_url = artwork_url_json.asString();
+                string_replace_all(artwork_url, "-large", "-t200x200");
+                body_item->thumbnail_url = std::move(artwork_url);
+                body_item->thumbnail_size.x = 200;
+                body_item->thumbnail_size.y = 200;
+            } else {
+                body_item->thumbnail_url = artwork_url_json.asString();
+                body_item->thumbnail_size.x = 100;
+                body_item->thumbnail_size.y = 100;
+            }
+            */
+            body_item->thumbnail_url = artwork_url_json.asString();
+            body_item->thumbnail_size.x = 100;
+            body_item->thumbnail_size.y = 100;
+        } else if(avatar_url_json.isString()) {
+            body_item->thumbnail_url = avatar_url_json.asString();
+            body_item->thumbnail_size.x = 100;
+            body_item->thumbnail_size.y = 100;
+            body_item->thumbnail_mask_type = ThumbnailMaskType::CIRCLE;
+        }
+
+        std::string duration_str = collection_item_get_duration(item_json);
+        if(!duration_str.empty()) {
+            if(!description.empty())
+                description += '\n';
+            description += std::move(duration_str);
+        }
+
+        body_item->set_description(std::move(description));
+        return body_item;
+    }
+
+    static PluginResult parse_user_page(const Json::Value &json_root, BodyItems &result_items, std::string &next_href) {
+        if(!json_root.isObject())
+            return PluginResult::ERR;
+
+        const Json::Value &next_href_json = json_root["next_href"];
+        if(next_href_json.isString())
+            next_href = next_href_json.asString();
+
+        const Json::Value &collection_json = json_root["collection"];
+        if(!collection_json.isArray())
+            return PluginResult::ERR;
+
+        for(const Json::Value &item_json : collection_json) {
+            if(!item_json.isObject())
+                continue;
+
+            const Json::Value &track_json = item_json["track"];
+            const Json::Value &playlist_json = item_json["playlist"];
+            if(track_json.isObject()) {
+                auto body_item = parse_collection_item(track_json);
+                if(body_item)
+                    result_items.push_back(std::move(body_item));
+            } else if(playlist_json.isObject()) {
+                auto body_item = parse_collection_item(playlist_json);
+                if(body_item)
+                    result_items.push_back(std::move(body_item));
+            }
+        }
+
+        return PluginResult::OK;
+    }
+
+    PluginResult SoundcloudPage::submit(const std::string &title, const std::string &url, std::vector<Tab> &result_tabs) {
+        if(url.empty())
+            return PluginResult::ERR;
+
+        if(url == "track") {
+            auto body = create_body();
+            body->items = static_cast<SoundcloudPlaylist*>(submit_body_item->extra.get())->tracks;
+            result_tabs.push_back(Tab{std::move(body), std::make_unique<SoundcloudPlaylistPage>(program, title), nullptr});
+        } else if(url.find("/stream/users/") != std::string::npos) {
+            std::string query_url = url + "?client_id=" + client_id + "&limit=20&offset=0&linked_partitioning=1&app_version=1616689516&app_locale=en";
+
+            Json::Value json_root;
+            DownloadResult result = download_json(json_root, query_url, {}, true);
+            if(result != DownloadResult::OK) return download_result_to_plugin_result(result);
+
+            auto body = create_body();
+            std::string next_href;
+            PluginResult pr = parse_user_page(json_root, body->items, next_href);
+            if(pr != PluginResult::OK) return pr;
+
+            result_tabs.push_back(Tab{std::move(body), std::make_unique<SoundcloudUserPage>(program, title, url, std::move(next_href)), nullptr});
+        } else {
+            std::string query_url = url + "?client_id=" + client_id;
+
+            Json::Value json_root;
+            DownloadResult result = download_json(json_root, query_url, {}, true);
+            if(result != DownloadResult::OK) return download_result_to_plugin_result(result);
+
+            if(!json_root.isObject())
+                return PluginResult::ERR;
+
+            const Json::Value &url_json = json_root["url"];
+            if(!url_json.isString())
+                return PluginResult::ERR;
+
+            result_tabs.push_back(Tab{create_body(), std::make_unique<SoundcloudAudioPage>(program, url_json.asString()), nullptr});
+        }
+
+        return PluginResult::OK;
+    }
+
+    SearchResult SoundcloudSearchPage::search(const std::string &str, BodyItems &result_items) {
+        query_urn.clear();
+        PluginResult result = get_page(str, 0, result_items);
+        if(result != PluginResult::OK)
+            return SearchResult::ERR;
+        return SearchResult::OK;
+    }
+
+    PluginResult SoundcloudSearchPage::get_page(const std::string &str, int page, BodyItems &result_items) {
+        std::string url = "https://api-v2.soundcloud.com/search?q=";
+        url += url_param_encode(str);
+        url += "&variant_ids=2227&facet=model&client_id=" + client_id + "&limit=20&offset=" + std::to_string(page * 20) + "&linked_partitioning=1&app_version=1616689516&app_locale=en";
+        if(!query_urn.empty())
+            url += "&query_url=" + url_param_encode(query_urn);
+        else if(page > 0)
+            return PluginResult::OK;
+
+        Json::Value json_root;
+        DownloadResult result = download_json(json_root, url, {}, true);
+        if(result != DownloadResult::OK) return download_result_to_plugin_result(result);
+
+        if(!json_root.isObject())
+            return PluginResult::ERR;
+
+        const Json::Value &query_urn_json = json_root["query_urn"];
+        if(query_urn_json.isString())
+            query_urn = query_urn_json.asString();
+
+        const Json::Value &collection_json = json_root["collection"];
+        if(!collection_json.isArray())
+            return PluginResult::ERR;
+
+        for(const Json::Value &item_json : collection_json) {
+            if(!item_json.isObject())
+                continue;
+
+            const Json::Value &kind_json = item_json["kind"];
+            if(!kind_json.isString())
+                continue;
+
+            if(strcmp(kind_json.asCString(), "user") == 0 || strcmp(kind_json.asCString(), "track") == 0 || strcmp(kind_json.asCString(), "playlist") == 0) {
+                auto body_item = parse_collection_item(item_json);
+                if(body_item)
+                    result_items.push_back(std::move(body_item));
+            }
+        }
+
+        return PluginResult::OK;
+    }
+
+    PluginResult SoundcloudUserPage::get_page(const std::string&, int page, BodyItems &result_items) {
+        while(current_page < page) {
+            PluginResult plugin_result = get_continuation_page(result_items);
+            if(plugin_result != PluginResult::OK) return plugin_result;
+            ++current_page;
+        }
+        return PluginResult::OK;
+    }
+
+    PluginResult SoundcloudUserPage::get_continuation_page(BodyItems &result_items) {
+        if(next_href.empty())
+            return PluginResult::OK;
+
+        Json::Value json_root;
+        DownloadResult result = download_json(json_root, next_href + "&client_id=" + client_id, {}, true);
+        if(result != DownloadResult::OK) return download_result_to_plugin_result(result);
+        return parse_user_page(json_root, result_items, next_href);
+    }
+}
\ No newline at end of file
diff --git a/src/plugins/Spotify.cpp b/src/plugins/Spotify.cpp
index 14f9831..f56ed6c 100644
--- a/src/plugins/Spotify.cpp
+++ b/src/plugins/Spotify.cpp
@@ -274,8 +274,8 @@ namespace QuickMedia {
         return PluginResult::OK;
     }
 
-    PluginResult SpotifyEpisodeListPage::submit(const std::string &, const std::string &, std::vector<Tab> &result_tabs) {
-        result_tabs.push_back(Tab{nullptr, std::make_unique<SpotifyAudioPage>(program), nullptr});
+    PluginResult SpotifyEpisodeListPage::submit(const std::string &, const std::string &url, std::vector<Tab> &result_tabs) {
+        result_tabs.push_back(Tab{nullptr, std::make_unique<SpotifyAudioPage>(program, url), nullptr});
         return PluginResult::OK;
     }
 }
\ No newline at end of file
diff --git a/src/plugins/Youtube.cpp b/src/plugins/Youtube.cpp
index 99227d5..3813068 100644
--- a/src/plugins/Youtube.cpp
+++ b/src/plugins/Youtube.cpp
@@ -572,7 +572,7 @@ namespace QuickMedia {
             // TODO: Make all pages (for all services) lazy fetch in a similar manner!
             result_tabs.push_back(Tab{create_body(), std::make_unique<YoutubeChannelPage>(program, url, "", title), create_search_bar("Search...", 350)});
         } else {
-            result_tabs.push_back(Tab{nullptr, std::make_unique<YoutubeVideoPage>(program), nullptr});
+            result_tabs.push_back(Tab{nullptr, std::make_unique<YoutubeVideoPage>(program, url), nullptr});
         }
         return PluginResult::OK;
     }
@@ -1099,10 +1099,10 @@ namespace QuickMedia {
         return PluginResult::OK;
     }
 
-    PluginResult YoutubeChannelPage::submit(const std::string&, const std::string&, std::vector<Tab> &result_tabs) {
+    PluginResult YoutubeChannelPage::submit(const std::string&, const std::string &url, std::vector<Tab> &result_tabs) {
         if(url.empty())
             return PluginResult::OK;
-        result_tabs.push_back(Tab{nullptr, std::make_unique<YoutubeVideoPage>(program), nullptr});
+        result_tabs.push_back(Tab{nullptr, std::make_unique<YoutubeVideoPage>(program, url), nullptr});
         return PluginResult::OK;
     }
 
@@ -1126,8 +1126,8 @@ namespace QuickMedia {
         return PluginResult::OK;
     }
 
-    PluginResult YoutubeRelatedVideosPage::submit(const std::string&, const std::string&, std::vector<Tab> &result_tabs) {
-        result_tabs.push_back(Tab{nullptr, std::make_unique<YoutubeVideoPage>(program), nullptr});
+    PluginResult YoutubeRelatedVideosPage::submit(const std::string&, const std::string &url, std::vector<Tab> &result_tabs) {
+        result_tabs.push_back(Tab{nullptr, std::make_unique<YoutubeVideoPage>(program, url), nullptr});
         return PluginResult::OK;
     }