mirror of
https://github.com/Myxelium/Lunaris2.0.git
synced 2026-04-13 16:10:36 +00:00
Add jellyfin source
This commit is contained in:
@@ -6,6 +6,7 @@ using Lavalink4NET;
|
||||
using Lavalink4NET.Events.Players;
|
||||
using Lavalink4NET.Integrations.SponsorBlock;
|
||||
using Lavalink4NET.Integrations.SponsorBlock.Extensions;
|
||||
using Lavalink4NET.Jellyfin;
|
||||
using Lavalink4NET.Players.Queued;
|
||||
using Lavalink4NET.Rest.Entities.Tracks;
|
||||
using Lavalink4NET.Tracks;
|
||||
@@ -19,8 +20,7 @@ public class PlayHandler : IRequestHandler<PlayCommand>
|
||||
private readonly MusicEmbed _musicEmbed;
|
||||
private readonly DiscordSocketClient _client;
|
||||
private readonly IAudioService _audioService;
|
||||
private SocketSlashCommand _context;
|
||||
private const int MaxTrackDuration = 30;
|
||||
private SocketSlashCommand _context = default!; // ensure initialized before use
|
||||
private LavalinkTrack? _previousTrack;
|
||||
private static readonly HashSet<ulong> SubscribedGuilds = new();
|
||||
|
||||
@@ -34,15 +34,16 @@ public class PlayHandler : IRequestHandler<PlayCommand>
|
||||
_audioService = audioService;
|
||||
}
|
||||
|
||||
private async Task OnTrackEnded(object sender, TrackEndedEventArgs eventargs)
|
||||
private Task OnTrackEnded(object sender, TrackEndedEventArgs eventargs)
|
||||
{
|
||||
// Reset the previous track when the track ends.
|
||||
_previousTrack = null;
|
||||
return Task.CompletedTask;
|
||||
}
|
||||
|
||||
private async Task OnTrackStarted(object sender, TrackStartedEventArgs eventargs)
|
||||
{
|
||||
var player = sender as QueuedLavalinkPlayer;
|
||||
var player = eventargs.Player as QueuedLavalinkPlayer;
|
||||
|
||||
if (player?.CurrentTrack is null)
|
||||
{
|
||||
@@ -60,119 +61,132 @@ public class PlayHandler : IRequestHandler<PlayCommand>
|
||||
|
||||
// Track has changed, update the previous track and send the embed
|
||||
_previousTrack = currentTrack;
|
||||
await _musicEmbed.NowPlayingEmbed(currentTrack, _context, _client);
|
||||
if (_context != null)
|
||||
{
|
||||
await _musicEmbed.NowPlayingEmbed(currentTrack, _context, _client);
|
||||
}
|
||||
}
|
||||
|
||||
public Task Handle(PlayCommand command, CancellationToken cancellationToken)
|
||||
public async Task Handle(PlayCommand command, CancellationToken cancellationToken)
|
||||
{
|
||||
new Thread(PlayMusic).Start();
|
||||
return Task.CompletedTask;
|
||||
|
||||
async void PlayMusic()
|
||||
try
|
||||
{
|
||||
try
|
||||
var context = command.Message;
|
||||
_context = context;
|
||||
|
||||
if ((context.User as SocketGuildUser)?.VoiceChannel == null)
|
||||
{
|
||||
RegisterTrackStartedEventListerner(command);
|
||||
|
||||
await _audioService.StartAsync(cancellationToken);
|
||||
|
||||
var context = command.Message;
|
||||
_context = context;
|
||||
|
||||
if ((context.User as SocketGuildUser)?.VoiceChannel == null)
|
||||
await context.SendMessageAsync("You must be in a voice channel to use this command.", _client);
|
||||
return;
|
||||
}
|
||||
|
||||
var searchQuery = context.GetOptionValueByName(Option.Input);
|
||||
|
||||
if (string.IsNullOrWhiteSpace(searchQuery))
|
||||
{
|
||||
await context.SendMessageAsync("Please provide search terms.", _client);
|
||||
return;
|
||||
}
|
||||
|
||||
RegisterTrackStartedEventListerner(command);
|
||||
|
||||
await _audioService.StartAsync(cancellationToken);
|
||||
|
||||
await context.SendMessageAsync("📻 Searching...", _client);
|
||||
|
||||
var player = await _audioService.GetPlayerAsync(_client, context, connectToVoiceChannel: true);
|
||||
|
||||
if (player is null)
|
||||
return;
|
||||
|
||||
await ApplyFilters(cancellationToken, player);
|
||||
await ConfigureSponsorBlock(cancellationToken, player);
|
||||
|
||||
// Parse the query to extract search mode and clean query
|
||||
// Supports prefixes like jfsearch:, ytsearch:, scsearch:, etc.
|
||||
// Default: Jellyfin (jfsearch:) when no prefix is specified
|
||||
var (searchMode, queryToSearch) = SearchQueryParser.Parse(searchQuery, JellyfinSearchMode.Jellyfin);
|
||||
|
||||
var trackLoadOptions = new TrackLoadOptions
|
||||
{
|
||||
SearchMode = searchMode,
|
||||
};
|
||||
|
||||
var trackCollection = await _audioService.Tracks.LoadTracksAsync(queryToSearch, trackLoadOptions, cancellationToken: cancellationToken);
|
||||
|
||||
// Check if the result is a playlist or just a single track from the search result
|
||||
if (trackCollection.IsPlaylist)
|
||||
{
|
||||
// If it's a playlist, check if it's a free-text input.
|
||||
if (!Uri.IsWellFormedUriString(searchQuery, UriKind.Absolute))
|
||||
{
|
||||
await context.SendMessageAsync("You must be in a voice channel to use this command.", _client);
|
||||
return;
|
||||
}
|
||||
|
||||
var searchQuery = context.GetOptionValueByName(Option.Input);
|
||||
|
||||
if (string.IsNullOrWhiteSpace(searchQuery))
|
||||
{
|
||||
await context.SendMessageAsync("Please provide search terms.", _client);
|
||||
return;
|
||||
}
|
||||
|
||||
await context.SendMessageAsync("📻 Searching...", _client);
|
||||
|
||||
var player = await _audioService.GetPlayerAsync(_client, context, connectToVoiceChannel: true);
|
||||
|
||||
if (player is null)
|
||||
return;
|
||||
|
||||
await ApplyFilters(cancellationToken, player);
|
||||
await ConfigureSponsorBlock(cancellationToken, player);
|
||||
|
||||
var trackLoadOptions = new TrackLoadOptions
|
||||
{
|
||||
SearchMode = TrackSearchMode.YouTubeMusic,
|
||||
};
|
||||
|
||||
var trackCollection = await _audioService.Tracks.LoadTracksAsync(searchQuery, trackLoadOptions, cancellationToken: cancellationToken);
|
||||
|
||||
// Check if the result is a playlist or just a single track from the search result
|
||||
if (trackCollection.IsPlaylist)
|
||||
{
|
||||
// If it's a playlist, check if it's a free-text input.
|
||||
if (!Uri.IsWellFormedUriString(searchQuery, UriKind.Absolute))
|
||||
// Free text was used (not a direct URL to a playlist), let's prevent queueing the whole playlist.
|
||||
// Queue only the first track of the search result
|
||||
// var firstTrack = trackCollection.Tracks.FirstOrDefault();
|
||||
if (trackCollection.Track != null)
|
||||
{
|
||||
// Free text was used (not a direct URL to a playlist), let's prevent queueing the whole playlist.
|
||||
// Queue only the first track of the search result
|
||||
// var firstTrack = trackCollection.Tracks.FirstOrDefault();
|
||||
if (trackCollection.Track != null)
|
||||
{
|
||||
await player.PlayAsync(trackCollection.Track, cancellationToken: cancellationToken).ConfigureAwait(false);
|
||||
await _musicEmbed.NowPlayingEmbed(trackCollection.Track, _context, _client);
|
||||
}
|
||||
else
|
||||
{
|
||||
await context.SendMessageAsync("No tracks found.", _client);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
// It's a playlist and a URL was used, so we can queue all tracks as usual
|
||||
var queueTracks = trackCollection.Tracks
|
||||
.Skip(1)
|
||||
.Select(t => new TrackQueueItem(t))
|
||||
.ToList();
|
||||
|
||||
await player.PlayAsync(trackCollection.Tracks.First(), cancellationToken: cancellationToken).ConfigureAwait(false);
|
||||
await player.Queue.AddRangeAsync(queueTracks, cancellationToken);
|
||||
await context.SendMessageAsync($"Queued playlist {trackCollection.Playlist?.Name} with {queueTracks.Count} tracks.", _client);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
// It's just a single track or a search result.
|
||||
var track = trackCollection.Track;
|
||||
|
||||
if (track != null)
|
||||
{
|
||||
await player.PlayAsync(track, cancellationToken: cancellationToken).ConfigureAwait(false);
|
||||
await _musicEmbed.NowPlayingEmbed(track, _context, _client);
|
||||
await player.PlayAsync(trackCollection.Track, cancellationToken: cancellationToken).ConfigureAwait(false);
|
||||
// rely on TrackStarted event to send Now Playing
|
||||
}
|
||||
else
|
||||
{
|
||||
await context.SendMessageAsync("No tracks found.", _client);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
// It's a playlist and a URL was used, so we can queue all tracks as usual
|
||||
var queueTracks = trackCollection.Tracks
|
||||
.Skip(1)
|
||||
.Select(t => new TrackQueueItem(t))
|
||||
.ToList();
|
||||
|
||||
await player.PlayAsync(trackCollection.Tracks.First(), cancellationToken: cancellationToken).ConfigureAwait(false);
|
||||
await player.Queue.AddRangeAsync(queueTracks, cancellationToken);
|
||||
await context.SendMessageAsync($"Queued playlist {trackCollection.Playlist?.Name} with {queueTracks.Count} tracks.", _client);
|
||||
}
|
||||
}
|
||||
catch (Exception error)
|
||||
else
|
||||
{
|
||||
throw new Exception("Error occured in the Play handler!", error);
|
||||
// It's just a single track or a search result.
|
||||
var track = trackCollection.Track;
|
||||
|
||||
if (track != null)
|
||||
{
|
||||
await player.PlayAsync(track, cancellationToken: cancellationToken).ConfigureAwait(false);
|
||||
// rely on TrackStarted event to send Now Playing
|
||||
}
|
||||
else
|
||||
{
|
||||
await context.SendMessageAsync("No tracks found.", _client);
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (Exception error)
|
||||
{
|
||||
throw new Exception("Error occured in the Play handler!", error);
|
||||
}
|
||||
}
|
||||
|
||||
private void RegisterTrackStartedEventListerner(PlayCommand command)
|
||||
{
|
||||
if (SubscribedGuilds.Contains((ulong)command.Message.GuildId!))
|
||||
var guildId = command.Message.GuildId;
|
||||
if (!guildId.HasValue)
|
||||
{
|
||||
// Ignore registration for DMs or contexts without a guild.
|
||||
return;
|
||||
|
||||
}
|
||||
|
||||
var gid = guildId.Value;
|
||||
if (SubscribedGuilds.Contains(gid))
|
||||
return;
|
||||
|
||||
// Ensure we don't accidentally double-subscribe.
|
||||
_audioService.TrackStarted -= OnTrackStarted;
|
||||
_audioService.TrackEnded -= OnTrackEnded;
|
||||
_audioService.TrackStarted += OnTrackStarted;
|
||||
_audioService.TrackEnded += OnTrackEnded;
|
||||
SubscribedGuilds.Add((ulong)command.Message.GuildId!);
|
||||
SubscribedGuilds.Add(gid);
|
||||
}
|
||||
|
||||
private static async Task ApplyFilters(CancellationToken cancellationToken, QueuedLavalinkPlayer player)
|
||||
@@ -193,4 +207,4 @@ public class PlayHandler : IRequestHandler<PlayCommand>
|
||||
|
||||
await player.UpdateSponsorBlockCategoriesAsync(categories, cancellationToken: cancellationToken);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user