pdfjs-reader-core 0.1.3 → 0.1.4

package/dist/index.cjs

@@ -338,13 +338,21 @@ function createViewerStore(initialOverrides = {}) {
     ...initialOverrides,
     // Document actions
     setDocument: (document2) => {
-      set({
-        document: document2,
-        numPages: document2.numPages,
-        isLoading: false,
-        error: null,
-        currentPage: 1
-      });
+      if (document2) {
+        set({
+          document: document2,
+          numPages: document2.numPages,
+          isLoading: false,
+          error: null,
+          currentPage: 1
+        });
+      } else {
+        set({
+          document: null,
+          numPages: 0,
+          isLoading: false
+        });
+      }
     },
     setLoading: (isLoading) => {
       set({ isLoading });
@@ -8777,6 +8785,17 @@ var PDFViewerClient_exports = {};
 __export(PDFViewerClient_exports, {
   PDFViewerClient: () => PDFViewerClient
 });
+function getSrcIdentifier(src) {
+  if (typeof src === "string") {
+    return src;
+  }
+  const data = src instanceof ArrayBuffer ? new Uint8Array(src) : src;
+  const len = data.byteLength;
+  if (len === 0) return "empty";
+  const first = Array.from(data.slice(0, 4)).map((b) => b.toString(16).padStart(2, "0")).join("");
+  const last = Array.from(data.slice(-4)).map((b) => b.toString(16).padStart(2, "0")).join("");
+  return `binary:${len}:${first}:${last}`;
+}
 var import_react39, import_jsx_runtime25, PDFViewerInner, PDFViewerClient;
 var init_PDFViewerClient = __esm({
   "src/components/PDFViewer/PDFViewerClient.tsx"() {
@@ -8808,8 +8827,8 @@ var init_PDFViewerClient = __esm({
       className
     }) {
       const { viewerStore } = usePDFViewerStores();
-      const loadingRef = (0, import_react39.useRef)(false);
-      const srcRef = (0, import_react39.useRef)(src);
+      const mountedRef = (0, import_react39.useRef)(true);
+      const [, setLoadState] = (0, import_react39.useState)("idle");
       const onDocumentLoadRef = (0, import_react39.useRef)(onDocumentLoad);
       const onErrorRef = (0, import_react39.useRef)(onError);
       const onPageChangeRef = (0, import_react39.useRef)(onPageChange);
@@ -8818,42 +8837,49 @@ var init_PDFViewerClient = __esm({
       onErrorRef.current = onError;
       onPageChangeRef.current = onPageChange;
       onScaleChangeRef.current = onScaleChange;
+      const srcIdRef = (0, import_react39.useRef)(null);
       const currentPage = useViewerStore((s) => s.currentPage);
       const scale = useViewerStore((s) => s.scale);
       const theme = useViewerStore((s) => s.theme);
       const isLoading = useViewerStore((s) => s.isLoading);
       const error = useViewerStore((s) => s.error);
       const sidebarOpen = useViewerStore((s) => s.sidebarOpen);
-      const [retryTrigger, setRetryTrigger] = (0, import_react39.useState)(0);
+      const srcId = getSrcIdentifier(src);
       const handleRetry = (0, import_react39.useCallback)(() => {
-        loadingRef.current = false;
-        srcRef.current = null;
+        srcIdRef.current = null;
         viewerStore.getState().setError(null);
-        setRetryTrigger((c) => c + 1);
-      }, [viewerStore, src]);
+        setLoadState("idle");
+      }, [viewerStore]);
       (0, import_react39.useEffect)(() => {
-        if (loadingRef.current) {
-          return;
-        }
-        if (srcRef.current === src && viewerStore.getState().document) {
+        mountedRef.current = true;
+        return () => {
+          mountedRef.current = false;
+        };
+      }, []);
+      (0, import_react39.useEffect)(() => {
+        if (srcIdRef.current === srcId && viewerStore.getState().document) {
           return;
         }
-        srcRef.current = src;
-        loadingRef.current = true;
+        const loadId = srcId;
+        srcIdRef.current = srcId;
         const currentDoc = viewerStore.getState().document;
         if (currentDoc) {
-          viewerStore.getState().reset();
+          currentDoc.destroy();
+          viewerStore.getState().setDocument(null);
         }
         const loadDoc = async () => {
+          if (!mountedRef.current) return;
           try {
             viewerStore.getState().setLoading(true);
             viewerStore.getState().setError(null);
+            setLoadState("loading");
             const { document: document2, numPages } = await loadDocument({
               src,
               workerSrc
             });
-            if (srcRef.current === src) {
+            if (mountedRef.current && srcIdRef.current === loadId) {
               viewerStore.getState().setDocument(document2);
+              setLoadState("loaded");
               if (initialPage !== 1) {
                 viewerStore.getState().goToPage(initialPage);
               }
@@ -8865,25 +8891,19 @@ var init_PDFViewerClient = __esm({
               document2.destroy();
             }
           } catch (err) {
-            if (srcRef.current === src) {
+            if (mountedRef.current && srcIdRef.current === loadId) {
               const error2 = err instanceof Error ? err : new Error("Failed to load document");
               viewerStore.getState().setError(error2);
+              viewerStore.getState().setLoading(false);
+              setLoadState("error");
               onErrorRef.current?.(error2);
             }
-          } finally {
-            loadingRef.current = false;
           }
         };
         loadDoc();
         return () => {
-          srcRef.current = null;
-        };
-      }, [src, workerSrc, initialPage, initialScale, viewerStore, retryTrigger]);
-      (0, import_react39.useEffect)(() => {
-        return () => {
-          viewerStore.getState().reset();
         };
-      }, [viewerStore]);
+      }, [srcId, src, workerSrc, initialPage, initialScale, viewerStore]);
       const prevPageRef = (0, import_react39.useRef)(currentPage);
       (0, import_react39.useEffect)(() => {
         if (prevPageRef.current !== currentPage) {