update main.tsx

Author: abvthecity

custom-app/src/main.tsx

@@ -1,39 +1,14 @@
 import './main.css'
 
-import ReactDOM from 'react-dom'
 import { NavFooter, NavHeader } from './NavHeader'
-import React from 'react'
+import { onDOMContentMutate, renderInContainer } from './utils'
 
 function render() {
-  ReactDOM.render(
-    React.createElement(NavHeader),
-    document.getElementById('fern-header'),
-  )
-  ReactDOM.render(
-    React.createElement(NavFooter),
-    document.getElementById('fern-footer'),
-  )
+  // render custom header
+  renderInContainer(NavHeader, document.getElementById('fern-header'))
+
+  // render custom footer
+  renderInContainer(NavFooter, document.getElementById('fern-footer'))
 }
 
-let observations = 0
-document.addEventListener('DOMContentLoaded', () => {
-  console.log('DOMContentLoaded')
-  render()
-  new MutationObserver((e, o) => {
-    render()
-    for (const item of e) {
-      if (item.target instanceof HTMLElement) {
-        const target = item.target
-        if (target.id === 'fern-header' || target.id === 'fern-footer') {
-          if (observations < 3) {
-            // react hydration will trigger a mutation event
-            observations++
-          } else {
-            o.disconnect()
-          }
-          break
-        }
-      }
-    }
-  }).observe(document.body, { childList: true, subtree: true })
-})
+onDOMContentMutate(render)

custom-app/src/utils.ts

@@ -0,0 +1,49 @@
+import { createRoot, type Root } from 'react-dom/client'
+import React, { FunctionComponent } from 'react'
+
+// avoid creating multiple roots for the same element
+const roots = new WeakMap<HTMLElement, Root>()
+function getOrCreateRoot(element: HTMLElement): Root {
+  let root = roots.get(element)
+  if (root == null) {
+    root = createRoot(element, {
+      identifierPrefix: 'custom-app',
+      onRecoverableError: console.error,
+    })
+    roots.set(element, root)
+  }
+  return root
+}
+
+export function renderInContainer(Component: FunctionComponent, container: HTMLElement | null | undefined) {
+  if (container != null) {
+    const root = getOrCreateRoot(container)
+    root.render(React.createElement(Component))
+  }
+}
+
+export function onDOMContentMutate(render: () => void) {
+    // observe DOM changes to re-render the custom header and footer
+    let observations = 0
+    document.addEventListener('DOMContentLoaded', () => {
+        console.log('DOMContentLoaded')
+        render()
+        new MutationObserver((e, o) => {
+            render()
+            for (const item of e) {
+            if (item.target instanceof HTMLElement) {
+                const target = item.target
+                if (target.id === 'fern-header' || target.id === 'fern-footer') {
+                if (observations < 3) {
+                    // react hydration will trigger a mutation event
+                    observations++
+                } else {
+                    o.disconnect()
+                }
+                break
+                }
+            }
+            }
+        }).observe(document.body, { childList: true, subtree: true })
+    })
+}