"use client";
import { useMutation, useQueryClient } from "@tanstack/react-query";
import { apiClient } from "@/lib/api/client";
import { showApiError } from "@/components/feedback/ApiErrorToast";

interface SnoozeArgs {
  conversation_id: string;
  duration_hours: 1 | 3 | 24;
}

interface CancelArgs {
  conversation_id: string;
}

export function useSnoozeConversation() {
  const qc = useQueryClient();

  const invalidate = (conversationId: string) => {
    qc.invalidateQueries({ queryKey: ["conversations"] });
    qc.invalidateQueries({ queryKey: ["conversation", conversationId] });
  };

  const snooze = useMutation({
    mutationFn: async (args: SnoozeArgs) =>
      apiClient.post<{ data: { snooze_until: string } }>(
        `/api/v1/conversations/${args.conversation_id}/snooze`,
        { duration_hours: args.duration_hours },
      ),
    onError: showApiError,
    onSuccess: (_data, args) => invalidate(args.conversation_id),
  });

  const cancel = useMutation({
    mutationFn: async (args: CancelArgs) =>
      apiClient.delete<void>(`/api/v1/conversations/${args.conversation_id}/snooze`),
    onError: showApiError,
    onSuccess: (_data, args) => invalidate(args.conversation_id),
  });

  return { snooze, cancel };
}