feat(vite): add VITE_* env caching for build and dev commands (#544)

branchseer · web-flow · commit aa6dddbe54d1 · 2026-02-05T21:18:07.000+08:00
Pass and fingerprint `VITE_*` environment variables to `build`/`dev`
diff --git a/packages/cli/binding/src/cli.rs b/packages/cli/binding/src/cli.rs
@@ -403,7 +403,16 @@ impl TaskSynthesizer<CustomTaskSubcommand> for VitePlusTaskSynthesizer {
                         .chain(iter::once(Str::from("build")))
                         .chain(args.into_iter().map(Str::from))
                         .collect(),
-                    task_options: Default::default(),
+                    task_options: UserTaskOptions {
+                        cache_config: UserCacheConfig::Enabled {
+                            cache: None,
+                            enabled_cache_config: EnabledCacheConfig {
+                                envs: Some(Box::new([Str::from("VITE_*")])),
+                                pass_through_envs: None,
+                            },
+                        },
+                        ..Default::default()
+                    },
                     direct_execution_cache_key,
                     envs: merge_resolved_envs(envs, resolved.envs),
                 })
@@ -479,7 +488,16 @@ impl TaskSynthesizer<CustomTaskSubcommand> for VitePlusTaskSynthesizer {
                         .chain(iter::once(Str::from("dev")))
                         .chain(args.into_iter().map(Str::from))
                         .collect(),
-                    task_options: Default::default(),
+                    task_options: UserTaskOptions {
+                        cache_config: UserCacheConfig::Enabled {
+                            cache: None,
+                            enabled_cache_config: EnabledCacheConfig {
+                                envs: Some(Box::new([Str::from("VITE_*")])),
+                                pass_through_envs: None,
+                            },
+                        },
+                        ..Default::default()
+                    },
                     direct_execution_cache_key,
                     envs: merge_resolved_envs(envs, resolved.envs),
                 })
diff --git a/packages/cli/snap-tests/build-vite-env/index.html b/packages/cli/snap-tests/build-vite-env/index.html
@@ -0,0 +1,8 @@
+<!doctype html>
+<html>
+  <body>
+    <script type="module">
+      console.log('hello');
+    </script>
+  </body>
+</html>
diff --git a/packages/cli/snap-tests/build-vite-env/package.json b/packages/cli/snap-tests/build-vite-env/package.json
@@ -0,0 +1,4 @@
+{
+  "name": "build-vite-env-test",
+  "private": true
+}
diff --git a/packages/cli/snap-tests/build-vite-env/snap.txt b/packages/cli/snap-tests/build-vite-env/snap.txt
@@ -0,0 +1,33 @@
+> VITE_MY_VAR=1 vite build
+vite v<semver>
+[2Ktransforming...✓ <variable> modules transformed.
+rendering chunks...
+computing gzip size...
+dist/index.html                <variable> kB │ gzip: <variable> kB
+dist/assets/index-BnIqjoTZ.js  <variable> kB │ gzip: <variable> kB
+
+✓ built in <variable>ms
+
+
+> VITE_MY_VAR=1 vite build # should hit cache
+vite v<semver>
+[2Ktransforming...✓ <variable> modules transformed.
+rendering chunks...
+computing gzip size...
+dist/index.html                <variable> kB │ gzip: <variable> kB
+dist/assets/index-BnIqjoTZ.js  <variable> kB │ gzip: <variable> kB
+
+✓ built in <variable>ms
+✓ cache hit, logs replayed
+
+
+> VITE_MY_VAR=2 vite build # env changed, should miss cache
+vite v<semver>
+[2Ktransforming...✓ <variable> modules transformed.
+rendering chunks...
+computing gzip size...
+dist/index.html                <variable> kB │ gzip: <variable> kB
+dist/assets/index-BnIqjoTZ.js  <variable> kB │ gzip: <variable> kB
+
+✓ built in <variable>ms
+
diff --git a/packages/cli/snap-tests/build-vite-env/steps.json b/packages/cli/snap-tests/build-vite-env/steps.json
@@ -0,0 +1,11 @@
+{
+  "ignoredPlatforms": ["win32"],
+  "env": {
+    "VITE_DISABLE_AUTO_INSTALL": "1"
+  },
+  "commands": [
+    "VITE_MY_VAR=1 vite build",
+    "VITE_MY_VAR=1 vite build # should hit cache",
+    "VITE_MY_VAR=2 vite build # env changed, should miss cache"
+  ]
+}
diff --git a/packages/tools/src/utils.ts b/packages/tools/src/utils.ts
@@ -82,6 +82,8 @@ export function replaceUnstableOutput(output: string, cwd?: string) {
         '$1$2\n',
       )
       .replaceAll(/\nfound \d+ vulnerabilities\n/g, '')
+      // vite modules transformed count
+      .replaceAll(/✓ \d+ modules? transformed/g, '✓ <variable> modules transformed')
       // replace size for tsdown
       .replaceAll(/ \d+(\.\d+)? ([km]?B)/g, ' <variable> $2')
       // replace npm notice size:

-Original file line number
+Diff line change
@@ @@ -0,0 +1,4 @@ @@
 +{
 +  "name": "build-vite-env-test",
 +  "private": true
 +}
Original file line number	Diff line number	Diff line change
`@@ -82,6 +82,8 @@ export function replaceUnstableOutput(output: string, cwd?: string) {`
`82`	`82`	`'$1$2\n',`
`83`	`83`	`)`
`84`	`84`	`.replaceAll(/\nfound \d+ vulnerabilities\n/g, '')`
	`85`	`+ // vite modules transformed count`
	`86`	`+ .replaceAll(/✓ \d+ modules? transformed/g, '✓ <variable> modules transformed')`
`85`	`87`	`// replace size for tsdown`
`86`	`88`	`.replaceAll(/ \d+(\.\d+)? ([km]?B)/g, ' <variable> $2')`
`87`	`89`	`// replace npm notice size:`