mctaylors/Cassette
mctaylors/Cassette
1
0
Fork 0
Code Issues Pull requests Activity

/play: Add playlist support

Browse source 
Signed-off-by: mctaylors <cantsendmails@mctaylors.ru>
This commit is contained in:
Macintxsh 2023-12-23 19:43:10 +03:00
parent 090d3e426d
commit b0e0d27b8c
Signed by: mctaylors
GPG key ID: 7181BEBE676903C1
1 changed files with 36 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
Cassette/Commands/ControlsCommandGroup.cs
View file

@ -5,6 +5,7 @@ using JetBrains.Annotations;
using Lavalink4NET;
using Lavalink4NET.Players.Queued;
using Lavalink4NET.Rest.Entities.Tracks;
using Lavalink4NET.Tracks;
using Remora.Commands.Attributes;
using Remora.Commands.Groups;
using Remora.Discord.API.Abstractions.Objects;
@ -26,7 +27,7 @@ public sealed class ControlsCommandGroup(
FeedbackService feedbackService) : CommandGroup
{
[Command("play")]
[Description("Plays or adds the requested track to the queue")]
[Description("Plays or adds the requested track/playlist to the queue")]
[RequireBotDiscordPermissions(DiscordPermission.Connect)]
[DiscordDefaultDMPermission(false)]
[UsedImplicitly]
@ -39,7 +40,23 @@ public sealed class ControlsCommandGroup(
return Result.FromSuccess();
}
var track = await audioService.Tracks.LoadTrackAsync(query, TrackSearchMode.YouTube);
var loadResult = await audioService.Tracks.LoadTracksAsync(query, TrackSearchMode.YouTube);
if (!loadResult.IsSuccess)
{
return await feedbackService.SendContextualMessageResult(
"Track loading error, try again later.", feedbackService.Theme.FaultOrDanger);
}
if (loadResult.IsPlaylist)
{
return await PlayListResultAsync(player, loadResult);
}
return await PlayTrackResultAsync(player, loadResult.Track);
}
private async Task<Result> PlayTrackResultAsync(IQueuedLavalinkPlayer player, LavalinkTrack? track)
{
if (track is null)
{
return await feedbackService.SendContextualMessageResult(
@ -64,6 +81,23 @@ public sealed class ControlsCommandGroup(
message.ToString(), feedbackService.Theme.Success);
}
private async Task<Result> PlayListResultAsync(IQueuedLavalinkPlayer player, TrackLoadResult loadResult)
{
var playlistInfo = loadResult.Playlist;
if (playlistInfo is null)
{
return Result.FromSuccess();
}
foreach (var track in loadResult.Tracks)
{
await player.PlayAsync(track);
}
return await feedbackService.SendContextualMessageResult(
$"Added {Markdown.Bold(playlistInfo.Name)} playlist to the queue", feedbackService.Theme.Success);
}
[Command("pause")]
[Description("Pauses the current track")]
[DiscordDefaultDMPermission(false)]