From a196f2489b6d2c7400b072d828cf6558070ddc66 Mon Sep 17 00:00:00 2001
From: bluwy <bjornlu.dev@gmail.com>
Date: Wed, 3 Nov 2021 12:02:30 +0800
Subject: [PATCH] feat(define): prevent assignment

---
 packages/playground/define/index.html     | 1 +
 packages/playground/define/vite.config.js | 1 +
 packages/vite/src/node/plugins/define.ts  | 3 ++-
 3 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/packages/playground/define/index.html b/packages/playground/define/index.html
index 590d92083c4c21..bf6a9c59689396 100644
--- a/packages/playground/define/index.html
+++ b/packages/playground/define/index.html
@@ -11,6 +11,7 @@ <h1>Define</h1>
 <p>spread array: <code class="spread-array"></code></p>
 
 <script type="module">
+  const __VAR_NAME__ = true // ensure define doesn't replace var name
   text('.exp', __EXP__)
   text('.string', __STRING__)
   text('.number', __NUMBER__)
diff --git a/packages/playground/define/vite.config.js b/packages/playground/define/vite.config.js
index 422f1c4daa9628..848abd09322df6 100644
--- a/packages/playground/define/vite.config.js
+++ b/packages/playground/define/vite.config.js
@@ -15,6 +15,7 @@ module.exports = {
         }
       }
     },
+    __VAR_NAME__: false,
     'process.env.SOMEVAR': '"SOMEVAR"'
   }
 }
diff --git a/packages/vite/src/node/plugins/define.ts b/packages/vite/src/node/plugins/define.ts
index e520a88890e293..ae1eded679cbfc 100644
--- a/packages/vite/src/node/plugins/define.ts
+++ b/packages/vite/src/node/plugins/define.ts
@@ -67,7 +67,8 @@ export function definePlugin(config: ResolvedConfig): Plugin {
             return str.replace(/[-[\]/{}()*+?.\\^$|]/g, '\\$&')
           })
           .join('|') +
-        ')\\b',
+        // prevent trailing assignments
+        ')\\b(?!\\s*?=[^=])',
       'g'
     )