Make switch npm registry and authentication configurable

packages/zpm-switch/Cargo.toml

@@ -8,6 +8,7 @@ name = "yarn"
 path = "src/main.rs"
 
 [dependencies]
+base64 = { workspace = true }
 clipanion = { workspace = true, features = ["serde"] }
 regex = { workspace = true }
 reqwest = { workspace = true, default-features = false, features = ["hickory-dns", "rustls-tls"] }

packages/zpm-switch/src/http.rs

@@ -17,6 +17,45 @@ pub fn http_client() -> Result<Client, Error> {
     HTTP_CLIENT.clone()
 }
 
+fn get_npm_registry_server() -> String {
+    std::env::var("YARNSW_NPM_REGISTRY_SERVER")
+        .unwrap_or_else(|_| "https://registry.npmjs.org".to_string())
+}
+
+fn is_npm_registry_request(url: &str) -> bool {
+    let registry = get_npm_registry_server();
+    let normalized_registry
+        = registry.trim_end_matches('/');
+    let normalized_url
+        = url.trim_end_matches('/');
+
+    normalized_url == normalized_registry
+        || normalized_url.starts_with(&format!("{normalized_registry}/"))
+}
+
+fn attach_auth_headers(request: &mut reqwest::RequestBuilder) {
+    if let Ok(token) = std::env::var("YARN_SWITCH_NPM_AUTH_TOKEN") {
+        if let Some(cloned_request) = request.try_clone() {
+            *request = cloned_request.bearer_auth(token);
+        }
+
+        return;
+    }
+
+    if let Ok(mut auth_ident) = std::env::var("YARN_SWITCH_NPM_AUTH_IDENT") {
+        if auth_ident.contains(':') {
+            auth_ident = base64::Engine::encode(&base64::engine::general_purpose::STANDARD, auth_ident);
+        }
+
+        if let Some(cloned_request) = request.try_clone() {
+            *request = cloned_request.header(
+                reqwest::header::AUTHORIZATION,
+                format!("Basic {auth_ident}")
+            );
+        }
+    }
+}
+
 pub async fn fetch(url: &str) -> Result<Vec<u8>, Error> {
     let client
         = http_client()?;
@@ -28,11 +67,17 @@ pub async fn fetch(url: &str) -> Result<Vec<u8>, Error> {
                 |provider| serde_plain::to_string(&provider).unwrap()
             );
 
-    let request
+    let mut request
         = client.get(url)
             .header("User-Agent", "zpm-switch")
-            .header("X-Switch-CI", is_ci_header)
-            .send().await?;
+            .header("X-Switch-CI", is_ci_header);
+
+    if is_npm_registry_request(url) {
+        attach_auth_headers(&mut request);
+    }
+
+    let request
+        = request.send().await?;
 
     let status
         = request.status();