Attempt 5

Author: Archmonger

src/js/packages/@reactpy/client/src/bind.tsx

@@ -30,31 +30,30 @@ function local_preact_bind(node: HTMLElement) {
   };
 }
 
+const roots = new WeakMap<HTMLElement, any>();
+
 function reactjs_bind(node: HTMLElement, React: any, ReactDOM: any) {
   let root: any = null;
   return {
     create: (type: any, props: any, children?: any[]) =>
       React.createElement(type, props, ...(children || [])),
     render: (element: any) => {
       if (!root) {
-        // @ts-ignore
-        if (node._reactPyRoot) {
-          // @ts-ignore
-          node._reactPyRoot.unmount();
+        const existingRoot = roots.get(node);
+        if (existingRoot) {
+          existingRoot.unmount();
+          roots.delete(node);
         }
         root = ReactDOM.createRoot(node);
-        // @ts-ignore
-        node._reactPyRoot = root;
+        roots.set(node, root);
       }
       root.render(element);
     },
     unmount: () => {
       if (root) {
         root.unmount();
-        // @ts-ignore
-        if (node._reactPyRoot === root) {
-          // @ts-ignore
-          node._reactPyRoot = null;
+        if (roots.get(node) === root) {
+          roots.delete(node);
         }
         root = null;
       }