from __future__ import annotations

import argparse
from unittest.mock import MagicMock

import pytest

from smriti_cli import main as cli_main
from smriti_cli.client import SmritiClient


def _worktree_dict(**overrides):
    base = {
        "id": "11111111-2222-3333-4444-555555555555",
        "repo_id": "space-uuid",
        "agent": "claude-1",
        "path": "/tmp/worktree",
        "branch_name": "smriti/claude-1/abc12345",
        "base_commit_sha": "abc123",
        "status": "active",
        "created_at": "2026-05-04T00:00:00Z",
        "closed_at": None,
    }
    base.update(overrides)
    return base


def test_worktree_open_parser_wiring():
    parser = cli_main._build_parser()

    args = parser.parse_args(
        [
            "worktree",
            "open",
            "my-project",
            "--agent",
            "claude-1",
            "--branch",
            "feature/wt",
            "--base-commit",
            "abc123",
            "--base-path",
            "/tmp/wt",
        ]
    )

    assert args.command == "worktree"
    assert args.subcommand == "open"
    assert args.space == "my-project"
    assert args.agent == "claude-1"
    assert args.branch == "feature/wt"
    assert args.base_commit == "abc123"
    assert args.base_path == "/tmp/wt"
    assert args.func is cli_main.cmd_worktree_open


def test_cmd_worktree_open_calls_client_and_prints_path(capsys: pytest.CaptureFixture[str]):
    client = MagicMock(spec=SmritiClient)
    client.resolve_space.return_value = {"id": "space-uuid", "name": "my-project"}
    client.create_worktree.return_value = _worktree_dict(path="/tmp/wt")
    args = argparse.Namespace(
        space="my-project",
        agent="claude-1",
        branch=None,
        base_commit=None,
        base_path=None,
        json=False,
    )

    cli_main.cmd_worktree_open(client, args)

    out = capsys.readouterr().out
    assert out == "/tmp/wt\n"
    client.resolve_space.assert_called_once_with("my-project")
    client.create_worktree.assert_called_once_with(
        space_id="space-uuid",
        agent="claude-1",
        branch_name=None,
        base_commit_sha=None,
        base_path=None,
    )


def test_cmd_worktree_open_json_path():
    client = MagicMock(spec=SmritiClient)
    client.resolve_space.return_value = {"id": "space-uuid", "name": "my-project"}
    worktree = _worktree_dict(path="/tmp/json-wt")
    client.create_worktree.return_value = worktree
    args = argparse.Namespace(
        space="my-project",
        agent="claude-1",
        branch="feature/wt",
        base_commit="abc123",
        base_path="/tmp/json-wt",
        json=True,
    )
    captured: list[dict] = []
    original = cli_main._print_json
    cli_main._print_json = captured.append
    try:
        cli_main.cmd_worktree_open(client, args)
    finally:
        cli_main._print_json = original

    assert captured == [worktree]
    client.create_worktree.assert_called_once_with(
        space_id="space-uuid",
        agent="claude-1",
        branch_name="feature/wt",
        base_commit_sha="abc123",
        base_path="/tmp/json-wt",
    )


def test_cmd_worktree_list_calls_client(capsys: pytest.CaptureFixture[str]):
    client = MagicMock(spec=SmritiClient)
    client.resolve_space.return_value = {"id": "space-uuid", "name": "my-project"}
    client.list_worktrees.return_value = [_worktree_dict()]
    args = argparse.Namespace(space="my-project", include_closed=False, json=False)

    cli_main.cmd_worktree_list(client, args)

    out = capsys.readouterr().out
    assert "AGENT" in out
    assert "claude-1" in out
    client.list_worktrees.assert_called_once_with("space-uuid", include_closed=False)


def test_list_renders_probe_data(capsys: pytest.CaptureFixture[str]):
    client = MagicMock(spec=SmritiClient)
    client.resolve_space.return_value = {"id": "space-uuid", "name": "my-project"}
    client.list_worktrees.return_value = [
        _worktree_dict(
            probe={
                "dirty_files": 3,
                "ahead": 1,
                "behind": 0,
                "last_commit_sha": "abc1234",
                "last_commit_relative": "5 minutes ago",
            },
        )
    ]
    args = argparse.Namespace(space="my-project", include_closed=False, json=False)

    cli_main.cmd_worktree_list(client, args)

    out = capsys.readouterr().out
    assert "DIRTY" in out
    assert "AHEAD" in out
    assert "3" in out
    assert "+1" in out


def test_list_renders_dash_when_probe_null(capsys: pytest.CaptureFixture[str]):
    client = MagicMock(spec=SmritiClient)
    client.resolve_space.return_value = {"id": "space-uuid", "name": "my-project"}
    client.list_worktrees.return_value = [_worktree_dict(probe=None)]
    args = argparse.Namespace(space="my-project", include_closed=False, json=False)

    cli_main.cmd_worktree_list(client, args)

    out = capsys.readouterr().out
    assert "—" in out


def test_list_renders_negative_ahead_for_behind_only(capsys: pytest.CaptureFixture[str]):
    client = MagicMock(spec=SmritiClient)
    client.resolve_space.return_value = {"id": "space-uuid", "name": "my-project"}
    client.list_worktrees.return_value = [
        _worktree_dict(
            probe={
                "dirty_files": 0,
                "ahead": 0,
                "behind": 2,
                "last_commit_sha": "abc1234",
                "last_commit_relative": "5 minutes ago",
            },
        )
    ]
    args = argparse.Namespace(space="my-project", include_closed=False, json=False)

    cli_main.cmd_worktree_list(client, args)

    out = capsys.readouterr().out
    assert "-2" in out


def test_cmd_worktree_show_calls_client(capsys: pytest.CaptureFixture[str]):
    client = MagicMock(spec=SmritiClient)
    client.get_worktree.return_value = _worktree_dict()
    args = argparse.Namespace(worktree_id="wt-uuid", json=False)

    cli_main.cmd_worktree_show(client, args)

    out = capsys.readouterr().out
    assert "branch: smriti/claude-1/abc12345" in out
    client.get_worktree.assert_called_once_with("wt-uuid")


def test_cmd_worktree_close_calls_client(capsys: pytest.CaptureFixture[str]):
    client = MagicMock(spec=SmritiClient)
    client.close_worktree.return_value = _worktree_dict(status="closed")
    args = argparse.Namespace(worktree_id="wt-uuid", force=True, json=False)

    cli_main.cmd_worktree_close(client, args)

    out = capsys.readouterr().out
    assert "Closed worktree" in out
    client.close_worktree.assert_called_once_with("wt-uuid", force=True)


def test_claim_create_parser_accepts_worktree_flag():
    parser = cli_main._build_parser()

    args = parser.parse_args(
        [
            "claim",
            "create",
            "my-project",
            "--agent",
            "codex-local",
            "--scope",
            "Implement V2",
            "--worktree",
            "wt-uuid",
        ]
    )

    assert args.worktree == "wt-uuid"


def test_cmd_claim_create_passes_worktree_id():
    client = MagicMock(spec=SmritiClient)
    client.resolve_space.return_value = {"id": "space-uuid", "name": "my-project"}
    client.get_head.return_value = {"commit_id": "checkpoint-uuid"}
    client.create_claim.return_value = {
        "id": "claim-uuid",
        "agent": "codex-local",
        "scope": "Implement V2",
        "branch_name": "worktree-v2-binding-and-enrichment",
        "intent_type": "implement",
    }
    args = argparse.Namespace(
        space="my-project",
        agent="codex-local",
        scope="Implement V2",
        branch="worktree-v2-binding-and-enrichment",
        task_id="v2-plan",
        worktree="wt-uuid",
        intent_type="implement",
        ttl=4.0,
        json=False,
    )

    cli_main.cmd_claim_create(client, args)

    client.create_claim.assert_called_once_with(
        space_id="space-uuid",
        agent="codex-local",
        scope="Implement V2",
        branch_name="worktree-v2-binding-and-enrichment",
        base_commit_id="checkpoint-uuid",
        task_id="v2-plan",
        worktree_id="wt-uuid",
        intent_type="implement",
        ttl_hours=4.0,
    )