using System.Runtime.Caching; using System.Text.RegularExpressions; using Humanizer; using KfChatDotNetBot.Models; using KfChatDotNetBot.Models.DbModels; using KfChatDotNetBot.Settings; using KfChatDotNetWsClient.Models.Events; using Microsoft.EntityFrameworkCore; using Newtonsoft.Json; namespace KfChatDotNetBot.Commands; public class SetRoleCommand : ICommand { public List Patterns => [ new Regex(@"^admin role set (?\d+) (?\d+)$"), new Regex(@"^admin right set (?\d+) (?\d+)$") ]; public string? HelpText => "Set a user's role"; public UserRight RequiredRight => UserRight.Admin; public TimeSpan Timeout => TimeSpan.FromSeconds(10); public async Task RunCommand(ChatBot botInstance, MessageModel message, UserDbModel user, GroupCollection arguments, CancellationToken ctx) { await using var db = new ApplicationDbContext(); var targetUserId = Convert.ToInt32(arguments["user"].Value); var role = (UserRight)Convert.ToInt32(arguments["role"].Value); var targetUser = await db.Users.FirstOrDefaultAsync(u => u.KfId == targetUserId, cancellationToken: ctx); if (targetUser == null) { await botInstance.SendChatMessageAsync($"User '{targetUserId}' does not exist", true); return; } targetUser.UserRight = role; await db.SaveChangesAsync(ctx); await botInstance.SendChatMessageAsync($"@{message.Author.Username}, {targetUser.KfUsername}'s role set to {role.Humanize()}", true); } } public class CacheClearAdminCommand : ICommand { public List Patterns => [ new Regex("^admin cache clear$") ]; public string? HelpText => null; public UserRight RequiredRight => UserRight.Admin; public TimeSpan Timeout => TimeSpan.FromSeconds(10); public async Task RunCommand(ChatBot botInstance, MessageModel message, UserDbModel user, GroupCollection arguments, CancellationToken ctx) { var cacheKeys = MemoryCache.Default.Select(kvp => kvp.Key).ToList(); foreach (var cacheKey in cacheKeys) { MemoryCache.Default.Remove(cacheKey); } await botInstance.SendChatMessageAsync("Cache wiped", true); } } public class NewKickChannelCommand : ICommand { public List Patterns => [ new Regex(@"^admin kick add (?\d+) (?\d+) (?\S+)$"), new Regex(@"^admin kick add (?\d+) (?\d+) (?\S+) (?true|false)$") ]; public string? HelpText => "Add a Kick channel to the bot's database"; public UserRight RequiredRight => UserRight.Admin; public TimeSpan Timeout => TimeSpan.FromSeconds(10); public async Task RunCommand(ChatBot botInstance, MessageModel message, UserDbModel user, GroupCollection arguments, CancellationToken ctx) { var autoCapture = false; if (arguments.TryGetValue("auto_capture", out var argument)) { autoCapture = argument.Value == "true"; } await using var db = new ApplicationDbContext(); var url = $"https://kick.com/{arguments["slug"].Value}"; if (await db.Streams.AnyAsync(s => s.StreamUrl == url, cancellationToken: ctx)) { await botInstance.SendChatMessageAsync("Channel is already in the database", true); return; } var forumUser = await db.Users.FirstOrDefaultAsync(u => u.KfId == Convert.ToInt32(arguments["forum_id"].Value), cancellationToken: ctx); var meta = JsonConvert.SerializeObject(new KickStreamMetaModel { ChannelId = Convert.ToInt32(arguments["channel_id"].Value) }); db.Streams.Add(new StreamDbModel { Service = StreamService.Kick, User = forumUser, Metadata = meta, StreamUrl = url, AutoCapture = autoCapture }); await db.SaveChangesAsync(ctx); await botInstance.SendChatMessageAsync("Updated list of channels", true); } } public class RemoveStreamChannelCommand : ICommand { public List Patterns => [ new Regex(@"^admin stream remove (?\d+)$") ]; public string? HelpText => "Remove a Kick channel from the bot's database"; public UserRight RequiredRight => UserRight.Admin; public TimeSpan Timeout => TimeSpan.FromSeconds(10); public async Task RunCommand(ChatBot botInstance, MessageModel message, UserDbModel user, GroupCollection arguments, CancellationToken ctx) { await using var db = new ApplicationDbContext(); var rowId = Convert.ToInt32(arguments["id"].Value); var channel = db.Streams.FirstOrDefault(ch => ch.Id == rowId); if (channel == null) { await botInstance.SendChatMessageAsync("Could not find this row in the database", true); return; } db.Streams.Remove(channel); await db.SaveChangesAsync(ctx); await botInstance.SendChatMessageAsync("Updated list of streams", true); } } public class ReconnectKickCommand : ICommand { public List Patterns => [ new Regex(@"^admin kick reconnect$") ]; public string? HelpText => "Disconnect from Kick so the watchdog can reconnect it"; public UserRight RequiredRight => UserRight.Admin; public TimeSpan Timeout => TimeSpan.FromSeconds(10); public async Task RunCommand(ChatBot botInstance, MessageModel message, UserDbModel user, GroupCollection arguments, CancellationToken ctx) { if (botInstance.BotServices.KickClient == null) { await botInstance.SendChatMessageAsync("Kick client is not initialized", true); return; } botInstance.BotServices.KickClient.Disconnect(); await botInstance.SendChatMessageAsync("Disconnected from Kick. Client should reconnect shortly.", true); } } public class NewPartiChannelCommand : ICommand { public List Patterns => [ new Regex(@"^admin parti add (?\d+) (?\S+) (?\S+) (?true|false)$"), new Regex(@"^admin parti add (?\d+) (?\S+) (?\S+)$") ]; public string? HelpText => "Add a Parti channel to the bot's database"; public UserRight RequiredRight => UserRight.Admin; public TimeSpan Timeout => TimeSpan.FromSeconds(10); public async Task RunCommand(ChatBot botInstance, MessageModel message, UserDbModel user, GroupCollection arguments, CancellationToken ctx) { var autoCapture = false; if (arguments.TryGetValue("auto_capture", out var argument)) { autoCapture = argument.Value == "true"; } await using var db = new ApplicationDbContext(); var url = $"https://parti.com/creator/{arguments["social"].Value}/{arguments["username"].Value}/"; if (arguments["social"].Value == "discord") { url += "0"; } if (await db.Streams.AnyAsync(s => s.StreamUrl == url, cancellationToken: ctx)) { await botInstance.SendChatMessageAsync("Channel is already in the database", true); return; } var forumUser = await db.Users.FirstOrDefaultAsync(u => u.KfId == Convert.ToInt32(arguments["forum_id"].Value), cancellationToken: ctx); db.Streams.Add(new StreamDbModel { Service = StreamService.Parti, User = forumUser, StreamUrl = url, AutoCapture = autoCapture }); await db.SaveChangesAsync(ctx); await botInstance.SendChatMessageAsync("Updated list of channels", true); } } public class NewDLiveChannelCommand : ICommand { public List Patterns => [ new Regex(@"^admin dlive add (?\d+) (?\S+) (?true|false)$"), new Regex(@"^admin dlive add (?\d+) (?\S+)$") ]; public string? HelpText => "Add a DLive channel to the bot's database"; public UserRight RequiredRight => UserRight.Admin; public TimeSpan Timeout => TimeSpan.FromSeconds(10); public async Task RunCommand(ChatBot botInstance, MessageModel message, UserDbModel user, GroupCollection arguments, CancellationToken ctx) { var autoCapture = false; if (arguments.TryGetValue("auto_capture", out var argument)) { autoCapture = argument.Value == "true"; } await using var db = new ApplicationDbContext(); var url = $"https://dlive.tv/{arguments["username"].Value}"; if (await db.Streams.AnyAsync(s => s.StreamUrl == url, cancellationToken: ctx)) { await botInstance.SendChatMessageAsync("Channel is already in the database", true); return; } var forumUser = await db.Users.FirstOrDefaultAsync(u => u.KfId == Convert.ToInt32(arguments["forum_id"].Value), cancellationToken: ctx); db.Streams.Add(new StreamDbModel { Service = StreamService.DLive, User = forumUser, StreamUrl = url, AutoCapture = autoCapture }); await db.SaveChangesAsync(ctx); await botInstance.SendChatMessageAsync("Updated list of channels", true); } } public class AddCourtHearingCommand : ICommand { public List Patterns => [ new Regex(@"^admin hearing add (?\S+) (?\S+) (?.+)$") ]; public string? HelpText => "Add a court hearing to the bot's calendar"; public UserRight RequiredRight => UserRight.TrueAndHonest; public TimeSpan Timeout => TimeSpan.FromSeconds(10); public async Task RunCommand(ChatBot botInstance, MessageModel message, UserDbModel user, GroupCollection arguments, CancellationToken ctx) { var hearings = (await SettingsProvider.GetValueAsync(BuiltIn.Keys.BotCourtCalendar)).JsonDeserialize>(); if (hearings == null) { await botInstance.SendChatMessageAsync("Hearings list was null", true); return; } var caseNumber = arguments["case"].Value; if (!DateTimeOffset.TryParse(arguments["date"].Value, out var date)) { await botInstance.SendChatMessageAsync("Failed to parse date", true); return; } hearings.Add(new CourtHearingModel {CaseNumber = caseNumber, Description = arguments["description"].Value, Time = date}); await SettingsProvider.SetValueAsJsonObjectAsync(BuiltIn.Keys.BotCourtCalendar, hearings); await botInstance.SendChatMessageAsync("Updated list of hearings", true); } } public class RemoveCourtHearingCommand : ICommand { public List Patterns => [ new Regex(@"^admin hearing remove (?\d+)$") ]; public string? HelpText => "Remove a hearing from the bot's calendar"; public UserRight RequiredRight => UserRight.TrueAndHonest; public TimeSpan Timeout => TimeSpan.FromSeconds(10); public async Task RunCommand(ChatBot botInstance, MessageModel message, UserDbModel user, GroupCollection arguments, CancellationToken ctx) { var hearings = (await SettingsProvider.GetValueAsync(BuiltIn.Keys.BotCourtCalendar)).JsonDeserialize>(); if (hearings == null) { await botInstance.SendChatMessageAsync("Hearings list was null", true); return; } var hearingIndex = Convert.ToInt32(arguments["index"].Value); if (hearings.Count < hearingIndex) { await botInstance.SendChatMessageAsync( $"Index supplied is out of range. There are only {hearings.Count} hearings in the database", true); return; } hearings.RemoveAt(hearingIndex - 1); await SettingsProvider.SetValueAsJsonObjectAsync(BuiltIn.Keys.BotCourtCalendar, hearings); await botInstance.SendChatMessageAsync("Updated list of hearings", true); } } public class DeleteMessagesCommand : ICommand { public List Patterns => [ new Regex(@"^admin delete (?\d+)$") ]; public string? HelpText => "Delete the most recent x number of messages"; public UserRight RequiredRight => UserRight.TrueAndHonest; public TimeSpan Timeout => TimeSpan.FromSeconds(10); public async Task RunCommand(ChatBot botInstance, MessageModel message, UserDbModel user, GroupCollection arguments, CancellationToken ctx) { var amount = int.Parse(arguments["msg_count"].Value); if (amount > 10) { await botInstance.SendChatMessageAsync("More than 10 messages seems like a bit much?", true); return; } var messages = botInstance.SentMessages.Where(msg => msg.Status == SentMessageTrackerStatus.ResponseReceived) .TakeLast(amount); foreach (var msg in messages) { if (msg.ChatMessageId == null) { continue; } await botInstance.KfClient.DeleteMessageAsync(msg.ChatMessageId.Value); } } } public class IgnoreCommand : ICommand { public List Patterns => [ new Regex(@"^admin ignore (?\d+)$") ]; public string? HelpText => "Ignore a user by ID"; public UserRight RequiredRight => UserRight.TrueAndHonest; public TimeSpan Timeout => TimeSpan.FromSeconds(10); public async Task RunCommand(ChatBot botInstance, MessageModel message, UserDbModel user, GroupCollection arguments, CancellationToken ctx) { await using var db = new ApplicationDbContext(); var targetUser = await db.Users.FirstOrDefaultAsync(u => u.KfId == int.Parse(arguments["user_id"].Value), cancellationToken: ctx); if (targetUser == null) { await botInstance.SendChatMessageAsync("Can't find user", true); return; } if (targetUser.Ignored) { await botInstance.SendChatMessageAsync($"User {targetUser.KfUsername} has already been ignored", true); return; } targetUser.Ignored = true; await db.SaveChangesAsync(ctx); await botInstance.SendChatMessageAsync($"Now ignoring {targetUser.KfUsername}", true); } } public class UnignoreCommand : ICommand { public List Patterns => [ new Regex(@"^admin unignore (?\d+)$") ]; public string? HelpText => "Unignore a user by ID"; public UserRight RequiredRight => UserRight.TrueAndHonest; public TimeSpan Timeout => TimeSpan.FromSeconds(10); public async Task RunCommand(ChatBot botInstance, MessageModel message, UserDbModel user, GroupCollection arguments, CancellationToken ctx) { await using var db = new ApplicationDbContext(); var targetUser = await db.Users.FirstOrDefaultAsync(u => u.KfId == int.Parse(arguments["user_id"].Value), cancellationToken: ctx); if (targetUser == null) { await botInstance.SendChatMessageAsync("Can't find user", true); return; } if (!targetUser.Ignored) { await botInstance.SendChatMessageAsync($"User {targetUser.KfUsername} is not ignored", true); return; } targetUser.Ignored = false; await db.SaveChangesAsync(ctx); await botInstance.SendChatMessageAsync($"No longer ignoring {targetUser.KfUsername}", true); } } public class SetAlmanacTextCommand : ICommand { public List Patterns => [ new Regex("^admin almanac set text (?.+)$") ]; public string? HelpText => "Set the almanac text to whatever"; public UserRight RequiredRight => UserRight.TrueAndHonest; public TimeSpan Timeout => TimeSpan.FromSeconds(10); public async Task RunCommand(ChatBot botInstance, MessageModel message, UserDbModel user, GroupCollection arguments, CancellationToken ctx) { await SettingsProvider.SetValueAsync(BuiltIn.Keys.BotAlmanacText, arguments["text"].Value); await botInstance.SendChatMessageAsync($"@{message.Author.Username}, updated text for the almanac shill", true); } } public class SetAlmanacIntervalCommand : ICommand { public List Patterns => [ new Regex("^admin almanac set interval (?.+)$") ]; public string? HelpText => "Set the almanac interval to whatever in seconds"; public UserRight RequiredRight => UserRight.TrueAndHonest; public TimeSpan Timeout => TimeSpan.FromSeconds(10); public async Task RunCommand(ChatBot botInstance, MessageModel message, UserDbModel user, GroupCollection arguments, CancellationToken ctx) { var interval = Convert.ToInt32(arguments["interval"].Value); if (interval < 300) { await botInstance.SendChatMessageAsync("Not going to let you use an interval below 300 seconds", true); return; } await SettingsProvider.SetValueAsync(BuiltIn.Keys.BotAlmanacInterval, arguments["interval"].Value); if (botInstance.BotServices.AlmanacShill == null) { await botInstance.SendChatMessageAsync("Value has been saved but almanac shill has not been initialized", true); return; } await botInstance.BotServices.AlmanacShill.StopShillTaskAsync(); botInstance.BotServices.AlmanacShill.StartShillTask(); await botInstance.SendChatMessageAsync($"@{message.Author.Username}, updated interval and restarted the shill task", true); } } public class StopAlmanacCommand : ICommand { public List Patterns => [ new Regex("^admin almanac stop$") ]; public string? HelpText => "Stop the almanac reminder"; public UserRight RequiredRight => UserRight.TrueAndHonest; public TimeSpan Timeout => TimeSpan.FromSeconds(10); public async Task RunCommand(ChatBot botInstance, MessageModel message, UserDbModel user, GroupCollection arguments, CancellationToken ctx) { if (botInstance.BotServices.AlmanacShill == null) { await botInstance.SendChatMessageAsync("AlmanacShill is null", true); return; } if (!botInstance.BotServices.AlmanacShill.IsShillTaskRunning()) { await botInstance.SendChatMessageAsync("Looks like the task isn't even running", true); return; } await botInstance.BotServices.AlmanacShill.StopShillTaskAsync(); await botInstance.SendChatMessageAsync("Asked it nicely to stop", true); } } public class StartAlmanacCommand : ICommand { public List Patterns => [ new Regex("^admin almanac start") ]; public string? HelpText => "Start the almanac reminder"; public UserRight RequiredRight => UserRight.TrueAndHonest; public TimeSpan Timeout => TimeSpan.FromSeconds(10); public async Task RunCommand(ChatBot botInstance, MessageModel message, UserDbModel user, GroupCollection arguments, CancellationToken ctx) { if (botInstance.BotServices.AlmanacShill == null) { await botInstance.SendChatMessageAsync("AlmanacShill is null", true); return; } if (botInstance.BotServices.AlmanacShill.IsShillTaskRunning()) { await botInstance.SendChatMessageAsync("Looks like the task is already running", true); return; } botInstance.BotServices.AlmanacShill.StartShillTask(); await botInstance.SendChatMessageAsync("Asked it nicely to start", true); } }