Compare commits
2 commits
0a12b87a2a
...
ed93609899
| Author | SHA1 | Date | |
|---|---|---|---|
| ed93609899 | |||
| 2aaae9239c |
2 changed files with 48 additions and 23 deletions
42
src/main.rs
42
src/main.rs
|
|
@ -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?;
|
||||||
|
|
|
||||||
29
src/utils.rs
29
src/utils.rs
|
|
@ -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
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue