mirror of
https://github.com/Bubuni-Team/telegram-bot.git
synced 2026-07-31 00:29:24 +03:00
58 lines
2.0 KiB
C#
58 lines
2.0 KiB
C#
using BotFramework.Attributes;
|
|
using Kruzya.TelegramBot.Core.Data;
|
|
using Kruzya.TelegramBot.Core.Extensions;
|
|
using Kruzya.TelegramBot.Core.Handler;
|
|
using Kruzya.TelegramBot.Core.Service;
|
|
using Kruzya.TelegramBot.CustomChatAddOns.Options;
|
|
using Telegram.Bot.Types.Enums;
|
|
|
|
namespace Kruzya.TelegramBot.CustomChatAddOns.Handler
|
|
{
|
|
public class Config : CommonHandler
|
|
{
|
|
private readonly CustomChatHandler _option;
|
|
private readonly UserService _userService;
|
|
|
|
public Config(CustomChatHandler option, UserService userService, CoreContext db) : base(db)
|
|
{
|
|
_option = option;
|
|
_userService = userService;
|
|
|
|
}
|
|
|
|
private bool IsAllowedChatType(ChatType chatType)
|
|
{
|
|
// TODO: move to class constant.
|
|
var allowedChatTypes = new[] { ChatType.Supergroup, ChatType.Group, ChatType.Channel };
|
|
return allowedChatTypes.Contains(chatType);
|
|
}
|
|
|
|
[ParametrizedCommand("set_custom_handler")]
|
|
public async Task Handle(Uri endpoint)
|
|
{
|
|
if (!IsAllowedChatType(Chat.Type))
|
|
{
|
|
return;
|
|
}
|
|
|
|
if (!await Bot.IsUserAdminAsync(Chat, From))
|
|
{
|
|
await Reply("Sorry, you cannot do this");
|
|
return;
|
|
}
|
|
|
|
// URLs to somewhere in local bot network should be allowed for using only super-admins.
|
|
// It may be bypassed via using, for example, "kitties" instead of "kitties.bot.svc.cluster.local". Maybe check resulting IP?
|
|
if (endpoint.Host.EndsWith(".local") && !_userService.IsUserSuper(From.Id))
|
|
{
|
|
await Reply("Sorry, you cannot do this");
|
|
return;
|
|
}
|
|
|
|
// TODO: perform a "ping" request to endpoint. "Ping" should contain chat id and user id who performs this action.
|
|
await _option.SetValueAsync(Chat.Id, endpoint.ToString());
|
|
await Reply("Done, configured");
|
|
}
|
|
}
|
|
}
|