chore: refactor runtime code

Author: baiwusanyu-c

packages/core/index.ts

@@ -130,6 +130,14 @@ const unplugin = createUnplugin<Options>(
             this.error(`[${NAME}] ${err}`)
           }
         },
+        buildEnd() {
+          if (context.isServer) {
+            if (context.framework === 'vite'
+                || context.framework === 'rollup'
+                || context.framework === 'esbuild')
+              context.isHMR = false
+          }
+        },
       },
     ]
   })

packages/core/runtime/handle-inject-css.ts

@@ -3,6 +3,7 @@ import { injectCSSVars } from '../inject'
 import type { MagicStringBase } from 'magic-string-ast'
 import type { IVueCSSVarsCtx } from '../types'
 
+// TODO: unit test
 export function handleInjectCss(
   id: string,
   code: string,
@@ -18,6 +19,5 @@ export function handleInjectCss(
   )
   mgcStr = injectRes.mgcStr
   injectRes.vbindVariableList && ctx.vbindVariableList.set(id, injectRes.vbindVariableList)
-  // TODO vite hmr close ? isHMR -> false
   return mgcStr
 }

packages/core/runtime/handle-variable.ts

@@ -4,6 +4,7 @@ import { getVariable, matchVariable } from '../parser'
 import { getVBindVariableListByPath } from './process-css'
 import type { IVueCSSVarsCtx } from '../types'
 
+// TODO: unit test
 export function handleVBindVariable(
   code: string,
   id: string,

packages/core/runtime/vite.ts

@@ -7,6 +7,7 @@ import type { IVueCSSVarsCtx } from '../types'
 import type { MagicStringBase } from 'magic-string-ast'
 import type { HmrContext, ResolvedConfig } from 'vite'
 
+// TODO: unit test
 export function transformPreVite(
   id: string,
   code: string,
@@ -30,6 +31,7 @@ export function transformPreVite(
   return mgcStr
 }
 
+// TODO: unit test
 export function transformPostVite(
   id: string,
   code: string,
@@ -50,6 +52,7 @@ export function transformPostVite(
   return mgcStr
 }
 
+// TODO: unit test
 export const vitePlugin = (ctx: IVueCSSVarsCtx) => {
   return {
     // Vite plugin

packages/core/runtime/webpack.ts

@@ -6,6 +6,7 @@ import { handleInjectCss } from './handle-inject-css'
 import type { IVueCSSVarsCtx } from '../types'
 import type { MagicStringBase } from 'magic-string-ast'
 
+// TODO: unit test
 export function transformPreWebpack(
   id: string,
   code: string,
@@ -21,6 +22,7 @@ export function transformPreWebpack(
   }
 }
 
+// TODO: unit test
 export function transformPostWebpack(
   id: string,
   code: string,
@@ -45,6 +47,7 @@ export function transformPostWebpack(
   return mgcStr
 }
 
+// TODO: unit test
 export const webpackPlugin = (ctx: IVueCSSVarsCtx, compiler: any) => {
   // mark webpack hmr
   let modifiedFile = ''