telegram-bot/Core/Handler/CommonHandler.cs

#nullable enable

using System;
using System.Threading.Tasks;
using BotFramework;
using Kruzya.TelegramBot.Core.Data;
using Kruzya.TelegramBot.Core.Extensions;
using Telegram.Bot;
using Telegram.Bot.Types;
using Telegram.Bot.Types.Enums;

namespace Kruzya.TelegramBot.Core.Handler;

public abstract class CommonHandler : BotEventHandler
{
    protected readonly CoreContext Db;

    protected Message? Message => RawUpdate.Message;

    protected User? RepliedUser => Message?.ReplyToMessage?.From;

    protected CommonHandler(CoreContext db)
    {
        Db = db;
    }

    protected virtual async Task<Message> Reply(string text)
    {
        return await Bot.SendTextMessageAsync(Chat.Id, text,
            replyToMessageId: Message?.MessageId, parseMode: ParseMode.Html);
    }

    protected async Task<bool> CanPunish()
    {
        return RepliedUser != null && await Bot.CanPunishMember(Chat, From, RepliedUser);
    }


    protected async Task BanMember(long userId, DateTime banEnd)
    {
        await Bot.RestrictChatMemberAsync(
            Chat.Id,
            userId,
            new ChatPermissions
            {
                CanSendMessages = false
            },
            banEnd
        );
    }

    protected async Task<BotUserValue> GetWarnOption(User user)
    {
        return await Db.UserValues.FindOrCreateOption(Chat.Id, user.Id, "warnCount");
    }

    protected async Task AnswerQuery(string id, string? message = null)
    {
        await Bot.AnswerCallbackQueryAsync(id, message);
    }
}