deerflow-code/frontend-web/src/core/thread-shares/hooks.ts

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

import {
  createShare,
  importShare,
  listShares,
  previewShare,
  revokeShare,
  type ThreadShareMode,
} from "./api";

export function useShares(enabled = true) {
  return useQuery({
    queryKey: ["thread-shares"],
    queryFn: listShares,
    enabled,
  });
}

export function useCreateShare() {
  const queryClient = useQueryClient();
  return useMutation({
    mutationFn: ({
      threadId,
      mode = "import",
    }: {
      threadId: string;
      mode?: ThreadShareMode;
    }) => createShare(threadId, mode),
    onSuccess: () => {
      void queryClient.invalidateQueries({ queryKey: ["thread-shares"] });
    },
  });
}

export function useRevokeShare() {
  const queryClient = useQueryClient();
  return useMutation({
    mutationFn: (shareCode: string) => revokeShare(shareCode),
    onSuccess: () => {
      void queryClient.invalidateQueries({ queryKey: ["thread-shares"] });
    },
  });
}

export function usePreviewShare() {
  return useMutation({
    mutationFn: (shareCode: string) => previewShare(shareCode),
  });
}

export function useImportShare() {
  const queryClient = useQueryClient();
  return useMutation({
    mutationFn: (shareCode: string) => importShare(shareCode),
    onSuccess: () => {
      // Imported conversations show up in the user's own thread list.
      void queryClient.invalidateQueries({ queryKey: ["threads"] });
    },
  });
}