from __future__ import annotations

from collections.abc import Awaitable, Callable

from adapter.matrix.agent_registry import AgentRegistry
from adapter.matrix.store import (
    get_platform_chat_id,
    get_selected_agent_id,
    get_room_meta,
    next_platform_chat_id,
    set_platform_chat_id,
    set_room_agent_id,
    set_selected_agent_id,
)
from core.protocol import IncomingCommand, OutgoingMessage


def make_handle_agent(store, registry: AgentRegistry) -> Callable[..., Awaitable[list]]:
    async def handle_agent(
        event: IncomingCommand, auth_mgr, platform, chat_mgr, settings_mgr
    ) -> list:
        if not event.args:
            selected_agent_id = await get_selected_agent_id(store, event.user_id)
            lines = ["Доступные агенты:"]
            for index, agent in enumerate(registry.agents, start=1):
                suffix = " [текущий]" if agent.agent_id == selected_agent_id else ""
                lines.append(f"{index}. {agent.label}{suffix}")
            lines.extend(["", "Выбери агент: !agent <номер>"])
            return [OutgoingMessage(chat_id=event.chat_id, text="\n".join(lines))]

        try:
            selected_index = int(event.args[0])
        except ValueError:
            return [
                OutgoingMessage(
                    chat_id=event.chat_id,
                    text="Укажи номер агента из списка: !agent <номер>.",
                )
            ]

        if selected_index < 1 or selected_index > len(registry.agents):
            return [
                OutgoingMessage(
                    chat_id=event.chat_id,
                    text="Такого агента нет. Открой список через !agent.",
                )
            ]

        agent = registry.agents[selected_index - 1]
        await set_selected_agent_id(store, event.user_id, agent.agent_id)

        current_chat = await chat_mgr.get(event.chat_id, user_id=event.user_id)
        if current_chat is not None and current_chat.surface_ref:
            room_id = current_chat.surface_ref
            room_meta = await get_room_meta(store, room_id)
            if room_meta is not None and not room_meta.get("agent_id"):
                await set_room_agent_id(store, room_id, agent.agent_id)
                if await get_platform_chat_id(store, room_id) is None:
                    await set_platform_chat_id(
                        store,
                        room_id,
                        await next_platform_chat_id(store),
                    )
                return [
                    OutgoingMessage(
                        chat_id=event.chat_id,
                        text=f"Агент {agent.label} выбран. Текущий чат готов к работе.",
                    )
                ]

        return [
            OutgoingMessage(
                chat_id=event.chat_id,
                text=f"Агент переключен на {agent.label}. Продолжай через !new.",
            )
        ]

    return handle_agent