pdfjs-reader-core 0.1.2 → 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,68 +8827,97 @@ 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);
+      const onScaleChangeRef = (0, import_react39.useRef)(onScaleChange);
+      onDocumentLoadRef.current = onDocumentLoad;
+      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 loadDoc = (0, import_react39.useCallback)(async () => {
-        if (loadingRef.current) return;
-        loadingRef.current = true;
-        try {
-          viewerStore.getState().setLoading(true);
-          viewerStore.getState().setError(null);
-          const { document: document2, numPages } = await loadDocument({
-            src,
-            workerSrc
-          });
-          if (srcRef.current === src) {
-            viewerStore.getState().setDocument(document2);
-            if (initialPage !== 1) {
-              viewerStore.getState().goToPage(initialPage);
-            }
-            if (typeof initialScale === "number" && initialScale !== 1) {
-              viewerStore.getState().setScale(initialScale);
-            }
-            onDocumentLoad?.({ document: document2, numPages });
-          } else {
-            document2.destroy();
-          }
-        } catch (err) {
-          if (srcRef.current === src) {
-            const error2 = err instanceof Error ? err : new Error("Failed to load document");
-            viewerStore.getState().setError(error2);
-            onError?.(error2);
-          }
-        } finally {
-          loadingRef.current = false;
-        }
-      }, [src, workerSrc, initialPage, initialScale, onDocumentLoad, onError, viewerStore]);
+      const srcId = getSrcIdentifier(src);
+      const handleRetry = (0, import_react39.useCallback)(() => {
+        srcIdRef.current = null;
+        viewerStore.getState().setError(null);
+        setLoadState("idle");
+      }, [viewerStore]);
       (0, import_react39.useEffect)(() => {
-        srcRef.current = src;
+        mountedRef.current = true;
+        return () => {
+          mountedRef.current = false;
+        };
+      }, []);
+      (0, import_react39.useEffect)(() => {
+        if (srcIdRef.current === srcId && viewerStore.getState().document) {
+          return;
+        }
+        const loadId = srcId;
+        srcIdRef.current = srcId;
         const currentDoc = viewerStore.getState().document;
         if (currentDoc) {
-          viewerStore.getState().reset();
+          currentDoc.destroy();
+          viewerStore.getState().setDocument(null);
         }
-        loadDoc();
-        return () => {
-          srcRef.current = 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 (mountedRef.current && srcIdRef.current === loadId) {
+              viewerStore.getState().setDocument(document2);
+              setLoadState("loaded");
+              if (initialPage !== 1) {
+                viewerStore.getState().goToPage(initialPage);
+              }
+              if (typeof initialScale === "number" && initialScale !== 1) {
+                viewerStore.getState().setScale(initialScale);
+              }
+              onDocumentLoadRef.current?.({ document: document2, numPages });
+            } else {
+              document2.destroy();
+            }
+          } catch (err) {
+            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);
+            }
+          }
         };
-      }, [src, loadDoc, viewerStore]);
-      (0, import_react39.useEffect)(() => {
+        loadDoc();
         return () => {
-          viewerStore.getState().reset();
         };
-      }, [viewerStore]);
+      }, [srcId, src, workerSrc, initialPage, initialScale, viewerStore]);
+      const prevPageRef = (0, import_react39.useRef)(currentPage);
       (0, import_react39.useEffect)(() => {
-        onPageChange?.(currentPage);
-      }, [currentPage, onPageChange]);
+        if (prevPageRef.current !== currentPage) {
+          prevPageRef.current = currentPage;
+          onPageChangeRef.current?.(currentPage);
+        }
+      }, [currentPage]);
+      const prevScaleRef = (0, import_react39.useRef)(scale);
       (0, import_react39.useEffect)(() => {
-        onScaleChange?.(scale);
-      }, [scale, onScaleChange]);
+        if (prevScaleRef.current !== scale) {
+          prevScaleRef.current = scale;
+          onScaleChangeRef.current?.(scale);
+        }
+      }, [scale]);
       const themeClass = theme === "dark" ? "dark" : "";
       if (error) {
         return /* @__PURE__ */ (0, import_jsx_runtime25.jsx)(
@@ -8888,7 +8936,7 @@ var init_PDFViewerClient = __esm({
               /* @__PURE__ */ (0, import_jsx_runtime25.jsx)(
                 "button",
                 {
-                  onClick: loadDoc,
+                  onClick: handleRetry,
                   className: "mt-4 px-4 py-2 bg-blue-500 text-white rounded hover:bg-blue-600",
                   children: "Retry"
                 }