"""CLI tests for tmuxp shell."""

from __future__ import annotations

import contextlib
import io
import subprocess
import typing as t

import pytest

from tmuxp import cli, exc

if t.TYPE_CHECKING:
    import pathlib

    from libtmux.server import Server
    from libtmux.session import Session


class CLIShellFixture(t.NamedTuple):
    """Test fixture for tmuxp shell tests."""

    # pytest (internal): Test fixture name
    test_id: str

    # test params
    cli_cmd: list[str]
    cli_args: list[str]
    inputs: list[t.Any]
    env: dict[str, str]
    expected_output: str


TEST_SHELL_FIXTURES: list[CLIShellFixture] = [
    # Regular shell command
    CLIShellFixture(
        test_id="print-socket-name",
        cli_cmd=["shell"],
        cli_args=["-L{SOCKET_NAME}", "-c", "print(str(server.socket_name))"],
        inputs=[],
        env={},
        expected_output="{SERVER_SOCKET_NAME}",
    ),
    CLIShellFixture(
        test_id="print-session-name",
        cli_cmd=["shell"],
        cli_args=[
            "-L{SOCKET_NAME}",
            "{SESSION_NAME}",
            "-c",
            "print(session.name)",
        ],
        inputs=[],
        env={},
        expected_output="{SESSION_NAME}",
    ),
    CLIShellFixture(
        test_id="print-has-session",
        cli_cmd=["shell"],
        cli_args=[
            "-L{SOCKET_NAME}",
            "{SESSION_NAME}",
            "{WINDOW_NAME}",
            "-c",
            "print(server.has_session(session.name))",
        ],
        inputs=[],
        env={},
        expected_output="True",
    ),
    CLIShellFixture(
        test_id="print-window-name",
        cli_cmd=["shell"],
        cli_args=[
            "-L{SOCKET_NAME}",
            "{SESSION_NAME}",
            "{WINDOW_NAME}",
            "-c",
            "print(window.name)",
        ],
        inputs=[],
        env={},
        expected_output="{WINDOW_NAME}",
    ),
    CLIShellFixture(
        test_id="print-pane-id",
        cli_cmd=["shell"],
        cli_args=[
            "-L{SOCKET_NAME}",
            "{SESSION_NAME}",
            "{WINDOW_NAME}",
            "-c",
            "print(pane.id)",
        ],
        inputs=[],
        env={},
        expected_output="{PANE_ID}",
    ),
    CLIShellFixture(
        test_id="print-pane-id-obeys-tmux-pane-env-var",
        cli_cmd=["shell"],
        cli_args=[
            "-L{SOCKET_NAME}",
            "-c",
            "print(pane.id)",
        ],
        inputs=[],
        env={"TMUX_PANE": "{PANE_ID}"},
        expected_output="{PANE_ID}",
    ),
    # Shell with --pdb
    CLIShellFixture(
        test_id="print-socket-name-pdb",
        cli_cmd=["shell", "--pdb"],
        cli_args=["-L{SOCKET_NAME}", "-c", "print(str(server.socket_name))"],
        inputs=[],
        env={},
        expected_output="{SERVER_SOCKET_NAME}",
    ),
    CLIShellFixture(
        test_id="print-session-name-pdb",
        cli_cmd=["shell", "--pdb"],
        cli_args=[
            "-L{SOCKET_NAME}",
            "{SESSION_NAME}",
            "-c",
            "print(session.name)",
        ],
        inputs=[],
        env={},
        expected_output="{SESSION_NAME}",
    ),
    CLIShellFixture(
        test_id="print-has-session-pdb",
        cli_cmd=["shell", "--pdb"],
        cli_args=[
            "-L{SOCKET_NAME}",
            "{SESSION_NAME}",
            "{WINDOW_NAME}",
            "-c",
            "print(server.has_session(session.name))",
        ],
        inputs=[],
        env={},
        expected_output="True",
    ),
    CLIShellFixture(
        test_id="print-window-name-pdb",
        cli_cmd=["shell", "--pdb"],
        cli_args=[
            "-L{SOCKET_NAME}",
            "{SESSION_NAME}",
            "{WINDOW_NAME}",
            "-c",
            "print(window.name)",
        ],
        inputs=[],
        env={},
        expected_output="{WINDOW_NAME}",
    ),
    CLIShellFixture(
        test_id="print-pane-id-pdb",
        cli_cmd=["shell", "--pdb"],
        cli_args=[
            "-L{SOCKET_NAME}",
            "{SESSION_NAME}",
            "{WINDOW_NAME}",
            "-c",
            "print(pane.id)",
        ],
        inputs=[],
        env={},
        expected_output="{PANE_ID}",
    ),
    CLIShellFixture(
        test_id="print-pane-id-obeys-tmux-pane-env-var-pdb",
        cli_cmd=["shell", "--pdb"],
        cli_args=[
            "-L{SOCKET_NAME}",
            "-c",
            "print(pane.id)",
        ],
        inputs=[],
        env={"TMUX_PANE": "{PANE_ID}"},
        expected_output="{PANE_ID}",
    ),
]


@pytest.mark.parametrize(
    list(CLIShellFixture._fields),
    TEST_SHELL_FIXTURES,
    ids=[test.test_id for test in TEST_SHELL_FIXTURES],
)
def test_shell(
    test_id: str,
    cli_cmd: list[str],
    cli_args: list[str],
    inputs: list[t.Any],
    env: dict[str, str],
    expected_output: str,
    server: Server,
    session: Session,
    tmp_path: pathlib.Path,
    monkeypatch: pytest.MonkeyPatch,
    capsys: pytest.CaptureFixture[str],
) -> None:
    """CLI tests for tmuxp shell."""
    monkeypatch.setenv("HOME", str(tmp_path))
    window_name = "my_window"
    window = session.new_window(window_name=window_name)
    window.split()

    assert window.active_pane is not None

    template_ctx = {
        "SOCKET_NAME": server.socket_name,
        "SESSION_NAME": session.name,
        "WINDOW_NAME": window_name,
        "PANE_ID": window.active_pane.id,
        "SERVER_SOCKET_NAME": server.socket_name,
    }

    cli_args = cli_cmd + [cli_arg.format(**template_ctx) for cli_arg in cli_args]

    for k, v in env.items():
        monkeypatch.setenv(k, v.format(**template_ctx))

    monkeypatch.chdir(tmp_path)

    cli.cli(cli_args)
    result = capsys.readouterr()
    assert expected_output.format(**template_ctx) in result.out


class CLIShellTargetMissingFixture(t.NamedTuple):
    """Test fixture for tmuxp shell target missing tests."""

    test_id: str
    cli_cmd: list[str]
    cli_args: list[str]
    inputs: list[t.Any]
    env: dict[t.Any, t.Any]
    template_ctx: dict[str, str]
    exception: type[exc.TmuxpException | subprocess.CalledProcessError]
    message: str


TEST_SHELL_TARGET_MISSING_FIXTURES: list[CLIShellTargetMissingFixture] = [
    # Regular shell command
    CLIShellTargetMissingFixture(
        test_id="nonexistent_socket",
        cli_cmd=["shell"],
        cli_args=["-LDoesNotExist", "-c", "print(str(server.socket_name))"],
        inputs=[],
        env={},
        template_ctx={},
        exception=subprocess.CalledProcessError,
        message=r".*DoesNotExist.*",
    ),
    CLIShellTargetMissingFixture(
        test_id="nonexistent_session",
        cli_cmd=["shell"],
        cli_args=[
            "-L{SOCKET_NAME}",
            "nonexistent_session",
            "-c",
            "print(str(server.socket_name))",
        ],
        inputs=[],
        env={},
        template_ctx={"session_name": "nonexistent_session"},
        exception=exc.TmuxpException,
        message="Session not found: nonexistent_session",
    ),
    CLIShellTargetMissingFixture(
        test_id="nonexistent_window",
        cli_cmd=["shell"],
        cli_args=[
            "-L{SOCKET_NAME}",
            "{SESSION_NAME}",
            "nonexistent_window",
            "-c",
            "print(str(server.socket_name))",
        ],
        inputs=[],
        env={},
        template_ctx={"window_name": "nonexistent_window"},
        exception=exc.TmuxpException,
        message="Window not found: {WINDOW_NAME}",
    ),
    # Shell with --pdb
    CLIShellTargetMissingFixture(
        test_id="nonexistent_socket_pdb",
        cli_cmd=["shell", "--pdb"],
        cli_args=["-LDoesNotExist", "-c", "print(str(server.socket_name))"],
        inputs=[],
        env={},
        template_ctx={},
        exception=subprocess.CalledProcessError,
        message=r".*DoesNotExist.*",
    ),
    CLIShellTargetMissingFixture(
        test_id="nonexistent_session_pdb",
        cli_cmd=["shell", "--pdb"],
        cli_args=[
            "-L{SOCKET_NAME}",
            "nonexistent_session",
            "-c",
            "print(str(server.socket_name))",
        ],
        inputs=[],
        env={},
        template_ctx={"session_name": "nonexistent_session"},
        exception=exc.TmuxpException,
        message="Session not found: nonexistent_session",
    ),
    CLIShellTargetMissingFixture(
        test_id="nonexistent_window_pdb",
        cli_cmd=["shell", "--pdb"],
        cli_args=[
            "-L{SOCKET_NAME}",
            "{SESSION_NAME}",
            "nonexistent_window",
            "-c",
            "print(str(server.socket_name))",
        ],
        inputs=[],
        env={},
        template_ctx={"window_name": "nonexistent_window"},
        exception=exc.TmuxpException,
        message="Window not found: {WINDOW_NAME}",
    ),
]


@pytest.mark.parametrize(
    list(CLIShellTargetMissingFixture._fields),
    TEST_SHELL_TARGET_MISSING_FIXTURES,
    ids=[test.test_id for test in TEST_SHELL_TARGET_MISSING_FIXTURES],
)
def test_shell_target_missing(
    test_id: str,
    cli_cmd: list[str],
    cli_args: list[str],
    inputs: list[t.Any],
    env: dict[t.Any, t.Any],
    template_ctx: dict[str, str],
    exception: type[exc.TmuxpException | subprocess.CalledProcessError],
    message: str,
    server: Server,
    session: Session,
    tmp_path: pathlib.Path,
    monkeypatch: pytest.MonkeyPatch,
    capsys: pytest.CaptureFixture[str],
) -> None:
    """CLI tests for tmuxp shell when target is not specified."""
    monkeypatch.setenv("HOME", str(tmp_path))
    window_name = "my_window"
    window = session.new_window(window_name=window_name)
    window.split()

    assert server.socket_name is not None
    assert session.name is not None

    template_ctx.update(
        {
            "SOCKET_NAME": server.socket_name,
            "SESSION_NAME": session.name,
            "WINDOW_NAME": template_ctx.get("window_name", window_name),
        },
    )
    cli_args = cli_cmd + [cli_arg.format(**template_ctx) for cli_arg in cli_args]

    for k, v in env.items():
        monkeypatch.setenv(k, v.format(**template_ctx))

    monkeypatch.chdir(tmp_path)

    if exception is not None:
        with pytest.raises(exception, match=message.format(**template_ctx)):
            cli.cli(cli_args)
    else:
        cli.cli(cli_args)
        result = capsys.readouterr()
        assert message.format(**template_ctx) in result.out


class CLIShellInteractiveFixture(t.NamedTuple):
    """Test fixture for tmuxp shell interactive tests."""

    test_id: str
    cli_cmd: list[str]
    cli_args: list[str]
    inputs: list[t.Any]
    env: dict[str, str]
    message: str


TEST_SHELL_INTERACTIVE_FIXTURES: list[CLIShellInteractiveFixture] = [
    CLIShellInteractiveFixture(
        test_id="basic_interactive",
        cli_cmd=["shell", "--code"],
        cli_args=[
            "-L{SOCKET_NAME}",
        ],
        inputs=[],
        env={},
        message="(InteractiveConsole)",
    ),
    CLIShellInteractiveFixture(
        test_id="interactive_with_pane_id",
        cli_cmd=["shell", "--code"],
        cli_args=[
            "-L{SOCKET_NAME}",
        ],
        inputs=[],
        env={"PANE_ID": "{PANE_ID}"},
        message="(InteractiveConsole)",
    ),
]


@pytest.mark.parametrize(
    list(CLIShellInteractiveFixture._fields),
    TEST_SHELL_INTERACTIVE_FIXTURES,
    ids=[test.test_id for test in TEST_SHELL_INTERACTIVE_FIXTURES],
)
def test_shell_interactive(
    test_id: str,
    cli_cmd: list[str],
    cli_args: list[str],
    inputs: list[t.Any],
    env: dict[str, str],
    message: str,
    server: Server,
    session: Session,
    tmp_path: pathlib.Path,
    monkeypatch: pytest.MonkeyPatch,
    capsys: pytest.CaptureFixture[str],
) -> None:
    """CLI tests for tmuxp shell when shell is specified."""
    monkeypatch.setenv("HOME", str(tmp_path))
    window_name = "my_window"
    window = session.new_window(window_name=window_name)
    window.split()

    assert window.active_pane is not None

    template_ctx = {
        "SOCKET_NAME": server.socket_name,
        "SESSION_NAME": session.name,
        "WINDOW_NAME": window_name,
        "PANE_ID": window.active_pane.id,
        "SERVER_SOCKET_NAME": server.socket_name,
    }

    cli_args = cli_cmd + [cli_arg.format(**template_ctx) for cli_arg in cli_args]

    for k, v in env.items():
        monkeypatch.setenv(k, v.format(**template_ctx))

    monkeypatch.chdir(tmp_path)
    monkeypatch.setattr("sys.stdin", io.StringIO("exit()\r"))
    with contextlib.suppress(SystemExit):
        cli.cli(cli_args)

    result = capsys.readouterr()
    assert message.format(**template_ctx) in result.err