nhentairs/src/commands/view.rs

50 lines
1.4 KiB
Rust

use crate::api;
use crate::utils;
use crate::structs;
use std::env;
use std::process::exit;
use tokio::task::JoinHandle;
extern crate tokio;
extern crate reqwest;
pub async fn run(args: env::Args) {
let sauces = utils::get_arg_sauces(args).unwrap();
if sauces.len() < 1 {
eprintln!("Missing sauce(s)");
exit(1);
}
let client = reqwest::Client::new();
let mut handles: Vec<JoinHandle<(structs::GalleryInfo, i32)>> = Vec::with_capacity(sauces.len());
for sauce in sauces {
let cloned_client = client.clone();
handles.push(tokio::spawn(async move {
(api::get_sauce_info(cloned_client, sauce).await.unwrap(), sauce)
}));
}
let mut fail = false;
let mut one_done = false;
for handle in handles {
let (sauce_info, sauce) = handle.await.unwrap();
match sauce_info {
structs::GalleryInfo::Info(sauce_info) => {
if one_done {
println!("");
}
println!("{}", &sauce_info);
},
structs::GalleryInfo::Error(sauce_error) => {
if one_done {
eprintln!("");
}
eprintln!("Sauce: {}\nError: {}", sauce, sauce_error.error);
fail = true;
}
};
one_done = true;
}
if fail {
exit(1);
}
}