from __future__ import annotations

from datetime import datetime

import pytest

from yuxi.channels.models import (
    ChannelIdentity,
    ChannelMessage,
    ChannelType,
    ChatType,
    EventType,
    MessageType,
    SessionScope,
)
from yuxi.channels.protocols.session_router import BaseSessionRouter, SessionRouter


def _make_message(chat_type: ChatType = ChatType.DIRECT, chat_id: str = "chat_1") -> ChannelMessage:
    return ChannelMessage(
        identity=ChannelIdentity(
            channel_id="test",
            channel_type=ChannelType.WEBCHAT,
            channel_user_id="u1",
            channel_chat_id=chat_id,
        ),
        content="hello",
        chat_type=chat_type,
    )


class TestBaseSessionRouter:
    def test_resolve_chat_id(self):
        router = BaseSessionRouter(channel_type="test")
        msg = _make_message(chat_id="chat_123")
        assert router.resolve_chat_id(msg) == "chat_123"

    def test_resolve_thread_key(self):
        router = BaseSessionRouter(channel_type="test")
        msg = _make_message(chat_type=ChatType.DIRECT, chat_id="dm_123")
        key = router.resolve_thread_key(msg)
        assert "test:default" in key
        assert "dm_123" in key

    def test_resolve_session_scope_direct(self):
        router = BaseSessionRouter(channel_type="test")
        msg = _make_message(chat_type=ChatType.DIRECT)
        assert router.resolve_session_scope(msg) == SessionScope.DIRECT

    def test_resolve_session_scope_group(self):
        router = BaseSessionRouter(channel_type="test")
        msg = _make_message(chat_type=ChatType.GROUP)
        assert router.resolve_session_scope(msg) == SessionScope.GROUP

    def test_resolve_session_scope_thread(self):
        router = BaseSessionRouter(channel_type="test")
        msg = _make_message(chat_type=ChatType.THREAD)
        assert router.resolve_session_scope(msg) == SessionScope.THREAD

    def test_normalize_target_without_prefix(self):
        router = BaseSessionRouter(channel_type="test")
        assert router.normalize_target("target_1") == "test:target_1"

    def test_normalize_target_with_prefix(self):
        router = BaseSessionRouter(channel_type="test")
        assert router.normalize_target("test:target_1") == "test:target_1"

    def test_resolve_agent_route(self):
        router = BaseSessionRouter(channel_type="test")
        msg = _make_message(chat_type=ChatType.DIRECT, chat_id="dm_123")
        route = router.resolve_agent_route(msg)
        assert route.startswith("agent:main:")
        assert "dm_123" in route

    def test_build_key(self):
        router = BaseSessionRouter(channel_type="test", account_id="acc1")
        key = router._build_key("chat", "direct", "id_123")
        assert key == "test:acc1:chat:direct:id_123"