fix: save changes to debug textDivs

watson-developer-cloud · Jun 3, 2024 · 0435a15 · 0435a15
1 parent 29f8d23
commit 0435a15
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 23 deletions.
diff --git a/...-react-components/src/components/DocumentPreview/components/PdfHighlight/PdfHighlight.tsx b/...-react-components/src/components/DocumentPreview/components/PdfHighlight/PdfHighlight.tsx
@@ -88,6 +88,8 @@ const PdfHighlight: FC<Props> = ({
 
   const { textDivs } = pdfRenderedText || {};
 
+  console.log('textDivs', textDivs);
+
   const highlightShapes = useMemo(() => {
     if (boxHighlights) {
       return getShapeFromBboxHighlight(boxHighlights, page);

diff --git a/...act-components/src/components/DocumentPreview/components/PdfViewer/PdfViewerTextLayer.tsx b/...act-components/src/components/DocumentPreview/components/PdfViewer/PdfViewerTextLayer.tsx
@@ -49,7 +49,7 @@ const PdfViewerTextLayer: FC<PdfViewerTextLayerProps> = ({
   setRenderedText = () => {}
 }) => {
   const textLayerRef = useRef<HTMLDivElement>(null);
-  const textLayerDiv = textLayerRef.current;
+  const textLayerWrapper = textLayerRef.current;
 
   // load text content from the page
   const loadedText = useAsyncFunctionCall(
@@ -67,23 +67,25 @@ const PdfViewerTextLayer: FC<PdfViewerTextLayerProps> = ({
   const renderedText = useAsyncFunctionCall(
     useCallback(
       async (signal: AbortSignal) => {
-        if (textLayerDiv && loadedText) {
+        if (textLayerWrapper && loadedText) {
           const { textContent, viewport, scale, page } = loadedText;
 
           const builder = new TextLayerBuilder({
-            textLayerDiv,
-            viewport,
-            eventBus: new EventBus(),
-            pageIndex: page - 1
+            pdfPage: loadedPage
           });
+          // trying to find a way to return textDivs
+          const textItems = textContent.items;
+          console.log('textItems', textItems);
+
+          textLayerWrapper.append(builder.div);
           signal.addEventListener('abort', () => builder.cancel());
 
-          await _renderTextLayer(builder, textContent, textLayerDiv, scale);
-          return { textContent, viewport, page, textDivs: builder.textDivs };
+          await _renderTextLayer(builder, textContent, textLayerWrapper, scale, viewport);
+          return { textContent, viewport, page, textDivs: [] };
         }
         return undefined;
       },
-      [loadedText, textLayerDiv]
+      [loadedPage, loadedText, textLayerWrapper]
     )
   );
 
@@ -112,24 +114,32 @@ async function _renderTextLayer(
   builder: TextLayerBuilder,
   textContent: TextContent,
   textLayerDiv: HTMLDivElement,
-  scale: number
+  scale: number,
+  viewport: PageViewport
+  // loadedPage: PDFPageProxy
 ) {
-  builder.setTextContent(textContent);
+  // TODO: delete if this works without it
+  // if (!builder.renderingDone) {
+  //   const readableStream = loadedPage.streamTextContent({
+  //     includeMarkedContent: true
+  //   })
+  //   builder.(readableStream);
+  // }
 
   // render
   textLayerDiv.innerHTML = '';
-  const deferredRenderEndPromise = new Promise<void>(resolve => {
-    const listener = () => {
-      resolve();
-      builder?.eventBus.off('textlayerrendered', listener);
-    };
-    builder?.eventBus.on('textlayerrendered', listener);
-  });
-
-  builder.render();
-  await deferredRenderEndPromise;
-
-  _adjustTextDivs(builder.textDivs, textContent.items as TextItem[], scale);
+  // const deferredRenderEndPromise = new Promise<void>(resolve => {
+  //   const listener = () => {
+  //     resolve();
+  //     builder?.eventBus.off('textlayerrendered', listener);
+  //   };
+  //   builder?.eventBus.on('textlayerrendered', listener);
+  // });
+
+  await builder.render(viewport);
+  // await deferredRenderEndPromise;
+
+  // _adjustTextDivs(builder.textDivs, textContent.items as TextItem[], scale);
 }
 
 /**