cargo-binstall/crates/binstalk/src/helpers/remote.rs

pub use binstalk_downloader::remote::*;
pub use url::ParseError as UrlParseError;

use std::sync::{
    atomic::{AtomicBool, Ordering::Relaxed},
    Once,
};
use tracing::{debug, warn};

use super::gh_api_client::{GhApiClient, GhReleaseArtifact, HasReleaseArtifact};
use crate::errors::BinstallError;

/// This function returns a future where its size should be at most size of
/// 2 pointers.
pub async fn does_url_exist(
    client: Client,
    gh_api_client: GhApiClient,
    url: &Url,
) -> Result<bool, BinstallError> {
    static GH_API_CLIENT_FAILED: AtomicBool = AtomicBool::new(false);
    static WARN_RATE_LIMIT_ONCE: Once = Once::new();
    static WARN_UNAUTHORIZED_ONCE: Once = Once::new();

    debug!("Checking for package at: '{url}'");

    if !GH_API_CLIENT_FAILED.load(Relaxed) {
        if let Some(artifact) = GhReleaseArtifact::try_extract_from_url(url) {
            debug!("Using GitHub API to check for existence of artifact, which will also cache the API response");

            // The future returned has the same size as a pointer
            match gh_api_client.has_release_artifact(artifact).await? {
                HasReleaseArtifact::Yes { .. } => return Ok(true),
                HasReleaseArtifact::No | HasReleaseArtifact::NoSuchRelease => return Ok(false),

                HasReleaseArtifact::RateLimit { retry_after } => {
                    WARN_RATE_LIMIT_ONCE.call_once(|| {
                        warn!("Your GitHub API token (if any) has reached its rate limit and cannot be used again until {retry_after:?}, so we will fallback to HEAD/GET on the url.");
                        warn!("If you did not supply a github token, consider doing so: GitHub limits unauthorized users to 60 requests per hour per origin IP address.");
                    });
                }
                HasReleaseArtifact::Unauthorized => {
                    WARN_UNAUTHORIZED_ONCE.call_once(|| {
                        warn!("GitHub API somehow requires a token for the API access, so we will fallback to HEAD/GET on the url.");
                        warn!("Please consider supplying a token to cargo-binstall to speedup resolution.");
                    });
                }
            }

            GH_API_CLIENT_FAILED.store(true, Relaxed);
        }
    }

    Ok(Box::pin(client.remote_gettable(url.clone())).await?)
}