139 lines
5.2 KiB
Python
139 lines
5.2 KiB
Python
from aiogram import Router, F, Bot, html
|
|
from aiogram.filters import CommandStart
|
|
from aiogram.types import Message, BufferedInputFile
|
|
from aiogram.fsm.context import FSMContext
|
|
from aiogram.fsm.state import State, StatesGroup
|
|
|
|
from bot.config import settings
|
|
from bot.utils.captcha import generate_captcha
|
|
from bot.utils.db import db
|
|
from bot.utils.i18n import _
|
|
|
|
router = Router()
|
|
|
|
class CaptchaStates(StatesGroup):
|
|
waiting_for_captcha = State()
|
|
|
|
@router.message(CommandStart())
|
|
async def cmd_start(message: Message, locale: str):
|
|
db.log_user(message.from_user)
|
|
await message.answer(_("welcome", locale=locale))
|
|
|
|
@router.message(CaptchaStates.waiting_for_captcha)
|
|
async def process_captcha(message: Message, state: FSMContext, bot: Bot, locale: str):
|
|
data = await state.get_data()
|
|
correct_captcha = data.get("captcha_text")
|
|
original_msg_id = data.get("original_msg_id")
|
|
|
|
if message.text and message.text.upper() == correct_captcha:
|
|
await state.clear()
|
|
|
|
try:
|
|
db_msg_id = data.get("db_msg_id", "N/A")
|
|
prefix = f"Forwarding, ID [{db_msg_id}]:"
|
|
target_chat_id = settings.group_chat_id
|
|
|
|
if data.get('has_media'):
|
|
new_caption = html.quote(prefix)
|
|
if data.get('original_caption'):
|
|
new_caption += f"\n{html.quote(data.get('original_caption'))}"
|
|
|
|
await bot.copy_message(
|
|
chat_id=target_chat_id,
|
|
from_chat_id=message.chat.id,
|
|
message_id=original_msg_id,
|
|
caption=new_caption,
|
|
parse_mode=None # Use default or specific? Default is HTML.
|
|
)
|
|
elif data.get('is_text'):
|
|
await bot.send_message(
|
|
chat_id=target_chat_id,
|
|
text=f"{html.quote(prefix)}\n{html.quote(data.get('original_text'))}"
|
|
)
|
|
else:
|
|
# Other types (stickers, etc.)
|
|
await bot.copy_message(
|
|
chat_id=target_chat_id,
|
|
from_chat_id=message.chat.id,
|
|
message_id=original_msg_id
|
|
)
|
|
await bot.send_message(chat_id=target_chat_id, text=html.quote(prefix))
|
|
|
|
await message.answer(_("captcha_solved", locale=locale))
|
|
|
|
except Exception as e:
|
|
await message.answer(_("captcha_error", locale=locale))
|
|
print(f"Error forwarding: {e}")
|
|
else:
|
|
# Generate new captcha
|
|
captcha_text, captcha_img = generate_captcha()
|
|
await state.update_data(captcha_text=captcha_text)
|
|
await message.answer_photo(
|
|
BufferedInputFile(captcha_img, filename="captcha.png"),
|
|
caption=_("incorrect_captcha", locale=locale)
|
|
)
|
|
|
|
@router.message(F.chat.type == "private")
|
|
async def handle_any_message(message: Message, state: FSMContext, bot: Bot, locale: str):
|
|
# Log everything
|
|
db_msg_id = db.log_message(message)
|
|
|
|
has_media = bool(message.photo or message.video or message.document or message.audio or message.voice)
|
|
is_text = bool(message.text)
|
|
|
|
if settings.disable_captcha:
|
|
try:
|
|
wait_message = await message.answer(_("processing_message", locale=locale))
|
|
prefix = f"Forwarding, ID [{db_msg_id}]:"
|
|
target_chat_id = settings.group_chat_id
|
|
|
|
if has_media:
|
|
new_caption = html.quote(prefix)
|
|
if message.caption:
|
|
new_caption += f"\n{html.quote(message.caption)}"
|
|
|
|
await bot.copy_message(
|
|
chat_id=target_chat_id,
|
|
from_chat_id=message.chat.id,
|
|
message_id=message.message_id,
|
|
caption=new_caption
|
|
)
|
|
elif is_text:
|
|
await bot.send_message(
|
|
chat_id=target_chat_id,
|
|
text=f"{html.quote(prefix)}\n{html.quote(message.text)}"
|
|
)
|
|
else:
|
|
await bot.copy_message(
|
|
chat_id=target_chat_id,
|
|
from_chat_id=message.chat.id,
|
|
message_id=message.message_id
|
|
)
|
|
await bot.send_message(chat_id=target_chat_id, text=html.quote(prefix))
|
|
await wait_message.edit_text(_("message_processed", locale=locale))
|
|
return
|
|
except Exception as e:
|
|
print(f"Error forwarding without captcha: {e}")
|
|
# Fallback to captcha if something goes wrong?
|
|
# Or just ignore. Usually, we want to proceed.
|
|
pass
|
|
|
|
# Generate captcha
|
|
captcha_text, captcha_img = generate_captcha()
|
|
|
|
await state.set_state(CaptchaStates.waiting_for_captcha)
|
|
await state.update_data(
|
|
captcha_text=captcha_text,
|
|
db_msg_id=db_msg_id,
|
|
original_msg_id=message.message_id,
|
|
original_text=message.text,
|
|
original_caption=message.caption,
|
|
has_media=has_media,
|
|
is_text=is_text
|
|
)
|
|
|
|
await message.answer_photo(
|
|
BufferedInputFile(captcha_img, filename="captcha.png"),
|
|
caption=_("solve_captcha", locale=locale)
|
|
)
|