From bef2d6e2eccb58976f45a5e1d093d471a055dddb Mon Sep 17 00:00:00 2001
From: Vitalij Ryndin <crashmax.dev@gmail.com>
Date: Tue, 2 Jul 2024 16:49:01 +0800
Subject: [PATCH] fix: unmount `createRoot` for demo component

---
 docs/src/components/demo.vue | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/docs/src/components/demo.vue b/docs/src/components/demo.vue
index 93624749..35db6e87 100644
--- a/docs/src/components/demo.vue
+++ b/docs/src/components/demo.vue
@@ -1,7 +1,7 @@
 <script setup>
 import { createElement } from 'react';
 import { createRoot } from 'react-dom/client';
-import { ref, computed, onMounted, defineAsyncComponent } from 'vue';
+import { shallowRef, computed, onMounted, onUnmounted, defineAsyncComponent } from 'vue';
 
 const props = defineProps({
   hook: {
@@ -13,12 +13,18 @@ const sourceLink = computed(() => {
   return `https://github.com/siberiacancode/reactuse/blob/main/src/hooks/${props.hook}/${props.hook}.demo.tsx`;
 });
 
-const demoRef = ref();
+const demoRef = shallowRef();
+const demoRoot = shallowRef();
 
 onMounted(async () => {
   const demoComponent = await import(`../../../src/hooks/${props.hook}/${props.hook}.demo.tsx`);
-  const root = createRoot(demoRef.value);
-  root.render(createElement(demoComponent.default, {}, null));
+  demoRoot.value = createRoot(demoRef.value);
+  demoRoot.value.render(createElement(demoComponent.default, {}, null));
+});
+
+onUnmounted(() => {
+  if (!demoRoot.value) return;
+  demoRoot.value.unmount();
 });
 </script>