import type { SidebarStatusWorkspacePlacement } from "@/hooks/sidebar-workspaces-view-model";

export type StatusBucket = SidebarStatusWorkspacePlacement["statusBucket"];

export const STATUS_BUCKET_ORDER: readonly StatusBucket[] = [
  "needs_input",
  "failed",
  "attention",
  "running",
  "done",
] as const;

export const STATUS_BUCKET_LABELS: Record<StatusBucket, string> = {
  needs_input: "Needs input",
  failed: "Failed",
  attention: "Ready to review",
  running: "Working",
  done: "Done",
};

export interface StatusGroup {
  bucket: StatusBucket;
  label: string;
  rows: SidebarStatusWorkspacePlacement[];
}

export function buildStatusGroups(
  workspaces: SidebarStatusWorkspacePlacement[],
  projectNamesByKey: Map<string, string>,
): StatusGroup[] {
  const bucketRows = new Map<StatusBucket, SidebarStatusWorkspacePlacement[]>();

  for (const ws of workspaces) {
    const bucket: StatusBucket = ws.statusBucket;
    let rows = bucketRows.get(bucket);
    if (!rows) {
      rows = [];
      bucketRows.set(bucket, rows);
    }
    rows.push(ws);
  }

  const groups: StatusGroup[] = [];

  for (const bucket of STATUS_BUCKET_ORDER) {
    const rows = bucketRows.get(bucket);
    if (!rows || rows.length === 0) continue;

    rows.sort((a, b) => compareStatusRows(a, b, projectNamesByKey));
    groups.push({ bucket, label: STATUS_BUCKET_LABELS[bucket], rows });
  }

  return groups;
}

function compareStatusRows(
  a: SidebarStatusWorkspacePlacement,
  b: SidebarStatusWorkspacePlacement,
  projectNamesByKey: Map<string, string>,
): number {
  const aTime = a.statusEnteredAt?.getTime() ?? null;
  const bTime = b.statusEnteredAt?.getTime() ?? null;

  if (aTime !== null && bTime !== null) {
    if (aTime !== bTime) return bTime - aTime;
  } else if (aTime !== null) {
    return -1;
  } else if (bTime !== null) {
    return 1;
  }

  const aProject = projectNamesByKey.get(a.projectKey) ?? "";
  const bProject = projectNamesByKey.get(b.projectKey) ?? "";
  const projectCmp = aProject.localeCompare(bProject);
  if (projectCmp !== 0) return projectCmp;

  const nameCmp = a.name.localeCompare(b.name);
  if (nameCmp !== 0) return nameCmp;

  return a.workspaceKey.localeCompare(b.workspaceKey);
}

export function buildStatusShortcutIndex(groups: StatusGroup[]): Map<string, number> {
  const index = new Map<string, number>();
  let shortcutNumber = 1;
  for (const group of groups) {
    for (const row of group.rows) {
      if (shortcutNumber > 9) return index;
      index.set(row.workspaceKey, shortcutNumber);
      shortcutNumber += 1;
    }
  }
  return index;
}