import { useQuery, useMutation, useQueryClient } from "@tanstack/react-query"
import { getAllMyBorrows, returnBookForMe } from "@/api/borrows"
import { toast } from "sonner"

export function useMyBorrows() {
  return useQuery({
    queryKey: ["myBorrows"],
    queryFn: getAllMyBorrows,
  })
}

export function useReturnBook() {
  const qc = useQueryClient()
  return useMutation({
    mutationFn: returnBookForMe,
    onSuccess: () => {
      qc.invalidateQueries({ queryKey: ["myBorrows"] })
      toast.success("归还成功")
    },
    onError: (err) => {
      toast.error(err instanceof Error ? err.message : "归还失败")
    },
  })
}