DEV :: same styles

src/app/share/[...shareId]/SharePage.tsx

@@ -1,7 +1,6 @@
 'use client';
 import WebsiteDetails from 'app/(main)/websites/[websiteId]/WebsiteDetails';
 import { useShareToken } from 'components/hooks';
-import { useEffect } from 'react';
 import Page from 'components/layout/Page';
 import Header from './Header';
 import styles from './SharePage.module.css';
@@ -12,26 +11,6 @@ export default function SharePage({ shareId }) {
   if (isLoading || !shareToken) {
     return null;
   }
-  useEffect(() => {
-    function sendHeightToParent() {
-      const height = document.body.scrollHeight;
-      window.parent.postMessage({ height: height }, '*');
-    }
-
-    sendHeightToParent();
-
-    // Add event listener for resize
-    window.addEventListener('resize', () => {
-      sendHeightToParent();
-    });
-
-    // Cleanup function to remove event listener
-    return () => {
-      window.removeEventListener('resize', () => {
-        sendHeightToParent();
-      });
-    };
-  }, [])
   return (
     <div className={styles.container}>
       <Page>

src/app/share/[...shareId]/page.tsx

@@ -1,5 +1,26 @@
+import { useEffect } from 'react';
 import SharePage from './SharePage';
 
 export default function ({ params: { shareId } }) {
+  useEffect(() => {
+    function sendHeightToParent() {
+      const height = document.body.scrollHeight;
+      window.parent.postMessage({ height: height }, '*');
+    }
+
+    sendHeightToParent();
+    
+    // Add event listener for resize
+    window.addEventListener('resize', () => {
+      sendHeightToParent();
+    });
+
+    // Cleanup function to remove event listener
+    return () => {
+      window.removeEventListener('resize', () => {
+        sendHeightToParent();
+      });
+    };
+  }, [])
   return <SharePage shareId={shareId[0]} />;
 }