SurfSense/surfsense_web/features/documents/viewer/document-content-query.ts

import { queryOptions } from "@tanstack/react-query";
import { z } from "zod";
import { authenticatedFetch } from "@/lib/auth-fetch";
import { buildBackendUrl } from "@/lib/env-config";

const DocumentContentSchema = z.object({
	document_id: z.number(),
	title: z.string(),
	document_type: z.string().optional(),
	source_markdown: z.string(),
	viewer_mode: z.enum(["plate", "monaco"]).optional(),
});

export function documentContentQueryOptions(workspaceId: number, documentId: number) {
	return queryOptions({
		queryKey: ["document-content", workspaceId, documentId] as const,
		queryFn: async () => {
			const response = await authenticatedFetch(
				buildBackendUrl(`/api/v1/workspaces/${workspaceId}/documents/${documentId}/editor-content`),
				{ skipAuthRedirect: true }
			);
			if (!response.ok) {
				const body = await response.json().catch(() => null);
				throw new Error(body?.detail ?? `Could not load document (${response.status})`);
			}
			return DocumentContentSchema.parse(await response.json());
		},
		staleTime: 30_000,
	});
}