"""发送服务回归测试。"""

from types import SimpleNamespace
from typing import Any, Dict, List

import pytest

from src.chat.message_receive.chat_manager import BotChatSession
from src.common.data_models.message_component_data_model import MessageSequence, TextComponent
from src.services import send_service


class _FakePlatformIOManager:
    """用于测试的 Platform IO 管理器假对象。"""

    def __init__(self, delivery_batch: Any) -> None:
        self._delivery_batch = delivery_batch
        self.ensure_calls = 0
        self.sent_messages: List[Dict[str, Any]] = []

    async def ensure_send_pipeline_ready(self) -> None:
        self.ensure_calls += 1

    def build_route_key_from_message(self, message: Any) -> Any:
        del message
        return SimpleNamespace(platform="qq")

    async def send_message(self, message: Any, route_key: Any, metadata: Dict[str, Any]) -> Any:
        self.sent_messages.append(
            {
                "message": message,
                "route_key": route_key,
                "metadata": metadata,
            }
        )
        return self._delivery_batch


def _build_private_stream() -> BotChatSession:
    return BotChatSession(
        session_id="test-session",
        platform="qq",
        user_id="target-user",
        group_id=None,
    )


def _build_group_stream() -> BotChatSession:
    return BotChatSession(
        session_id="group-session",
        platform="qq",
        user_id="target-user",
        group_id="target-group",
    )


def test_inherit_platform_io_route_metadata_falls_back_to_bot_account(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.setattr(send_service, "get_bot_account", lambda platform: "bot-qq" if platform == "qq" else "")

    metadata = send_service._inherit_platform_io_route_metadata(_build_private_stream())

    assert metadata["platform_io_account_id"] == "bot-qq"
    assert metadata["platform_io_target_user_id"] == "target-user"


@pytest.mark.asyncio
async def test_text_to_stream_delegates_to_platform_io(monkeypatch: pytest.MonkeyPatch) -> None:
    fake_manager = _FakePlatformIOManager(
        delivery_batch=SimpleNamespace(
            has_success=True,
            sent_receipts=[SimpleNamespace(driver_id="plugin.qq.sender")],
            failed_receipts=[],
            route_key=SimpleNamespace(platform="qq"),
        )
    )
    stored_messages: List[Any] = []

    monkeypatch.setattr(send_service, "get_platform_io_manager", lambda: fake_manager)
    monkeypatch.setattr(send_service, "get_bot_account", lambda platform: "bot-qq")
    monkeypatch.setattr(
        send_service._chat_manager,
        "get_session_by_session_id",
        lambda stream_id: _build_private_stream() if stream_id == "test-session" else None,
    )
    monkeypatch.setattr(
        send_service.MessageUtils,
        "store_message_to_db",
        lambda message: stored_messages.append(message),
    )

    result = await send_service.text_to_stream(text="你好", stream_id="test-session")

    assert result is True
    assert fake_manager.ensure_calls == 1
    assert len(fake_manager.sent_messages) == 1
    assert fake_manager.sent_messages[0]["metadata"] == {"show_log": False}
    assert len(stored_messages) == 1


@pytest.mark.asyncio
async def test_text_to_stream_returns_false_when_platform_io_fails(monkeypatch: pytest.MonkeyPatch) -> None:
    fake_manager = _FakePlatformIOManager(
        delivery_batch=SimpleNamespace(
            has_success=False,
            sent_receipts=[],
            failed_receipts=[
                SimpleNamespace(
                    driver_id="plugin.qq.sender",
                    status="failed",
                    error="network error",
                )
            ],
            route_key=SimpleNamespace(platform="qq"),
        )
    )

    monkeypatch.setattr(send_service, "get_platform_io_manager", lambda: fake_manager)
    monkeypatch.setattr(send_service, "get_bot_account", lambda platform: "bot-qq")
    monkeypatch.setattr(
        send_service._chat_manager,
        "get_session_by_session_id",
        lambda stream_id: _build_private_stream() if stream_id == "test-session" else None,
    )

    result = await send_service.text_to_stream(text="发送失败", stream_id="test-session")

    assert result is False
    assert fake_manager.ensure_calls == 1
    assert len(fake_manager.sent_messages) == 1


@pytest.mark.asyncio
async def test_private_outbound_message_preserves_bot_sender_and_receiver_user(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.setattr(send_service, "get_bot_account", lambda platform: "bot-qq")
    monkeypatch.setattr(
        send_service._chat_manager,
        "get_session_by_session_id",
        lambda stream_id: _build_private_stream() if stream_id == "test-session" else None,
    )

    outbound_message = send_service._build_outbound_session_message(
        message_sequence=MessageSequence(components=[TextComponent(text="你好")]),
        stream_id="test-session",
        display_message="你好",
    )

    assert outbound_message is not None
    maim_message = await outbound_message.to_maim_message()

    assert maim_message.message_info.user_info is not None
    assert maim_message.message_info.user_info.user_id == "bot-qq"
    assert maim_message.message_info.group_info is None
    assert maim_message.message_info.sender_info is not None
    assert maim_message.message_info.sender_info.user_info is not None
    assert maim_message.message_info.sender_info.user_info.user_id == "bot-qq"
    assert maim_message.message_info.receiver_info is not None
    assert maim_message.message_info.receiver_info.user_info is not None
    assert maim_message.message_info.receiver_info.user_info.user_id == "target-user"


@pytest.mark.asyncio
async def test_group_outbound_message_preserves_bot_sender_and_target_group(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.setattr(send_service, "get_bot_account", lambda platform: "bot-qq")
    monkeypatch.setattr(
        send_service._chat_manager,
        "get_session_by_session_id",
        lambda stream_id: _build_group_stream() if stream_id == "group-session" else None,
    )

    outbound_message = send_service._build_outbound_session_message(
        message_sequence=MessageSequence(components=[TextComponent(text="大家好")]),
        stream_id="group-session",
        display_message="大家好",
    )

    assert outbound_message is not None
    maim_message = await outbound_message.to_maim_message()

    assert maim_message.message_info.user_info is not None
    assert maim_message.message_info.user_info.user_id == "bot-qq"
    assert maim_message.message_info.group_info is not None
    assert maim_message.message_info.group_info.group_id == "target-group"
    assert maim_message.message_info.receiver_info is not None
    assert maim_message.message_info.receiver_info.group_info is not None
    assert maim_message.message_info.receiver_info.group_info.group_id == "target-group"