You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
84 lines
1.8 KiB
84 lines
1.8 KiB
#![cfg_attr(nightly, feature(int_error_matching))]
|
|
#![cfg_attr(nightly, feature(linked_list_remove))]
|
|
#![cfg_attr(nightly, feature(const_fn))]
|
|
#![cfg_attr(nightly, feature(never_type))]
|
|
|
|
|
|
#![allow(dead_code)]
|
|
|
|
use cfg_if::cfg_if;
|
|
|
|
mod defer;
|
|
mod ext;
|
|
pub use ext::JoinStrsExt as _;
|
|
|
|
#[cfg(feature="splash")]
|
|
mod splash;
|
|
mod flags;
|
|
mod arg;
|
|
mod error;
|
|
pub use error::{ErrorExt as _, ResultExt as _};
|
|
|
|
mod stage;
|
|
mod leanify;
|
|
|
|
mod colour;
|
|
mod dir;
|
|
mod fixed_stack;
|
|
mod process;
|
|
mod work;
|
|
|
|
//mod timeout;
|
|
mod maybe_single;
|
|
#[cfg(feature="progress")] mod task_list;
|
|
#[cfg(feature="progress")] mod progress;
|
|
|
|
async fn work() -> Result<(), Box<dyn std::error::Error>>
|
|
{
|
|
|
|
//println!("umm {}", colour::style(colour!(Color::Blue), "hiii"));
|
|
let args = arg::parse_args().await.with_prefix("failed to parse args")?;
|
|
let leanify = leanify::find_binary().with_prefix("Couldn't find leanify binary")?;
|
|
|
|
#[cfg(feature="colour")] {
|
|
match args.flags.coloured {
|
|
Some(x) => recolored::control::set_override(x),
|
|
_ => {},
|
|
}
|
|
}
|
|
|
|
work::work(&args.flags, leanify, args.files, match args.flags.hard_limit {
|
|
Some(hard_limit) => {
|
|
// We have hard limit
|
|
match args.max_children {
|
|
// We also have max children, and it's higher than hard limit,
|
|
Some(max_children) if max_children > hard_limit => Some(hard_limit),
|
|
// We have no max children
|
|
None => Some(hard_limit),
|
|
//Max children is lower than hard limit
|
|
_ => args.max_children,
|
|
}
|
|
},
|
|
_ => args.max_children,
|
|
}).await
|
|
}
|
|
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
prettify_expect(work().await.map_err(|e| e.to_string()), "exited with error");
|
|
}
|
|
|
|
#[inline] fn prettify_expect<T,E,S>(res: Result<T,E>, msg: S) -> T
|
|
where S: AsRef<str>,
|
|
E: std::fmt::Display
|
|
{
|
|
match res {
|
|
Ok(v) => v,
|
|
Err(e) => {
|
|
eprintln!("\n{}: {}", msg.as_ref(), e);
|
|
std::process::exit(1)
|
|
},
|
|
}
|
|
}
|