import { useInfiniteQuery, useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
import type { CursorPage, EngineRun, EngineRunDetail, EngineStats } from "@umt/contracts";
import { apiData } from "~/lib/api.ts";
import { qk } from "~/lib/query-keys.ts";

export type EngineRunFilters = { status?: string; workflowName?: string };

/** 终态:不能再 cancel */
export const ENGINE_TERMINAL = new Set(["completed", "succeeded", "failed", "canceled"]);

export function useEngineRuns(filters: EngineRunFilters) {
  return useInfiniteQuery({
    queryKey: qk.list("engine", filters),
    initialPageParam: undefined as string | undefined,
    queryFn: ({ pageParam, signal }) =>
      apiData<CursorPage<EngineRun>>("/engine/runs", { params: { ...filters, cursor: pageParam, limit: 50 }, signal }),
    getNextPageParam: (last) => last.nextCursor ?? undefined,
  });
}

export function useEngineRun(id: string | undefined) {
  return useQuery({
    queryKey: qk.detail("engine", id ?? ""),
    queryFn: ({ signal }) => apiData<EngineRunDetail>("/engine/runs/{id}", { pathParams: { id: id ?? "" }, signal }),
    enabled: !!id,
    // 非终态每 5 秒兜底刷新(SSE 事件驱动是主路径)
    refetchInterval: (q) => (q.state.data && !ENGINE_TERMINAL.has(q.state.data.status) ? 5_000 : false),
  });
}

export function useEngineStats() {
  return useQuery({
    queryKey: qk.engineStats,
    queryFn: ({ signal }) => apiData<EngineStats>("/engine/stats", { signal }),
    refetchInterval: 30_000,
  });
}

export function useCancelEngineRun() {
  const qc = useQueryClient();
  return useMutation({
    mutationFn: ({ id, reason }: { id: string; reason?: string }) =>
      apiData<{ canceled: boolean }>("/engine/runs/{id}/cancel", {
        method: "POST",
        pathParams: { id },
        body: { reason },
      }),
    onSettled: () => qc.invalidateQueries({ queryKey: qk.all("engine") }),
  });
}