telegram-bot/modules/ChatManagement/Handler/Warn.cs

#nullable enable

using System;
using System.Threading.Tasks;
using BotFramework.Attributes;
using BotFramework.Enums;
using Kruzya.TelegramBot.Core;
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 Warn : CommonHandler
{
    public Warn(CoreContext db) : base(db) {}

    [Command("warn", CommandParseMode.Both)]
    public async Task HandleWarn()
    {
        if (!await CanPunish()) return;
            
        var warnUser = RepliedUser!;
        var warnOption = await GetWarnOption(warnUser);
        var warnCount = warnOption!.GetValue<int>() + 1;
        warnOption.SetValue(warnCount);
        Db.AddOrUpdate(warnOption);
            
        await Bot.SendTextMessageAsync(Chat.Id, $"{warnUser.ToHtml()}<code>, Вам выдано предупреждение! " +
                                                $"Текущее кол-во: {warnCount}/3</code>", 
            disableNotification: true, 
            parseMode: ParseMode.Html);
            
        if (warnCount == 3)
        {
            await BanMember(warnUser);
            warnOption.SetValue(0);
            Db.AddOrUpdate(warnOption);
        }
    }

    [Command("unwarn", CommandParseMode.Both)]
    public async Task HandleUnwarn()
    {
        if (!await CanPunish()) return;

        var warnOption = await GetWarnOption(RepliedUser!);
        var warnCount = warnOption.GetValue<int>();
        if (warnCount == 0) return;
            
        warnCount = Math.Max(warnCount - 1, 0);
        warnOption.SetValue(warnCount);
        Db.AddOrUpdate(warnOption);
            
        await Bot.SendTextMessageAsync(Chat.Id, 
            $"{RepliedUser.ToHtml()}<code>, с Вас снято предупреждение! " +
            $"Текущее кол-во: {warnCount}/3</code>", 
            disableNotification: true, 
            parseMode: ParseMode.Html);
    }
}