fix: use custom registry if exist

Author: daflyinbed

crates/vite_setup/src/install.rs

@@ -284,7 +284,7 @@ pub async fn install_production_deps(
             format!("pnpm entry not found at {}", pnpm_entry.as_path().display()).into(),
         ));
     }
-    let output = run_pnpm_install(version_dir, &node_runtime, &pnpm_entry, &args).await?;
+    let output = run_pnpm_install(version_dir, &node_runtime, &pnpm_entry, &args, registry).await?;
 
     if !output.status.success() {
         let log_path = write_upgrade_log(version_dir, &output.stdout, &output.stderr).await;
@@ -316,7 +316,7 @@ pub async fn install_production_deps(
         // Only create the local override after explicit consent. This preserves
         // minimumReleaseAge protection for the default and non-interactive paths.
         write_release_age_overrides(version_dir).await?;
-        let retry_output = run_pnpm_install(version_dir, &node_runtime, &pnpm_entry, &args).await?;
+        let retry_output = run_pnpm_install(version_dir, &node_runtime, &pnpm_entry, &args, registry).await?;
         if !retry_output.status.success() {
             let retry_log_path =
                 write_upgrade_log(version_dir, &retry_output.stdout, &retry_output.stderr).await;
@@ -339,6 +339,7 @@ async fn run_pnpm_install(
     node_runtime: &vite_js_runtime::JsRuntime,
     pnpm_entry: &AbsolutePath,
     args: &[&str],
+    registry: Option<&str>,
 ) -> Result<Output, Error> {
     let node_bin = node_runtime.get_bin_prefix();
     let pnpm_bin = pnpm_entry.parent().ok_or_else(|| {
@@ -350,15 +351,18 @@ async fn run_pnpm_install(
     let path = env::join_paths(path_entries)
         .map_err(|error| Error::Setup(format!("Failed to build PATH for pnpm: {error}").into()))?;
 
-    let output = tokio::process::Command::new(node_runtime.get_binary_path().as_path())
-        .arg(pnpm_entry.as_path())
+    let mut cmd = tokio::process::Command::new(node_runtime.get_binary_path().as_path());
+    cmd.arg(pnpm_entry.as_path())
         .args(args)
         .current_dir(version_dir)
         .env("CI", "true")
-        .env("PATH", path)
-        .output()
-        .await?;
+        .env("PATH", path);
+
+    if let Some(registry_url) = registry {
+        cmd.env("npm_config_registry", registry_url);
+    }
 
+    let output = cmd.output().await?;
     Ok(output)
 }