import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
import { api } from "@/lib/api";
import type { MatchResponse, UnmatchedClaim, UnmatchedResponse } from "@/types";

/**
 * React Query bindings for the reconciliation surface. Exposes:
 *   - `unmatched`: polled query for the GET /api/reconciliation/unmatched
 *     bucket (refreshes every 30s so the page reflects backend auto-matches
 *     without a manual reload).
 *   - `match`: mutation that POSTs /api/reconciliation/match. On success it
 *     invalidates every cache that could be affected by a state change:
 *     the unmatched bucket, the claims list, the remittances list, and the
 *     activity feed.
 *   - `unmatch`: mutation that POSTs /api/reconciliation/unmatch, with the
 *     same invalidation cascade.
 */
export function useReconciliation() {
  const qc = useQueryClient();

  const unmatched = useQuery<UnmatchedResponse>({
    queryKey: ["reconciliation", "unmatched"],
    queryFn: () => api.listUnmatched(),
    refetchInterval: 30_000,
  });

  const match = useMutation<
    MatchResponse,
    Error,
    { claimId: string; remitId: string }
  >({
    mutationFn: (args) => api.matchRemit(args.claimId, args.remitId),
    onSuccess: () => {
      qc.invalidateQueries({ queryKey: ["reconciliation"] });
      qc.invalidateQueries({ queryKey: ["claims"] });
      qc.invalidateQueries({ queryKey: ["remittances"] });
      qc.invalidateQueries({ queryKey: ["activity"] });
    },
  });

  const unmatch = useMutation<
    { claim: UnmatchedClaim },
    Error,
    { claimId: string }
  >({
    mutationFn: (args) => api.unmatchClaim(args.claimId),
    onSuccess: () => {
      qc.invalidateQueries({ queryKey: ["reconciliation"] });
      qc.invalidateQueries({ queryKey: ["claims"] });
      qc.invalidateQueries({ queryKey: ["remittances"] });
      qc.invalidateQueries({ queryKey: ["activity"] });
    },
  });

  return { unmatched, match, unmatch };
}