+

lifeart · Jan 15, 2024 · c4870bd · c4870bd
1 parent 991b651
commit c4870bd
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/src/utils/component.ts b/src/utils/component.ts
@@ -32,8 +32,10 @@ export type GenericReturnType =
   | null[];
 
 // this is workaround for `if` case, where we don't have stable root, and to remove it properly we need to look into last rendered part
-export const relatedRoots: WeakMap<DocumentFragment, GenericReturnType> =
-  new WeakMap();
+export const relatedRoots: WeakMap<
+  DocumentFragment | HTMLElement,
+  GenericReturnType
+> = new WeakMap();
 
 function renderNode(parent: Node, target: Node, placeholder: Node | Comment) {
   if (import.meta.env.DEV) {

diff --git a/src/utils/ssr.ts b/src/utils/ssr.ts
@@ -1,4 +1,8 @@
-import { renderComponent, runDestructors, type ComponentReturnType } from '@/utils/component';
+import {
+  renderComponent,
+  runDestructors,
+  type ComponentReturnType,
+} from '@/utils/component';
 import { setDocument, getDocument } from './dom-api';
 import { getRoot, resetNodeCounter, resetRoot } from '@/utils/dom';