from __future__ import annotations

from pyrogram.types import CallbackQuery, Message

from wbb import app
from wbb.services.message_cleanup import (
    begin_group_callback_interaction,
    begin_group_message_interaction,
    install_group_interaction_tracking,
)
from wbb.services.point_settings import CHECKIN_BUTTON_TEXT

__MODULE__ = "群聊交互清理"
__HELP__ = "群内机器人指令和按钮交互消息会在 1 分钟后自动清理。"

install_group_interaction_tracking(app)


@app.on_message(group=-2000)
async def begin_message_cleanup_context(_, message: Message):
    begin_group_message_interaction(message, menu_texts={CHECKIN_BUTTON_TEXT})


@app.on_callback_query(group=-2000)
async def begin_callback_cleanup_context(_, query: CallbackQuery):
    begin_group_callback_interaction(query)