CrazyHackGUT/telegram-bot
1
0
Fork 0
mirror of https://github.com/Bubuni-Team/telegram-bot.git synced 2026-07-31 00:29:24 +03:00
Code Issues Packages Projects Releases Wiki Activity

[cm] warn decomposition

Browse Source
This commit is contained in:
West14
2022-05-05 13:26:17 +03:00
parent d90556d445
commit df43c45f02
4 changed files with 119 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/ChatManagement/ChatManagement.cs
+4
View File
@@ -15,6 +15,10 @@ public class ChatManagement : Module
public override void ConfigureServices(IServiceCollection services)
{
services.AddScoped<RulesService>();
services.AddScoped<WarnService>();
services.AddTelegramBotParameterParser<BanLength.Param, BanLength.Parser>();
WarnService.WarnEvent += Notifier.WarnNotify;
WarnService.UnWarnEvent += Notifier.UnWarnNotify;
}
}
modules/ChatManagement/Handler/Warn.cs
+8 -48
View File
@@ -1,56 +1,28 @@
#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;
using West.TelegramBot.ChatManagement.Service;
namespace West.TelegramBot.ChatManagement.Handler;
public class Warn : CommonHandler
{
public Warn(CoreContext db) : base(db) {}
private readonly WarnService _warnService;
public Warn(CoreContext db, WarnService warnService) : base(db)
{
_warnService = warnService;
}
[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;
await Bot.SendTextMessageAsync(Chat.Id, $"{warnUser.ToHtml()}<code>, Вам выдано предупреждение! " +
$"Текущее кол-во: {warnCount}/3</code>",
disableNotification: true,
parseMode: ParseMode.Html);
if (warnCount >= 3)
{
try
{
await BanMember(warnUser.Id, DateTime.Now.AddDays(7));
warnCount = 0;
}
catch
{
await Bot.SendTextMessageAsync(Chat.Id, "Нет прав на блокировку пользователя.");
return;
}
await Bot.SendTextMessageAsync(
Chat.Id,
$"{From.ToHtml()} <code>заблокировал</code> {RepliedUser!.ToHtml()} <code>на 7 дней</code>",
disableNotification: true, parseMode: ParseMode.Html);
}
warnOption.SetValue(warnCount);
Db.AddOrUpdate(warnOption);
await _warnService.WarnChatUserAsync(Chat, From, RepliedUser!);
}
[Command("unwarn", CommandParseMode.Both)]
@@ -58,18 +30,6 @@ public class Warn : CommonHandler
{
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);
await _warnService.UnWarnChatUserAsync(Chat, From, RepliedUser!);
}
}
modules/ChatManagement/Notifier.cs
+29
View File
@@ -0,0 +1,29 @@
namespace West.TelegramBot.ChatManagement;
public class Notifier
{
public static async void WarnNotify(ITelegramBotClient bot, Chat chat, User from, User to, int count, bool isBanned = false)
{
await bot.SendTextMessageAsync(chat.Id, $"{to.ToHtml()}<code>, Вам выдано предупреждение! " +
$"Текущее кол-во: {count}/3</code>",
disableNotification: true,
parseMode: ParseMode.Html);
if (isBanned)
{
await bot.SendTextMessageAsync(
chat.Id,
$"{from.ToHtml()} <code>заблокировал</code> {to.ToHtml()} <code>на 7 дней</code>",
disableNotification: true, parseMode: ParseMode.Html);
}
}
public static async void UnWarnNotify(ITelegramBotClient bot, Chat chat, User from, User to, int count)
{
await bot.SendTextMessageAsync(chat.Id,
$"{to.ToHtml()}<code>, с Вас снято предупреждение! " +
$"Текущее кол-во: {count}/3</code>",
disableNotification: true,
parseMode: ParseMode.Html);
}
}
modules/ChatManagement/Service/WarnService.cs
+78
View File
@@ -0,0 +1,78 @@
#nullable enable
using System;
using System.Threading.Tasks;
using BotFramework.Abstractions;
using Kruzya.TelegramBot.Core.Data;
using Kruzya.TelegramBot.Core.Extensions;
using Telegram.Bot;
using Telegram.Bot.Types;
namespace West.TelegramBot.ChatManagement.Service;
public class WarnService
{
public delegate void WarnHandler(ITelegramBotClient bot, Chat chat, User from, User to, int count, bool isBanned = false);
public delegate void UnWarnHandler(ITelegramBotClient bot, Chat chat, User from, User to, int count);
public static event WarnHandler? WarnEvent;
public static event UnWarnHandler? UnWarnEvent;
private readonly CoreContext _db;
private readonly ITelegramBotClient _bot;
public WarnService(CoreContext db, IBotInstance bot)
{
_db = db;
_bot = bot.BotClient;
}
public async Task WarnChatUserAsync(Chat chat, User from, User to)
{
var warnOption = await GetWarnOption(chat, to);
var warnCount = warnOption.GetValue<int>() + 1;
if (warnCount >= 3)
{
try
{
await _bot.RestrictChatMemberAsync(
chat.Id,
to.Id,
new ChatPermissions
{
CanSendMessages = false
},
DateTime.Now.AddDays(7)
);
warnCount = 0;
}
catch
{
await _bot.SendTextMessageAsync(chat.Id, "У меня нет прав на блокировку пользователя.");
}
}
warnOption.SetValue(warnCount);
_db.AddOrUpdate(warnOption);
WarnEvent?.Invoke(_bot, chat, from, to, warnCount == 0 ? 3 : warnCount, warnCount == 0);
}
public async Task UnWarnChatUserAsync(Chat chat, User from, User user)
{
var warnOption = await GetWarnOption(chat, user);
var warnCount = warnOption.GetValue<int>();
if (warnCount == 0) return;
warnCount = Math.Max(warnCount - 1, 0);
warnOption.SetValue(warnCount);
_db.AddOrUpdate(warnOption);
UnWarnEvent?.Invoke(_bot, chat, from, user, warnCount);
}
protected async Task<BotUserValue> GetWarnOption(Chat chat, User user)
{
return await _db.UserValues.FindOrCreateOption(chat.Id, user.Id, "warnCount");
}
}