mirror of
https://github.com/nexus-stc/hyperboria
synced 2024-10-14 16:29:34 +02:00
83 lines
2.9 KiB
Python
83 lines
2.9 KiB
Python
from datetime import (
|
|
datetime,
|
|
timedelta,
|
|
)
|
|
|
|
from aiobaseclient.exceptions import ClientError
|
|
from library.telegram.base import RequestContext
|
|
from nexus.bot.widgets.banlist_widget import BanlistWidget
|
|
from pytimeparse.timeparse import timeparse
|
|
from telethon import events
|
|
|
|
from .admin import BaseAdminHandler
|
|
|
|
|
|
class BanHandler(BaseAdminHandler):
|
|
filter = events.NewMessage(incoming=True, pattern='^/ban ([0-9]+) ([A-Za-z0-9]+)\\s?(.*)?$')
|
|
|
|
async def handler(self, event: events.ChatAction, request_context: RequestContext):
|
|
chat_id = int(event.pattern_match.group(1))
|
|
ban_duration = event.pattern_match.group(2)
|
|
ban_message = event.pattern_match.group(3)
|
|
ban_end_date = datetime.utcnow() + timedelta(seconds=timeparse(ban_duration))
|
|
|
|
try:
|
|
await self.application.idm_client.update_chat(
|
|
chat_id=chat_id,
|
|
ban_until=int(ban_end_date.timestamp()),
|
|
ban_message=ban_message,
|
|
request_id=request_context.request_id,
|
|
)
|
|
request_context.statbox(
|
|
action='banned',
|
|
ban_message=ban_message,
|
|
ban_until=ban_end_date.timestamp(),
|
|
banned_chat_id=chat_id,
|
|
)
|
|
except ClientError as e:
|
|
if e.code == 'nonexistent_entity_error':
|
|
await event.reply('Chat not found')
|
|
return
|
|
raise
|
|
|
|
await event.reply('User banned until ' + ban_end_date.strftime("%Y-%m-%d %H:%M") + ' UTC')
|
|
|
|
|
|
class UnbanHandler(BaseAdminHandler):
|
|
filter = events.NewMessage(incoming=True, pattern='^/unban(?:_|\\s)([0-9]+)$')
|
|
|
|
async def handler(self, event, request_context: RequestContext):
|
|
chat_id = int(event.pattern_match.group(1))
|
|
|
|
try:
|
|
await self.application.idm_client.update_chat(
|
|
chat_id=chat_id,
|
|
ban_until=0,
|
|
request_id=request_context.request_id,
|
|
)
|
|
request_context.statbox(
|
|
action='unbanned',
|
|
unbanned_chat_id=chat_id,
|
|
)
|
|
except ClientError as e:
|
|
if e.code == 'nonexistent_entity_error':
|
|
await event.reply('Chat not found')
|
|
return
|
|
raise
|
|
|
|
await event.reply('User unbanned')
|
|
|
|
|
|
class BanlistHandler(BaseAdminHandler):
|
|
filter = events.NewMessage(incoming=True, pattern='^/banlist$')
|
|
|
|
async def handler(self, event, request_context: RequestContext):
|
|
request_context.statbox(action='show', mode='banlist')
|
|
chat_list = (await self.application.idm_client.list_chats(
|
|
banned_at_moment=int(datetime.utcnow().timestamp()),
|
|
request_id=request_context.request_id,
|
|
)).chats
|
|
banlist_widget_view = BanlistWidget(application=self.application, chat=request_context.chat)
|
|
widget_content = await banlist_widget_view.render(chat_list=chat_list)
|
|
await event.reply(widget_content)
|