fix max-bot, add tests

2026-05-15 10:22:43 +03:00 · 2026-05-15 10:22:43 +03:00 · 2ad1438e1c
commit 2ad1438e1c
parent 7abbaf7e7a
17 changed files with 1621 additions and 494 deletions
--- a/adapter/max/handlers/commands.py
+++ b/adapter/max/handlers/commands.py
@ -0,0 +1,112 @@
+from __future__ import annotations
+
+from adapter.max.handlers.chat import ChatHandler as MaxChatHandler
+from adapter.max.handlers.help import get_help
+from adapter.max.store import ChatStore
+from sdk.prototype_state import PrototypeStateStore
+
+from core.handler import EventDispatcher
+from core.protocol import IncomingCallback, IncomingCommand, OutgoingMessage
+
+_SINGLE_DIALOG_HINT = (
+    "В MAX один диалог с ботом. Чтобы сбросить контекст агента, используйте /clear или /reset."
+)
+
+
+def register_max_handlers(
+    dispatcher: EventDispatcher,
+    *,
+    chat_store: ChatStore,
+    max_chat_handler: MaxChatHandler,
+    prototype_state: PrototypeStateStore,
+) -> None:
+    async def _room_or_error(
+        event: IncomingCommand,
+    ) -> tuple[str, str] | list[OutgoingMessage]:
+        room = chat_store.get_room_by_platform_chat_id(event.chat_id)
+        if room is None:
+            return [
+                OutgoingMessage(
+                    chat_id=event.chat_id,
+                    text="Состояние ещё не готово. Напишите сообщение ещё раз.",
+                )
+            ]
+        return room.max_chat_id, room.platform_chat_id
+
+    async def handle_max_help(event: IncomingCommand, auth_mgr, platform, chat_mgr, settings_mgr):  # noqa: ARG001
+        if not await auth_mgr.is_authenticated(event.user_id):
+            return [
+                OutgoingMessage(
+                    chat_id=event.chat_id,
+                    text="Введите /start чтобы начать.",
+                )
+            ]
+        return [OutgoingMessage(chat_id=event.chat_id, text=get_help())]
+
+    async def handle_max_no_multichat(event: IncomingCommand, auth_mgr, platform, chat_mgr, settings_mgr):  # noqa: ARG001
+        if not await auth_mgr.is_authenticated(event.user_id):
+            return [
+                OutgoingMessage(
+                    chat_id=event.chat_id,
+                    text="Введите /start чтобы начать.",
+                )
+            ]
+        _ = event
+        return [OutgoingMessage(chat_id=event.chat_id, text=_SINGLE_DIALOG_HINT)]
+
+    async def handle_max_clear(event: IncomingCommand, auth_mgr, platform, chat_mgr, settings_mgr):  # noqa: ARG001
+        if not await auth_mgr.is_authenticated(event.user_id):
+            return [
+                OutgoingMessage(
+                    chat_id=event.chat_id,
+                    text="Введите /start чтобы начать.",
+                )
+            ]
+        routed = await _room_or_error(event)
+        if isinstance(routed, list):
+            return routed
+        max_chat_id, platform_chat_old = routed
+
+        await prototype_state.clear_current_session(platform_chat_old)
+        await chat_mgr.archive(platform_chat_old, event.user_id)
+        max_chat_handler.handle_clear(max_chat_id)
+        room = chat_store.get_room_by_max_chat_id(max_chat_id)
+        if room is None:
+            return [
+                OutgoingMessage(
+                    chat_id=platform_chat_old,
+                    text="Не удалось сбросить контекст.",
+                )
+            ]
+        platform_chat_new = room.platform_chat_id
+        await chat_mgr.get_or_create(
+            user_id=event.user_id,
+            chat_id=platform_chat_new,
+            platform="max",
+            surface_ref=max_chat_id,
+            name=room.name,
+        )
+        return [
+            OutgoingMessage(
+                chat_id=platform_chat_new,
+                text="Контекст агента сброшен.",
+            )
+        ]
+
+    for cmd in ("new", "rename", "archive", "chats"):
+        dispatcher.register(IncomingCommand, cmd, handle_max_no_multichat)
+
+    dispatcher.register(IncomingCommand, "reset", handle_max_clear)
+    dispatcher.register(IncomingCommand, "clear", handle_max_clear)
+    dispatcher.register(IncomingCommand, "help", handle_max_help)
+
+    async def handle_max_plain_callback(event: IncomingCallback, auth_mgr, platform, chat_mgr, settings_mgr):  # noqa: ARG001
+        payload = str(event.payload.get("payload", ""))
+        return [
+            OutgoingMessage(
+                chat_id=event.chat_id,
+                text=f"Неизвестное действие кнопки: {payload}",
+            )
+        ]
+
+    dispatcher.register(IncomingCallback, "max_callback", handle_max_plain_callback)