change return signature of renderUI

WorldQL · Nov 9, 2023 · d961a15 · d961a15
1 parent 77c1c5d
commit d961a15
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/react/render.tsx b/src/react/render.tsx
@@ -15,15 +15,17 @@ export const renderUI = (
   game: Game<false>,
   ui: ReactNode,
   container?: HTMLDivElement,
-): (() => void) => {
+): { unmount(): void } => {
   const div = container ?? document.createElement('div')
   game.client.ui.add(div)
 
   const root = createRoot(div)
   root.render(<GameContext.Provider value={game}>{ui}</GameContext.Provider>)
 
-  return () => {
-    game.client.ui.remove(div)
-    root.unmount()
+  return {
+    unmount: () => {
+      game.client.ui.remove(div)
+      root.unmount()
+    },
   }
 }