Add ChatManagement commands. Drop appsettings from version control.

2026-07-31 00:29:24 +03:00 · 2021-12-26 15:40:14 +02:00
parent 685caf52b7
commit 7b37e2f35d
7 changed files with 190 additions and 98 deletions
@@ -0,0 +1,66 @@
+using System;
+using System.Threading.Tasks;
+using BotFramework.Attributes;
+using BotFramework.Enums;
+using Kruzya.TelegramBot.Core.Data;
+using Kruzya.TelegramBot.Core.Extensions;
+using Telegram.Bot;
+using Telegram.Bot.Types.Enums;
+
+namespace West.TelegramBot.ChatManagement.Handler
+{
+    public class Command : ManagementHandler
+    {
+        private readonly CoreContext _db;
+        
+        public Command(CoreContext db)
+        {
+            _db = db;
+        }
+        
+        [Command("warn", CommandParseMode.Both)]
+        public async Task HandleWarn()
+        {
+            if (!await CanPunish()) return;
+            
+            var warnUser = GetVictim();
+            var warnOption = await GetWarnOption();
+            var warnCount = warnOption.GetValue<int>() + 1;
+            warnOption.SetValue(warnCount);
+            _db.MarkAsModified(warnOption);
+            
+            await Bot.SendTextMessageAsync(Chat.Id, $"{warnUser.ToHtml(false)}<code>, Вам выдано предупреждение! " +
+                                                    $"Текущее кол-во: {warnCount}/3</code>", 
+                disableNotification: true, 
+                parseMode: ParseMode.Html);
+            
+            if (warnCount == 3)
+            {
+                await BanMember(warnUser);
+                warnOption.SetValue(0);
+            }
+        }
+
+        [Command("unwarn", CommandParseMode.Both)]
+        public async Task HandleUnwarn()
+        {
+            if (!await CanPunish()) return;
+
+            var warnOption = await GetWarnOption();
+            var warnCount = Math.Max(warnOption.GetValue<int>() - 1, 0);
+            warnOption.SetValue(warnCount);
+            _db.MarkAsModified(warnOption);
+            
+            await Bot.SendTextMessageAsync(Chat.Id, 
+                $"{GetVictim().ToHtml(false)}<code>, с Вас снято предупреждение! " +
+                $"Текущее кол-во: {warnCount}/3</code>", 
+                disableNotification: true, 
+                parseMode: ParseMode.Html);
+        }
+        
+        private async Task<BotUserValue> GetWarnOption()
+        {
+            return await _db.UserValues.FindOrCreateOption(Chat.Id, GetVictim().Id, "warnCount");
+        }
+    }
+}