use std::fmt::Debug; use std::path::{Path, PathBuf}; use bytes::Bytes; use cargo_toml::Manifest; use futures_util::stream::Stream; use log::debug; use once_cell::sync::OnceCell; use reqwest::{Client, ClientBuilder, Method, Response}; use serde::Serialize; use tinytemplate::TinyTemplate; use url::Url; use crate::{BinstallError, Meta, PkgFmt, PkgFmtDecomposed, TarBasedFmt}; mod async_extracter; pub use async_extracter::*; mod auto_abort_join_handle; pub use auto_abort_join_handle::AutoAbortJoinHandle; mod ui_thread; pub use ui_thread::UIThread; mod extracter; mod stream_readable; mod path_ext; pub use path_ext::*; mod tls_version; pub use tls_version::TLSVersion; /// (enable https only mode, min TLS version_option) pub static REQWESTGLOBALCONFIG: OnceCell<(bool, Option)> = OnceCell::new(); /// Load binstall metadata from the crate `Cargo.toml` at the provided path pub fn load_manifest_path>( manifest_path: P, ) -> Result, BinstallError> { debug!("Reading manifest: {}", manifest_path.as_ref().display()); // Load and parse manifest (this checks file system for binary output names) let manifest = Manifest::::from_path_with_metadata(manifest_path)?; // Return metadata Ok(manifest) } pub fn new_reqwest_client_builder() -> ClientBuilder { let mut builder = ClientBuilder::new(); if let Some((https_only, min_tls_ver_opt)) = REQWESTGLOBALCONFIG.get() { if *https_only { builder = builder.http2_prior_knowledge(); } if let Some(min_tls_ver) = *min_tls_ver_opt { builder = builder.min_tls_version(min_tls_ver.into()); } } builder } pub fn new_reqwest_client() -> reqwest::Result { new_reqwest_client_builder().build() } pub async fn remote_exists(url: Url, method: Method) -> Result { let req = new_reqwest_client()? .request(method.clone(), url.clone()) .send() .await .map_err(|err| BinstallError::Http { method, url, err })?; Ok(req.status().is_success()) } async fn create_request( url: Url, ) -> Result>, BinstallError> { debug!("Downloading from: '{url}'"); new_reqwest_client()? .get(url.clone()) .send() .await .and_then(|r| r.error_for_status()) .map_err(|err| BinstallError::Http { method: Method::GET, url, err, }) .map(Response::bytes_stream) } /// Download a file from the provided URL and extract it to the provided path. pub async fn download_and_extract>( url: Url, fmt: PkgFmt, path: P, ) -> Result<(), BinstallError> { let stream = create_request(url).await?; let path = path.as_ref(); debug!("Downloading and extracting to: '{}'", path.display()); match fmt.decompose() { PkgFmtDecomposed::Tar(fmt) => extract_tar_based_stream(stream, path, fmt).await?, PkgFmtDecomposed::Bin => extract_bin(stream, path).await?, PkgFmtDecomposed::Zip => extract_zip(stream, path).await?, } debug!("Download OK, extracted to: '{}'", path.display()); Ok(()) } /// Download a file from the provided URL and extract part of it to /// the provided path. /// /// * `filter` - If Some, then it will pass the path of the file to it /// and only extract ones which filter returns `true`. pub async fn download_tar_based_and_visit( url: Url, fmt: TarBasedFmt, visitor: V, ) -> Result { let stream = create_request(url).await?; debug!("Downloading and extracting then in-memory processing"); let visitor = extract_tar_based_stream_and_visit(stream, fmt, visitor).await?; debug!("Download, extraction and in-memory procession OK"); Ok(visitor) } /// Fetch install path from environment /// roughly follows pub fn get_install_path>(install_path: Option

) -> Option { // Command line override first first if let Some(p) = install_path { return Some(PathBuf::from(p.as_ref())); } // Environmental variables if let Ok(p) = std::env::var("CARGO_INSTALL_ROOT") { debug!("using CARGO_INSTALL_ROOT ({p})"); let b = PathBuf::from(p); return Some(b.join("bin")); } if let Ok(p) = std::env::var("CARGO_HOME") { debug!("using CARGO_HOME ({p})"); let b = PathBuf::from(p); return Some(b.join("bin")); } // Standard $HOME/.cargo/bin if let Some(d) = dirs::home_dir() { let d = d.join(".cargo/bin"); if d.exists() { debug!("using $HOME/.cargo/bin"); return Some(d); } } // Local executable dir if no cargo is found let dir = dirs::executable_dir(); if let Some(d) = &dir { debug!("Fallback to {}", d.display()); } dir } pub trait Template: Serialize { fn render(&self, template: &str) -> Result where Self: Sized, { // Create template instance let mut tt = TinyTemplate::new(); // Add template to instance tt.add_template("path", template)?; // Render output Ok(tt.render("path", self)?) } }