5.1

src/NadekoBot/Modules/Games/ChatterBot/ChatterBotCommands.cs

@@ -0,0 +1,49 @@
+#nullable disable
+using NadekoBot.Db;
+using NadekoBot.Modules.Games.Services;
+using NadekoBot.Db.Models;
+
+namespace NadekoBot.Modules.Games;
+
+public partial class Games
+{
+    [Group]
+    public partial class ChatterBotCommands : NadekoModule<ChatterBotService>
+    {
+        private readonly DbService _db;
+
+        public ChatterBotCommands(DbService db)
+            => _db = db;
+
+        [Cmd]
+        [RequireContext(ContextType.Guild)]
+        [UserPerm(GuildPerm.ManageMessages)]
+        [NoPublicBot]
+        public async Task CleverBot()
+        {
+            var channel = (ITextChannel)ctx.Channel;
+
+            if (_service.ChatterBotGuilds.TryRemove(channel.Guild.Id, out _))
+            {
+                await using (var uow = _db.GetDbContext())
+                {
+                    uow.Set<GuildConfig>().SetCleverbotEnabled(ctx.Guild.Id, false);
+                    await uow.SaveChangesAsync();
+                }
+
+                await Response().Confirm(strs.chatbot_disabled).SendAsync();
+                return;
+            }
+
+            _service.ChatterBotGuilds.TryAdd(channel.Guild.Id, new(() => _service.CreateSession(), true));
+
+            await using (var uow = _db.GetDbContext())
+            {
+                uow.Set<GuildConfig>().SetCleverbotEnabled(ctx.Guild.Id, true);
+                await uow.SaveChangesAsync();
+            }
+
+            await Response().Confirm(strs.chatbot_enabled).SendAsync();
+        }
+    }
+}