from __future__ import annotations

from unittest.mock import MagicMock, patch

from pyratatui import Rect

from pygittools.tui.keys import key
from pygittools.tui.tasks_list import TasksBoardList, _Row
from pygittools.tui.types import PageContext


def test_draw_error_offsets_list_selection() -> None:
    tasks = TasksBoardList()
    tasks._rows = [
        _Row(kind="board", status=None, task_ref=None, label="Board: Tasks (0 tasks)"),
        _Row(kind="status_header", status="todo", task_ref=None, label="[v] Todo (0)"),
    ]
    tasks._list_state.select(0)
    tasks._error = "sync failed"
    frame = MagicMock()
    tasks.draw(frame, Rect(0, 0, 80, 20), highlight=True)
    assert frame.render_stateful_list.called
    list_state = frame.render_stateful_list.call_args.args[2]
    assert list_state.selected == 1
    assert (tasks._list_state.selected or 0) == 0


def test_collapse_rebuilds_without_git(page_ctx: PageContext) -> None:
    tasks = TasksBoardList()
    tasks._boards = [{"name": "Tasks", "description": "", "task_count": 1}]
    tasks._board_index = 0
    tasks._board_name = "Tasks"
    tasks._columns = [
        {
            "status": "TODO",
            "label": "Todo",
            "tasks": [{"title": "one", "ref": "refs/tasks/one"}],
        },
    ]
    tasks._rebuild_rows()
    header = next(index for index, row in enumerate(tasks._rows) if row.kind == "status_header")
    tasks._list_state.select(header)
    with (
        patch("pygittools.tui.tasks_list.list_boards") as list_boards,
        patch("pygittools.tui.tasks_list.get_board_tasks_grouped") as grouped,
    ):
        tasks.handle_nav_key(key(" "), page_ctx.repo)
    list_boards.assert_not_called()
    grouped.assert_not_called()
    assert all(row.kind != "task" for row in tasks._rows)