"use client";

import { useEffect, useCallback } from "react";
import { listTaskPRs, getTaskPR } from "@/lib/api/domains/github-api";
import { useAppStore } from "@/components/state-provider";
import type { TaskPR } from "@/lib/types/github";

/** Fetch and cache PR associations for a batch of task IDs. */
export function useTaskPRs(taskIds: string[]) {
  const byTaskId = useAppStore((state) => state.taskPRs.byTaskId);
  const loading = useAppStore((state) => state.taskPRs.loading);
  const setTaskPRs = useAppStore((state) => state.setTaskPRs);
  const setTaskPRsLoading = useAppStore((state) => state.setTaskPRsLoading);

  useEffect(() => {
    if (taskIds.length === 0 || loading) return;
    setTaskPRsLoading(true);
    listTaskPRs(taskIds, { cache: "no-store" })
      .then((response) => {
        setTaskPRs(response?.task_prs ?? {});
      })
      .catch(() => {
        // Keep existing data on error
      })
      .finally(() => {
        setTaskPRsLoading(false);
      });
    // eslint-disable-next-line react-hooks/exhaustive-deps
  }, [taskIds.join(",")]);

  return { byTaskId, loading };
}

/** Fetch a single task's PR association. */
export function useTaskPR(taskId: string | null) {
  const pr = useAppStore((state) => (taskId ? (state.taskPRs.byTaskId[taskId] ?? null) : null));
  const setTaskPR = useAppStore((state) => state.setTaskPR);

  const refresh = useCallback(() => {
    if (!taskId) return;
    getTaskPR(taskId, { cache: "no-store" })
      .then((response) => {
        if (response) setTaskPR(taskId, response);
      })
      .catch(() => {
        // Ignore - PR may not exist for this task
      });
  }, [taskId, setTaskPR]);

  useEffect(() => {
    if (!taskId || pr) return;
    refresh();
  }, [taskId, pr, refresh]);

  return { pr, refresh } as { pr: TaskPR | null; refresh: () => void };
}

/** Read the active task's PR from the store (no fetching). */
export function useActiveTaskPR(): TaskPR | null {
  return useAppStore((s) => {
    const taskId = s.tasks.activeTaskId;
    return taskId ? (s.taskPRs.byTaskId[taskId] ?? null) : null;
  });
}