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,11 +6,14 @@ namespace Lunaris2.Service
|
||||
public class VoiceChannelMonitorService
|
||||
{
|
||||
private readonly DiscordSocketClient _client;
|
||||
private readonly Dictionary<ulong, Timer> _timers = new();
|
||||
// Track a cancellation source per voice channel when the bot is alone
|
||||
private readonly Dictionary<ulong, CancellationTokenSource> _leaveCtsByChannel = new();
|
||||
|
||||
public VoiceChannelMonitorService(DiscordSocketClient client)
|
||||
{
|
||||
_client = client;
|
||||
// Subscribe to voice state updates to react immediately
|
||||
_client.UserVoiceStateUpdated += OnUserVoiceStateUpdated;
|
||||
}
|
||||
|
||||
public void StartMonitoring()
|
||||
@@ -20,7 +23,7 @@ namespace Lunaris2.Service
|
||||
while (true)
|
||||
{
|
||||
await CheckVoiceChannels();
|
||||
await Task.Delay(TimeSpan.FromMinutes(1)); // Monitor every minute
|
||||
await Task.Delay(TimeSpan.FromMinutes(1)); // Status refresh every minute
|
||||
}
|
||||
});
|
||||
}
|
||||
@@ -28,7 +31,7 @@ namespace Lunaris2.Service
|
||||
private async Task CheckVoiceChannels()
|
||||
{
|
||||
SetStatus();
|
||||
await LeaveOnAlone();
|
||||
await EnsureCurrentAloneStatesScheduled();
|
||||
}
|
||||
|
||||
private void SetStatus()
|
||||
@@ -49,48 +52,137 @@ namespace Lunaris2.Service
|
||||
_client.SetGameAsync($" in {channels} servers", type: ActivityType.Playing);
|
||||
}
|
||||
|
||||
private async Task LeaveOnAlone()
|
||||
// Monitor existing alone states during the periodic check to ensure timers exist
|
||||
private async Task EnsureCurrentAloneStatesScheduled()
|
||||
{
|
||||
foreach (var guild in _client.Guilds)
|
||||
{
|
||||
// Find voice channels where only the bot is left
|
||||
var voiceChannel = guild.VoiceChannels.FirstOrDefault(vc =>
|
||||
vc.ConnectedUsers.Count == 1 &&
|
||||
vc.Users.Any(u => u.Id == _client.CurrentUser.Id));
|
||||
foreach (var voiceChannel in guild.VoiceChannels)
|
||||
{
|
||||
var botInChannel = voiceChannel.ConnectedUsers.Any(u => u.Id == _client.CurrentUser.Id);
|
||||
var userCount = voiceChannel.ConnectedUsers.Count;
|
||||
|
||||
if (voiceChannel != null)
|
||||
{
|
||||
// If timer not set for this channel, start one
|
||||
if (!_timers.ContainsKey(voiceChannel.Id))
|
||||
if (botInChannel && userCount == 1)
|
||||
{
|
||||
Console.WriteLine($"Bot is alone in channel {voiceChannel.Name}, starting timer...");
|
||||
_timers[voiceChannel.Id] = new Timer(async _ => await LeaveChannel(voiceChannel), null,
|
||||
TimeSpan.FromMinutes(3), Timeout.InfiniteTimeSpan); // Set delay before leaving
|
||||
// Schedule leave if not already scheduled
|
||||
if (!_leaveCtsByChannel.ContainsKey(voiceChannel.Id))
|
||||
{
|
||||
ScheduleLeave(voiceChannel);
|
||||
}
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
// Clean up timer if channel is no longer active
|
||||
var timersToDispose = _timers.Where(t => guild.VoiceChannels.All(vc => vc.Id != t.Key)).ToList();
|
||||
foreach (var timer in timersToDispose)
|
||||
else
|
||||
{
|
||||
await timer.Value.DisposeAsync();
|
||||
_timers.Remove(timer.Key);
|
||||
Console.WriteLine($"Disposed timer for inactive voice channel ID: {timer.Key}");
|
||||
// Cancel if a schedule exists but the bot is not alone anymore
|
||||
if (_leaveCtsByChannel.TryGetValue(voiceChannel.Id, out var cts))
|
||||
{
|
||||
cts.Cancel();
|
||||
_leaveCtsByChannel.Remove(voiceChannel.Id);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
await Task.CompletedTask;
|
||||
}
|
||||
|
||||
private async Task LeaveChannel(SocketVoiceChannel voiceChannel)
|
||||
private Task OnUserVoiceStateUpdated(SocketUser user, SocketVoiceState before, SocketVoiceState after)
|
||||
{
|
||||
if (voiceChannel.ConnectedUsers.Count == 1 && voiceChannel.Users.Any(u => u.Id == _client.CurrentUser.Id))
|
||||
// React only when events relate to the guild(s) and voice channels where the bot might be
|
||||
var botId = _client.CurrentUser.Id;
|
||||
|
||||
// Determine affected channels
|
||||
var beforeChannelId = before.VoiceChannel?.Id;
|
||||
var afterChannelId = after.VoiceChannel?.Id;
|
||||
|
||||
// If the bot itself moved, we should cancel any old schedule and possibly set a new one
|
||||
if (user.Id == botId)
|
||||
{
|
||||
Console.WriteLine($"Leaving channel {voiceChannel.Name} due to inactivity...");
|
||||
await voiceChannel.DisconnectAsync();
|
||||
await _timers[voiceChannel.Id].DisposeAsync();
|
||||
_timers.Remove(voiceChannel.Id); // Clean up after leaving
|
||||
if (beforeChannelId.HasValue && _leaveCtsByChannel.TryGetValue(beforeChannelId.Value, out var oldCts))
|
||||
{
|
||||
oldCts.Cancel();
|
||||
_leaveCtsByChannel.Remove(beforeChannelId.Value);
|
||||
}
|
||||
|
||||
if (afterChannelId.HasValue)
|
||||
{
|
||||
var channel = after.VoiceChannel!;
|
||||
var isAlone = channel.ConnectedUsers.Count == 1 && channel.ConnectedUsers.Any(u => u.Id == botId);
|
||||
if (isAlone && !_leaveCtsByChannel.ContainsKey(channel.Id))
|
||||
{
|
||||
ScheduleLeave(channel);
|
||||
}
|
||||
}
|
||||
|
||||
return Task.CompletedTask;
|
||||
}
|
||||
|
||||
// For other users, if they join the bot's channel, cancel the leave; if they leave and bot becomes alone, schedule leave
|
||||
if (afterChannelId.HasValue)
|
||||
{
|
||||
var channel = after.VoiceChannel!;
|
||||
var botInChannel = channel.ConnectedUsers.Any(u => u.Id == botId);
|
||||
var userCount = channel.ConnectedUsers.Count;
|
||||
|
||||
if (botInChannel && userCount > 1)
|
||||
{
|
||||
// Cancel any pending leave
|
||||
if (_leaveCtsByChannel.TryGetValue(channel.Id, out var cts))
|
||||
{
|
||||
cts.Cancel();
|
||||
_leaveCtsByChannel.Remove(channel.Id);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (beforeChannelId.HasValue)
|
||||
{
|
||||
var channel = before.VoiceChannel!; // user left this channel
|
||||
var botInChannel = channel.ConnectedUsers.Any(u => u.Id == botId);
|
||||
var userCount = channel.ConnectedUsers.Count;
|
||||
|
||||
if (botInChannel && userCount == 1)
|
||||
{
|
||||
// Bot became alone, schedule leave
|
||||
if (!_leaveCtsByChannel.ContainsKey(channel.Id))
|
||||
{
|
||||
ScheduleLeave(channel);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return Task.CompletedTask;
|
||||
}
|
||||
|
||||
private void ScheduleLeave(SocketVoiceChannel voiceChannel)
|
||||
{
|
||||
var cts = new CancellationTokenSource();
|
||||
_leaveCtsByChannel[voiceChannel.Id] = cts;
|
||||
|
||||
_ = Task.Run(async () =>
|
||||
{
|
||||
try
|
||||
{
|
||||
await Task.Delay(TimeSpan.FromMinutes(3), cts.Token);
|
||||
|
||||
// After delay, verify still alone
|
||||
var botId = _client.CurrentUser.Id;
|
||||
var isStillAlone = voiceChannel.ConnectedUsers.Count == 1 && voiceChannel.ConnectedUsers.Any(u => u.Id == botId);
|
||||
if (isStillAlone)
|
||||
{
|
||||
Console.WriteLine($"Leaving channel {voiceChannel.Name} due to inactivity...");
|
||||
await voiceChannel.DisconnectAsync();
|
||||
}
|
||||
}
|
||||
catch (OperationCanceledException)
|
||||
{
|
||||
// Cancelled because someone joined or bot moved
|
||||
}
|
||||
finally
|
||||
{
|
||||
_leaveCtsByChannel.Remove(voiceChannel.Id);
|
||||
cts.Dispose();
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user