1 files changed, 42 insertions, 1 deletions
diff --git a/apps/web/lib/hooks/useBookmarkImport.ts b/apps/web/lib/hooks/useBookmarkImport.ts
index d4ffda4e..0d9bbaaf 100644
--- a/apps/web/lib/hooks/useBookmarkImport.ts
+++ b/apps/web/lib/hooks/useBookmarkImport.ts
@@ -13,10 +13,12 @@ import {
   useAddBookmarkToList,
   useCreateBookmarkList,
 } from "@karakeep/shared-react/hooks/lists";
+import { api } from "@karakeep/shared-react/trpc";
 import {
   importBookmarksFromFile,
   ImportSource,
   ParsedBookmark,
+  parseImportFile,
 } from "@karakeep/shared/import-export";
 import {
   BookmarkTypes,
@@ -36,7 +38,9 @@ export function useBookmarkImport() {
   const [importProgress, setImportProgress] = useState<ImportProgress | null>(
     null,
   );
+  const [quotaError, setQuotaError] = useState<string | null>(null);
 
+  const apiUtils = api.useUtils();
   const { mutateAsync: createImportSession } = useCreateImportSession();
   const { mutateAsync: createBookmark } = useCreateBookmarkWithPostHook();
   const { mutateAsync: createList } = useCreateBookmarkList();
@@ -51,6 +55,36 @@ export function useBookmarkImport() {
       file: File;
       source: ImportSource;
     }) => {
+      // Clear any previous quota error
+      setQuotaError(null);
+
+      // First, parse the file to count bookmarks
+      const textContent = await file.text();
+      const parsedBookmarks = parseImportFile(source, textContent);
+      const bookmarkCount = parsedBookmarks.length;
+
+      // Check quota before proceeding
+      if (bookmarkCount > 0) {
+        const quotaUsage =
+          await apiUtils.client.subscriptions.getQuotaUsage.query();
+
+        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
+      // Use a custom parser to avoid re-parsing the file
       const result = await importBookmarksFromFile(
         {
           file,
@@ -122,7 +156,12 @@ export function useBookmarkImport() {
           },
           onProgress: (done, total) => setImportProgress({ done, total }),
         },
-        {},
+        {
+          // Use a custom parser to avoid re-parsing the file
+          parsers: {
+            [source]: () => parsedBookmarks,
+          },
+        },
       );
       return result;
     },
@@ -158,6 +197,8 @@ export function useBookmarkImport() {
 
   return {
     importProgress,
+    quotaError,
+    clearQuotaError: () => setQuotaError(null),
     runUploadBookmarkFile: uploadBookmarkFileMutation.mutateAsync,
     isImporting: uploadBookmarkFileMutation.isPending,
   };