import { afterEach, describe, expect, it, vi } from "vitest";
import { cleanup, fireEvent, render, screen, waitFor, within } from "@testing-library/react";
import { StateProvider } from "@/components/state-provider";
import { ToastProvider } from "@/components/toast-provider";
import { TaskSwitcher, type TaskSwitcherItem } from "./task-switcher";
import { createTaskLinkSelectAction } from "./task-switcher-context-menu";
import type { GroupedSidebarList } from "@/lib/sidebar/apply-view";

afterEach(() => cleanup());

function Providers({ children }: { children: React.ReactNode }) {
  return (
    <StateProvider>
      <ToastProvider>{children}</ToastProvider>
    </StateProvider>
  );
}

function item(
  id: string,
  parentTaskId?: string,
  overrides: Partial<TaskSwitcherItem> = {},
): TaskSwitcherItem {
  return { id, title: id, state: "IN_PROGRESS", parentTaskId, ...overrides };
}

// root → child → grandchild (depth 2)
const ROOT = item("Root");
const CHILD = item("Child", "Root");
const GRANDCHILD = item("Grandchild", "Child");
const TASK_A = item("Task A");
const TASK_B = item("Task B");

function grouped(): GroupedSidebarList {
  return {
    groups: [{ key: "__all__", label: "All", tasks: [ROOT] }],
    subTasksByParentId: new Map([
      ["Root", [CHILD]],
      ["Child", [GRANDCHILD]],
    ]),
  };
}

const TEST_WORKFLOW_ID = "workflow-1";
const DETACH_MENU_LABEL = "Detach from parent";

function renderSwitcher(collapsedSubtaskParentIds: string[] = []) {
  return render(
    <Providers>
      <TaskSwitcher
        grouped={grouped()}
        activeTaskId={null}
        selectedTaskId={null}
        onSelectTask={vi.fn()}
        onToggleSubtasks={vi.fn()}
        collapsedSubtaskParentIds={collapsedSubtaskParentIds}
      />
    </Providers>,
  );
}

function blockDepth(container: HTMLElement, taskId: string): string | null {
  return (
    container
      .querySelector(`[data-testid='sortable-task-block'][data-task-id='${taskId}']`)
      ?.getAttribute("data-depth") ?? null
  );
}

describe("TaskSwitcher — nested subtasks beyond depth 1", () => {
  it("renders the full tree (root, child, grandchild)", () => {
    renderSwitcher();
    expect(screen.queryByText("Root")).not.toBeNull();
    expect(screen.queryByText("Child")).not.toBeNull();
    expect(screen.queryByText("Grandchild")).not.toBeNull();
  });

  it("tags each row with its tree depth", () => {
    const { container } = renderSwitcher();
    expect(blockDepth(container, "Root")).toBe("0");
    expect(blockDepth(container, "Child")).toBe("1");
    expect(blockDepth(container, "Grandchild")).toBe("2");
  });

  it("collapsing a mid-level parent hides its whole subtree", () => {
    renderSwitcher(["Child"]);
    expect(screen.queryByText("Root")).not.toBeNull();
    expect(screen.queryByText("Child")).not.toBeNull();
    // Grandchild lives under the collapsed Child, so it must not render.
    expect(screen.queryByText("Grandchild")).toBeNull();
  });

  it("group header counts the whole subtree, not just direct children", () => {
    renderSwitcher();
    // Header only shows for >1 group or a non-default key. Force it by using a
    // keyed group instead of the implicit __all__ bucket.
    cleanup();
    render(
      <Providers>
        <TaskSwitcher
          grouped={{
            groups: [{ key: "wf1", label: "Workflow 1", tasks: [ROOT] }],
            subTasksByParentId: grouped().subTasksByParentId,
          }}
          activeTaskId={null}
          selectedTaskId={null}
          onSelectTask={vi.fn()}
          onToggleSubtasks={vi.fn()}
          collapsedSubtaskParentIds={[]}
        />
      </Providers>,
    );
    // Root + Child + Grandchild = 3
    expect(screen.queryByText("3")).not.toBeNull();
  });

  it("parent subtask toggle counts all descendants, not just direct children", () => {
    const { container } = renderSwitcher();
    const rootToggle = container.querySelector(
      "[data-testid='sidebar-subtask-toggle'][data-task-id='Root']",
    );
    expect(rootToggle).not.toBeNull();
    // Root → Child → Grandchild: badge should reflect 2 hidden rows, not 1.
    expect(rootToggle!.textContent).toContain("2");
  });

  it("omits grab cursor on nested rows when subtask reorder is disabled", () => {
    const { container } = render(
      <Providers>
        <TaskSwitcher
          grouped={grouped()}
          activeTaskId={null}
          selectedTaskId={null}
          onSelectTask={vi.fn()}
          onToggleSubtasks={vi.fn()}
          collapsedSubtaskParentIds={[]}
        />
      </Providers>,
    );
    for (const taskId of ["Child", "Grandchild"]) {
      const handle = container.querySelector(
        `[data-testid='sortable-task-block'][data-task-id='${taskId}'] > [data-testid='sortable-task-handle']`,
      );
      expect(handle).not.toBeNull();
      expect(handle!.className).not.toContain("cursor-grab");
    }
  });
});

describe("TaskSwitcher — workflow completion icons", () => {
  it("derives workflow completion from the final ordered step for each task", () => {
    const finalStepId = "step-final";
    const groupedTasks = [
      item("Turn finished", undefined, {
        state: "REVIEW",
        sessionState: "COMPLETED",
        workflowId: TEST_WORKFLOW_ID,
        workflowStepId: "step-middle",
      }),
      item("Workflow complete", undefined, {
        state: "REVIEW",
        sessionState: "COMPLETED",
        workflowId: TEST_WORKFLOW_ID,
        workflowStepId: finalStepId,
      }),
      item("Missing workflow metadata", undefined, {
        state: "REVIEW",
        sessionState: "COMPLETED",
      }),
    ];

    render(
      <Providers>
        <TaskSwitcher
          grouped={{
            groups: [{ key: "__all__", label: "All", tasks: groupedTasks }],
            subTasksByParentId: new Map(),
          }}
          stepsByWorkflowId={{
            [TEST_WORKFLOW_ID]: [
              { id: "step-start", title: "Start" },
              { id: "step-middle", title: "Middle" },
              { id: finalStepId, title: "Done" },
            ],
          }}
          activeTaskId={null}
          selectedTaskId={null}
          onSelectTask={vi.fn()}
        />
      </Providers>,
    );

    const turnFinishedRow = screen
      .getByText("Turn finished")
      .closest<HTMLElement>("[data-testid='sidebar-task-item']");
    const workflowCompleteRow = screen
      .getByText("Workflow complete")
      .closest<HTMLElement>("[data-testid='sidebar-task-item']");
    const missingMetadataRow = screen
      .getByText("Missing workflow metadata")
      .closest<HTMLElement>("[data-testid='sidebar-task-item']");

    expect(within(turnFinishedRow!).getByTestId("task-state-turn-finished")).toBeTruthy();
    expect(within(turnFinishedRow!).queryByTestId("task-state-workflow-complete")).toBeNull();
    expect(within(workflowCompleteRow!).getByTestId("task-state-workflow-complete")).toBeTruthy();
    expect(within(workflowCompleteRow!).queryByTestId("task-state-turn-finished")).toBeNull();
    expect(within(missingMetadataRow!).getByTestId("task-state-turn-finished")).toBeTruthy();
    expect(within(missingMetadataRow!).queryByTestId("task-state-workflow-complete")).toBeNull();
  });
});

describe("TaskSwitcher — bulk pin menu", () => {
  it("offers bulk unpin when every selected root task is pinned", () => {
    render(
      <Providers>
        <TaskSwitcher
          grouped={{
            groups: [{ key: "__all__", label: "All", tasks: [TASK_A, TASK_B] }],
            subTasksByParentId: new Map(),
          }}
          activeTaskId={null}
          selectedTaskId={null}
          onSelectTask={vi.fn()}
          onBulkPin={vi.fn()}
          pinnedTaskIds={[TASK_A.id, TASK_B.id]}
          selectedTaskIds={new Set([TASK_A.id, TASK_B.id])}
        />
      </Providers>,
    );

    fireEvent.contextMenu(screen.getByText(TASK_A.title));

    expect(screen.getByText("Unpin 2 tasks")).toBeTruthy();
  });
});

describe("TaskSwitcher — detach menu", () => {
  it("offers detach for a single subtask and invokes the action", () => {
    const onDetachTask = vi.fn();
    render(
      <Providers>
        <TaskSwitcher
          grouped={grouped()}
          activeTaskId={null}
          selectedTaskId={null}
          onSelectTask={vi.fn()}
          onDetachTask={onDetachTask}
        />
      </Providers>,
    );

    fireEvent.contextMenu(screen.getByText(CHILD.title));
    fireEvent.click(screen.getByRole("menuitem", { name: DETACH_MENU_LABEL }));

    expect(onDetachTask).toHaveBeenCalledWith(CHILD.id);
  });

  it("clears a lone selected task before detaching it", () => {
    const calls: string[] = [];
    const onClearSelection = vi.fn(() => calls.push("clear"));
    const onDetachTask = vi.fn(() => calls.push("detach"));
    render(
      <Providers>
        <TaskSwitcher
          grouped={grouped()}
          activeTaskId={null}
          selectedTaskId={null}
          onSelectTask={vi.fn()}
          onDetachTask={onDetachTask}
          onClearSelection={onClearSelection}
          selectedTaskIds={new Set([CHILD.id])}
        />
      </Providers>,
    );

    fireEvent.contextMenu(screen.getByText(CHILD.title));
    fireEvent.click(screen.getByRole("menuitem", { name: DETACH_MENU_LABEL }));

    expect(calls).toEqual(["clear", "detach"]);
  });

  it("omits detach for root and multi-selection menus", () => {
    const { rerender } = render(
      <Providers>
        <TaskSwitcher
          grouped={grouped()}
          activeTaskId={null}
          selectedTaskId={null}
          onSelectTask={vi.fn()}
          onDetachTask={vi.fn()}
        />
      </Providers>,
    );

    fireEvent.contextMenu(screen.getByText(ROOT.title));
    expect(screen.queryByRole("menuitem", { name: DETACH_MENU_LABEL })).toBeNull();

    rerender(
      <Providers>
        <TaskSwitcher
          grouped={grouped()}
          activeTaskId={null}
          selectedTaskId={null}
          onSelectTask={vi.fn()}
          onDetachTask={vi.fn()}
          selectedTaskIds={new Set([CHILD.id, GRANDCHILD.id])}
        />
      </Providers>,
    );
    fireEvent.contextMenu(screen.getByText(CHILD.title));
    expect(screen.queryByRole("menuitem", { name: DETACH_MENU_LABEL })).toBeNull();
  });
});

describe("TaskSwitcher — create subtask menu", () => {
  it("passes the right-clicked task to the create-subtask action", () => {
    const onCreateSubtask = vi.fn();
    render(
      <Providers>
        <TaskSwitcher
          grouped={grouped()}
          activeTaskId={null}
          selectedTaskId={null}
          onSelectTask={vi.fn()}
          onCreateSubtask={onCreateSubtask}
        />
      </Providers>,
    );

    fireEvent.contextMenu(screen.getByText(CHILD.title));
    fireEvent.click(screen.getByRole("menuitem", { name: "Create Subtask" }));

    expect(onCreateSubtask).toHaveBeenCalledWith(CHILD.id, CHILD.title);
  });
});

describe("TaskSwitcher — edit menu", () => {
  it("passes the clicked task to the edit action", () => {
    const editableTask = item("Editable task", undefined, {
      workflowId: TEST_WORKFLOW_ID,
      workflowStepId: "step-1",
    });
    const onEditTask = vi.fn();

    render(
      <Providers>
        <TaskSwitcher
          grouped={{
            groups: [{ key: "__all__", label: "All", tasks: [editableTask] }],
            subTasksByParentId: new Map(),
          }}
          activeTaskId={null}
          selectedTaskId={null}
          onSelectTask={vi.fn()}
          onEditTask={onEditTask}
        />
      </Providers>,
    );

    fireEvent.contextMenu(screen.getByText(editableTask.title));
    fireEvent.click(screen.getByRole("menuitem", { name: "Edit" }));

    expect(onEditTask).toHaveBeenCalledWith(editableTask);
  });

  it("omits edit for archived rows and multi-selection menus", () => {
    const archivedTask = item("Archived task", undefined, {
      isArchived: true,
      workflowId: TEST_WORKFLOW_ID,
      workflowStepId: "step-1",
    });
    const editableTask = item("Editable task", undefined, {
      workflowId: TEST_WORKFLOW_ID,
      workflowStepId: "step-1",
    });
    const onEditTask = vi.fn();
    const { rerender } = render(
      <Providers>
        <TaskSwitcher
          grouped={{
            groups: [{ key: "__all__", label: "All", tasks: [archivedTask, editableTask] }],
            subTasksByParentId: new Map(),
          }}
          activeTaskId={null}
          selectedTaskId={null}
          onSelectTask={vi.fn()}
          onEditTask={onEditTask}
        />
      </Providers>,
    );

    fireEvent.contextMenu(screen.getByText(archivedTask.title));
    expect(screen.queryByRole("menuitem", { name: "Edit" })).toBeNull();
    expect(screen.queryByTestId("task-context-priority")).toBeNull();

    rerender(
      <Providers>
        <TaskSwitcher
          grouped={{
            groups: [{ key: "__all__", label: "All", tasks: [archivedTask, editableTask] }],
            subTasksByParentId: new Map(),
          }}
          activeTaskId={null}
          selectedTaskId={null}
          onSelectTask={vi.fn()}
          onEditTask={onEditTask}
          selectedTaskIds={new Set([editableTask.id, archivedTask.id])}
        />
      </Providers>,
    );
    fireEvent.contextMenu(screen.getByText(editableTask.title));
    expect(screen.queryByRole("menuitem", { name: "Edit" })).toBeNull();
    expect(screen.queryByTestId("task-context-priority")).toBeNull();
  });
});

describe("TaskSwitcher — archived rows action menu", () => {
  it("keeps archived rows navigation-only apart from delete", () => {
    const archivedTask = item("Archived task", undefined, {
      isArchived: true,
      workflowId: TEST_WORKFLOW_ID,
      workflowStepId: "step-1",
    });
    render(
      <Providers>
        <TaskSwitcher
          grouped={{
            groups: [{ key: "__all__", label: "All", tasks: [archivedTask] }],
            subTasksByParentId: new Map(),
          }}
          activeTaskId={null}
          selectedTaskId={null}
          onSelectTask={vi.fn()}
          onEditTask={vi.fn()}
          onRenameTask={vi.fn()}
          onArchiveTask={vi.fn()}
          onCreateSubtask={vi.fn()}
          onDeleteTask={vi.fn()}
          onDetachTask={vi.fn()}
          onLinkIssue={vi.fn()}
          onMoveToStep={vi.fn()}
          onTogglePin={vi.fn()}
        />
      </Providers>,
    );

    fireEvent.contextMenu(screen.getByText(archivedTask.title));
    expect(screen.queryByRole("menuitem", { name: "Edit" })).toBeNull();
    expect(screen.queryByRole("menuitem", { name: "Rename" })).toBeNull();
    expect(screen.queryByRole("menuitem", { name: "Archive" })).toBeNull();
    expect(screen.queryByRole("menuitem", { name: "Pin" })).toBeNull();
    expect(screen.queryByRole("menuitem", { name: "Create Subtask" })).toBeNull();
    expect(screen.queryByRole("menuitem", { name: "Detach from parent" })).toBeNull();
    expect(screen.queryByText("Link")).toBeNull();
    expect(screen.queryByText("Duplicate")).toBeNull();
    expect(screen.queryByText("Color")).toBeNull();
    expect(screen.queryByTestId("task-context-priority")).toBeNull();
    expect(screen.queryByText("Change workflow...")).toBeNull();
    expect(screen.getByRole("menuitem", { name: "Delete" })).toBeTruthy();
  });
});

describe("TaskSwitcher — edit menu prerequisites", () => {
  it("omits edit when a task lacks workflow metadata", () => {
    const task = item("No workflow task");
    render(
      <Providers>
        <TaskSwitcher
          grouped={{
            groups: [{ key: "__all__", label: "All", tasks: [task] }],
            subTasksByParentId: new Map(),
          }}
          activeTaskId={null}
          selectedTaskId={null}
          onSelectTask={vi.fn()}
          onEditTask={vi.fn()}
        />
      </Providers>,
    );

    fireEvent.contextMenu(screen.getByText(task.title));
    expect(screen.queryByRole("menuitem", { name: "Edit" })).toBeNull();
  });
});

describe("TaskSwitcher — external issue link menu", () => {
  it("offers configured external issue providers from the task context menu", async () => {
    const onLinkMergeRequest = vi.fn();
    render(
      <Providers>
        <TaskSwitcher
          grouped={{
            groups: [{ key: "__all__", label: "All", tasks: [TASK_A] }],
            subTasksByParentId: new Map(),
          }}
          activeTaskId={null}
          selectedTaskId={null}
          onSelectTask={vi.fn()}
          onLinkPullRequest={vi.fn()}
          onLinkIssue={vi.fn()}
          onLinkMergeRequest={onLinkMergeRequest}
          onLinkJiraTicket={vi.fn()}
          onLinkLinearIssue={vi.fn()}
          onLinkSentryIssue={vi.fn()}
        />
      </Providers>,
    );

    fireEvent.contextMenu(screen.getByText(TASK_A.title));
    const linkTrigger = screen.getByText("Link");
    fireEvent.focus(linkTrigger);
    fireEvent.keyDown(linkTrigger, { key: "ArrowRight" });

    await waitFor(() => {
      expect(screen.getByText("Jira Ticket")).toBeTruthy();
      expect(screen.getByText("Linear Issue")).toBeTruthy();
      expect(screen.getByText("Sentry Issue")).toBeTruthy();
      expect(screen.getByText("GitLab Merge Request")).toBeTruthy();
    });
  });

  it("targets the selected task when linking a GitLab merge request", () => {
    const archivedTask: TaskSwitcherItem = {
      id: "archived-task",
      title: "Archived task title",
      isArchived: true,
    };
    const closeMenu = vi.fn();
    const onLinkMergeRequest = vi.fn();

    createTaskLinkSelectAction(archivedTask, onLinkMergeRequest, closeMenu)?.();

    expect(onLinkMergeRequest).toHaveBeenCalledWith(archivedTask.id, archivedTask.title);
    expect(closeMenu).toHaveBeenCalledOnce();
  });
});

describe("TaskSwitcher repository labels", () => {
  const REPOSITORY_TEST_ID = "sidebar-task-repository";
  const REPOSITORY_SLUG = "kdlbs/kandev";

  function renderGroupedBy(
    groupKey: GroupedSidebarList["groupKey"],
    groups: GroupedSidebarList["groups"],
  ) {
    return render(
      <Providers>
        <TaskSwitcher
          grouped={{ groups, subTasksByParentId: new Map(), groupKey }}
          activeTaskId={null}
          selectedTaskId={null}
          onSelectTask={vi.fn()}
        />
      </Providers>,
    );
  }

  const REPO_TASK = item("Task A", undefined, { repositoryPath: REPOSITORY_SLUG });

  it("names each task's repository in a flat, ungrouped list", () => {
    renderGroupedBy("none", [{ key: "__all__", label: "All", tasks: [REPO_TASK] }]);

    expect(screen.getByTestId(REPOSITORY_TEST_ID).textContent).toBe(REPOSITORY_SLUG);
  });

  it("names each task's repository when grouped by a dimension other than repository", () => {
    renderGroupedBy("workflowStep", [{ key: "step-1", label: "In progress", tasks: [REPO_TASK] }]);

    expect(screen.getByTestId(REPOSITORY_TEST_ID).textContent).toBe(REPOSITORY_SLUG);
  });

  it("leaves the repository off the rows when the group header already names it", () => {
    renderGroupedBy("repository", [
      { key: REPOSITORY_SLUG, label: REPOSITORY_SLUG, tasks: [REPO_TASK] },
    ]);

    expect(screen.queryByTestId(REPOSITORY_TEST_ID)).toBeNull();
  });
});