"use client";

import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";

import { toast } from "sonner";

import { api, ApiError } from "@/lib/api";

import { queryKeys } from "@/lib/query";

import type { WatchlistCreate, WatchlistRule } from "@/lib/types";

export function useWatchlist() {

  return useQuery({

    queryKey: queryKeys.watchlist,

    queryFn: () => api.watchlist(),

  });

}

export function useCreateWatch() {

  const qc = useQueryClient();

  return useMutation({

    mutationFn: (body: WatchlistCreate) => api.createWatch(body),

    onSuccess: (rule) => {

      qc.invalidateQueries({ queryKey: queryKeys.watchlist });

      toast.success(`Now watching “${rule.label}”`);

    },

    onError: (err) => {

      if (err instanceof ApiError && err.status === 409) {

        toast.error("You're already watching that.");

      } else {

        toast.error(err instanceof Error ? err.message : "Could not add watch.");

      }

    },

  });

}

export function usePatchWatch() {

  const qc = useQueryClient();

  return useMutation({

    mutationFn: ({ id, ...body }: { id: number } & Partial<Pick<WatchlistRule, "active" | "priority" | "label">>) =>

      api.patchWatch(id, body),

    onSuccess: () => qc.invalidateQueries({ queryKey: queryKeys.watchlist }),

  });

}

export function useDeleteWatch() {

  const qc = useQueryClient();

  return useMutation({

    mutationFn: (id: number) => api.deleteWatch(id),

    onSuccess: () => {

      qc.invalidateQueries({ queryKey: queryKeys.watchlist });

      toast("Removed from watchlist");

    },

  });

}