From 489c4cbfe8a5cce37909e0be6652ee8dcb1bdab2 Mon Sep 17 00:00:00 2001
From: hamed musallam <hamed.musallam@gmail.com>
Date: Wed, 10 Jul 2024 18:56:04 +0200
Subject: [PATCH] fix: trigger printing after the NMRium viewer is rendered and
 rescaled to an A4 page

---
 src/component/main/InnerNMRiumContents.tsx | 18 +++++++++++-----
 src/component/main/NMRiumViewer.tsx        | 25 +++++++++++++++++++---
 2 files changed, 35 insertions(+), 8 deletions(-)

diff --git a/src/component/main/InnerNMRiumContents.tsx b/src/component/main/InnerNMRiumContents.tsx
index 33dd6cfd3..daac68a15 100644
--- a/src/component/main/InnerNMRiumContents.tsx
+++ b/src/component/main/InnerNMRiumContents.tsx
@@ -98,19 +98,23 @@ export function InnerNMRiumContents(props: InnerNMRiumContentsProps) {
   const [printFlag, setPrintFlag] = useState(false);
 
   useEffect(() => {
-    function handleBeforePrint() {
-      setPrintFlag(true);
-    }
     function handleAfterPrint() {
       setPrintFlag(false);
     }
 
-    window.addEventListener('beforeprint', handleBeforePrint);
+    function handleKeyDow(event) {
+      if ((event.ctrlKey || event.metaKey) && event.key === 'p') {
+        event.preventDefault();
+        setPrintFlag(true);
+      }
+    }
+
+    window.addEventListener('keydown', handleKeyDow);
     window.addEventListener('afterprint', handleAfterPrint);
 
     return () => {
-      window.removeEventListener('beforeprint', handleBeforePrint);
       window.removeEventListener('afterprint', handleAfterPrint);
+      window.removeEventListener('keydown', handleKeyDow);
     };
   }, []);
 
@@ -187,6 +191,10 @@ export function InnerNMRiumContents(props: InnerNMRiumContentsProps) {
             style={{
               width: '297mm',
               height: '210mm',
+              padding: '0.5cm',
+            }}
+            onRender={() => {
+              window.print();
             }}
           />
         )}
diff --git a/src/component/main/NMRiumViewer.tsx b/src/component/main/NMRiumViewer.tsx
index e82af10b2..9123398c9 100644
--- a/src/component/main/NMRiumViewer.tsx
+++ b/src/component/main/NMRiumViewer.tsx
@@ -1,4 +1,4 @@
-import { CSSProperties, RefObject } from 'react';
+import { CSSProperties, RefObject, useDeferredValue, useEffect } from 'react';
 
 import Viewer1D from '../1d/Viewer1D';
 import FloatMoleculeStructures from '../1d-2d/components/FloatMoleculeStructures';
@@ -11,11 +11,29 @@ interface NMRiumViewerProps {
   emptyText: NMRiumProps['emptyText'];
   viewerRef: RefObject<HTMLDivElement>;
   style?: CSSProperties;
+  onRender?: () => void;
 }
 
 export function NMRiumViewer(props: NMRiumViewerProps) {
-  const { emptyText, viewerRef, style = {} } = props;
-  const { displayerMode } = useChartData();
+  const { emptyText, viewerRef, onRender, style = {} } = props;
+  const { displayerMode, width, height } = useChartData();
+  const renderDimension = useDeferredValue({ width, height });
+  useEffect(() => {
+    if (typeof onRender !== 'function') {
+      return;
+    }
+
+    requestAnimationFrame(() => {
+      setTimeout(() => {
+        if (
+          renderDimension.width !== width ||
+          renderDimension.height !== height
+        ) {
+          onRender();
+        }
+      }, 0);
+    });
+  }, [onRender, width, height, renderDimension.width, renderDimension.height]);
 
   return (
     <div
@@ -26,6 +44,7 @@ export function NMRiumViewer(props: NMRiumViewerProps) {
         width: '100%',
         height: '100%',
         position: 'relative',
+        backgroundColor: 'white',
         ...style,
       }}
     >