telegram-bot/modules/UrlLimitations/MessageHandler.cs

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

namespace Kruzya.TelegramBot.UrlLimitations;

public class MessageHandler : BotEventHandler
{
    protected ILogger Logger;
    protected readonly CoreContext dbCtx;

    public MessageHandler(ILogger<MessageHandler> logger, CoreContext coreContext)
    {
        Logger = logger;
        dbCtx = coreContext;
    }

    [InChat(InChat.Public)]
    [Message(MessageFlag.HasText)]
    public async Task OnMessageReceived()
    {
        var option = await dbCtx.UserValues.FindOption(Chat.Id, From.Id, "urlLimitations.messagesAfterJoin");
        if (option == null)
        {
            return;
        }

        var messageCount = option.GetValue<UInt16>();
        if (messageCount > 10)
        {
            dbCtx.UserValues.Remove(option);
            return;
        }

        option.SetValue(messageCount + 1);
        var message = RawUpdate.Message;
        if (message.Entities.Count(e => e.Type == MessageEntityType.Url || e.Type == MessageEntityType.TextLink) >
            0 || await message.HasUnknownMention(Bot))
        {
            var targetChat = new ChatId(Chat.Id);
            Task.WaitAll(
                Bot.BanChatMemberAsync(new ChatId(Chat.Id), From.Id, DateTime.Now.AddDays(1)),
                Bot.DeleteMessageAsync(targetChat, message.MessageId),
                Bot.SendTextMessageAsync(targetChat,
                    $"Member {From.ToHtml(true)} has been banned.\n<b>Reason</b>: <pre>Spam suspicion</pre>", parseMode: ParseMode.Html)
            );
        }
    }
}