fix(global): respect custom VP_HOME (#2029)

At present, Vite+ allows users to specific `VP_HOME` as an installation
directory. However, for any future `vp` runs without manual `VP_HOME`
environment, we still use default `~/.vite-plus` or `vp`'s grandparent
directory if its name is `.vite-plus` (It's related to #1185 by me).

It will cause mess, like multiple Vite+ instances, vp binary not
finding, packages unusable and a lot of unexpected behavior.

This PR improves all related logic, include:

- Set `VP_HOME` in env files
- Improve prompt and implode check related to `VP_HOME`
- Respect `VP_HOME` in `pnpm boostrap-cli` at devtime.

The new `install.sh` prompt looks like that:
<img width="977" height="270" alt="image"
src="https://github.com/user-attachments/assets/c96435d1-4b49-4e51-bb75-3da06205f23e"
/>


🤖 Generated by Codex

---------

Co-authored-by: MK (fengmk2) <fengmk2@gmail.com>

Author: liangmiQwQ

Cargo.lock

@@ -526,6 +526,7 @@ pub(crate) async fn cleanup_legacy_windows_shim(bin_dir: &vite_path::AbsolutePat
 // Includes shell completion support
 const ENV_TEMPLATE_POSIX: &str = r#"#!/bin/sh
 # Vite+ environment setup (https://viteplus.dev)
+export VP_HOME="__VP_HOME__"
 __vp_bin="__VP_BIN__"
 case ":${PATH}:" in
     *":${__vp_bin}:"*)
@@ -573,6 +574,7 @@ fi
 "#;
 
 const ENV_TEMPLATE_FISH: &str = r#"# Vite+ environment setup (https://viteplus.dev)
+set -gx VP_HOME "__VP_HOME__"
 set -l __vp_idx (contains -i -- __VP_BIN__ $PATH)
 and set -e PATH[$__vp_idx]
 set -gx PATH __VP_BIN__ $PATH
@@ -608,6 +610,7 @@ complete -c vpr --keep-order --exclusive --arguments "(__vpr_complete)"
 // Completions delegate to Fish dynamically (VP_COMPLETE=fish) because clap_complete_nushell
 // generates multiple rest params (e.g. for `vp install`), which Nushell does not support.
 const ENV_TEMPLATE_NU: &str = r#"# Vite+ environment setup (https://viteplus.dev)
+$env.VP_HOME = ("__VP_HOME__" | path expand --no-symlink)
 $env.PATH = ($env.PATH | where { $in != "__VP_BIN__" } | prepend "__VP_BIN__")
 
 # Shell function wrapper: intercepts `vp env use` to parse its stdout,
@@ -668,6 +671,7 @@ export extern "vpr" [...args: string@"nu-complete vpr"]
 "#;
 
 const ENV_TEMPLATE_PS1: &str = r#"# Vite+ environment setup (https://viteplus.dev)
+$env:VP_HOME = "__VP_HOME_WIN__"
 $__vp_bin = "__VP_BIN_WIN__"
 if ($env:Path -split ';' -notcontains $__vp_bin) {
     $env:Path = "$__vp_bin;$env:Path"
@@ -728,37 +732,63 @@ Register-ArgumentCompleter -Native -CommandName vpr -ScriptBlock $__vpr_comp
 
 // cmd.exe wrapper for `vp env use` (cmd.exe cannot define shell functions).
 // Users run `vp-use 24` in cmd.exe instead of `vp env use 24`.
-const VP_USE_CMD_CONTENT: &str = "@echo off\r\nset VP_ENV_USE_EVAL_ENABLE=1\r\nfor /f \"delims=\" %%i in ('%~dp0..\\current\\bin\\vp.exe env use %*') do %%i\r\nset VP_ENV_USE_EVAL_ENABLE=\r\n";
-
-/// Render the env-file content for `shell` against `vite_plus_home`.
-fn render_env_content(shell: EnvShell, vite_plus_home: &vite_path::AbsolutePath) -> String {
-    let bin_path = vite_plus_home.join("bin");
+const VP_USE_CMD_CONTENT: &str = "@echo off\r\nset VP_ENV_USE_EVAL_ENABLE=1\r\nset VP_HOME=%~dp0..\r\nfor /f \"delims=\" %%i in ('%~dp0..\\current\\bin\\vp.exe env use %*') do %%i\r\nset VP_ENV_USE_EVAL_ENABLE=\r\n";
 
+fn render_home_relative_path(path: &std::path::Path, home_dir: Option<&std::path::Path>) -> String {
     // Use $HOME-relative path if install dir is under HOME (like rustup's ~/.cargo/env).
     // This makes the env file portable across sessions where HOME may differ.
-    let home_dir = vite_shared::EnvConfig::get().user_home;
-    let bin_path_ref = home_dir
-        .as_ref()
-        .and_then(|h| bin_path.as_path().strip_prefix(h).ok())
+    home_dir
+        .and_then(|h| path.strip_prefix(h).ok())
         .map(|s| {
-            // Normalize to forward slashes for $HOME/... paths (POSIX-style)
-            format!("$HOME/{}", s.display().to_string().replace('\\', "/"))
+            if s.as_os_str().is_empty() {
+                "$HOME".to_string()
+            } else {
+                // Normalize to forward slashes for $HOME/... paths (POSIX-style)
+                format!("$HOME/{}", s.display().to_string().replace('\\', "/"))
+            }
         })
-        .unwrap_or_else(|| bin_path.as_path().display().to_string().replace('\\', "/"));
+        .unwrap_or_else(|| path.display().to_string().replace('\\', "/"))
+}
+
+fn render_nu_path_ref(path_ref: &str) -> String {
+    match path_ref.strip_prefix("$HOME") {
+        Some("") => "~".to_string(),
+        Some(suffix) if suffix.starts_with('/') => format!("~{suffix}"),
+        _ => path_ref.to_string(),
+    }
+}
+
+/// Render the env-file content for `shell` against `vite_plus_home`.
+fn render_env_content(shell: EnvShell, vite_plus_home: &vite_path::AbsolutePath) -> String {
+    let bin_path = vite_plus_home.join("bin");
+    let home_dir = vite_shared::EnvConfig::get().user_home;
+    let home_dir = home_dir.as_deref();
+    let home_path_ref = render_home_relative_path(vite_plus_home.as_path(), home_dir);
+    let bin_path_ref = render_home_relative_path(bin_path.as_path(), home_dir);
 
     match shell {
-        EnvShell::Posix => ENV_TEMPLATE_POSIX.replace("__VP_BIN__", &bin_path_ref),
-        EnvShell::Fish => ENV_TEMPLATE_FISH.replace("__VP_BIN__", &bin_path_ref),
+        EnvShell::Posix => ENV_TEMPLATE_POSIX
+            .replace("__VP_HOME__", &home_path_ref)
+            .replace("__VP_BIN__", &bin_path_ref),
+        EnvShell::Fish => ENV_TEMPLATE_FISH
+            .replace("__VP_HOME__", &home_path_ref)
+            .replace("__VP_BIN__", &bin_path_ref),
         EnvShell::Nu => {
             // Nushell requires `~` instead of `$HOME` in string literals — `$HOME` is not
             // expanded at parse time, so PATH entries would contain a literal "$HOME/...".
-            let bin_path_ref_nu = bin_path_ref.replace("$HOME/", "~/");
-            ENV_TEMPLATE_NU.replace("__VP_BIN__", &bin_path_ref_nu)
+            let home_path_ref_nu = render_nu_path_ref(&home_path_ref);
+            let bin_path_ref_nu = render_nu_path_ref(&bin_path_ref);
+            ENV_TEMPLATE_NU
+                .replace("__VP_HOME__", &home_path_ref_nu)
+                .replace("__VP_BIN__", &bin_path_ref_nu)
         }
         EnvShell::Powershell => {
             // PowerShell uses the actual absolute path (not $HOME-relative)
+            let home_path_win = vite_plus_home.as_path().display().to_string();
             let bin_path_win = bin_path.as_path().display().to_string();
-            ENV_TEMPLATE_PS1.replace("__VP_BIN_WIN__", &bin_path_win)
+            ENV_TEMPLATE_PS1
+                .replace("__VP_HOME_WIN__", &home_path_win)
+                .replace("__VP_BIN_WIN__", &bin_path_win)
         }
     }
 }
@@ -919,13 +949,16 @@ mod tests {
     #[tokio::test]
     async fn test_create_env_files_replaces_placeholder_with_home_relative_path() {
         let temp_dir = TempDir::new().unwrap();
-        let home = AbsolutePathBuf::new(temp_dir.path().to_path_buf()).unwrap();
+        let home = AbsolutePathBuf::new(temp_dir.path().join("vp_home")).unwrap();
         let _guard = home_guard(temp_dir.path());
+        tokio::fs::create_dir_all(&home).await.unwrap();
 
         create_env_files(&home).await.unwrap();
 
         let env_content = tokio::fs::read_to_string(home.join("env")).await.unwrap();
         let fish_content = tokio::fs::read_to_string(home.join("env.fish")).await.unwrap();
+        let nu_content = tokio::fs::read_to_string(home.join("env.nu")).await.unwrap();
+        let ps1_content = tokio::fs::read_to_string(home.join("env.ps1")).await.unwrap();
 
         // Placeholder should be fully replaced
         assert!(
@@ -936,15 +969,45 @@ mod tests {
             !fish_content.contains("__VP_BIN__"),
             "env.fish file should not contain __VP_BIN__ placeholder"
         );
+        assert!(
+            !env_content.contains("__VP_HOME__") && !fish_content.contains("__VP_HOME__"),
+            "env files should not contain __VP_HOME__ placeholder"
+        );
+        assert!(
+            !nu_content.contains("__VP_HOME__") && !ps1_content.contains("__VP_HOME_WIN__"),
+            "env files should not contain VP_HOME placeholders"
+        );
 
         // Should use $HOME-relative path since install dir is under HOME
         assert!(
-            env_content.contains("$HOME/bin"),
-            "env file should reference $HOME/bin, got: {env_content}"
+            env_content.contains("$HOME/vp_home/bin"),
+            "env file should reference $HOME/vp_home/bin, got: {env_content}"
+        );
+        assert!(
+            fish_content.contains("$HOME/vp_home/bin"),
+            "env.fish file should reference $HOME/vp_home/bin, got: {fish_content}"
+        );
+        assert!(
+            env_content.contains("export VP_HOME=\"$HOME/vp_home\""),
+            "env file should export VP_HOME, got: {env_content}"
         );
         assert!(
-            fish_content.contains("$HOME/bin"),
-            "env.fish file should reference $HOME/bin, got: {fish_content}"
+            fish_content.contains("set -gx VP_HOME \"$HOME/vp_home\""),
+            "env.fish file should export VP_HOME, got: {fish_content}"
+        );
+        assert!(
+            nu_content.contains("$env.VP_HOME = (\"~/vp_home\" | path expand --no-symlink)"),
+            "env.nu file should set home-relative VP_HOME, got: {nu_content}"
+        );
+        assert!(
+            nu_content.contains("~/vp_home/bin"),
+            "env.nu file should reference ~/vp_home/bin, got: {nu_content}"
+        );
+
+        let expected_home = home.as_path().display().to_string();
+        assert!(
+            ps1_content.contains(&format!("$env:VP_HOME = \"{expected_home}\"")),
+            "env.ps1 file should set VP_HOME, got: {ps1_content}"
         );
     }
 
@@ -963,6 +1026,7 @@ mod tests {
         // Should use absolute path since install dir is not under HOME
         let expected_bin = home.join("bin");
         let expected_str = expected_bin.as_path().display().to_string().replace('\\', "/");
+        let expected_home = home.as_path().display().to_string().replace('\\', "/");
         assert!(
             env_content.contains(&expected_str),
             "env file should use absolute path {expected_str}, got: {env_content}"
@@ -971,6 +1035,14 @@ mod tests {
             fish_content.contains(&expected_str),
             "env.fish file should use absolute path {expected_str}, got: {fish_content}"
         );
+        assert!(
+            env_content.contains(&format!("export VP_HOME=\"{expected_home}\"")),
+            "env file should export absolute VP_HOME {expected_home}, got: {env_content}"
+        );
+        assert!(
+            fish_content.contains(&format!("set -gx VP_HOME \"{expected_home}\"")),
+            "env.fish file should export absolute VP_HOME {expected_home}, got: {fish_content}"
+        );
 
         // Should NOT use $HOME-relative path
         assert!(!env_content.contains("$HOME/bin"), "env file should not reference $HOME/bin");
@@ -1120,6 +1192,27 @@ mod tests {
         );
     }
 
+    #[tokio::test]
+    async fn test_create_env_files_cmd_wrapper_sets_vp_home_before_env_use() {
+        let temp_dir = TempDir::new().unwrap();
+        let home = AbsolutePathBuf::new(temp_dir.path().to_path_buf()).unwrap();
+        let _guard = home_guard(temp_dir.path());
+        let bin_dir = home.join("bin");
+        tokio::fs::create_dir_all(&bin_dir).await.unwrap();
+
+        create_env_files(&home).await.unwrap();
+
+        let cmd_content = tokio::fs::read_to_string(bin_dir.join("vp-use.cmd")).await.unwrap();
+        assert!(
+            cmd_content.contains("set VP_HOME=%~dp0..\r\nfor /f"),
+            "vp-use.cmd should set VP_HOME before invoking vp env use, got: {cmd_content}"
+        );
+        assert!(
+            cmd_content.contains("%~dp0..\\current\\bin\\vp.exe env use %*"),
+            "vp-use.cmd should invoke the install-local vp.exe"
+        );
+    }
+
     #[tokio::test]
     async fn test_execute_env_only_creates_home_dir_and_env_files() {
         let temp_dir = TempDir::new().unwrap();