Merge remote-tracking branch 'origin/feature/warns-net5' into feature/new-dotnet-5

modules/ChatManagement/ChatManagement.cs

@@ -0,0 +1,11 @@
+using Kruzya.TelegramBot.Core;
+
+namespace West.TelegramBot.ChatManagement
+{
+    public class ChatManagement : Module
+    {
+        public ChatManagement(Core core) : base(core)
+        {
+        }
+    }
+}

modules/ChatManagement/ChatManagement.csproj

@@ -0,0 +1,13 @@
+<Project Sdk="Microsoft.NET.Sdk">
+
+    <PropertyGroup>
+        <TargetFramework>net5.0</TargetFramework>
+        <AssemblyName>TelegramBot.ChatManagement</AssemblyName>
+        <RootNamespace>West.TelegramBot.ChatManagement</RootNamespace>
+    </PropertyGroup>
+
+    <ItemGroup>
+      <ProjectReference Include="..\..\Core\Core.csproj" />
+    </ItemGroup>
+
+</Project>

modules/ChatManagement/Extensions/BotExtension.cs

@@ -0,0 +1,28 @@
+using System.Threading.Tasks;
+using Telegram.Bot;
+using Telegram.Bot.Types;
+
+namespace West.TelegramBot.ChatManagement.Extensions
+{
+    public static class BotExtension
+    {
+        public static async Task<bool> CanPunishMember(this ITelegramBotClient bot, Chat chat, User user, User victim = null)
+        {
+            var callerMember = await bot.GetChatMemberAsync(chat, user.Id);
+            var canUse = callerMember is ChatMemberOwner;
+
+            if (callerMember is ChatMemberAdministrator callerAdmin)
+            {
+                canUse = callerAdmin.CanRestrictMembers;
+            }
+
+            if (victim != null && canUse)
+            {
+                var victimMember = await bot.GetChatMemberAsync(chat, victim.Id);
+                canUse = victimMember is not ChatMemberOwner && victimMember is not ChatMemberAdministrator;
+            }
+
+            return canUse;
+        }
+    }
+}

modules/ChatManagement/Handler/Ban.cs

@@ -0,0 +1,38 @@
+using System.Threading.Tasks;
+using BotFramework.Attributes;
+using BotFramework.Enums;
+using Telegram.Bot;
+using Telegram.Bot.Types;
+
+namespace West.TelegramBot.ChatManagement.Handler
+{
+    public class Ban : ManagementHandler
+    {
+        [Command("ban", CommandParseMode.Both)]
+        public async Task HandleBan()
+        {
+            if (!await CanPunish()) return;
+
+            await BanMember(GetVictim());
+        }
+
+        [Command("unban", CommandParseMode.Both)]
+        public async Task HandleUnban()
+        {
+            if (!await CanPunish()) return;
+
+            await Bot.RestrictChatMemberAsync(Chat.Id, GetVictim().Id,
+                new ChatPermissions
+                {
+                    CanSendMessages = true,
+                    CanChangeInfo = true,
+                    CanInviteUsers = true,
+                    CanPinMessages = true,
+                    CanSendPolls = true,
+                    CanSendMediaMessages = true,
+                    CanSendOtherMessages = true,
+                    CanAddWebPagePreviews = true
+                });
+        }
+    }
+}

modules/ChatManagement/Handler/ManagementHandler.cs

@@ -0,0 +1,50 @@
+using System;
+using System.Threading.Tasks;
+using BotFramework;
+using Kruzya.TelegramBot.Core.Extensions;
+using Telegram.Bot;
+using Telegram.Bot.Types;
+using Telegram.Bot.Types.Enums;
+using West.TelegramBot.ChatManagement.Extensions;
+
+namespace West.TelegramBot.ChatManagement.Handler
+{
+    public class ManagementHandler : BotEventHandler
+    {
+        protected User GetVictim()
+        {
+            return RawUpdate.Message?.ReplyToMessage?.From;
+        }
+        
+        protected async Task<bool> CanPunish()
+        {
+            return GetVictim() != null && await Bot.CanPunishMember(Chat, From, GetVictim());
+        }
+
+        protected async Task BanMember(User user, DateTime? banEnd = null, bool sendMessage = true)
+        {
+            await BanMember(user.Id, banEnd, sendMessage);
+
+            if (sendMessage)
+            {
+                await Bot.SendTextMessageAsync(
+                    Chat.Id, 
+                    $"{From.ToHtml(false)} <code>заблокировал</code> {user.ToHtml(false)} <code>на 7 дней</code>",
+                    disableNotification: true, parseMode: ParseMode.Html);
+            }
+        }
+        
+        protected async Task BanMember(long userId, DateTime? banEnd = null, bool sendMessage = true)
+        {
+            await Bot.RestrictChatMemberAsync(
+                                Chat.Id,
+                                userId,
+                                new ChatPermissions
+                                {
+                                    CanSendMessages = false
+                                },
+                                banEnd ?? DateTime.Now.AddDays(7)
+                                );
+        }
+    }
+}

modules/ChatManagement/Handler/Warn.cs

@@ -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");
+        }
+    }
+}