From e3d29ed67dd79a5e0611f9ebaccf70c82914587f Mon Sep 17 00:00:00 2001
From: owjs3901 <owjs3901@naver.com>
Date: Tue, 28 Jan 2025 02:58:04 +0900
Subject: [PATCH] Fix hmr issue

---
 .changeset/stale-cherries-drum.md  |  5 +++++
 packages/vite-plugin/src/plugin.ts | 13 ++++++++++++-
 2 files changed, 17 insertions(+), 1 deletion(-)
 create mode 100644 .changeset/stale-cherries-drum.md

diff --git a/.changeset/stale-cherries-drum.md b/.changeset/stale-cherries-drum.md
new file mode 100644
index 00000000..8e9cc97b
--- /dev/null
+++ b/.changeset/stale-cherries-drum.md
@@ -0,0 +1,5 @@
+---
+"@devup-ui/vite-plugin": patch
+---
+
+Fix hmr issue
diff --git a/packages/vite-plugin/src/plugin.ts b/packages/vite-plugin/src/plugin.ts
index c132238b..91db8b8c 100644
--- a/packages/vite-plugin/src/plugin.ts
+++ b/packages/vite-plugin/src/plugin.ts
@@ -61,7 +61,7 @@ export function DevupUI({
   }
   return {
     name: 'devup-ui',
-    config: () => {
+    config() {
       return {
         server: {
           watch: {
@@ -98,6 +98,17 @@ export function DevupUI({
         writeFileSync(cssFile, css, {
           encoding: 'utf-8',
         })
+        return {
+          code: `${retCode}
+            const exists = !!document.getElementById('devup-ui');
+            const style = document.getElementById('devup-ui') || document.createElement('style');
+            style.id = 'devup-ui';
+            style.textContent = \`
+            ${css}
+            \`;
+            if (!exists) document.head.appendChild(style);
+          `,
+        }
       }
       return {
         code: retCode,