rsup_utils 0.1.3

use std::{error::Error, fs::File, path::Path};

use flate2::read::GzDecoder;
use futures_util::StreamExt;
use indicatif::{ProgressBar, ProgressStyle};
use reqwest::Client;
use tar::Archive;
use tokio::{fs, io::AsyncWriteExt};

/// 解压文件
///
/// # Arguments
/// * url 下载地址
/// * target_dir 解压目录
///
/// # Returns
/// * `Result<(), Box<dyn Error>>`
///     
///     解压成功返回`Ok(())`
///
///     解压失败返回`Err(Box<dyn Error>)`
///
pub async fn decompress_file(url: &str, target_dir: &str) -> Result<(), Box<dyn Error>> {
    let tar_gz = File::open(url)?;

    let decomppress = GzDecoder::new(tar_gz);
    let mut archive = Archive::new(decomppress);

    // 目录不存在时创建目录
    if !Path::new(target_dir).exists() {
        // 处理解压目录，不存在则创建目录
        fs::create_dir_all(target_dir).await?;
    }

    archive.unpack(target_dir)?;

    Ok(())
}
///
/// 下载文件
///
/// # Arguments
/// * client 请求客户端
/// * url 下载地址
/// * output 保存路径
///
/// # Returns
/// * `Result<(), Box<dyn Error>>`
///     
///     下载成功返回`Ok(())`
///
///     下载失败返回`Err(Box<dyn Error>)`
pub async fn download_file(url: &str, output: &str) -> Result<(), Box<dyn Error>> {
    let client = Client::new();
    // 下载地址
    let res = client.get(url).send().await?;

    if res.status().is_success() {
        // 获取文件大小
        let content_size = res.content_length().ok_or("无法获取文件大小")?;

        // 下载成功
        // 保存文件到指定目录
        // 文件路径
        let mut file = fs::File::create(output).await?;

        // 创建进度条
        let pb = ProgressBar::new(content_size);
        pb.set_style(
            ProgressStyle::default_bar()
                .template("{msg} [{elapsed_precise}] {bar:80} {percent}%")?
                .progress_chars("##-"),
        );

        // 创建流式响应体
        let mut downloaded = 0;
        let mut stream = res.bytes_stream();
        while let Some(item) = stream.next().await {
            let chunk = item?;
            file.write_all(&chunk).await?;

            let len = chunk.len() as u64;
            downloaded += len;
            pb.set_position(downloaded);
        }
        file.flush().await?;

        pb.finish_with_message("下载完成");
        // 保存文件
        // let bytes = res.bytes().await?;
        // file.write_all(&bytes).await?;
        Ok(())
    } else {
        let error_message = format!("Request failed with status code: {}", res.status());
        Err(Box::new(std::io::Error::new(
            std::io::ErrorKind::Other,
            error_message,
        )))
    }
}