//! ref: composer/src/Composer/Util/Http/ProxyManager.php use std::sync::{Mutex, OnceLock}; use crate::downloader::transport_exception::TransportException; use crate::util::http::proxy_item::ProxyItem; use crate::util::http::request_proxy::RequestProxy; use crate::util::no_proxy_pattern::NoProxyPattern; static INSTANCE: OnceLock>> = OnceLock::new(); #[derive(Debug)] pub struct ProxyManager { error: Option, http_proxy: Option, https_proxy: Option, no_proxy_handler: Option, } impl ProxyManager { fn new() -> Self { let mut instance = Self { error: None, http_proxy: None, https_proxy: None, no_proxy_handler: None, }; if let Err(e) = instance.get_proxy_data() { instance.error = Some(e.to_string()); } instance } pub fn get_instance() -> &'static Mutex> { INSTANCE.get_or_init(|| Mutex::new(Some(ProxyManager::new()))) } pub fn reset() { if let Some(mutex) = INSTANCE.get() { *mutex.lock().unwrap() = Some(ProxyManager::new()); } } pub fn has_proxy(&self) -> bool { self.http_proxy.is_some() || self.https_proxy.is_some() } pub fn get_proxy_for_request( &self, request_url: &str, ) -> Result { if let Some(ref error) = self.error { return Err(TransportException::new(format!( "Unable to use a proxy: {}", error ))); } let scheme = request_url.split("://").next().unwrap_or("").to_string(); let proxy = self.get_proxy_for_scheme(&scheme); if proxy.is_none() { return Ok(RequestProxy::none()); } if self.no_proxy(request_url) { return Ok(RequestProxy::no_proxy()); } Ok(proxy.unwrap().to_request_proxy(&scheme)) } fn get_proxy_for_scheme(&self, scheme: &str) -> Option<&ProxyItem> { if scheme == "http" { return self.http_proxy.as_ref(); } if scheme == "https" { return self.https_proxy.as_ref(); } None } fn get_proxy_data(&mut self) -> anyhow::Result<()> { // handle http_proxy/HTTP_PROXY on CLI only for security reasons // PHP_SAPI is always 'cli' for this application let (env, name) = Self::get_proxy_env("http_proxy"); if let Some(env) = env { self.http_proxy = Some(ProxyItem::new(env, name)); } if self.http_proxy.is_none() { let (env, name) = Self::get_proxy_env("cgi_http_proxy"); if let Some(env) = env { self.http_proxy = Some(ProxyItem::new(env, name)); } } let (env, name) = Self::get_proxy_env("https_proxy"); if let Some(env) = env { self.https_proxy = Some(ProxyItem::new(env, name)); } let (env, _name) = Self::get_proxy_env("no_proxy"); if let Some(env) = env { self.no_proxy_handler = Some(NoProxyPattern::new(env)); } Ok(()) } fn get_proxy_env(env_name: &str) -> (Option, String) { for name in [env_name.to_lowercase(), env_name.to_uppercase()] { if let Ok(val) = std::env::var(&name) { if !val.is_empty() { return (Some(val), name); } } } (None, String::new()) } fn no_proxy(&self, request_url: &str) -> bool { match &self.no_proxy_handler { None => false, Some(handler) => handler.test(request_url), } } }