Compare commits

..

2 commits

2 changed files with 48 additions and 23 deletions

View file

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

View file

@ -1,5 +1,5 @@
use std::{error::Error, future::Future};
use indicatif::ProgressBar;
use indicatif::{ProgressBar, ProgressStyle};
pub async fn with_spinner<F, Fut, T>(
start_msg: &str,
@ -17,7 +17,32 @@ where
let result = f().await;
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
}