use std::{path::Path, sync::Arc}; use compact_str::CompactString; use tracing::{debug, warn}; use url::Url; use crate::{ errors::BinstallError, helpers::{ download::Download, remote::{Client, Method}, signal::wait_on_cancellation_signal, tasks::AutoAbortJoinHandle, }, manifests::cargo_toml_binstall::{PkgFmt, PkgMeta}, }; use super::{Data, TargetData}; const BASE_URL: &str = "https://github.com/alsuren/cargo-quickinstall/releases/download"; const STATS_URL: &str = "https://warehouse-clerk-tmp.vercel.app/api/crate"; pub struct QuickInstall { client: Client, package: String, target_data: Arc, } #[async_trait::async_trait] impl super::Fetcher for QuickInstall { fn new( client: Client, data: Arc, target_data: Arc, ) -> Arc { let crate_name = &data.name; let version = &data.version; let target = &target_data.target; Arc::new(Self { client, package: format!("{crate_name}-{version}-{target}"), target_data, }) } fn find(self: Arc) -> AutoAbortJoinHandle> { AutoAbortJoinHandle::spawn(async move { if cfg!(debug_assertions) { debug!("Not sending quickinstall report in debug mode"); } else { let this = self.clone(); tokio::spawn(async move { if let Err(err) = this.report().await { warn!( "Failed to send quickinstall report for package {}: {err}", this.package ) } }); } let url = self.package_url(); debug!("Checking for package at: '{url}'"); Ok(self .client .remote_exists(Url::parse(&url)?, Method::HEAD) .await?) }) } async fn fetch_and_extract(&self, dst: &Path) -> Result<(), BinstallError> { let url = self.package_url(); debug!("Downloading package from: '{url}'"); Ok(Download::new(self.client.clone(), Url::parse(&url)?) .and_extract( self.pkg_fmt(), dst, Some(Box::pin(wait_on_cancellation_signal())), ) .await?) } fn pkg_fmt(&self) -> PkgFmt { PkgFmt::Tgz } fn target_meta(&self) -> PkgMeta { let mut meta = self.target_data.meta.clone(); meta.pkg_fmt = Some(self.pkg_fmt()); meta.bin_dir = Some("{ bin }{ binary-ext }".to_string()); meta } fn source_name(&self) -> CompactString { CompactString::from("QuickInstall") } fn fetcher_name(&self) -> &'static str { "QuickInstall" } fn is_third_party(&self) -> bool { true } fn target(&self) -> &str { &self.target_data.target } } impl QuickInstall { fn package_url(&self) -> String { format!( "{base_url}/{package}/{package}.tar.gz", base_url = BASE_URL, package = self.package ) } fn stats_url(&self) -> String { format!( "{stats_url}/{package}.tar.gz", stats_url = STATS_URL, package = self.package ) } pub async fn report(&self) -> Result<(), BinstallError> { let url = Url::parse(&self.stats_url())?; debug!("Sending installation report to quickinstall ({url})"); self.client.remote_exists(url, Method::HEAD).await?; Ok(()) } }