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

[rules] Fix permission check on setrules.

Browse Source
This commit is contained in:
West14
2022-01-18 21:59:26 +02:00
parent 1444e3282f
commit b1876edf13
2 changed files with 16 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Core/Extensions/BotExtension.cs
+14 -7
View File
@@ -8,13 +8,7 @@ namespace Kruzya.TelegramBot.Core.Extensions
{
public static async Task<bool> CanPunishMember(this ITelegramBotClient bot, Chat chat, User user, User victim = null)
{
var callerMember = await bot.GetChatMemberAsync(chat, user.Id);
var canUse = callerMember is ChatMemberOwner;
if (callerMember is ChatMemberAdministrator callerAdmin)
{
canUse = callerAdmin.CanRestrictMembers;
}
var canUse = await IsUserAdminAsync(bot, chat, user);
if (victim != null && canUse)
{
@@ -24,5 +18,18 @@ namespace Kruzya.TelegramBot.Core.Extensions
return canUse;
}
public static async Task<bool> IsUserAdminAsync(this ITelegramBotClient bot, Chat chat, User user)
{
var callerMember = await bot.GetChatMemberAsync(chat, user.Id);
var canUse = callerMember is ChatMemberOwner;
if (callerMember is ChatMemberAdministrator callerAdmin)
{
canUse = callerAdmin.CanRestrictMembers;
}
return canUse;
}
}
}
modules/ChatManagement/Handler/Rules.cs
+2 -2
View File
@@ -8,14 +8,14 @@ using Telegram.Bot.Exceptions;
namespace West.TelegramBot.ChatManagement.Handler
{
internal class Rules : ManagementHandler
class Rules : ManagementHandler
{
[Command(InChat.Public, "setrules", CommandParseMode.Both)]
public async Task HandleSetRules()
{
var replyToMessage = Message?.ReplyToMessage;
if (replyToMessage == null || !await CanPunish()) return;
if (replyToMessage == null || !await Bot.IsUserAdminAsync(Chat, From)) return;
var opt = await Db.UserValues.FindOrCreateOption(Chat.Id, "rulesMsgId");
opt.SetValue(replyToMessage.MessageId);