@osdk/react-components 0.2.0-beta.9 → 0.2.1-main-20260407074313

package/build/browser/pdf-viewer/hooks/usePdfHighlightMode.js

@@ -0,0 +1,227 @@
+/*
+ * Copyright 2025 Palantir Technologies, Inc. All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { AnnotationEditorType } from "pdfjs-dist";
+import { useCallback, useEffect, useRef, useState } from "react";
+/**
+ * Converts an RGB array (0.0–1.0 per channel) to a CSS hex color string.
+ */
+export function rgbArrayToHex(color) {
+  const r = Math.round(color[0] * 255);
+  const g = Math.round(color[1] * 255);
+  const b = Math.round(color[2] * 255);
+  return `#${r.toString(16).padStart(2, "0")}${g.toString(16).padStart(2, "0")}${b.toString(16).padStart(2, "0")}`;
+}
+
+/**
+ * Converts quadPoints (PDF spec format) back to PdfRect[].
+ * QuadPoints are 8 values per quad: [x1,y1, x2,y2, x3,y3, x4,y4]
+ * representing the four corners of each highlight rect.
+ *
+ * PDF.js highlight editor produces axis-aligned quads with ordering:
+ *   (x1,y1)=top-left, (x2,y2)=top-right, (x3,y3)=bottom-left, (x4,y4)=bottom-right
+ * We derive the bounding rect from x1, y1 (top), x2 (right edge), y3 (bottom).
+ */
+export function quadPointsToRects(quadPoints) {
+  const rects = [];
+  for (let i = 0; i < quadPoints.length; i += 8) {
+    const x1 = quadPoints[i];
+    const y1 = quadPoints[i + 1];
+    const x2 = quadPoints[i + 2];
+    const y3 = quadPoints[i + 5];
+    const x = Math.min(x1, x2);
+    const y = Math.min(y1, y3);
+    const width = Math.abs(x2 - x1);
+    const height = Math.abs(y1 - y3);
+    rects.push({
+      x,
+      y,
+      width,
+      height
+    });
+  }
+  return rects;
+}
+
+/**
+ * Hook that manages the PDF.js built-in highlight annotation editor mode.
+ * When active, users can select text to create highlights.
+ * A callback fires with the annotation details for persistence.
+ */
+export function usePdfHighlightMode({
+  pdfViewerRef,
+  document,
+  enabled,
+  onTextHighlight,
+  onHighlightDelete
+}) {
+  const [highlightModeActive, setHighlightModeActive] = useState(false);
+  const onTextHighlightRef = useRef(onTextHighlight);
+  const onHighlightDeleteRef = useRef(onHighlightDelete);
+  const knownEditorIdsRef = useRef(new Set());
+  const editorEventsRef = useRef(new Map());
+
+  // Keep callback refs in sync
+  useEffect(function () {
+    onTextHighlightRef.current = onTextHighlight;
+    onHighlightDeleteRef.current = onHighlightDelete;
+  }, [onTextHighlight, onHighlightDelete]);
+
+  // Toggle the PDF.js annotation editor mode
+  useEffect(function () {
+    const pdfViewer = pdfViewerRef.current;
+    if (pdfViewer == null || document == null) {
+      return;
+    }
+    try {
+      if (highlightModeActive && enabled) {
+        pdfViewer.annotationEditorMode = {
+          mode: AnnotationEditorType.HIGHLIGHT
+        };
+      } else {
+        pdfViewer.annotationEditorMode = {
+          mode: AnnotationEditorType.NONE
+        };
+      }
+    } catch {
+      // AnnotationEditorUIManager may not be initialized yet
+    }
+  }, [pdfViewerRef, document, highlightModeActive, enabled]);
+
+  // Deactivate highlight mode when the feature is disabled
+  useEffect(function () {
+    if (!enabled) {
+      setHighlightModeActive(false);
+    }
+  }, [enabled]);
+
+  // Clear tracked editors when the document changes (but not on highlight toggle)
+  const prevDocumentRef = useRef(undefined);
+  useEffect(function () {
+    if (document !== prevDocumentRef.current) {
+      knownEditorIdsRef.current.clear();
+      editorEventsRef.current.clear();
+      prevDocumentRef.current = document;
+    }
+  }, [document]);
+
+  // Listen for new highlights being added to annotation storage
+  useEffect(function () {
+    if (document == null || !highlightModeActive || !enabled) {
+      return;
+    }
+    const storage = document.annotationStorage;
+    const previousOnAnnotationEditor = storage.onAnnotationEditor;
+
+    // Monkey-patch storage.remove to detect highlight deletions.
+    // This is fragile but necessary because PDF.js does not emit events
+    // when editors are removed. Restore the original on cleanup.
+    const originalRemove = storage.remove.bind(storage);
+    storage.remove = key => {
+      const savedEvent = editorEventsRef.current.get(key);
+      if (savedEvent != null) {
+        editorEventsRef.current.delete(key);
+        knownEditorIdsRef.current.delete(key);
+        onHighlightDeleteRef.current?.(savedEvent);
+      }
+      originalRemove(key);
+    };
+    storage.onAnnotationEditor = type => {
+      // onAnnotationEditor fires with the editor's static _type (string "highlight")
+      if (type !== "highlight") {
+        return;
+      }
+
+      // Find new editors in storage
+      const allEntries = storage.getAll();
+      if (allEntries == null) {
+        return;
+      }
+      for (const [id, entry] of Object.entries(allEntries)) {
+        if (knownEditorIdsRef.current.has(id)) {
+          continue;
+        }
+
+        // Check if this is a highlight editor with serializable data
+        if (typeof entry === "object" && entry != null && "serialize" in entry && typeof entry.serialize === "function") {
+          knownEditorIdsRef.current.add(id);
+          const serialized = entry.serialize(false);
+          if (serialized == null || serialized.annotationType !== AnnotationEditorType.HIGHLIGHT) {
+            continue;
+          }
+          const rects = serialized.quadPoints != null ? quadPointsToRects(serialized.quadPoints) : [];
+          if (rects.length === 0) {
+            continue;
+          }
+          const color = Array.isArray(serialized.color) ? rgbArrayToHex(serialized.color) : "#fff066";
+
+          // Extract text from the editor's aria-label (stores selected text)
+          const selectedText = typeof entry.div?.getAttribute === "function" ? entry.div.getAttribute("aria-label") ?? "" : "";
+          const event = {
+            editorId: id,
+            page: serialized.pageIndex + 1,
+            rects,
+            selectedText,
+            color
+          };
+
+          // Store the event so we can provide it back on delete
+          editorEventsRef.current.set(id, event);
+          onTextHighlightRef.current?.(event);
+        }
+      }
+    };
+    return () => {
+      storage.onAnnotationEditor = previousOnAnnotationEditor;
+      storage.remove = originalRemove;
+    };
+  }, [document, highlightModeActive, enabled]);
+  const toggleHighlightMode = useCallback(() => {
+    setHighlightModeActive(prev => !prev);
+  }, []);
+  const deleteHighlight = useCallback(editorId => {
+    if (document == null) return;
+    const storage = document.annotationStorage;
+
+    // Fire the delete callback directly. The monkey-patch on storage.remove
+    // only exists while highlight mode is active, so we handle it here to
+    // ensure the callback fires regardless of highlight mode state.
+    // Clean up refs first so the monkey-patch (if active) won't double-fire.
+    const savedEvent = editorEventsRef.current.get(editorId);
+    if (savedEvent != null) {
+      editorEventsRef.current.delete(editorId);
+      knownEditorIdsRef.current.delete(editorId);
+      onHighlightDeleteRef.current?.(savedEvent);
+    }
+
+    // Remove the editor's DOM element if it exists
+    const allEntries = storage.getAll();
+    const entry = allEntries?.[editorId];
+    if (hasDiv(entry) && entry.div instanceof HTMLElement) {
+      entry.div.remove();
+    }
+    storage.remove(editorId);
+  }, [document]);
+  return {
+    highlightModeActive,
+    toggleHighlightMode,
+    deleteHighlight
+  };
+}
+function hasDiv(obj) {
+  return obj != null && typeof obj === "object" && "div" in obj;
+}
+//# sourceMappingURL=usePdfHighlightMode.js.map

package/build/browser/pdf-viewer/hooks/usePdfHighlightMode.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usePdfHighlightMode.js","names":["AnnotationEditorType","useCallback","useEffect","useRef","useState","rgbArrayToHex","color","r","Math","round","g","b","toString","padStart","quadPointsToRects","quadPoints","rects","i","length","x1","y1","x2","y3","x","min","y","width","abs","height","push","usePdfHighlightMode","pdfViewerRef","document","enabled","onTextHighlight","onHighlightDelete","highlightModeActive","setHighlightModeActive","onTextHighlightRef","onHighlightDeleteRef","knownEditorIdsRef","Set","editorEventsRef","Map","current","pdfViewer","annotationEditorMode","mode","HIGHLIGHT","NONE","prevDocumentRef","undefined","clear","storage","annotationStorage","previousOnAnnotationEditor","onAnnotationEditor","originalRemove","remove","bind","key","savedEvent","get","delete","type","allEntries","getAll","id","entry","Object","entries","has","serialize","add","serialized","annotationType","Array","isArray","selectedText","div","getAttribute","event","editorId","page","pageIndex","set","toggleHighlightMode","prev","deleteHighlight","hasDiv","HTMLElement","obj"],"sources":["usePdfHighlightMode.ts"],"sourcesContent":["/*\n * Copyright 2025 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { AnnotationEditorType } from \"pdfjs-dist\";\nimport type { PDFDocumentProxy } from \"pdfjs-dist\";\nimport type { PDFViewer } from \"pdfjs-dist/web/pdf_viewer.mjs\";\nimport type { RefObject } from \"react\";\nimport { useCallback, useEffect, useRef, useState } from \"react\";\nimport type { PdfRect, PdfTextHighlightEvent } from \"../types.js\";\n\nexport interface UsePdfHighlightModeOptions {\n  pdfViewerRef: RefObject<PDFViewer | null>;\n  document: PDFDocumentProxy | undefined;\n  enabled: boolean;\n  onTextHighlight?: (event: PdfTextHighlightEvent) => void;\n  onHighlightDelete?: (event: PdfTextHighlightEvent) => void;\n}\n\nexport interface UsePdfHighlightModeResult {\n  highlightModeActive: boolean;\n  toggleHighlightMode: () => void;\n  deleteHighlight: (editorId: string) => void;\n}\n\n/**\n * Converts an RGB array (0.0–1.0 per channel) to a CSS hex color string.\n */\nexport function rgbArrayToHex(color: number[]): string {\n  const r = Math.round(color[0] * 255);\n  const g = Math.round(color[1] * 255);\n  const b = Math.round(color[2] * 255);\n  return `#${r.toString(16).padStart(2, \"0\")}${\n    g.toString(16).padStart(2, \"0\")\n  }${b.toString(16).padStart(2, \"0\")}`;\n}\n\n/**\n * Converts quadPoints (PDF spec format) back to PdfRect[].\n * QuadPoints are 8 values per quad: [x1,y1, x2,y2, x3,y3, x4,y4]\n * representing the four corners of each highlight rect.\n *\n * PDF.js highlight editor produces axis-aligned quads with ordering:\n *   (x1,y1)=top-left, (x2,y2)=top-right, (x3,y3)=bottom-left, (x4,y4)=bottom-right\n * We derive the bounding rect from x1, y1 (top), x2 (right edge), y3 (bottom).\n */\nexport function quadPointsToRects(\n  quadPoints: Float32Array,\n): PdfRect[] {\n  const rects: PdfRect[] = [];\n  for (let i = 0; i < quadPoints.length; i += 8) {\n    const x1 = quadPoints[i];\n    const y1 = quadPoints[i + 1];\n    const x2 = quadPoints[i + 2];\n    const y3 = quadPoints[i + 5];\n\n    const x = Math.min(x1, x2);\n    const y = Math.min(y1, y3);\n    const width = Math.abs(x2 - x1);\n    const height = Math.abs(y1 - y3);\n\n    rects.push({ x, y, width, height });\n  }\n  return rects;\n}\n\n/**\n * Hook that manages the PDF.js built-in highlight annotation editor mode.\n * When active, users can select text to create highlights.\n * A callback fires with the annotation details for persistence.\n */\nexport function usePdfHighlightMode({\n  pdfViewerRef,\n  document,\n  enabled,\n  onTextHighlight,\n  onHighlightDelete,\n}: UsePdfHighlightModeOptions): UsePdfHighlightModeResult {\n  const [highlightModeActive, setHighlightModeActive] = useState(false);\n  const onTextHighlightRef = useRef(onTextHighlight);\n  const onHighlightDeleteRef = useRef(onHighlightDelete);\n  const knownEditorIdsRef = useRef<Set<string>>(new Set());\n  const editorEventsRef = useRef<Map<string, PdfTextHighlightEvent>>(new Map());\n\n  // Keep callback refs in sync\n  useEffect(function syncCallbackRef() {\n    onTextHighlightRef.current = onTextHighlight;\n    onHighlightDeleteRef.current = onHighlightDelete;\n  }, [onTextHighlight, onHighlightDelete]);\n\n  // Toggle the PDF.js annotation editor mode\n  useEffect(function syncHighlightMode() {\n    const pdfViewer = pdfViewerRef.current;\n    if (pdfViewer == null || document == null) {\n      return;\n    }\n\n    try {\n      if (highlightModeActive && enabled) {\n        pdfViewer.annotationEditorMode = {\n          mode: AnnotationEditorType.HIGHLIGHT,\n        };\n      } else {\n        pdfViewer.annotationEditorMode = {\n          mode: AnnotationEditorType.NONE,\n        };\n      }\n    } catch {\n      // AnnotationEditorUIManager may not be initialized yet\n    }\n  }, [pdfViewerRef, document, highlightModeActive, enabled]);\n\n  // Deactivate highlight mode when the feature is disabled\n  useEffect(function deactivateOnDisable() {\n    if (!enabled) {\n      setHighlightModeActive(false);\n    }\n  }, [enabled]);\n\n  // Clear tracked editors when the document changes (but not on highlight toggle)\n  const prevDocumentRef = useRef<PDFDocumentProxy | undefined>(undefined);\n  useEffect(function clearOnDocumentChange() {\n    if (document !== prevDocumentRef.current) {\n      knownEditorIdsRef.current.clear();\n      editorEventsRef.current.clear();\n      prevDocumentRef.current = document;\n    }\n  }, [document]);\n\n  // Listen for new highlights being added to annotation storage\n  useEffect(function listenForHighlights() {\n    if (document == null || !highlightModeActive || !enabled) {\n      return;\n    }\n\n    const storage = document.annotationStorage;\n    const previousOnAnnotationEditor = storage.onAnnotationEditor;\n\n    // Monkey-patch storage.remove to detect highlight deletions.\n    // This is fragile but necessary because PDF.js does not emit events\n    // when editors are removed. Restore the original on cleanup.\n    const originalRemove = storage.remove.bind(storage);\n    storage.remove = (key: string) => {\n      const savedEvent = editorEventsRef.current.get(key);\n      if (savedEvent != null) {\n        editorEventsRef.current.delete(key);\n        knownEditorIdsRef.current.delete(key);\n        onHighlightDeleteRef.current?.(savedEvent);\n      }\n      originalRemove(key);\n    };\n\n    storage.onAnnotationEditor = (type: string | number | null) => {\n      // onAnnotationEditor fires with the editor's static _type (string \"highlight\")\n      if (type !== \"highlight\") {\n        return;\n      }\n\n      // Find new editors in storage\n      const allEntries = storage.getAll();\n      if (allEntries == null) {\n        return;\n      }\n\n      for (const [id, entry] of Object.entries(allEntries)) {\n        if (knownEditorIdsRef.current.has(id)) {\n          continue;\n        }\n\n        // Check if this is a highlight editor with serializable data\n        if (\n          typeof entry === \"object\" && entry != null\n          && \"serialize\" in entry\n          && typeof entry.serialize === \"function\"\n        ) {\n          knownEditorIdsRef.current.add(id);\n          const serialized = entry.serialize(false);\n          if (\n            serialized == null\n            || serialized.annotationType !== AnnotationEditorType.HIGHLIGHT\n          ) {\n            continue;\n          }\n\n          const rects = serialized.quadPoints != null\n            ? quadPointsToRects(serialized.quadPoints)\n            : [];\n\n          if (rects.length === 0) {\n            continue;\n          }\n\n          const color = Array.isArray(serialized.color)\n            ? rgbArrayToHex(serialized.color)\n            : \"#fff066\";\n\n          // Extract text from the editor's aria-label (stores selected text)\n          const selectedText = typeof entry.div?.getAttribute === \"function\"\n            ? entry.div.getAttribute(\"aria-label\") ?? \"\"\n            : \"\";\n\n          const event: PdfTextHighlightEvent = {\n            editorId: id,\n            page: serialized.pageIndex + 1,\n            rects,\n            selectedText,\n            color,\n          };\n\n          // Store the event so we can provide it back on delete\n          editorEventsRef.current.set(id, event);\n\n          onTextHighlightRef.current?.(event);\n        }\n      }\n    };\n\n    return () => {\n      storage.onAnnotationEditor = previousOnAnnotationEditor;\n      storage.remove = originalRemove;\n    };\n  }, [document, highlightModeActive, enabled]);\n\n  const toggleHighlightMode = useCallback(() => {\n    setHighlightModeActive((prev) => !prev);\n  }, []);\n\n  const deleteHighlight = useCallback((editorId: string) => {\n    if (document == null) return;\n    const storage = document.annotationStorage;\n\n    // Fire the delete callback directly. The monkey-patch on storage.remove\n    // only exists while highlight mode is active, so we handle it here to\n    // ensure the callback fires regardless of highlight mode state.\n    // Clean up refs first so the monkey-patch (if active) won't double-fire.\n    const savedEvent = editorEventsRef.current.get(editorId);\n    if (savedEvent != null) {\n      editorEventsRef.current.delete(editorId);\n      knownEditorIdsRef.current.delete(editorId);\n      onHighlightDeleteRef.current?.(savedEvent);\n    }\n\n    // Remove the editor's DOM element if it exists\n    const allEntries = storage.getAll() as Record<string, unknown> | null;\n    const entry = allEntries?.[editorId];\n    if (hasDiv(entry) && entry.div instanceof HTMLElement) {\n      entry.div.remove();\n    }\n\n    storage.remove(editorId);\n  }, [document]);\n\n  return {\n    highlightModeActive,\n    toggleHighlightMode,\n    deleteHighlight,\n  };\n}\n\nfunction hasDiv(obj: unknown): obj is { div: unknown } {\n  return obj != null && typeof obj === \"object\" && \"div\" in obj;\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAEA,SAASA,oBAAoB,QAAQ,YAAY;AAIjD,SAASC,WAAW,EAAEC,SAAS,EAAEC,MAAM,EAAEC,QAAQ,QAAQ,OAAO;AAiBhE;AACA;AACA;AACA,OAAO,SAASC,aAAaA,CAACC,KAAe,EAAU;EACrD,MAAMC,CAAC,GAAGC,IAAI,CAACC,KAAK,CAACH,KAAK,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC;EACpC,MAAMI,CAAC,GAAGF,IAAI,CAACC,KAAK,CAACH,KAAK,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC;EACpC,MAAMK,CAAC,GAAGH,IAAI,CAACC,KAAK,CAACH,KAAK,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC;EACpC,OAAO,IAAIC,CAAC,CAACK,QAAQ,CAAC,EAAE,CAAC,CAACC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,GACxCH,CAAC,CAACE,QAAQ,CAAC,EAAE,CAAC,CAACC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,GAC9BF,CAAC,CAACC,QAAQ,CAAC,EAAE,CAAC,CAACC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE;AACtC;;AAEA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,OAAO,SAASC,iBAAiBA,CAC/BC,UAAwB,EACb;EACX,MAAMC,KAAgB,GAAG,EAAE;EAC3B,KAAK,IAAIC,CAAC,GAAG,CAAC,EAAEA,CAAC,GAAGF,UAAU,CAACG,MAAM,EAAED,CAAC,IAAI,CAAC,EAAE;IAC7C,MAAME,EAAE,GAAGJ,UAAU,CAACE,CAAC,CAAC;IACxB,MAAMG,EAAE,GAAGL,UAAU,CAACE,CAAC,GAAG,CAAC,CAAC;IAC5B,MAAMI,EAAE,GAAGN,UAAU,CAACE,CAAC,GAAG,CAAC,CAAC;IAC5B,MAAMK,EAAE,GAAGP,UAAU,CAACE,CAAC,GAAG,CAAC,CAAC;IAE5B,MAAMM,CAAC,GAAGf,IAAI,CAACgB,GAAG,CAACL,EAAE,EAAEE,EAAE,CAAC;IAC1B,MAAMI,CAAC,GAAGjB,IAAI,CAACgB,GAAG,CAACJ,EAAE,EAAEE,EAAE,CAAC;IAC1B,MAAMI,KAAK,GAAGlB,IAAI,CAACmB,GAAG,CAACN,EAAE,GAAGF,EAAE,CAAC;IAC/B,MAAMS,MAAM,GAAGpB,IAAI,CAACmB,GAAG,CAACP,EAAE,GAAGE,EAAE,CAAC;IAEhCN,KAAK,CAACa,IAAI,CAAC;MAAEN,CAAC;MAAEE,CAAC;MAAEC,KAAK;MAAEE;IAAO,CAAC,CAAC;EACrC;EACA,OAAOZ,KAAK;AACd;;AAEA;AACA;AACA;AACA;AACA;AACA,OAAO,SAASc,mBAAmBA,CAAC;EAClCC,YAAY;EACZC,QAAQ;EACRC,OAAO;EACPC,eAAe;EACfC;AAC0B,CAAC,EAA6B;EACxD,MAAM,CAACC,mBAAmB,EAAEC,sBAAsB,CAAC,GAAGjC,QAAQ,CAAC,KAAK,CAAC;EACrE,MAAMkC,kBAAkB,GAAGnC,MAAM,CAAC+B,eAAe,CAAC;EAClD,MAAMK,oBAAoB,GAAGpC,MAAM,CAACgC,iBAAiB,CAAC;EACtD,MAAMK,iBAAiB,GAAGrC,MAAM,CAAc,IAAIsC,GAAG,CAAC,CAAC,CAAC;EACxD,MAAMC,eAAe,GAAGvC,MAAM,CAAqC,IAAIwC,GAAG,CAAC,CAAC,CAAC;;EAE7E;EACAzC,SAAS,CAAC,YAA2B;IACnCoC,kBAAkB,CAACM,OAAO,GAAGV,eAAe;IAC5CK,oBAAoB,CAACK,OAAO,GAAGT,iBAAiB;EAClD,CAAC,EAAE,CAACD,eAAe,EAAEC,iBAAiB,CAAC,CAAC;;EAExC;EACAjC,SAAS,CAAC,YAA6B;IACrC,MAAM2C,SAAS,GAAGd,YAAY,CAACa,OAAO;IACtC,IAAIC,SAAS,IAAI,IAAI,IAAIb,QAAQ,IAAI,IAAI,EAAE;MACzC;IACF;IAEA,IAAI;MACF,IAAII,mBAAmB,IAAIH,OAAO,EAAE;QAClCY,SAAS,CAACC,oBAAoB,GAAG;UAC/BC,IAAI,EAAE/C,oBAAoB,CAACgD;QAC7B,CAAC;MACH,CAAC,MAAM;QACLH,SAAS,CAACC,oBAAoB,GAAG;UAC/BC,IAAI,EAAE/C,oBAAoB,CAACiD;QAC7B,CAAC;MACH;IACF,CAAC,CAAC,MAAM;MACN;IAAA;EAEJ,CAAC,EAAE,CAAClB,YAAY,EAAEC,QAAQ,EAAEI,mBAAmB,EAAEH,OAAO,CAAC,CAAC;;EAE1D;EACA/B,SAAS,CAAC,YAA+B;IACvC,IAAI,CAAC+B,OAAO,EAAE;MACZI,sBAAsB,CAAC,KAAK,CAAC;IAC/B;EACF,CAAC,EAAE,CAACJ,OAAO,CAAC,CAAC;;EAEb;EACA,MAAMiB,eAAe,GAAG/C,MAAM,CAA+BgD,SAAS,CAAC;EACvEjD,SAAS,CAAC,YAAiC;IACzC,IAAI8B,QAAQ,KAAKkB,eAAe,CAACN,OAAO,EAAE;MACxCJ,iBAAiB,CAACI,OAAO,CAACQ,KAAK,CAAC,CAAC;MACjCV,eAAe,CAACE,OAAO,CAACQ,KAAK,CAAC,CAAC;MAC/BF,eAAe,CAACN,OAAO,GAAGZ,QAAQ;IACpC;EACF,CAAC,EAAE,CAACA,QAAQ,CAAC,CAAC;;EAEd;EACA9B,SAAS,CAAC,YAA+B;IACvC,IAAI8B,QAAQ,IAAI,IAAI,IAAI,CAACI,mBAAmB,IAAI,CAACH,OAAO,EAAE;MACxD;IACF;IAEA,MAAMoB,OAAO,GAAGrB,QAAQ,CAACsB,iBAAiB;IAC1C,MAAMC,0BAA0B,GAAGF,OAAO,CAACG,kBAAkB;;IAE7D;IACA;IACA;IACA,MAAMC,cAAc,GAAGJ,OAAO,CAACK,MAAM,CAACC,IAAI,CAACN,OAAO,CAAC;IACnDA,OAAO,CAACK,MAAM,GAAIE,GAAW,IAAK;MAChC,MAAMC,UAAU,GAAGnB,eAAe,CAACE,OAAO,CAACkB,GAAG,CAACF,GAAG,CAAC;MACnD,IAAIC,UAAU,IAAI,IAAI,EAAE;QACtBnB,eAAe,CAACE,OAAO,CAACmB,MAAM,CAACH,GAAG,CAAC;QACnCpB,iBAAiB,CAACI,OAAO,CAACmB,MAAM,CAACH,GAAG,CAAC;QACrCrB,oBAAoB,CAACK,OAAO,GAAGiB,UAAU,CAAC;MAC5C;MACAJ,cAAc,CAACG,GAAG,CAAC;IACrB,CAAC;IAEDP,OAAO,CAACG,kBAAkB,GAAIQ,IAA4B,IAAK;MAC7D;MACA,IAAIA,IAAI,KAAK,WAAW,EAAE;QACxB;MACF;;MAEA;MACA,MAAMC,UAAU,GAAGZ,OAAO,CAACa,MAAM,CAAC,CAAC;MACnC,IAAID,UAAU,IAAI,IAAI,EAAE;QACtB;MACF;MAEA,KAAK,MAAM,CAACE,EAAE,EAAEC,KAAK,CAAC,IAAIC,MAAM,CAACC,OAAO,CAACL,UAAU,CAAC,EAAE;QACpD,IAAIzB,iBAAiB,CAACI,OAAO,CAAC2B,GAAG,CAACJ,EAAE,CAAC,EAAE;UACrC;QACF;;QAEA;QACA,IACE,OAAOC,KAAK,KAAK,QAAQ,IAAIA,KAAK,IAAI,IAAI,IACvC,WAAW,IAAIA,KAAK,IACpB,OAAOA,KAAK,CAACI,SAAS,KAAK,UAAU,EACxC;UACAhC,iBAAiB,CAACI,OAAO,CAAC6B,GAAG,CAACN,EAAE,CAAC;UACjC,MAAMO,UAAU,GAAGN,KAAK,CAACI,SAAS,CAAC,KAAK,CAAC;UACzC,IACEE,UAAU,IAAI,IAAI,IACfA,UAAU,CAACC,cAAc,KAAK3E,oBAAoB,CAACgD,SAAS,EAC/D;YACA;UACF;UAEA,MAAMhC,KAAK,GAAG0D,UAAU,CAAC3D,UAAU,IAAI,IAAI,GACvCD,iBAAiB,CAAC4D,UAAU,CAAC3D,UAAU,CAAC,GACxC,EAAE;UAEN,IAAIC,KAAK,CAACE,MAAM,KAAK,CAAC,EAAE;YACtB;UACF;UAEA,MAAMZ,KAAK,GAAGsE,KAAK,CAACC,OAAO,CAACH,UAAU,CAACpE,KAAK,CAAC,GACzCD,aAAa,CAACqE,UAAU,CAACpE,KAAK,CAAC,GAC/B,SAAS;;UAEb;UACA,MAAMwE,YAAY,GAAG,OAAOV,KAAK,CAACW,GAAG,EAAEC,YAAY,KAAK,UAAU,GAC9DZ,KAAK,CAACW,GAAG,CAACC,YAAY,CAAC,YAAY,CAAC,IAAI,EAAE,GAC1C,EAAE;UAEN,MAAMC,KAA4B,GAAG;YACnCC,QAAQ,EAAEf,EAAE;YACZgB,IAAI,EAAET,UAAU,CAACU,SAAS,GAAG,CAAC;YAC9BpE,KAAK;YACL8D,YAAY;YACZxE;UACF,CAAC;;UAED;UACAoC,eAAe,CAACE,OAAO,CAACyC,GAAG,CAAClB,EAAE,EAAEc,KAAK,CAAC;UAEtC3C,kBAAkB,CAACM,OAAO,GAAGqC,KAAK,CAAC;QACrC;MACF;IACF,CAAC;IAED,OAAO,MAAM;MACX5B,OAAO,CAACG,kBAAkB,GAAGD,0BAA0B;MACvDF,OAAO,CAACK,MAAM,GAAGD,cAAc;IACjC,CAAC;EACH,CAAC,EAAE,CAACzB,QAAQ,EAAEI,mBAAmB,EAAEH,OAAO,CAAC,CAAC;EAE5C,MAAMqD,mBAAmB,GAAGrF,WAAW,CAAC,MAAM;IAC5CoC,sBAAsB,CAAEkD,IAAI,IAAK,CAACA,IAAI,CAAC;EACzC,CAAC,EAAE,EAAE,CAAC;EAEN,MAAMC,eAAe,GAAGvF,WAAW,CAAEiF,QAAgB,IAAK;IACxD,IAAIlD,QAAQ,IAAI,IAAI,EAAE;IACtB,MAAMqB,OAAO,GAAGrB,QAAQ,CAACsB,iBAAiB;;IAE1C;IACA;IACA;IACA;IACA,MAAMO,UAAU,GAAGnB,eAAe,CAACE,OAAO,CAACkB,GAAG,CAACoB,QAAQ,CAAC;IACxD,IAAIrB,UAAU,IAAI,IAAI,EAAE;MACtBnB,eAAe,CAACE,OAAO,CAACmB,MAAM,CAACmB,QAAQ,CAAC;MACxC1C,iBAAiB,CAACI,OAAO,CAACmB,MAAM,CAACmB,QAAQ,CAAC;MAC1C3C,oBAAoB,CAACK,OAAO,GAAGiB,UAAU,CAAC;IAC5C;;IAEA;IACA,MAAMI,UAAU,GAAGZ,OAAO,CAACa,MAAM,CAAC,CAAmC;IACrE,MAAME,KAAK,GAAGH,UAAU,GAAGiB,QAAQ,CAAC;IACpC,IAAIO,MAAM,CAACrB,KAAK,CAAC,IAAIA,KAAK,CAACW,GAAG,YAAYW,WAAW,EAAE;MACrDtB,KAAK,CAACW,GAAG,CAACrB,MAAM,CAAC,CAAC;IACpB;IAEAL,OAAO,CAACK,MAAM,CAACwB,QAAQ,CAAC;EAC1B,CAAC,EAAE,CAAClD,QAAQ,CAAC,CAAC;EAEd,OAAO;IACLI,mBAAmB;IACnBkD,mBAAmB;IACnBE;EACF,CAAC;AACH;AAEA,SAASC,MAAMA,CAACE,GAAY,EAA2B;EACrD,OAAOA,GAAG,IAAI,IAAI,IAAI,OAAOA,GAAG,KAAK,QAAQ,IAAI,KAAK,IAAIA,GAAG;AAC/D","ignoreList":[]}

package/build/browser/pdf-viewer/hooks/usePdfOutline.js

@@ -0,0 +1,268 @@
+/*
+ * Copyright 2025 Palantir Technologies, Inc. All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { useEffect, useState } from "react";
+import { OUTLINE_HEADING_SIZE_RATIO, OUTLINE_MAX_HEADING_LENGTH } from "../constants.js";
+const EMPTY_OUTLINE = [];
+export function usePdfOutline(document) {
+  const [outlineItems, setOutlineItems] = useState(EMPTY_OUTLINE);
+  useEffect(function () {
+    if (document == null) {
+      setOutlineItems(EMPTY_OUTLINE);
+      return;
+    }
+    let cancelled = false;
+    void (async () => {
+      try {
+        // Try embedded bookmarks first
+        const outline = await document.getOutline();
+        if (cancelled) return;
+        if (outline != null && outline.length > 0) {
+          const items = await resolveBookmarkOutline(document, outline);
+          if (!cancelled) {
+            setOutlineItems(items);
+          }
+          return;
+        }
+
+        // Fall back to extracting headings from text content
+        const items = await extractHeadingsFromText(document);
+        if (!cancelled) {
+          setOutlineItems(items.length > 0 ? items : EMPTY_OUTLINE);
+        }
+      } catch {
+        // If extraction fails entirely, leave outline empty
+        if (!cancelled) {
+          setOutlineItems(EMPTY_OUTLINE);
+        }
+      }
+    })();
+    return () => {
+      cancelled = true;
+    };
+  }, [document]);
+  return outlineItems;
+}
+async function resolveBookmarkOutline(document, outline) {
+  const items = [];
+  const resolveItems = async (nodes, depth) => {
+    for (const node of nodes) {
+      let pageNumber = 1;
+      try {
+        if (typeof node.dest === "string") {
+          const dest = await document.getDestination(node.dest);
+          if (dest != null) {
+            const pageIndex = await document.getPageIndex(dest[0]);
+            pageNumber = pageIndex + 1;
+          }
+        } else if (Array.isArray(node.dest) && node.dest.length > 0) {
+          const pageIndex = await document.getPageIndex(node.dest[0]);
+          pageNumber = pageIndex + 1;
+        }
+      } catch {
+        // If destination resolution fails, default to page 1
+      }
+      items.push({
+        title: node.title,
+        depth,
+        pageNumber,
+        bold: node.bold,
+        italic: node.italic
+      });
+      if (node.items.length > 0) {
+        await resolveItems(node.items, depth + 1);
+      }
+    }
+  };
+  await resolveItems(outline, 0);
+  return items;
+}
+function isTextItem(item) {
+  return "str" in item && "transform" in item;
+}
+function roundFontSize(size) {
+  return Math.round(size * 10) / 10;
+}
+
+/**
+ * Extract the font size from a pdfjs TextItem.
+ * Prefer `height` when available, otherwise derive from the transform matrix.
+ * The transform is [scaleX, skewY, skewX, scaleY, translateX, translateY].
+ * Font size = sqrt(skewX² + scaleY²), which handles rotated text too.
+ */
+function getFontSize(item) {
+  if (item.height > 0) {
+    return item.height;
+  }
+  const transform = item.transform;
+  if (Array.isArray(transform) && transform.length >= 4) {
+    return Math.sqrt(transform[2] * transform[2] + transform[3] * transform[3]);
+  }
+  return 0;
+}
+async function extractHeadingsFromText(document) {
+  const numPages = document.numPages;
+
+  // Fetch all pages' text content in parallel
+  const pageContents = await Promise.all(Array.from({
+    length: numPages
+  }, (_, i) => document.getPage(i + 1).then(page => page.getTextContent())));
+
+  // Collect font size frequencies across all pages
+  const sizeCounts = new Map();
+  for (const textContent of pageContents) {
+    for (const rawItem of textContent.items) {
+      const item = rawItem;
+      if (!isTextItem(item) || item.str.trim().length === 0) {
+        continue;
+      }
+      const fontSize = getFontSize(item);
+      if (fontSize > 0) {
+        const rounded = roundFontSize(fontSize);
+        sizeCounts.set(rounded, (sizeCounts.get(rounded) ?? 0) + 1);
+      }
+    }
+  }
+
+  // Find the most common font size (body text)
+  let bodySize = 0;
+  let maxCount = 0;
+  for (const [size, count] of sizeCounts) {
+    if (count > maxCount) {
+      maxCount = count;
+      bodySize = size;
+    }
+  }
+  if (bodySize === 0) {
+    return [];
+  }
+
+  // Heading threshold: text significantly larger than body text
+  const headingThreshold = bodySize * OUTLINE_HEADING_SIZE_RATIO;
+
+  // Collect all unique heading sizes above threshold
+  const headingSizes = new Set();
+  for (const [size] of sizeCounts) {
+    if (size >= headingThreshold) {
+      headingSizes.add(size);
+    }
+  }
+  if (headingSizes.size === 0) {
+    return [];
+  }
+
+  // Sort heading sizes descending (largest = depth 0)
+  const sortedSizes = [...headingSizes].sort((a, b) => b - a);
+  const sizeToDepth = new Map();
+  for (let i = 0; i < sortedSizes.length; i++) {
+    sizeToDepth.set(sortedSizes[i], i);
+  }
+
+  // Extract headings with page numbers.
+  // pdfjs splits visual lines into multiple TextItems (e.g. "1." and
+  // "Introduction" are separate items). We accumulate all items on the
+  // same visual line and treat the line as a heading if most of its
+  // content is at a heading font size.
+  const candidateHeadings = [];
+  for (let pageIdx = 0; pageIdx < pageContents.length; pageIdx++) {
+    const textContent = pageContents[pageIdx];
+    const pageNumber = pageIdx + 1;
+    let lineItems = [];
+    let linePageNumber = pageNumber;
+    let lineFontName = "";
+    let lineHeadingFontSize = 0;
+    let lastFlushWasHeading = false;
+    const flushLine = () => {
+      if (lineItems.length === 0) return;
+
+      // Check if the majority of the line's non-whitespace content is heading-sized
+      let headingChars = 0;
+      let totalChars = 0;
+      for (const li of lineItems) {
+        const len = li.text.trim().length;
+        totalChars += len;
+        if (sizeToDepth.has(roundFontSize(li.fontSize))) {
+          headingChars += len;
+        }
+      }
+      if (totalChars > 0 && headingChars / totalChars > 0.5) {
+        const fullText = lineItems.map(li => li.text).join("").trim();
+        if (fullText.length > 0) {
+          candidateHeadings.push({
+            text: fullText,
+            fontSize: lineHeadingFontSize,
+            pageNumber: linePageNumber,
+            fontName: lineFontName,
+            contiguous: lastFlushWasHeading
+          });
+        }
+        lastFlushWasHeading = true;
+      } else {
+        lastFlushWasHeading = false;
+      }
+      lineItems = [];
+      lineHeadingFontSize = 0;
+    };
+    for (const rawItem of textContent.items) {
+      const item = rawItem;
+      if (!isTextItem(item)) {
+        continue;
+      }
+      const fontSize = getFontSize(item);
+      const rounded = roundFontSize(fontSize);
+      const isHeadingSized = sizeToDepth.has(rounded);
+      if (isHeadingSized && lineHeadingFontSize === 0) {
+        lineHeadingFontSize = rounded;
+        linePageNumber = pageNumber;
+        lineFontName = item.fontName;
+      }
+      if (lineHeadingFontSize > 0) {
+        lineItems.push({
+          text: item.str,
+          fontSize
+        });
+      }
+      if (item.hasEOL) {
+        flushLine();
+      }
+    }
+
+    // Flush remaining items at end of page
+    flushLine();
+  }
+
+  // Merge consecutive heading lines that were contiguous (no body text between them)
+  // and at the same font size. This joins multi-line titles that wrap in the PDF.
+  const mergedHeadings = [];
+  for (const heading of candidateHeadings) {
+    const prev = mergedHeadings[mergedHeadings.length - 1];
+    if (prev != null && heading.contiguous && Math.abs(prev.fontSize - heading.fontSize) < 0.1) {
+      prev.text += " " + heading.text;
+    } else {
+      mergedHeadings.push({
+        ...heading
+      });
+    }
+  }
+  return mergedHeadings.filter(heading => heading.text.trim().length <= OUTLINE_MAX_HEADING_LENGTH).map(heading => ({
+    title: heading.text.trim(),
+    depth: sizeToDepth.get(heading.fontSize) ?? 0,
+    pageNumber: heading.pageNumber,
+    bold: false,
+    italic: false
+  }));
+}
+//# sourceMappingURL=usePdfOutline.js.map

package/build/browser/pdf-viewer/hooks/usePdfOutline.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usePdfOutline.js","names":["useEffect","useState","OUTLINE_HEADING_SIZE_RATIO","OUTLINE_MAX_HEADING_LENGTH","EMPTY_OUTLINE","usePdfOutline","document","outlineItems","setOutlineItems","cancelled","outline","getOutline","length","items","resolveBookmarkOutline","extractHeadingsFromText","resolveItems","nodes","depth","node","pageNumber","dest","getDestination","pageIndex","getPageIndex","Array","isArray","push","title","bold","italic","isTextItem","item","roundFontSize","size","Math","round","getFontSize","height","transform","sqrt","numPages","pageContents","Promise","all","from","_","i","getPage","then","page","getTextContent","sizeCounts","Map","textContent","rawItem","str","trim","fontSize","rounded","set","get","bodySize","maxCount","count","headingThreshold","headingSizes","Set","add","sortedSizes","sort","a","b","sizeToDepth","candidateHeadings","pageIdx","lineItems","linePageNumber","lineFontName","lineHeadingFontSize","lastFlushWasHeading","flushLine","headingChars","totalChars","li","len","text","has","fullText","map","join","fontName","contiguous","isHeadingSized","hasEOL","mergedHeadings","heading","prev","abs","filter"],"sources":["usePdfOutline.ts"],"sourcesContent":["/*\n * Copyright 2025 Palantir Technologies, Inc. All rights reserved.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { PDFDocumentProxy } from \"pdfjs-dist\";\nimport { useEffect, useState } from \"react\";\nimport {\n  OUTLINE_HEADING_SIZE_RATIO,\n  OUTLINE_MAX_HEADING_LENGTH,\n} from \"../constants.js\";\nimport type { OutlineItem } from \"../types.js\";\n\nconst EMPTY_OUTLINE: OutlineItem[] = [];\n\nexport function usePdfOutline(\n  document: PDFDocumentProxy | undefined,\n): OutlineItem[] {\n  const [outlineItems, setOutlineItems] = useState<OutlineItem[]>(\n    EMPTY_OUTLINE,\n  );\n\n  useEffect(\n    function loadOutline() {\n      if (document == null) {\n        setOutlineItems(EMPTY_OUTLINE);\n        return;\n      }\n\n      let cancelled = false;\n\n      void (async () => {\n        try {\n          // Try embedded bookmarks first\n          const outline = await document.getOutline();\n          if (cancelled) return;\n\n          if (outline != null && outline.length > 0) {\n            const items = await resolveBookmarkOutline(document, outline);\n            if (!cancelled) {\n              setOutlineItems(items);\n            }\n            return;\n          }\n\n          // Fall back to extracting headings from text content\n          const items = await extractHeadingsFromText(document);\n          if (!cancelled) {\n            setOutlineItems(items.length > 0 ? items : EMPTY_OUTLINE);\n          }\n        } catch {\n          // If extraction fails entirely, leave outline empty\n          if (!cancelled) {\n            setOutlineItems(EMPTY_OUTLINE);\n          }\n        }\n      })();\n\n      return () => {\n        cancelled = true;\n      };\n    },\n    [document],\n  );\n\n  return outlineItems;\n}\n\ntype BookmarkNode = Awaited<ReturnType<PDFDocumentProxy[\"getOutline\"]>>[number];\n\nasync function resolveBookmarkOutline(\n  document: PDFDocumentProxy,\n  outline: BookmarkNode[],\n): Promise<OutlineItem[]> {\n  const items: OutlineItem[] = [];\n\n  const resolveItems = async (\n    nodes: BookmarkNode[],\n    depth: number,\n  ): Promise<void> => {\n    for (const node of nodes) {\n      let pageNumber = 1;\n      try {\n        if (typeof node.dest === \"string\") {\n          const dest = await document.getDestination(node.dest);\n          if (dest != null) {\n            const pageIndex = await document.getPageIndex(dest[0]);\n            pageNumber = pageIndex + 1;\n          }\n        } else if (Array.isArray(node.dest) && node.dest.length > 0) {\n          const pageIndex = await document.getPageIndex(node.dest[0]);\n          pageNumber = pageIndex + 1;\n        }\n      } catch {\n        // If destination resolution fails, default to page 1\n      }\n\n      items.push({\n        title: node.title,\n        depth,\n        pageNumber,\n        bold: node.bold,\n        italic: node.italic,\n      });\n\n      if (node.items.length > 0) {\n        await resolveItems(node.items, depth + 1);\n      }\n    }\n  };\n\n  await resolveItems(outline, 0);\n  return items;\n}\n\ninterface RawHeading {\n  text: string;\n  fontSize: number;\n  pageNumber: number;\n  fontName: string;\n  /** True if this heading line immediately followed another heading line (no body text in between). */\n  contiguous: boolean;\n}\n\ninterface TextItemLike {\n  str: string;\n  height: number;\n  fontName: string;\n  transform: number[];\n  hasEOL: boolean;\n}\n\nfunction isTextItem(\n  item: Record<string, unknown>,\n): item is Record<string, unknown> & TextItemLike {\n  return \"str\" in item && \"transform\" in item;\n}\n\nfunction roundFontSize(size: number): number {\n  return Math.round(size * 10) / 10;\n}\n\n/**\n * Extract the font size from a pdfjs TextItem.\n * Prefer `height` when available, otherwise derive from the transform matrix.\n * The transform is [scaleX, skewY, skewX, scaleY, translateX, translateY].\n * Font size = sqrt(skewX² + scaleY²), which handles rotated text too.\n */\nfunction getFontSize(item: TextItemLike): number {\n  if (item.height > 0) {\n    return item.height;\n  }\n  const transform = item.transform;\n  if (Array.isArray(transform) && transform.length >= 4) {\n    return Math.sqrt(\n      (transform[2]) * (transform[2])\n        + (transform[3]) * (transform[3]),\n    );\n  }\n  return 0;\n}\n\nasync function extractHeadingsFromText(\n  document: PDFDocumentProxy,\n): Promise<OutlineItem[]> {\n  const numPages = document.numPages;\n\n  // Fetch all pages' text content in parallel\n  const pageContents = await Promise.all(\n    Array.from(\n      { length: numPages },\n      (_, i) => document.getPage(i + 1).then(page => page.getTextContent()),\n    ),\n  );\n\n  // Collect font size frequencies across all pages\n  const sizeCounts = new Map<number, number>();\n  for (const textContent of pageContents) {\n    for (const rawItem of textContent.items) {\n      const item = rawItem as Record<string, unknown>;\n      if (!isTextItem(item) || item.str.trim().length === 0) {\n        continue;\n      }\n      const fontSize = getFontSize(item);\n      if (fontSize > 0) {\n        const rounded = roundFontSize(fontSize);\n        sizeCounts.set(rounded, (sizeCounts.get(rounded) ?? 0) + 1);\n      }\n    }\n  }\n\n  // Find the most common font size (body text)\n  let bodySize = 0;\n  let maxCount = 0;\n  for (const [size, count] of sizeCounts) {\n    if (count > maxCount) {\n      maxCount = count;\n      bodySize = size;\n    }\n  }\n\n  if (bodySize === 0) {\n    return [];\n  }\n\n  // Heading threshold: text significantly larger than body text\n  const headingThreshold = bodySize * OUTLINE_HEADING_SIZE_RATIO;\n\n  // Collect all unique heading sizes above threshold\n  const headingSizes = new Set<number>();\n  for (const [size] of sizeCounts) {\n    if (size >= headingThreshold) {\n      headingSizes.add(size);\n    }\n  }\n\n  if (headingSizes.size === 0) {\n    return [];\n  }\n\n  // Sort heading sizes descending (largest = depth 0)\n  const sortedSizes = [...headingSizes].sort((a, b) => b - a);\n  const sizeToDepth = new Map<number, number>();\n  for (let i = 0; i < sortedSizes.length; i++) {\n    sizeToDepth.set(sortedSizes[i], i);\n  }\n\n  // Extract headings with page numbers.\n  // pdfjs splits visual lines into multiple TextItems (e.g. \"1.\" and\n  // \"Introduction\" are separate items). We accumulate all items on the\n  // same visual line and treat the line as a heading if most of its\n  // content is at a heading font size.\n  const candidateHeadings: RawHeading[] = [];\n\n  for (let pageIdx = 0; pageIdx < pageContents.length; pageIdx++) {\n    const textContent = pageContents[pageIdx];\n    const pageNumber = pageIdx + 1;\n\n    let lineItems: Array<{ text: string; fontSize: number }> = [];\n    let linePageNumber = pageNumber;\n    let lineFontName = \"\";\n    let lineHeadingFontSize = 0;\n    let lastFlushWasHeading = false;\n\n    const flushLine = (): void => {\n      if (lineItems.length === 0) return;\n\n      // Check if the majority of the line's non-whitespace content is heading-sized\n      let headingChars = 0;\n      let totalChars = 0;\n      for (const li of lineItems) {\n        const len = li.text.trim().length;\n        totalChars += len;\n        if (sizeToDepth.has(roundFontSize(li.fontSize))) {\n          headingChars += len;\n        }\n      }\n\n      if (totalChars > 0 && headingChars / totalChars > 0.5) {\n        const fullText = lineItems.map((li) => li.text).join(\"\").trim();\n        if (fullText.length > 0) {\n          candidateHeadings.push({\n            text: fullText,\n            fontSize: lineHeadingFontSize,\n            pageNumber: linePageNumber,\n            fontName: lineFontName,\n            contiguous: lastFlushWasHeading,\n          });\n        }\n        lastFlushWasHeading = true;\n      } else {\n        lastFlushWasHeading = false;\n      }\n\n      lineItems = [];\n      lineHeadingFontSize = 0;\n    };\n\n    for (const rawItem of textContent.items) {\n      const item = rawItem as Record<string, unknown>;\n      if (!isTextItem(item)) {\n        continue;\n      }\n\n      const fontSize = getFontSize(item);\n      const rounded = roundFontSize(fontSize);\n      const isHeadingSized = sizeToDepth.has(rounded);\n\n      if (isHeadingSized && lineHeadingFontSize === 0) {\n        lineHeadingFontSize = rounded;\n        linePageNumber = pageNumber;\n        lineFontName = item.fontName;\n      }\n\n      if (lineHeadingFontSize > 0) {\n        lineItems.push({ text: item.str, fontSize });\n      }\n\n      if (item.hasEOL) {\n        flushLine();\n      }\n    }\n\n    // Flush remaining items at end of page\n    flushLine();\n  }\n\n  // Merge consecutive heading lines that were contiguous (no body text between them)\n  // and at the same font size. This joins multi-line titles that wrap in the PDF.\n  const mergedHeadings: RawHeading[] = [];\n  for (const heading of candidateHeadings) {\n    const prev = mergedHeadings[mergedHeadings.length - 1];\n    if (\n      prev != null\n      && heading.contiguous\n      && Math.abs(prev.fontSize - heading.fontSize) < 0.1\n    ) {\n      prev.text += \" \" + heading.text;\n    } else {\n      mergedHeadings.push({ ...heading });\n    }\n  }\n\n  return mergedHeadings\n    .filter((heading) =>\n      heading.text.trim().length <= OUTLINE_MAX_HEADING_LENGTH\n    )\n    .map((heading) => ({\n      title: heading.text.trim(),\n      depth: sizeToDepth.get(heading.fontSize) ?? 0,\n      pageNumber: heading.pageNumber,\n      bold: false,\n      italic: false,\n    }));\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAGA,SAASA,SAAS,EAAEC,QAAQ,QAAQ,OAAO;AAC3C,SACEC,0BAA0B,EAC1BC,0BAA0B,QACrB,iBAAiB;AAGxB,MAAMC,aAA4B,GAAG,EAAE;AAEvC,OAAO,SAASC,aAAaA,CAC3BC,QAAsC,EACvB;EACf,MAAM,CAACC,YAAY,EAAEC,eAAe,CAAC,GAAGP,QAAQ,CAC9CG,aACF,CAAC;EAEDJ,SAAS,CACP,YAAuB;IACrB,IAAIM,QAAQ,IAAI,IAAI,EAAE;MACpBE,eAAe,CAACJ,aAAa,CAAC;MAC9B;IACF;IAEA,IAAIK,SAAS,GAAG,KAAK;IAErB,KAAK,CAAC,YAAY;MAChB,IAAI;QACF;QACA,MAAMC,OAAO,GAAG,MAAMJ,QAAQ,CAACK,UAAU,CAAC,CAAC;QAC3C,IAAIF,SAAS,EAAE;QAEf,IAAIC,OAAO,IAAI,IAAI,IAAIA,OAAO,CAACE,MAAM,GAAG,CAAC,EAAE;UACzC,MAAMC,KAAK,GAAG,MAAMC,sBAAsB,CAACR,QAAQ,EAAEI,OAAO,CAAC;UAC7D,IAAI,CAACD,SAAS,EAAE;YACdD,eAAe,CAACK,KAAK,CAAC;UACxB;UACA;QACF;;QAEA;QACA,MAAMA,KAAK,GAAG,MAAME,uBAAuB,CAACT,QAAQ,CAAC;QACrD,IAAI,CAACG,SAAS,EAAE;UACdD,eAAe,CAACK,KAAK,CAACD,MAAM,GAAG,CAAC,GAAGC,KAAK,GAAGT,aAAa,CAAC;QAC3D;MACF,CAAC,CAAC,MAAM;QACN;QACA,IAAI,CAACK,SAAS,EAAE;UACdD,eAAe,CAACJ,aAAa,CAAC;QAChC;MACF;IACF,CAAC,EAAE,CAAC;IAEJ,OAAO,MAAM;MACXK,SAAS,GAAG,IAAI;IAClB,CAAC;EACH,CAAC,EACD,CAACH,QAAQ,CACX,CAAC;EAED,OAAOC,YAAY;AACrB;AAIA,eAAeO,sBAAsBA,CACnCR,QAA0B,EAC1BI,OAAuB,EACC;EACxB,MAAMG,KAAoB,GAAG,EAAE;EAE/B,MAAMG,YAAY,GAAG,MAAAA,CACnBC,KAAqB,EACrBC,KAAa,KACK;IAClB,KAAK,MAAMC,IAAI,IAAIF,KAAK,EAAE;MACxB,IAAIG,UAAU,GAAG,CAAC;MAClB,IAAI;QACF,IAAI,OAAOD,IAAI,CAACE,IAAI,KAAK,QAAQ,EAAE;UACjC,MAAMA,IAAI,GAAG,MAAMf,QAAQ,CAACgB,cAAc,CAACH,IAAI,CAACE,IAAI,CAAC;UACrD,IAAIA,IAAI,IAAI,IAAI,EAAE;YAChB,MAAME,SAAS,GAAG,MAAMjB,QAAQ,CAACkB,YAAY,CAACH,IAAI,CAAC,CAAC,CAAC,CAAC;YACtDD,UAAU,GAAGG,SAAS,GAAG,CAAC;UAC5B;QACF,CAAC,MAAM,IAAIE,KAAK,CAACC,OAAO,CAACP,IAAI,CAACE,IAAI,CAAC,IAAIF,IAAI,CAACE,IAAI,CAACT,MAAM,GAAG,CAAC,EAAE;UAC3D,MAAMW,SAAS,GAAG,MAAMjB,QAAQ,CAACkB,YAAY,CAACL,IAAI,CAACE,IAAI,CAAC,CAAC,CAAC,CAAC;UAC3DD,UAAU,GAAGG,SAAS,GAAG,CAAC;QAC5B;MACF,CAAC,CAAC,MAAM;QACN;MAAA;MAGFV,KAAK,CAACc,IAAI,CAAC;QACTC,KAAK,EAAET,IAAI,CAACS,KAAK;QACjBV,KAAK;QACLE,UAAU;QACVS,IAAI,EAAEV,IAAI,CAACU,IAAI;QACfC,MAAM,EAAEX,IAAI,CAACW;MACf,CAAC,CAAC;MAEF,IAAIX,IAAI,CAACN,KAAK,CAACD,MAAM,GAAG,CAAC,EAAE;QACzB,MAAMI,YAAY,CAACG,IAAI,CAACN,KAAK,EAAEK,KAAK,GAAG,CAAC,CAAC;MAC3C;IACF;EACF,CAAC;EAED,MAAMF,YAAY,CAACN,OAAO,EAAE,CAAC,CAAC;EAC9B,OAAOG,KAAK;AACd;AAmBA,SAASkB,UAAUA,CACjBC,IAA6B,EACmB;EAChD,OAAO,KAAK,IAAIA,IAAI,IAAI,WAAW,IAAIA,IAAI;AAC7C;AAEA,SAASC,aAAaA,CAACC,IAAY,EAAU;EAC3C,OAAOC,IAAI,CAACC,KAAK,CAACF,IAAI,GAAG,EAAE,CAAC,GAAG,EAAE;AACnC;;AAEA;AACA;AACA;AACA;AACA;AACA;AACA,SAASG,WAAWA,CAACL,IAAkB,EAAU;EAC/C,IAAIA,IAAI,CAACM,MAAM,GAAG,CAAC,EAAE;IACnB,OAAON,IAAI,CAACM,MAAM;EACpB;EACA,MAAMC,SAAS,GAAGP,IAAI,CAACO,SAAS;EAChC,IAAId,KAAK,CAACC,OAAO,CAACa,SAAS,CAAC,IAAIA,SAAS,CAAC3B,MAAM,IAAI,CAAC,EAAE;IACrD,OAAOuB,IAAI,CAACK,IAAI,CACbD,SAAS,CAAC,CAAC,CAAC,GAAKA,SAAS,CAAC,CAAC,CAAE,GAC1BA,SAAS,CAAC,CAAC,CAAC,GAAKA,SAAS,CAAC,CAAC,CACnC,CAAC;EACH;EACA,OAAO,CAAC;AACV;AAEA,eAAexB,uBAAuBA,CACpCT,QAA0B,EACF;EACxB,MAAMmC,QAAQ,GAAGnC,QAAQ,CAACmC,QAAQ;;EAElC;EACA,MAAMC,YAAY,GAAG,MAAMC,OAAO,CAACC,GAAG,CACpCnB,KAAK,CAACoB,IAAI,CACR;IAAEjC,MAAM,EAAE6B;EAAS,CAAC,EACpB,CAACK,CAAC,EAAEC,CAAC,KAAKzC,QAAQ,CAAC0C,OAAO,CAACD,CAAC,GAAG,CAAC,CAAC,CAACE,IAAI,CAACC,IAAI,IAAIA,IAAI,CAACC,cAAc,CAAC,CAAC,CACtE,CACF,CAAC;;EAED;EACA,MAAMC,UAAU,GAAG,IAAIC,GAAG,CAAiB,CAAC;EAC5C,KAAK,MAAMC,WAAW,IAAIZ,YAAY,EAAE;IACtC,KAAK,MAAMa,OAAO,IAAID,WAAW,CAACzC,KAAK,EAAE;MACvC,MAAMmB,IAAI,GAAGuB,OAAkC;MAC/C,IAAI,CAACxB,UAAU,CAACC,IAAI,CAAC,IAAIA,IAAI,CAACwB,GAAG,CAACC,IAAI,CAAC,CAAC,CAAC7C,MAAM,KAAK,CAAC,EAAE;QACrD;MACF;MACA,MAAM8C,QAAQ,GAAGrB,WAAW,CAACL,IAAI,CAAC;MAClC,IAAI0B,QAAQ,GAAG,CAAC,EAAE;QAChB,MAAMC,OAAO,GAAG1B,aAAa,CAACyB,QAAQ,CAAC;QACvCN,UAAU,CAACQ,GAAG,CAACD,OAAO,EAAE,CAACP,UAAU,CAACS,GAAG,CAACF,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;MAC7D;IACF;EACF;;EAEA;EACA,IAAIG,QAAQ,GAAG,CAAC;EAChB,IAAIC,QAAQ,GAAG,CAAC;EAChB,KAAK,MAAM,CAAC7B,IAAI,EAAE8B,KAAK,CAAC,IAAIZ,UAAU,EAAE;IACtC,IAAIY,KAAK,GAAGD,QAAQ,EAAE;MACpBA,QAAQ,GAAGC,KAAK;MAChBF,QAAQ,GAAG5B,IAAI;IACjB;EACF;EAEA,IAAI4B,QAAQ,KAAK,CAAC,EAAE;IAClB,OAAO,EAAE;EACX;;EAEA;EACA,MAAMG,gBAAgB,GAAGH,QAAQ,GAAG5D,0BAA0B;;EAE9D;EACA,MAAMgE,YAAY,GAAG,IAAIC,GAAG,CAAS,CAAC;EACtC,KAAK,MAAM,CAACjC,IAAI,CAAC,IAAIkB,UAAU,EAAE;IAC/B,IAAIlB,IAAI,IAAI+B,gBAAgB,EAAE;MAC5BC,YAAY,CAACE,GAAG,CAAClC,IAAI,CAAC;IACxB;EACF;EAEA,IAAIgC,YAAY,CAAChC,IAAI,KAAK,CAAC,EAAE;IAC3B,OAAO,EAAE;EACX;;EAEA;EACA,MAAMmC,WAAW,GAAG,CAAC,GAAGH,YAAY,CAAC,CAACI,IAAI,CAAC,CAACC,CAAC,EAAEC,CAAC,KAAKA,CAAC,GAAGD,CAAC,CAAC;EAC3D,MAAME,WAAW,GAAG,IAAIpB,GAAG,CAAiB,CAAC;EAC7C,KAAK,IAAIN,CAAC,GAAG,CAAC,EAAEA,CAAC,GAAGsB,WAAW,CAACzD,MAAM,EAAEmC,CAAC,EAAE,EAAE;IAC3C0B,WAAW,CAACb,GAAG,CAACS,WAAW,CAACtB,CAAC,CAAC,EAAEA,CAAC,CAAC;EACpC;;EAEA;EACA;EACA;EACA;EACA;EACA,MAAM2B,iBAA+B,GAAG,EAAE;EAE1C,KAAK,IAAIC,OAAO,GAAG,CAAC,EAAEA,OAAO,GAAGjC,YAAY,CAAC9B,MAAM,EAAE+D,OAAO,EAAE,EAAE;IAC9D,MAAMrB,WAAW,GAAGZ,YAAY,CAACiC,OAAO,CAAC;IACzC,MAAMvD,UAAU,GAAGuD,OAAO,GAAG,CAAC;IAE9B,IAAIC,SAAoD,GAAG,EAAE;IAC7D,IAAIC,cAAc,GAAGzD,UAAU;IAC/B,IAAI0D,YAAY,GAAG,EAAE;IACrB,IAAIC,mBAAmB,GAAG,CAAC;IAC3B,IAAIC,mBAAmB,GAAG,KAAK;IAE/B,MAAMC,SAAS,GAAGA,CAAA,KAAY;MAC5B,IAAIL,SAAS,CAAChE,MAAM,KAAK,CAAC,EAAE;;MAE5B;MACA,IAAIsE,YAAY,GAAG,CAAC;MACpB,IAAIC,UAAU,GAAG,CAAC;MAClB,KAAK,MAAMC,EAAE,IAAIR,SAAS,EAAE;QAC1B,MAAMS,GAAG,GAAGD,EAAE,CAACE,IAAI,CAAC7B,IAAI,CAAC,CAAC,CAAC7C,MAAM;QACjCuE,UAAU,IAAIE,GAAG;QACjB,IAAIZ,WAAW,CAACc,GAAG,CAACtD,aAAa,CAACmD,EAAE,CAAC1B,QAAQ,CAAC,CAAC,EAAE;UAC/CwB,YAAY,IAAIG,GAAG;QACrB;MACF;MAEA,IAAIF,UAAU,GAAG,CAAC,IAAID,YAAY,GAAGC,UAAU,GAAG,GAAG,EAAE;QACrD,MAAMK,QAAQ,GAAGZ,SAAS,CAACa,GAAG,CAAEL,EAAE,IAAKA,EAAE,CAACE,IAAI,CAAC,CAACI,IAAI,CAAC,EAAE,CAAC,CAACjC,IAAI,CAAC,CAAC;QAC/D,IAAI+B,QAAQ,CAAC5E,MAAM,GAAG,CAAC,EAAE;UACvB8D,iBAAiB,CAAC/C,IAAI,CAAC;YACrB2D,IAAI,EAAEE,QAAQ;YACd9B,QAAQ,EAAEqB,mBAAmB;YAC7B3D,UAAU,EAAEyD,cAAc;YAC1Bc,QAAQ,EAAEb,YAAY;YACtBc,UAAU,EAAEZ;UACd,CAAC,CAAC;QACJ;QACAA,mBAAmB,GAAG,IAAI;MAC5B,CAAC,MAAM;QACLA,mBAAmB,GAAG,KAAK;MAC7B;MAEAJ,SAAS,GAAG,EAAE;MACdG,mBAAmB,GAAG,CAAC;IACzB,CAAC;IAED,KAAK,MAAMxB,OAAO,IAAID,WAAW,CAACzC,KAAK,EAAE;MACvC,MAAMmB,IAAI,GAAGuB,OAAkC;MAC/C,IAAI,CAACxB,UAAU,CAACC,IAAI,CAAC,EAAE;QACrB;MACF;MAEA,MAAM0B,QAAQ,GAAGrB,WAAW,CAACL,IAAI,CAAC;MAClC,MAAM2B,OAAO,GAAG1B,aAAa,CAACyB,QAAQ,CAAC;MACvC,MAAMmC,cAAc,GAAGpB,WAAW,CAACc,GAAG,CAAC5B,OAAO,CAAC;MAE/C,IAAIkC,cAAc,IAAId,mBAAmB,KAAK,CAAC,EAAE;QAC/CA,mBAAmB,GAAGpB,OAAO;QAC7BkB,cAAc,GAAGzD,UAAU;QAC3B0D,YAAY,GAAG9C,IAAI,CAAC2D,QAAQ;MAC9B;MAEA,IAAIZ,mBAAmB,GAAG,CAAC,EAAE;QAC3BH,SAAS,CAACjD,IAAI,CAAC;UAAE2D,IAAI,EAAEtD,IAAI,CAACwB,GAAG;UAAEE;QAAS,CAAC,CAAC;MAC9C;MAEA,IAAI1B,IAAI,CAAC8D,MAAM,EAAE;QACfb,SAAS,CAAC,CAAC;MACb;IACF;;IAEA;IACAA,SAAS,CAAC,CAAC;EACb;;EAEA;EACA;EACA,MAAMc,cAA4B,GAAG,EAAE;EACvC,KAAK,MAAMC,OAAO,IAAItB,iBAAiB,EAAE;IACvC,MAAMuB,IAAI,GAAGF,cAAc,CAACA,cAAc,CAACnF,MAAM,GAAG,CAAC,CAAC;IACtD,IACEqF,IAAI,IAAI,IAAI,IACTD,OAAO,CAACJ,UAAU,IAClBzD,IAAI,CAAC+D,GAAG,CAACD,IAAI,CAACvC,QAAQ,GAAGsC,OAAO,CAACtC,QAAQ,CAAC,GAAG,GAAG,EACnD;MACAuC,IAAI,CAACX,IAAI,IAAI,GAAG,GAAGU,OAAO,CAACV,IAAI;IACjC,CAAC,MAAM;MACLS,cAAc,CAACpE,IAAI,CAAC;QAAE,GAAGqE;MAAQ,CAAC,CAAC;IACrC;EACF;EAEA,OAAOD,cAAc,CAClBI,MAAM,CAAEH,OAAO,IACdA,OAAO,CAACV,IAAI,CAAC7B,IAAI,CAAC,CAAC,CAAC7C,MAAM,IAAIT,0BAChC,CAAC,CACAsF,GAAG,CAAEO,OAAO,KAAM;IACjBpE,KAAK,EAAEoE,OAAO,CAACV,IAAI,CAAC7B,IAAI,CAAC,CAAC;IAC1BvC,KAAK,EAAEuD,WAAW,CAACZ,GAAG,CAACmC,OAAO,CAACtC,QAAQ,CAAC,IAAI,CAAC;IAC7CtC,UAAU,EAAE4E,OAAO,CAAC5E,UAAU;IAC9BS,IAAI,EAAE,KAAK;IACXC,MAAM,EAAE;EACV,CAAC,CAAC,CAAC;AACP","ignoreList":[]}

package/build/browser/pdf-viewer/hooks/usePdfViewer.js

@@ -0,0 +1,86 @@
+/*
+ * Copyright 2025 Palantir Technologies, Inc. All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { AnnotationEditorType } from "pdfjs-dist";
+import { EventBus, PDFFindController, PDFLinkService, PDFViewer } from "pdfjs-dist/web/pdf_viewer.mjs";
+import { useEffect, useRef } from "react";
+import { PAGES_LOADED_EVENT } from "../constants.js";
+export function usePdfViewer(containerRef, viewerRef, document, initialScale, initialPage) {
+  const pdfViewerRef = useRef(null);
+  const eventBusRef = useRef(null);
+  const findControllerRef = useRef(null);
+  useEffect(function () {
+    const container = containerRef.current;
+    const viewer = viewerRef.current;
+    if (container == null || viewer == null || document == null) {
+      return;
+    }
+    const eventBus = new EventBus();
+    const linkService = new PDFLinkService({
+      eventBus
+    });
+    const findController = new PDFFindController({
+      linkService,
+      eventBus,
+      updateMatchesCountOnProgress: true
+    });
+    const pdfViewer = new PDFViewer({
+      container,
+      viewer,
+      eventBus,
+      linkService,
+      findController,
+      removePageBorders: true,
+      annotationEditorMode: AnnotationEditorType.NONE,
+      annotationEditorHighlightColors: "yellow=#FFFF98,green=#53FFBC,blue=#80EBFF,pink=#FFCBE6,red=#FF4F5F"
+    });
+    linkService.setViewer(pdfViewer);
+    linkService.setDocument(document);
+    findController.setDocument(document);
+    pdfViewer.setDocument(document);
+
+    // Apply initial scale and page after pages are loaded to avoid
+    // "scrollPageIntoView: not a valid pageNumber" console errors.
+    const onPagesLoaded = () => {
+      if (initialScale != null) {
+        pdfViewer.currentScale = initialScale;
+      }
+      if (initialPage != null && initialPage > 1) {
+        pdfViewer.currentPageNumber = initialPage;
+        pdfViewer.scrollPageIntoView({
+          pageNumber: initialPage
+        });
+      }
+      eventBus.off(PAGES_LOADED_EVENT, onPagesLoaded);
+    };
+    eventBus.on(PAGES_LOADED_EVENT, onPagesLoaded);
+    eventBusRef.current = eventBus;
+    findControllerRef.current = findController;
+    pdfViewerRef.current = pdfViewer;
+    return () => {
+      pdfViewerRef.current = null;
+      eventBusRef.current = null;
+      findControllerRef.current = null;
+      pdfViewer.cleanup();
+    };
+  }, [containerRef, viewerRef, document]);
+  return {
+    pdfViewerRef,
+    eventBusRef,
+    findControllerRef
+  };
+}
+//# sourceMappingURL=usePdfViewer.js.map