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

ArnaudBarre · ArnaudBarre · commit c8dd1d6c3613 · 2023-01-25T14:55:55.000+01:00
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 {

Original file line number	Diff line number	Diff line change
`@@ -88,6 +88,7 @@ declare module 'vite' {`
`88`	`88`	`}`
`89`	`89`
`90`	`90`	`const prependReactImportCode = "import React from 'react'; "`
	`91`	`+const refreshContentRE = /\$Refresh(?:Reg\|Sig)\$\(/`
`91`	`92`
`92`	`93`	`export default function viteReact(opts: Options = {}): PluginOption[] {`
`93`	`94`	`// Provide default values for Rollup compat.`
`@@ -335,7 +336,7 @@ export default function viteReact(opts: Options = {}): PluginOption[] {`
`335`	`336`
`336`	`337`	`if (result) {`
`337`	`338`	`let code = result.code!`
`338`		`- if (useFastRefresh && /\$RefreshReg\$\(/.test(code)) {`
	`339`	`+ if (useFastRefresh && refreshContentRE.test(code)) {`
`339`	`340`	`code = addRefreshWrapper(code, id)`
`340`	`341`	`}`
`341`	`342`	`return {`