fix: add RefreshSig to refresh content regex (closes #52)

vitejs · Jan 25, 2023 · c8dd1d6 · c8dd1d6
1 parent 48017b7
commit c8dd1d6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/packages/plugin-react/src/index.ts b/packages/plugin-react/src/index.ts
@@ -88,6 +88,7 @@ declare module 'vite' {
 }
 
 const prependReactImportCode = "import React from 'react'; "
+const refreshContentRE = /\$Refresh(?:Reg|Sig)\$\(/
 
 export default function viteReact(opts: Options = {}): PluginOption[] {
   // Provide default values for Rollup compat.
@@ -335,7 +336,7 @@ export default function viteReact(opts: Options = {}): PluginOption[] {
 
         if (result) {
           let code = result.code!
-          if (useFastRefresh && /\$RefreshReg\$\(/.test(code)) {
+          if (useFastRefresh && refreshContentRE.test(code)) {
             code = addRefreshWrapper(code, id)
           }
           return {