Fortunevale/ProjectMakoto.Plugins.Music
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Initial commit

Browse source
This commit is contained in:
Mira 2025-01-27 18:36:22 +01:00
commit 79e7cffb4e
Signed by untrusted user who does not match committer: Xorog
GPG key ID: 983798ED9C3E7C36
49 changed files with 6399 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

192
Commands/MusicModuleAbstractions.cs Normal file
View file

@ -0,0 +1,192 @@
// Project Makoto
// Copyright (C) 2024 Fortunevale
// This program 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.
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY
using DisCatSharp.Lavalink.Entities;
using DisCatSharp.Lavalink.Enums;
using Xorog.UniversalExtensions;
using Xorog.UniversalExtensions.Enums;
namespace ProjectMakoto.Plugins.Music;
internal static class MusicModuleAbstractions
{
public static async Task<(List<LavalinkTrack> Tracks, LavalinkTrackLoadingResult oriResult, bool Continue)> GetLoadResult(SharedCommandContext ctx, string searchQuery)
{
var CommandKey = ((Entities.Translations)MusicPlugin.Plugin!.Translations).Commands.Music.Play;
if (Regex.IsMatch(searchQuery, "{jndi:(ldap[s]?|rmi):\\/\\/[^\n]+") || searchQuery.Contains("localhost", StringComparison.OrdinalIgnoreCase) || searchQuery.Contains("127.0.0.1", StringComparison.OrdinalIgnoreCase))
throw new Exception();
List<LavalinkTrack> Tracks = new();
var lava = ctx.Client.GetLavalink();
var session = lava.ConnectedSessions.Values.First(x => x.IsConnected);
var embed = new DiscordEmbedBuilder(ctx.ResponseMessage.Embeds[0]);
_ = await ctx.BaseCommand.RespondOrEdit(embed.WithDescription(CommandKey.LookingFor.Get(ctx.DbUser).Build(true, new TVar("Search", searchQuery))).AsLoading(ctx));
LavalinkTrackLoadingResult loadResult;
if (RegexTemplates.YouTubeUrl.IsMatch(searchQuery))
{
if (Regex.IsMatch(searchQuery, @"((\?|&)list=RDMM\w+)(&*)"))
{
Group group = Regex.Match(searchQuery, @"((\?|&)list=RDMM\w+)(&*)", RegexOptions.ExplicitCapture);
var value = group.Value;
if (value.EndsWith('&'))
value = value[..^1];
searchQuery = searchQuery.Replace(value, "");
}
if (Regex.IsMatch(searchQuery, @"((\?|&)start_radio=\d+)(&*)"))
{
Group group = Regex.Match(searchQuery, @"((\?|&)start_radio=\d+)(&*)", RegexOptions.ExplicitCapture);
var value = group.Value;
if (value.EndsWith('&'))
value = value[..^1];
searchQuery = searchQuery.Replace(value, "");
}
var AndIndex = searchQuery.IndexOf('&');
if (!searchQuery.Contains('?') && AndIndex != -1)
{
searchQuery = searchQuery.Remove(AndIndex, 1);
searchQuery = searchQuery.Insert(AndIndex, "?");
}
loadResult = await session.LoadTracksAsync(LavalinkSearchType.Plain, searchQuery);
}
else if (RegexTemplates.SoundcloudUrl.IsMatch(searchQuery))
{
loadResult = await session.LoadTracksAsync(LavalinkSearchType.Plain, searchQuery);
}
else if (RegexTemplates.BandcampUrl.IsMatch(searchQuery))
{
loadResult = await session.LoadTracksAsync(LavalinkSearchType.Plain, searchQuery);
}
else if (RegexTemplates.SpotifyUrl.IsMatch(searchQuery))
{
loadResult = await session.LoadTracksAsync(LavalinkSearchType.Plain, searchQuery);
}
else
{
embed.Description = CommandKey.PlatformSelect.Get(ctx.DbUser).Build(true);
_ = embed.AsAwaitingInput(ctx);
var YouTube = new DiscordButtonComponent(ButtonStyle.Primary, Guid.NewGuid().ToString(), "YouTube", false, EmojiTemplates.GetYouTube(ctx.Bot).ToComponent());
var SoundCloud = new DiscordButtonComponent(ButtonStyle.Primary, Guid.NewGuid().ToString(), "Soundcloud", false, EmojiTemplates.GetSoundcloud(ctx.Bot).ToComponent());
var Spotify = new DiscordButtonComponent(ButtonStyle.Primary, Guid.NewGuid().ToString(), "Spotify", false, EmojiTemplates.GetSpotify(ctx.Bot).ToComponent());
_ = await ctx.BaseCommand.RespondOrEdit(
new DiscordMessageBuilder().WithEmbed(embed)
.AddComponents(new List<DiscordComponent> { YouTube, SoundCloud, Spotify }));
var Menu1 = await ctx.WaitForButtonAsync(TimeSpan.FromMinutes(2));
if (Menu1.TimedOut)
{
ctx.BaseCommand.ModifyToTimedOut();
return (null, null, false);
}
_ = Menu1.Result.Interaction.CreateResponseAsync(InteractionResponseType.DeferredMessageUpdate);
var platformString = string.Empty;
var searchType = LavalinkSearchType.Plain;
if (Menu1.GetCustomId() == YouTube.CustomId)
{
platformString = "YouTube";
searchType = LavalinkSearchType.Youtube;
}
else if (Menu1.GetCustomId() == SoundCloud.CustomId)
{
platformString = "SoundCloud";
searchType = LavalinkSearchType.SoundCloud;
}
else if (Menu1.GetCustomId() == Spotify.CustomId)
{
platformString = "Spotify";
searchType = LavalinkSearchType.Spotify;
}
_ = await ctx.BaseCommand.RespondOrEdit(embed.WithDescription(CommandKey.LookingFor.Get(ctx.DbUser).Build(true,
new TVar("Search", searchQuery),
new TVar("Platform", platformString))).AsLoading(ctx));
loadResult = await session.LoadTracksAsync(searchType, searchQuery);
}
if (loadResult.LoadType == LavalinkLoadResultType.Error)
{
MusicPlugin.Plugin._logger.LogError("An exception occurred while trying to load lavalink track.");
embed.Description = CommandKey.FailedToLoad.Get(ctx.DbUser).Build(true,
new TVar("Search", searchQuery));
_ = embed.AsError(ctx);
_ = await ctx.BaseCommand.RespondOrEdit(embed.Build());
return (null, loadResult, false);
}
else if (loadResult.LoadType == LavalinkLoadResultType.Empty)
{
embed.Description = CommandKey.NoMatches.Get(ctx.DbUser).Build(true,
new TVar("Search", searchQuery));
_ = embed.AsError(ctx);
_ = await ctx.BaseCommand.RespondOrEdit(embed.Build());
return (null, loadResult, false);
}
else if (loadResult.LoadType == LavalinkLoadResultType.Playlist)
{
return (loadResult.GetResultAs<LavalinkPlaylist>().Tracks.ToList(), loadResult, true);
}
else if (loadResult.LoadType == LavalinkLoadResultType.Track)
{
Tracks.Add(loadResult.GetResultAs<LavalinkTrack>());
return (Tracks, loadResult, true);
}
else if (loadResult.LoadType == LavalinkLoadResultType.Search)
{
var searchResults = loadResult.GetResultAs<List<LavalinkTrack>>();
embed.Description = CommandKey.SearchSuccess.Get(ctx.DbUser).Build(true,
new TVar("Count", searchResults.Count));
_ = embed.AsAwaitingInput(ctx);
_ = await ctx.BaseCommand.RespondOrEdit(embed.Build());
var UriResult = await ctx.BaseCommand.PromptCustomSelection(searchResults
.Select(x => new DiscordStringSelectComponentOption(x.Info.Title.TruncateWithIndication(100), x.Info.Uri.ToString(), $"🔼 {x.Info.Author} | 🕒 {x.Info.Length.GetHumanReadable(TimeFormat.Minutes)}")).ToList());
if (UriResult.TimedOut)
{
ctx.BaseCommand.ModifyToTimedOut();
return (null, loadResult, false);
}
else if (UriResult.Cancelled)
{
return (null, loadResult, false);
}
else if (UriResult.Errored)
{
throw UriResult.Exception;
}
Tracks.Add(searchResults.First(x => x.Info.Uri.ToString() == UriResult.Result));
return (Tracks, loadResult, true);
}
else
{
throw new Exception($"Unknown Load Result Type: {loadResult.LoadType}");
}
}
}