Compare commits

...

2 commits

2 changed files with 48 additions and 23 deletions

View file

@ -1,14 +1,13 @@
mod models; mod models;
mod utils; mod utils;
use crate::{models::vanilla::{VanillaVersionDetail, VanillaVersionType, VanillaVersions}, utils::with_spinner}; use crate::{models::vanilla::{VanillaVersionDetail, VanillaVersionType, VanillaVersions}, utils::{with_progressbar, with_spinner}};
use clap::{ use clap::{
Parser, Subcommand, Parser, Subcommand,
builder::styling::{self}, builder::styling::{self},
}; };
use dialoguer::{FuzzySelect, Select, theme::ColorfulTheme}; use dialoguer::{FuzzySelect, Select, theme::ColorfulTheme};
use futures_util::StreamExt; use futures_util::StreamExt;
use indicatif::{ProgressBar, ProgressStyle};
use sha1::{Digest, Sha1}; use sha1::{Digest, Sha1};
use std::error::Error; use std::error::Error;
use tokio::io::AsyncWriteExt; use tokio::io::AsyncWriteExt;
@ -74,30 +73,31 @@ async fn main() -> Result<(), Box<dyn Error>> {
.interact()?; .interact()?;
let version_detail: VanillaVersionDetail = let version_detail: VanillaVersionDetail =
reqwest::get(release_versions[version_index].url.clone()) with_spinner("", "", || async {
.await? let version_detail: VanillaVersionDetail =
.json() reqwest::get(release_versions[version_index].url.clone())
.await?; .await?
.json()
.await?;
Ok(version_detail)
}).await?;
println!( let start_msg = format!(
"downloading server.jar for {} {}", "downloading server.jar for {} {}",
loaders[loader_index], release_version_names[version_index] loaders[loader_index], release_version_names[version_index]
); );
let pb = ProgressBar::new(version_detail.downloads.server.size); with_progressbar(&start_msg, "download finished", version_detail.downloads.server.size, |pb| async move {
pb.set_style(ProgressStyle::default_bar().template( let server_bytes = reqwest::get(version_detail.downloads.server.url).await?;
"{bar:40.green/white} {bytes}/{total_bytes} ({bytes_per_sec})", let mut file = tokio::fs::File::create("server.jar").await?;
)?); let mut stream = server_bytes.bytes_stream();
let server_bytes = reqwest::get(version_detail.downloads.server.url).await?; while let Some(chunk) = stream.next().await {
let mut file = tokio::fs::File::create("server.jar").await?; let chunk = chunk?;
let mut stream = server_bytes.bytes_stream(); file.write_all(&chunk).await?;
pb.inc(chunk.len() as u64);
while let Some(chunk) = stream.next().await { }
let chunk = chunk?; Ok(())
file.write_all(&chunk).await?; }).await?;
pb.inc(chunk.len() as u64);
}
pb.finish_with_message("download finished");
with_spinner("verifying sha1 hash", "sha1 hash verified", || async { with_spinner("verifying sha1 hash", "sha1 hash verified", || async {
let data = tokio::fs::read("server.jar").await?; let data = tokio::fs::read("server.jar").await?;

View file

@ -1,5 +1,5 @@
use std::{error::Error, future::Future}; use std::{error::Error, future::Future};
use indicatif::ProgressBar; use indicatif::{ProgressBar, ProgressStyle};
pub async fn with_spinner<F, Fut, T>( pub async fn with_spinner<F, Fut, T>(
start_msg: &str, start_msg: &str,
@ -17,7 +17,32 @@ where
let result = f().await; let result = f().await;
pb.finish_and_clear(); pb.finish_and_clear();
println!("{} {}", dialoguer::console::style("✔").green(), finish_msg); if !finish_msg.is_empty() {
println!("{} {}", dialoguer::console::style("✔").green(), finish_msg);
}
result
}
pub async fn with_progressbar<F, Fut, T>(
start_msg: &str,
finish_msg: &str,
len: u64,
f: F,
) -> Result<T, Box<dyn Error>>
where
F: FnOnce(ProgressBar) -> Fut,
Fut: Future<Output = Result<T, Box<dyn Error>>>
{
let pb = ProgressBar::new(len);
pb.set_style(ProgressStyle::default_bar().template(
"{msg}\n{bar:40.green/white} {bytes}/{total_bytes} ({bytes_per_sec})",
)?);
pb.set_message(start_msg.to_string());
let result = f(pb.clone()).await;
pb.finish_with_message(finish_msg.to_string());
result result
} }