"use client";
import { useState } from "react";
import { toast } from "@/components/ui/sonner";
import { useTranslation } from "@/lib/i18n/client";
import { useMutation, useQueryClient } from "@tanstack/react-query";
import { useCreateBookmarkList } from "@karakeep/shared-react/hooks/lists";
import { useTRPC } from "@karakeep/shared-react/trpc";
import {
importBookmarksFromFile,
ImportSource,
parseImportFile,
} from "@karakeep/shared/import-export";
import { useCreateImportSession } from "./useImportSessions";
export interface ImportProgress {
done: number;
total: number;
}
export function useBookmarkImport() {
const { t } = useTranslation();
const api = useTRPC();
const [importProgress, setImportProgress] = useState<ImportProgress | null>(
null,
);
const [quotaError, setQuotaError] = useState<string | null>(null);
const queryClient = useQueryClient();
const { mutateAsync: createImportSession } = useCreateImportSession();
const { mutateAsync: createList } = useCreateBookmarkList();
const { mutateAsync: stageImportedBookmarks } = useMutation(
api.importSessions.stageImportedBookmarks.mutationOptions(),
);
const { mutateAsync: finalizeImportStaging } = useMutation(
api.importSessions.finalizeImportStaging.mutationOptions(),
);
const uploadBookmarkFileMutation = useMutation({
mutationFn: async ({
file,
source,
}: {
file: File;
source: ImportSource;
}) => {
// Clear any previous quota error
setQuotaError(null);
// First, parse the file to count bookmarks
const textContent = await file.text();
const parsedImport = parseImportFile(source, textContent);
const bookmarkCount = parsedImport.bookmarks.length;
// Check quota before proceeding
if (bookmarkCount > 0) {
const quotaUsage = await queryClient.fetchQuery(
api.subscriptions.getQuotaUsage.queryOptions(),
);
if (
!quotaUsage.bookmarks.unlimited &&
quotaUsage.bookmarks.quota !== null
) {
const remaining =
quotaUsage.bookmarks.quota - quotaUsage.bookmarks.used;
if (remaining < bookmarkCount) {
const errorMsg = `Cannot import ${bookmarkCount} bookmarks. You have ${remaining} bookmark${remaining === 1 ? "" : "s"} remaining in your quota of ${quotaUsage.bookmarks.quota}.`;
setQuotaError(errorMsg);
throw new Error(errorMsg);
}
}
}
// Proceed with import if quota check passes
const result = await importBookmarksFromFile(
{
file,
source,
rootListName: t("settings.import.imported_bookmarks"),
deps: {
createImportSession,
createList,
stageImportedBookmarks,
finalizeImportStaging: async (sessionId: string) => {
await finalizeImportStaging({ importSessionId: sessionId });
},
},
onProgress: (done, total) => setImportProgress({ done, total }),
},
{
// Use a custom parser to avoid re-parsing the file
parsers: {
[source]: () => parsedImport,
},
},
);
return result;
},
onSuccess: async (result) => {
setImportProgress(null);
if (result.counts.total === 0) {
toast({ description: "No bookmarks found in the file." });
return;
}
toast({
description: `Staged ${result.counts.total} bookmarks for import. Background processing will start automatically.`,
variant: "default",
});
},
onError: (error) => {
setImportProgress(null);
toast({
description: error.message,
variant: "destructive",
});
},
});
return {
importProgress,
quotaError,
clearQuotaError: () => setQuotaError(null),
runUploadBookmarkFile: uploadBookmarkFileMutation.mutateAsync,
isImporting: uploadBookmarkFileMutation.isPending,
};
}
