nenv/src/web_api/mod.rs

use std::{
    cmp::min,
    fmt::{Debug, Display},
    time::Duration,
};

use crate::consts::NODE_ARCHIVE_SUFFIX;

use self::error::{ApiError, ApiResult};
use indicatif::{ProgressBar, ProgressStyle};
use reqwest::Client;

pub mod error;
mod model;
use futures_util::StreamExt;
pub use model::*;
use tokio::io::{AsyncWrite, AsyncWriteExt};

#[cfg(test)]
mod test;

#[derive(Clone, Debug)]
pub struct NodejsAccess {
    base_url: String,
    client: Client,
}

impl Default for NodejsAccess {
    fn default() -> Self {
        Self::new("https://nodejs.org/dist")
    }
}

impl NodejsAccess {
    /// Creates a new instance to access the nodejs website
    pub fn new<S: ToString>(base_url: S) -> Self {
        Self {
            base_url: base_url.to_string(),
            client: Client::new(),
        }
    }

    /// Returns the list of available node versions
    #[tracing::instrument(level = "trace")]
    pub async fn get_versions(&self) -> ApiResult<Vec<VersionInfo>> {
        let versions = self
            .client
            .get(format!("{}/index.json", self.base_url))
            .send()
            .await?
            .json()
            .await?;

        Ok(versions)
    }

    /// Downloads a specific node version
    /// and writes it to the given writer
    #[tracing::instrument(level = "trace", skip(writer))]
    pub async fn download_version<W: AsyncWrite + Unpin, S: Display + Debug>(
        &self,
        version: S,
        writer: &mut W,
    ) -> ApiResult<u64> {
        let res = self
            .client
            .get(format!(
                "{}/v{version}/node-v{version}{}",
                self.base_url, *NODE_ARCHIVE_SUFFIX
            ))
            .send()
            .await?;
        let total_size = res
            .content_length()
            .ok_or_else(|| ApiError::other("Missing content length"))?;
        let pb = ProgressBar::new(total_size);
        pb.set_message(format!("Downloading node v{version}"));
        pb.set_style(
            ProgressStyle::default_bar()
                .template(
                    "{msg} {spinner}\n[{wide_bar}] {bytes}/{total_bytes} ({bytes_per_sec}, {eta})",
                )
                .unwrap(),
        );
        pb.enable_steady_tick(Duration::from_millis(50));
        let mut stream = res.bytes_stream();
        let mut total_downloaded = 0;

        while let Some(item) = stream.next().await {
            let chunk = item?;
            writer.write_all(&chunk).await?;
            total_downloaded = min(chunk.len() as u64 + total_downloaded, total_size);
            pb.set_position(total_downloaded);
        }

        writer.flush().await?;
        pb.finish_with_message(format!("Downloaded node v{version}."));

        Ok(total_downloaded)
    }
}