"""End-to-end unit tests for deepagents-code with fake LLM models."""

import uuid
from collections.abc import Callable, Generator, Sequence
from contextlib import contextmanager
from pathlib import Path
from typing import Any
from unittest.mock import patch

from deepagents.backends import CompositeBackend
from deepagents.backends.filesystem import FilesystemBackend
from langchain_core.callbacks import CallbackManagerForLLMRun
from langchain_core.language_models import LanguageModelInput
from langchain_core.language_models.fake_chat_models import GenericFakeChatModel
from langchain_core.messages import AIMessage, BaseMessage, HumanMessage, SystemMessage
from langchain_core.outputs import ChatResult
from langchain_core.runnables import Runnable
from langchain_core.tools import BaseTool, tool
from langgraph.checkpoint.memory import InMemorySaver
from pydantic import Field

from deepagents_code.agent import create_cli_agent


@tool(description="Sample tool")
def sample_tool(sample_input: str) -> str:
    """A sample tool that returns the input string."""
    return sample_input


class FixedGenericFakeChatModel(GenericFakeChatModel):
    """Fixed version of GenericFakeChatModel that properly handles bind_tools."""

    captured_calls: list[tuple[list[Any], Any]] = Field(default_factory=list)

    def bind_tools(
        self,
        tools: Sequence[dict[str, Any] | type | Callable | BaseTool],  # noqa: ARG002
        *,
        tool_choice: str | None = None,  # noqa: ARG002
        **kwargs: Any,  # noqa: ARG002
    ) -> Runnable[LanguageModelInput, AIMessage]:
        """Override bind_tools to return self."""
        return self

    def _generate(
        self,
        messages: list[BaseMessage],
        stop: list[str] | None = None,
        run_manager: CallbackManagerForLLMRun | None = None,
        **kwargs: Any,
    ) -> ChatResult:
        """Override _generate to capture inputs and outputs."""
        result = super()._generate(
            messages, stop=stop, run_manager=run_manager, **kwargs
        )
        self.captured_calls.append((messages, result))
        return result


@contextmanager
def mock_settings(
    tmp_path: Path, assistant_id: str = "test-agent"
) -> Generator[Path, None, None]:
    """Context manager for patching CLI settings with temporary directories.

    Args:
        tmp_path: Temporary directory path (typically from pytest's tmp_path fixture)
        assistant_id: Agent identifier for directory setup

    Yields:
        The agent directory path
    """
    # Setup directory structure
    agent_dir = tmp_path / "agents" / assistant_id
    agent_dir.mkdir(parents=True)
    agent_md = agent_dir / "agent.md"
    agent_md.write_text("# Test Agent\nTest agent instructions.")

    skills_dir = tmp_path / "skills"
    skills_dir.mkdir(parents=True)

    # Patch settings
    with patch("deepagents_code.agent.settings") as mock_settings_obj:
        mock_settings_obj.user_deepagents_dir = tmp_path / "agents"
        mock_settings_obj.ensure_agent_dir.return_value = agent_dir
        mock_settings_obj.ensure_user_skills_dir.return_value = skills_dir
        mock_settings_obj.get_project_skills_dir.return_value = None

        # Mock methods that get called during agent execution to return
        # real Path objects. This prevents MagicMock objects from being
        # stored in state (which would fail serialization)
        def get_user_agent_md_path(agent_id: str) -> Path:
            return tmp_path / "agents" / agent_id / "agent.md"

        def get_agent_dir(agent_id: str) -> Path:
            return tmp_path / "agents" / agent_id

        mock_settings_obj.get_user_agent_md_path = get_user_agent_md_path
        mock_settings_obj.get_project_agent_md_path.return_value = []
        mock_settings_obj.get_agent_dir = get_agent_dir
        mock_settings_obj.project_root = None

        # Model identity settings (used in system prompt generation)
        mock_settings_obj.model_name = None
        mock_settings_obj.model_provider = None
        mock_settings_obj.model_context_limit = None

        yield agent_dir


class TestDeepAgentsCLIEndToEnd:
    """Test suite for end-to-end deepagents-code functionality with fake LLM."""

    def test_cli_agent_with_fake_llm_basic(self, tmp_path: Path) -> None:
        """Test basic CLI agent functionality with a fake LLM model.

        This test verifies that a CLI agent can be created and invoked with
        a fake LLM model that returns predefined responses.
        """
        with mock_settings(tmp_path):
            # Create a fake model that returns predefined messages
            model = FixedGenericFakeChatModel(
                messages=iter(
                    [
                        AIMessage(
                            content="I'll help you with that.",
                            tool_calls=[
                                {
                                    "name": "write_todos",
                                    "args": {"todos": []},
                                    "id": "call_1",
                                    "type": "tool_call",
                                }
                            ],
                        ),
                        AIMessage(
                            content="Task completed successfully!",
                        ),
                    ]
                )
            )

            # Create a CLI agent with the fake model
            agent, _ = create_cli_agent(
                model=model,
                assistant_id="test-agent",
                tools=[],
                checkpointer=InMemorySaver(),
            )

            # Invoke the agent with a simple message
            result = agent.invoke(
                {"messages": [HumanMessage(content="Hello, agent!")]},
                {"configurable": {"thread_id": str(uuid.uuid4())}},
            )

            # Verify the agent executed correctly
            assert "messages" in result
            assert len(result["messages"]) > 0

            # Verify we got AI responses
            ai_messages = [msg for msg in result["messages"] if msg.type == "ai"]
            assert len(ai_messages) > 0

            # Verify the final AI message contains our expected content
            final_ai_message = ai_messages[-1]
            assert "Task completed successfully!" in final_ai_message.content

    def test_cli_agent_summarizes(self, tmp_path: Path) -> None:
        """Test summarization."""
        with mock_settings(tmp_path):
            model = FixedGenericFakeChatModel(
                messages=iter(
                    [
                        AIMessage(content="summary goes here"),
                        AIMessage(content="response"),
                    ]
                )
            )
            model.profile = {"max_input_tokens": 200_000}

            # Create a CLI agent with the fake model
            agent, backend = create_cli_agent(
                model=model,
                assistant_id="test-agent",
                tools=[],
                checkpointer=InMemorySaver(),
            )

            # Invoke the agent
            thread_id = str(uuid.uuid4())
            text_10_000_tokens = "x" * 10_000 * 4
            text_50_000_tokens = "x" * 50_000 * 4
            input_messages = [
                HumanMessage(content=text_10_000_tokens),
                AIMessage(content=text_50_000_tokens),  # 60,000 tokens
                HumanMessage(content=text_10_000_tokens),
                AIMessage(content=text_50_000_tokens),  # 120,000 tokens
                HumanMessage(content=text_10_000_tokens),
                AIMessage(content=text_50_000_tokens),  # 180,000 tokens (summarizes)
                HumanMessage(content="query"),
            ]
            result = agent.invoke(
                {"messages": input_messages},
                {"configurable": {"thread_id": thread_id}},
            )
            assert len(result["messages"]) == 8  # 7 inputs + response
            assert result["messages"][-1].content == "response"

            # two calls: one to summarize, one for response
            assert len(model.captured_calls) == 2

            # summarization call
            summarization_input_messages, summarization_response = model.captured_calls[
                0
            ]
            assert len(summarization_input_messages) == 1
            assert "Messages to summarize:" in summarization_input_messages[0].content
            assert (
                summarization_response.generations[0].message.content
                == "summary goes here"
            )

            # model call on reduced context
            summarized_messages, agent_response = model.captured_calls[1]
            assert len(summarized_messages) < len(input_messages)
            assert isinstance(summarized_messages[0], SystemMessage)
            summary_message = summarized_messages[1]
            assert isinstance(summary_message, HumanMessage)
            assert "summary goes here" in summary_message.content
            assert agent_response.generations[0].message.content == "response"

            # Verify conversation history was offloaded to backend
            assert backend.ls("/conversation_history/").entries

    def test_cli_agent_with_fake_llm_with_tools(self, tmp_path: Path) -> None:
        """Test CLI agent with tools using a fake LLM model.

        This test verifies that a CLI agent can handle tool calls correctly
        when using a fake LLM model.
        """
        with mock_settings(tmp_path):
            # Create a fake model that calls sample_tool
            model = FixedGenericFakeChatModel(
                messages=iter(
                    [
                        AIMessage(
                            content="",
                            tool_calls=[
                                {
                                    "name": "sample_tool",
                                    "args": {"sample_input": "test input"},
                                    "id": "call_1",
                                    "type": "tool_call",
                                }
                            ],
                        ),
                        AIMessage(
                            content="I called the sample_tool with 'test input'.",
                        ),
                    ]
                )
            )

            # Create a CLI agent with the fake model and sample_tool
            agent, _ = create_cli_agent(
                model=model,
                assistant_id="test-agent",
                tools=[sample_tool],
                checkpointer=InMemorySaver(),
            )

            # Invoke the agent
            result = agent.invoke(
                {"messages": [HumanMessage(content="Use the sample tool")]},
                {"configurable": {"thread_id": "test-thread-2"}},
            )

            # Verify the agent executed correctly
            assert "messages" in result

            # Verify tool was called
            tool_messages = [msg for msg in result["messages"] if msg.type == "tool"]
            assert len(tool_messages) > 0

            # Verify the tool message contains our expected input
            assert any("test input" in msg.content for msg in tool_messages)

    def test_cli_agent_with_fake_llm_filesystem_tool(self, tmp_path: Path) -> None:
        """Test CLI agent with filesystem tools using a fake LLM model.

        This test verifies that a CLI agent can use the built-in filesystem
        tools (ls, read_file, etc.) with a fake LLM model.
        """
        with mock_settings(tmp_path):
            # Create a test file to list
            test_file = tmp_path / "test.txt"
            test_file.write_text("test content")

            # Create a fake model that uses filesystem tools
            model = FixedGenericFakeChatModel(
                messages=iter(
                    [
                        AIMessage(
                            content="",
                            tool_calls=[
                                {
                                    "name": "ls",
                                    "args": {"path": str(tmp_path)},
                                    "id": "call_1",
                                    "type": "tool_call",
                                }
                            ],
                        ),
                        AIMessage(
                            content="I've listed the files in the directory.",
                        ),
                    ]
                )
            )

            # Create a CLI agent with the fake model
            agent, _ = create_cli_agent(
                model=model,
                assistant_id="test-agent",
                tools=[],
                checkpointer=InMemorySaver(),
            )

            # Invoke the agent
            result = agent.invoke(
                {"messages": [HumanMessage(content="List files")]},
                {"configurable": {"thread_id": "test-thread-3"}},
            )

            # Verify the agent executed correctly
            assert "messages" in result

            # Verify ls tool was called
            tool_messages = [msg for msg in result["messages"] if msg.type == "tool"]
            assert len(tool_messages) > 0

    def test_cli_agent_with_fake_llm_multiple_tool_calls(self, tmp_path: Path) -> None:
        """Test CLI agent with multiple tool calls using a fake LLM model.

        This test verifies that a CLI agent can handle multiple sequential
        tool calls with a fake LLM model.
        """
        with mock_settings(tmp_path):
            # Create a fake model that makes multiple tool calls
            model = FixedGenericFakeChatModel(
                messages=iter(
                    [
                        AIMessage(
                            content="",
                            tool_calls=[
                                {
                                    "name": "sample_tool",
                                    "args": {"sample_input": "first call"},
                                    "id": "call_1",
                                    "type": "tool_call",
                                }
                            ],
                        ),
                        AIMessage(
                            content="",
                            tool_calls=[
                                {
                                    "name": "sample_tool",
                                    "args": {"sample_input": "second call"},
                                    "id": "call_2",
                                    "type": "tool_call",
                                }
                            ],
                        ),
                        AIMessage(
                            content="I completed both tool calls successfully.",
                        ),
                    ]
                )
            )

            # Create a CLI agent with the fake model and sample_tool
            agent, _ = create_cli_agent(
                model=model,
                assistant_id="test-agent",
                tools=[sample_tool],
                checkpointer=InMemorySaver(),
            )

            # Invoke the agent
            result = agent.invoke(
                {"messages": [HumanMessage(content="Use sample tool twice")]},
                {"configurable": {"thread_id": "test-thread-4"}},
            )

            # Verify the agent executed correctly
            assert "messages" in result

            # Verify multiple tool calls occurred
            tool_messages = [msg for msg in result["messages"] if msg.type == "tool"]
            assert len(tool_messages) >= 2

            # Verify both inputs were used
            tool_contents = [msg.content for msg in tool_messages]
            assert any("first call" in content for content in tool_contents)
            assert any("second call" in content for content in tool_contents)

    def test_cli_agent_backend_setup(self, tmp_path: Path) -> None:
        """Test that CLI agent creates the correct backend setup.

        This test verifies that the backend is properly configured with
        a CompositeBackend containing a FilesystemBackend.
        """
        with mock_settings(tmp_path):
            # Create a simple fake model
            model = FixedGenericFakeChatModel(
                messages=iter(
                    [
                        AIMessage(content="Done."),
                    ]
                )
            )

            # Create a CLI agent
            _, backend = create_cli_agent(
                model=model,
                assistant_id="test-agent",
                tools=[],
                checkpointer=InMemorySaver(),
            )

            assert isinstance(backend, CompositeBackend)
            assert isinstance(backend.default, FilesystemBackend)
