chore: refactor

unocss · Jan 12, 2023 · a48f743 · a48f743
1 parent c706caa
commit a48f743
Showing 1 changed file with 17 additions and 14 deletions.
diff --git a/packages/vite/src/modes/global/dev.ts b/packages/vite/src/modes/global/dev.ts
@@ -168,20 +168,23 @@ export function GlobalModeDevPlugin({ uno, tokens, tasks, flushTasks, affectedMo
 
         // inject css modules to send callback on css load
         if (layer && code.includes('import.meta.hot')) {
-          const immediateFunction = (code: string) => `;(async function() {\n${code}\n})()`
-          const importMetaHot = `if (import.meta.hot) {
-  try { await import.meta.hot.send('${WS_EVENT_PREFIX}', ['${layer}', __vite__css.slice(2,${2 + HASH_LENGTH})]); }
-  catch (e) { console.warn('[unocss-hmr]', e) }
-  if (!import.meta.url.includes('?'))
-    await new Promise(resolve => setTimeout(resolve, 100))
-}`
-          const { hmrTopLevelAwait } = await getConfig() as VitePluginConfig
-
-          return `${code}\n${
-            hmrTopLevelAwait !== false
-              ? importMetaHot
-              : immediateFunction(importMetaHot)
-          }`
+          let hmr = `
+try {
+  await import.meta.hot.send('${WS_EVENT_PREFIX}', ['${layer}', __vite__css.slice(2,${2 + HASH_LENGTH})]);
+} catch (e) {
+  console.warn('[unocss-hmr]', e)
+}
+if (!import.meta.url.includes('?'))
+  await new Promise(resolve => setTimeout(resolve, 100))`
+
+          const config = await getConfig() as VitePluginConfig
+
+          if (config.hmrTopLevelAwait === false)
+            hmr = `;(async function() {${hmr}\n})()`
+
+          hmr = `\nif (import.meta.hot) {${hmr}}`
+
+          return code + hmr
         }
       },
     },