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

Whois command. Some refactorings.

Browse Source
This commit is contained in:
West14
2021-12-27 22:53:13 +02:00
parent 9bc08c58b7
commit b91d6cfdb5
6 changed files with 94 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/ChatManagement/Handler/Warn.cs
+13 -21
View File
@@ -1,4 +1,6 @@
using System;
#nullable enable
using System;
using System.Threading.Tasks;
using BotFramework.Attributes;
using BotFramework.Enums;
@@ -9,25 +11,20 @@ using Telegram.Bot.Types.Enums;
namespace West.TelegramBot.ChatManagement.Handler
{
public class Command : ManagementHandler
public class Warn : ManagementHandler
{
private readonly CoreContext _db;
public Command(CoreContext db)
{
_db = db;
}
public Warn(CoreContext db) : base(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;
var warnUser = RepliedUser!;
var warnOption = await GetWarnOption(warnUser);
var warnCount = warnOption!.GetValue<int>() + 1;
warnOption.SetValue(warnCount);
_db.MarkAsModified(warnOption);
Db.MarkAsModified(warnOption);
await Bot.SendTextMessageAsync(Chat.Id, $"{warnUser.ToHtml(false)}<code>, Вам выдано предупреждение! " +
$"Текущее кол-во: {warnCount}/3</code>",
@@ -46,21 +43,16 @@ namespace West.TelegramBot.ChatManagement.Handler
{
if (!await CanPunish()) return;
var warnOption = await GetWarnOption();
var warnOption = await GetWarnOption(RepliedUser!);
var warnCount = Math.Max(warnOption.GetValue<int>() - 1, 0);
warnOption.SetValue(warnCount);
_db.MarkAsModified(warnOption);
Db.MarkAsModified(warnOption);
await Bot.SendTextMessageAsync(Chat.Id,
$"{GetVictim().ToHtml(false)}<code>, с Вас снято предупреждение! " +
$"{RepliedUser.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");
}
}
}