TomatenMusic-V2/TomatenMusicCore/Services/YoutubeService.cs

using Google.Apis.Services;
using Google.Apis.YouTube.v3;
using System;
using System.Collections.Generic;
using System.Text;
using System.Threading.Tasks;
using TomatenMusic.Music.Entitites;
using System.Linq;
using Google.Apis.YouTube.v3.Data;
using Microsoft.Extensions.Logging;
using static TomatenMusic.TomatenMusicBot;
using Lavalink4NET.Player;
using Microsoft.Extensions.DependencyInjection;
using Lavalink4NET;

namespace TomatenMusic.Services
{
    public class YoutubeService
    {
        public YouTubeService Service { get; }
        public ILogger<YoutubeService> _logger { get; set; }
        public YoutubeService(ILogger<YoutubeService> logger, ConfigJson config)
        {
            Service = new YouTubeService(new BaseClientService.Initializer()
            {
                ApiKey = config.YoutubeAPIKey,
                ApplicationName = "TomatenMusic"
            });
            _logger = logger;
        }

        public async Task<LavalinkTrack> PopulateTrackInfoAsync(LavalinkTrack track)
        {
            var video = await GetVideoAsync(track.TrackIdentifier);
            var channel = await GetChannelAsync(video.Snippet.ChannelId);
            FullTrackContext context = track.Context == null ? new FullTrackContext() : (FullTrackContext)track.Context;

            if (channel.Statistics.SubscriberCount != null)
                context.YoutubeAuthorSubs = (ulong) channel.Statistics.SubscriberCount;
            context.YoutubeAuthorThumbnail = new Uri(channel.Snippet.Thumbnails.High.Url);
            context.YoutubeAuthorUri = new Uri($"https://www.youtube.com/channel/{channel.Id}");
            context.YoutubeDescription = video.Snippet.Description;
            if (video.Statistics.LikeCount != null)
                context.YoutubeLikes = (ulong) video.Statistics.LikeCount;
            context.YoutubeTags = video.Snippet.Tags;
            context.YoutubeThumbnail = new Uri(video.Snippet.Thumbnails.High.Url);
            context.YoutubeUploadDate = (DateTime)video.Snippet.PublishedAt;
            context.YoutubeViews = (ulong)video.Statistics.ViewCount;
            context.YoutubeCommentCount = video.Statistics.CommentCount;
            track.Context = context;
            return track;
        }

        public async Task<List<LavalinkTrack>> PopulateMultiTrackListAsync(IEnumerable<LavalinkTrack> tracks)
        {
            List<LavalinkTrack> newTracks = new List<LavalinkTrack>();
            foreach (var track in tracks)
                newTracks.Add(await PopulateTrackInfoAsync(track));

            return newTracks;
        }
        public async Task<LavalinkPlaylist> PopulatePlaylistAsync(YoutubePlaylist playlist)
        {
            var list = await GetPlaylistAsync(playlist.Identifier);
            var channel = await GetChannelAsync(list.Snippet.ChannelId);

            string desc = list.Snippet.Description;

            playlist.Description = desc.Substring(0, Math.Min(desc.Length, 4092)) + (desc.Length > 4092 ? "..." : " ");
            playlist.Thumbnail = new Uri(list.Snippet.Thumbnails.High.Url);
            playlist.CreationTime = (DateTime)list.Snippet.PublishedAt;
            playlist.YoutubeItem = list;
            playlist.AuthorThumbnail = new Uri(channel.Snippet.Thumbnails.High.Url);
            playlist.AuthorUri = new Uri($"https://www.youtube.com/playlist?list={playlist.Identifier}");

            return playlist;
        }

        public async Task<Video> GetVideoAsync(string id)
        {
            var search = Service.Videos.List("contentDetails,id,liveStreamingDetails,localizations,player,recordingDetails,snippet,statistics,status,topicDetails");
            search.Id = id;
            var response = await search.ExecuteAsync();
            return response.Items.First();
        }

        public async Task<Channel> GetChannelAsync(string id)
        {
            var search = Service.Channels.List("brandingSettings,contentDetails,contentOwnerDetails,id,localizations,snippet,statistics,status,topicDetails");
            search.Id = id;
            var response = await search.ExecuteAsync();

            return response.Items.First();
        }
        public async Task<Playlist> GetPlaylistAsync(string id)
        {
            var search = Service.Playlists.List("snippet,contentDetails,status");
            search.Id = id;
            var response = await search.ExecuteAsync();

            return response.Items.First();
        }

        public async Task<SearchResult> GetRelatedVideoAsync(string id)
        {
            var search = Service.Search.List("snippet");
            search.RelatedToVideoId = id;
            search.Type = "video";
            var response = await search.ExecuteAsync();
            return response.Items.First(s => s.Snippet != null);
        }

        public async Task<LavalinkTrack> GetRelatedTrackAsync(string id)
        {
            var audioService = TomatenMusicBot.ServiceProvider.GetRequiredService<IAudioService>();

            var video = await GetRelatedVideoAsync(id);
            var loadResult = await audioService.GetTrackAsync($"https://youtu.be/{video.Id.VideoId}");

            if (loadResult == null)
                throw new Exception("An Error occurred while processing the Request");

            return await FullTrackContext.PopulateAsync(loadResult);
        }

    }
}