place image below in document preview

NASA-IMPACT · Oct 18, 2023 · 4e9563d · 4e9563d
1 parent 665cd00
commit 4e9563d
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 5 deletions.
diff --git a/app/assets/scripts/components/documents/pdf-preview/index.js b/app/assets/scripts/components/documents/pdf-preview/index.js
@@ -9,7 +9,11 @@ import DocumentBody from '../single-view/document-body';
 import DocumentTitle from '../single-view/document-title';
 import { DocumentProse } from '../single-view/document-content';
 import { ScrollAnchorProvider } from '../single-view/scroll-manager';
-import { applyNumberCaptionsToDocument } from '../../../utils/apply-number-captions-to-document';
+import {
+  PLACEMENTS,
+  applyNumberCaptionsToDocument
+} from '../../../utils/apply-number-captions-to-document';
+import { IMAGE_BLOCK } from '../../slate/plugins/constants';
 
 const TocHeader = styled.h1`
   border-bottom: 3px solid #000;
@@ -240,7 +244,11 @@ function PdfPreview() {
     }
 
     if (atbd.status === 'succeeded') {
-      setDocument(applyNumberCaptionsToDocument(atbd.data.document));
+      setDocument(
+        applyNumberCaptionsToDocument(atbd.data.document, {
+          imageCaptionBelow: true
+        })
+      );
       waitForImages();
     }
   }, [atbd.status]);

diff --git a/app/assets/scripts/utils/apply-number-captions-to-document.js b/app/assets/scripts/utils/apply-number-captions-to-document.js
@@ -7,7 +7,7 @@ import {
 /**
  * Include table numbers and move captions before the table in the document.
  */
-export function applyNumberCaptionsToDocument(document) {
+export function applyNumberCaptionsToDocument(document, placement) {
   // Section id list in the order they should appear in the document
   const documentSectionIds = [
     'key_points',
@@ -73,9 +73,9 @@ export function applyNumberCaptionsToDocument(document) {
 
               // Reverse the table rows to make caption appear first
               // and add the table number to the caption
-              return {
+              let nextChild = {
                 ...child,
-                children: child.children.reverse().map((c) => {
+                children: child.children.map((c) => {
                   if (c.type !== 'caption') {
                     return c;
                   }
@@ -93,6 +93,15 @@ export function applyNumberCaptionsToDocument(document) {
                   };
                 })
               };
+
+              if (child.type === IMAGE_BLOCK && placement?.imageCaptionBelow) {
+                return nextChild;
+              } else {
+                return {
+                  ...nextChild,
+                  children: nextChild.children.reverse()
+                };
+              }
             }
 
             return child;