Menu test

2024-03-27 16:04:48 +01:00
parent 4f0402d71b
commit 5e85b74224
3 changed files with 242 additions and 4 deletions
--- a/YouTubeMusic/API.pm
+++ b/YouTubeMusic/API.pm
@@ -0,0 +1,179 @@
+package Plugins::YouTubeMusic::API;
+
+use strict;
+
+use Digest::MD5 qw(md5_hex);
+use File::Spec::Functions qw(catdir);
+use JSON::XS::VersionOneAndTwo;
+use List::Util qw(min max);
+use URI::Escape qw(uri_escape uri_escape_utf8);
+use MIME::Base64 qw(decode_base64);
+
+use constant API_URL => 'https://music.youtube.com/youtubei/v1/';
+use constant DEFAULT_CACHE_TTL => 24 * 3600;
+use constant DEFAULT_API_KEY => 'QUl6YVN5Qi1wd1B0RGt4RjZKUW1BOHFxOWgxbWQ2ME15STVRNWlB';
+
+use Slim::Utils::Cache;
+use Slim::Utils::Log;
+use Slim::Utils::Prefs;
+
+my $prefs = preferences('plugin.youtubemusic');
+my $log   = logger('plugin.youtubemusic');
+my $cache = Slim::Utils::Cache->new();
+
+sub flushCache { $cache->cleanup(); }
+
+sub search {
+	my ( $class, $cb, $args ) = @_;
+
+	$args ||= {};
+	$args->{part} = 'snippet';
+	$args->{type} ||= 'video';
+	$args->{order} ||= $prefs->get('search_rank');
+	$args->{relevanceLanguage} = Slim::Utils::Strings::getLanguage();
+
+	_pagedCall('search', $args, $cb);
+}
+
+sub searchDirect {
+	my ( $class, $type, $cb, $args ) = @_;
+
+	$args ||= {};
+	$args->{_noRegion} = 1;
+
+	_pagedCall( $type, $args, $cb);
+}
+
+sub getCategories {
+	my ( $class, $type, $cb, $args ) = @_;
+
+	$args ||= {};
+	$args->{hl} = Slim::Utils::Strings::getLanguage();
+	$args->{_cache_ttl} = 7 * 86400;
+
+	_pagedCall($type, $args, $cb);
+}
+
+sub getVideoDetails {
+	my ( $class, $cb, $ids ) = @_;
+
+	_call('videos', {
+		part => 'snippet,contentDetails',
+		id   => $ids,
+		# cache video details a bit longer
+		_cache_ttl => 7 * 86400,
+	}, $cb);
+}
+
+sub _pagedCall {
+	my ( $method, $args, $cb ) = @_;
+	my $wantedItems = $args->{_quantity} || $prefs->get('max_items');
+	# ignore $args->{_index} and let LMS handle offset
+	my $wantedIndex = 0;
+	my @items;
+	my $pagingCb;
+	my $pageIndex = 0;
+	
+	# we want them starting from current index
+	$wantedItems += $args->{_index} || 0;
+
+	main::INFOLOG && $log->info("Searching by [$args->{order}]");
+	main::INFOLOG && $log->info("Querying [$args->{_quantity}] from [$args->{_index}] to [", $wantedItems-1, "] using [$method]");
+
+	# doing a search with a display order, so need to give precedence to 'query_size'
+	if ($prefs->get('search_sort') || $prefs->get('channel_sort') || $prefs->get('playlist_sort')) {
+		$wantedItems = (int(($wantedItems - 1) / $prefs->get('query_size')) + 1) * $prefs->get('query_size');
+		main::INFOLOG && $log->info("Stretching quantity to [$wantedItems] due to sorting");
+	}
+	
+	# that the maximum we'll get anyway
+	$wantedItems = $prefs->get('max_items') if $wantedItems > $prefs->get('max_items');
+
+	$pagingCb = sub {
+		my $results = shift;
+
+		if ( $results->{error} || !$results->{items} ) {
+			$log->error("no results");
+			$cb->( { items => undef, total => 0 } ) if ( $results->{error} );
+			return;
+		}
+
+		push @items, @{$results->{items}};
+		$pageIndex += scalar @{$results->{items}};
+
+		main::INFOLOG && $log->info("Want $wantedItems items from offset ", $wantedIndex, ", have " . scalar @items . " so far [acquired $pageIndex]");
+
+		if (@items < $wantedItems && $results->{nextPageToken}) {
+			$args->{pageToken} = $results->{nextPageToken};
+			main::INFOLOG && $log->info("Get next page using token " . $args->{pageToken});
+			_call($method, $args, $pagingCb);
+		} else {
+			my $total = min($results->{'pageInfo'}->{'totalResults'} || $pageIndex, $prefs->get('max_items'));
+			main::INFOLOG && $log->info("Got all we wanted, return " . scalar @items . "/$total. (YT total ", $results->{'pageInfo'}->{'totalResults'} || 'N/A', ")");
+			$cb->( { items => \@items, offset => $wantedIndex, total  => $total } );
+		}
+	};
+
+	_call($method, $args, $pagingCb);
+}
+
+sub _call {
+	my ( $method, $args, $cb ) = @_;
+
+	my $API_KEY = $prefs->get('APIkey') || MIME::Base64::decode_base64(DEFAULT_API_KEY);
+	my $url = '?' . ($args->{_noKey} ? '' : 'key=' . $API_KEY . '&');
+
+	$args->{regionCode} ||= $prefs->get('country') unless $args->{_noRegion};
+	$args->{part}       ||= 'snippet' unless $args->{_noPart};
+	$args->{maxResults} ||= 50;
+
+	for my $k ( sort keys %{$args} ) {
+		next if $k =~ /^_/;
+		$url .= $k . '=' . URI::Escape::uri_escape_utf8( Encode::decode( 'utf8', $args->{$k} ) ) . '&';
+	}
+
+	$url =~ s/&$//;
+	$url = API_URL . $method . $url;
+
+	my $cacheKey = $args->{_noCache} ? '' : md5_hex($url);
+
+	if ( $cacheKey && (my $cached = $cache->get($cacheKey)) ) {
+		main::INFOLOG && $log->info("Returning cached data for: $url");
+		$cb->($cached);
+		return;
+	}
+
+	main::INFOLOG && $log->info("Calling API (will cache for ", $args->{_cache_ttl} || DEFAULT_CACHE_TTL, "s): $url");
+
+	Slim::Networking::SimpleAsyncHTTP->new(
+		sub {
+			my $response = shift;
+			my $result = eval { from_json($response->content) };
+
+			if ($@) {
+				$log->error(Data::Dump::dump($response)) unless main::DEBUGLOG && $log->is_debug;
+				$log->error($@);
+			}
+
+			main::DEBUGLOG && $log->is_debug && warn Data::Dump::dump($result);
+
+			$result ||= {};
+			$cache->set($cacheKey, $result, $args->{_cache_ttl} || DEFAULT_CACHE_TTL);
+
+			$cb->($result);
+		},
+
+		sub {
+			warn Data::Dump::dump(@_);
+			$log->error($_[1]);
+			$cb->( { error => $_[1] } );
+		},
+
+		{
+			timeout => 15,
+		}
+
+	)->get($url);
+}
+
+1;