"""Tests for predefined team mentions and routing behavior.

These tests ensure that mentioning a predefined team:
- Does NOT trigger router routing
- Does cause the TeamBot to respond using its configured team members
"""

from __future__ import annotations

import asyncio
from typing import TYPE_CHECKING, Any
from unittest.mock import AsyncMock, MagicMock, patch

import nio
import pytest

from mindroom.bot import AgentBot, TeamBot
from mindroom.config.agent import AgentConfig, TeamConfig
from mindroom.config.main import Config
from mindroom.config.models import RouterConfig
from mindroom.matrix.identity import MatrixID
from mindroom.matrix.users import AgentMatrixUser
from mindroom.tool_system.worker_routing import get_tool_execution_identity

if TYPE_CHECKING:
    from pathlib import Path


@pytest.fixture
def config_with_team() -> Config:
    """Minimal config with two agents and one predefined team in a room."""
    return Config(
        agents={
            "a1": AgentConfig(display_name="Agent One", role="", rooms=["room_x"]),
            "a2": AgentConfig(display_name="Agent Two", role="", rooms=["room_x"]),
        },
        teams={
            "t1": TeamConfig(
                display_name="Team One",
                role="Test preformed team",
                agents=["a1", "a2"],
                rooms=["room_x"],
                mode="coordinate",
            ),
        },
        router=RouterConfig(model="default"),
    )


def _mock_room(room_id: str, member_ids: list[str]) -> MagicMock:
    room = MagicMock()
    room.room_id = room_id
    room.name = room_id
    room.users = member_ids
    return room


def _mock_event_with_team_mention(team_user_id: str, body: str = "@team please help") -> MagicMock:
    ev = MagicMock()
    ev.sender = "@user:localhost"
    ev.body = body
    ev.event_id = "$evt1"
    ev.source = {
        "content": {
            "body": body,
            "m.mentions": {"user_ids": [team_user_id]},
        },
    }
    return ev


@pytest.mark.asyncio
async def test_router_does_not_route_when_preformed_team_is_mentioned(config_with_team: Config, tmp_path: Path) -> None:
    """Router must not route if the message mentions a predefined team."""
    # Router bot setup
    # Use config-derived IDs to match domain in this environment
    router_user = AgentMatrixUser(
        agent_name="router",
        user_id=config_with_team.ids["router"].full_id,
        display_name="Router",
        password="p",  # noqa: S106
    )
    router = AgentBot(router_user, tmp_path, config_with_team)
    router.client = AsyncMock()

    # Room has router + team + two agents and the human user
    team_user_id = config_with_team.ids["t1"].full_id
    a1_id = config_with_team.ids["a1"].full_id
    a2_id = config_with_team.ids["a2"].full_id
    room = _mock_room("!room:localhost", [router_user.user_id, team_user_id, a1_id, a2_id, "@user:localhost"])

    # Event mentions the team
    event = _mock_event_with_team_mention(team_user_id)

    # Ensure no thread history fetch is attempted
    with (
        patch("mindroom.bot.fetch_thread_history", new=AsyncMock(return_value=[])),
        # Also patch suggest_agent_for_message to detect accidental routing
        patch("mindroom.bot.suggest_agent_for_message", new=AsyncMock(return_value="a1")),
    ):
        await router._on_message(room, event)

    # Router must not send any message (i.e., must not route)
    router.client.room_send.assert_not_called()


@pytest.mark.asyncio
async def test_preformed_team_bot_responds_when_mentioned(config_with_team: Config, tmp_path: Path) -> None:
    """TeamBot should respond with team response when the team is mentioned."""
    team_user = AgentMatrixUser(
        agent_name="t1",
        user_id=config_with_team.ids["t1"].full_id,
        display_name="Team One",
        password="p",  # noqa: S106
    )
    # Convert agent names to MatrixID objects
    team_matrix_ids = [
        MatrixID.from_agent("a1", config_with_team.domain),
        MatrixID.from_agent("a2", config_with_team.domain),
    ]
    bot = TeamBot(
        agent_user=team_user,
        storage_path=tmp_path,
        config=config_with_team,
        rooms=["!room:localhost"],
        team_agents=team_matrix_ids,
        team_mode="coordinate",
        enable_streaming=False,
    )
    bot.client = AsyncMock()

    # Minimal orchestrator stub is fine because we patch team_response
    bot.orchestrator = MagicMock()

    team_user_id = config_with_team.ids["t1"].full_id
    room = _mock_room("!room:localhost", [team_user_id, "@user:localhost"])
    event = _mock_event_with_team_mention(team_user_id)

    # No thread context in this test
    with patch("mindroom.bot.fetch_thread_history", new=AsyncMock(return_value=[])):
        # Patch team_response to avoid invoking Agno, return deterministic text
        async def fake_team_response(*_args: Any, **_kwargs: Any) -> str:  # noqa: ANN401
            return "🤝 Team Response (a1, a2):\n\n**a1**: ok\n\n**a2**: ok"

        with (
            patch("mindroom.bot.team_response", new=fake_team_response),
            patch(
                "mindroom.bot.should_agent_respond",
                return_value=True,
            ),
        ):
            await bot._on_message(room, event)

    # Team bot should have sent exactly one message
    assert bot.client.room_send.call_count == 2  # initial + streaming updates for team
    args, kwargs = bot.client.room_send.call_args
    # kwargs contains content with formatted body
    assert "🤝 Team Response" in kwargs["content"]["formatted_body"]


@pytest.mark.asyncio
async def test_preformed_team_bot_schedules_memory_save_for_all_file_members(
    config_with_team: Config,
    tmp_path: Path,
) -> None:
    """TeamBot should always schedule conversation memory storage for team replies."""
    config_with_team.memory.backend = "mem0"
    config_with_team.agents["a1"].memory_backend = "file"
    config_with_team.agents["a2"].memory_backend = "file"

    team_user = AgentMatrixUser(
        agent_name="t1",
        user_id=config_with_team.ids["t1"].full_id,
        display_name="Team One",
        password="p",  # noqa: S106
    )
    team_matrix_ids = [
        MatrixID.from_agent("a1", config_with_team.domain),
        MatrixID.from_agent("a2", config_with_team.domain),
    ]
    bot = TeamBot(
        agent_user=team_user,
        storage_path=tmp_path,
        config=config_with_team,
        rooms=["!room:localhost"],
        team_agents=team_matrix_ids,
        team_mode="coordinate",
        enable_streaming=False,
    )
    bot.client = AsyncMock()
    bot.orchestrator = MagicMock()
    bot._generate_team_response_helper = AsyncMock()

    store_calls: list[tuple[tuple[Any, ...], dict[str, Any]]] = []
    seen_requesters: list[str | None] = []
    scheduled_tasks: list[asyncio.Task[Any]] = []

    async def fake_store_conversation_memory(*args: Any, **kwargs: Any) -> None:  # noqa: ANN401
        identity = get_tool_execution_identity()
        seen_requesters.append(identity.requester_id if identity is not None else None)
        store_calls.append((args, kwargs))

    def schedule_background_task(
        coro: object,
        name: str | None = None,
        error_handler: object | None = None,  # noqa: ARG001
    ) -> asyncio.Task[Any]:
        assert asyncio.iscoroutine(coro)
        task = asyncio.create_task(coro, name=name)
        scheduled_tasks.append(task)
        return task

    with (
        patch("mindroom.bot.store_conversation_memory", new=fake_store_conversation_memory),
        patch("mindroom.bot.create_background_task", side_effect=schedule_background_task),
    ):
        await bot._generate_response(
            room_id="!room:localhost",
            prompt="@team remember this",
            reply_to_event_id="$evt1",
            thread_id=None,
            thread_history=[],
            user_id="@user:localhost",
        )

    if scheduled_tasks:
        await asyncio.gather(*scheduled_tasks)

    bot._generate_team_response_helper.assert_awaited_once()
    assert len(store_calls) == 1
    assert store_calls[0][0][1] == ["a1", "a2"]
    assert seen_requesters == ["@user:localhost"]


@pytest.mark.asyncio
async def test_preformed_team_reply_chain_uses_existing_thread_root(config_with_team: Config, tmp_path: Path) -> None:
    """TeamBot should continue the resolved thread when mention comes as a plain reply."""
    team_user = AgentMatrixUser(
        agent_name="t1",
        user_id=config_with_team.ids["t1"].full_id,
        display_name="Team One",
        password="p",  # noqa: S106
    )
    team_matrix_ids = [
        MatrixID.from_agent("a1", config_with_team.domain),
        MatrixID.from_agent("a2", config_with_team.domain),
    ]
    bot = TeamBot(
        agent_user=team_user,
        storage_path=tmp_path,
        config=config_with_team,
        rooms=["!room:localhost"],
        team_agents=team_matrix_ids,
        team_mode="coordinate",
        enable_streaming=False,
    )
    bot.client = AsyncMock()
    bot.orchestrator = MagicMock()

    team_user_id = config_with_team.ids["t1"].full_id
    room = _mock_room("!room:localhost", [team_user_id, "@user:localhost"])
    event = MagicMock()
    event.sender = "@user:localhost"
    event.body = "@t1 please continue"
    event.event_id = "$evt_plain_reply"
    event.source = {
        "content": {
            "body": event.body,
            "m.mentions": {"user_ids": [team_user_id]},
            "m.relates_to": {"m.in_reply_to": {"event_id": "$thread_msg"}},
        },
    }

    bot.client.room_get_event = AsyncMock(
        return_value=nio.RoomGetEventResponse.from_dict(
            {
                "content": {
                    "body": "Earlier team message",
                    "msgtype": "m.text",
                    "m.relates_to": {"rel_type": "m.thread", "event_id": "$thread_root"},
                },
                "event_id": "$thread_msg",
                "sender": "@mindroom_t1:localhost",
                "origin_server_ts": 1234567890,
                "room_id": "!room:localhost",
                "type": "m.room.message",
            },
        ),
    )

    with patch("mindroom.bot.fetch_thread_history", new=AsyncMock(return_value=[])):

        async def fake_team_response(*_args: Any, **_kwargs: Any) -> str:  # noqa: ANN401
            return "🤝 Team Response (a1, a2):\n\n**a1**: ok\n\n**a2**: ok"

        with (
            patch("mindroom.bot.team_response", new=fake_team_response),
            patch("mindroom.bot.should_agent_respond", return_value=True),
        ):
            await bot._on_message(room, event)

    assert bot.client.room_send.call_count >= 1
    first_content = bot.client.room_send.call_args_list[0].kwargs["content"]
    assert first_content["m.relates_to"]["rel_type"] == "m.thread"
    assert first_content["m.relates_to"]["event_id"] == "$thread_root"
    assert first_content["m.relates_to"]["m.in_reply_to"]["event_id"] == "$evt_plain_reply"


@pytest.mark.asyncio
async def test_team_does_not_respond_to_different_domain_mention(config_with_team: Config, tmp_path: Path) -> None:
    """TeamBot should NOT respond to mentions of the same username on a different domain.

    This is a security test - @mindroom_t1:evil.org should not trigger @mindroom_t1:localhost.
    """
    team_user = AgentMatrixUser(
        agent_name="t1",
        user_id=config_with_team.ids["t1"].full_id,
        display_name="Team One",
        password="p",  # noqa: S106
    )
    # Convert agent names to MatrixID objects
    team_matrix_ids = [
        MatrixID.from_agent("a1", config_with_team.domain),
        MatrixID.from_agent("a2", config_with_team.domain),
    ]
    bot = TeamBot(
        agent_user=team_user,
        storage_path=tmp_path,
        config=config_with_team,
        rooms=["!room:localhost"],
        team_agents=team_matrix_ids,
        team_mode="coordinate",
        enable_streaming=False,
    )
    bot.client = AsyncMock()
    bot.orchestrator = MagicMock()

    # Craft a mention using a DIFFERENT domain than the bot's MatrixID
    # This simulates someone trying to impersonate the team
    other_domain = "evil.org"
    if team_user.matrix_id.domain == other_domain:
        other_domain = "attacker.com"
    mentioned_id = f"@mindroom_t1:{other_domain}"

    room = _mock_room("!room:localhost", [team_user.user_id, "@user:localhost"])
    event = _mock_event_with_team_mention(mentioned_id, body=f"{mentioned_id} ping")

    with patch("mindroom.bot.fetch_thread_history", new=AsyncMock(return_value=[])):

        async def fake_team_response(*_args: Any, **_kwargs: Any) -> str:  # noqa: ANN401
            return "🤝 Team Response (a1, a2): ok"

        with patch("mindroom.bot.team_response", new=fake_team_response):
            await bot._on_message(room, event)

    # Team bot should NOT have responded - different domain!
    assert bot.client.room_send.call_count == 0
