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.
99 lines
1.9 KiB
99 lines
1.9 KiB
#![allow(dead_code)]
|
|
#![allow(unused_imports)]
|
|
|
|
#[cfg(feature="parallel")] #[macro_use] extern crate pin_project;
|
|
#[macro_use] extern crate log;
|
|
#[macro_use] mod macros;
|
|
|
|
use color_eyre::{
|
|
eyre::{self, eyre, WrapErr},
|
|
Help, SectionExt,
|
|
};
|
|
use lazy_static::lazy_static;
|
|
use cfg_if::cfg_if;
|
|
|
|
fn init() -> eyre::Result<()>
|
|
{
|
|
color_eyre::install()?;
|
|
if let None = std::env::var_os("RUST_LOG") {
|
|
std::env::set_var("RUST_LOG", "info");
|
|
}
|
|
pretty_env_logger::init();
|
|
trace!("Initialised");
|
|
Ok(())
|
|
}
|
|
|
|
mod ext;
|
|
use ext::*;
|
|
mod map;
|
|
use map::*;
|
|
mod temp;
|
|
mod error;
|
|
mod arg;
|
|
#[cfg(feature="recursive")]
|
|
mod recurse;
|
|
|
|
cfg_if!{
|
|
if #[cfg(feature="splash")] {
|
|
mod splash;
|
|
} else {
|
|
mod splash {
|
|
#[inline(always)] pub fn splash() -> ! {
|
|
super::arg::usage();
|
|
std::process::exit(1)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(feature="parallel")]
|
|
mod parallel;
|
|
|
|
fn args_or_out<T: ExactSizeIterator>(i: T, low: usize) -> T
|
|
{
|
|
if i.len() < low {
|
|
splash::splash();
|
|
} else {
|
|
i
|
|
}
|
|
}
|
|
|
|
#[cfg(feature="parallel")]
|
|
#[cfg_attr(feature="parallel", tokio::main)]
|
|
async fn main() -> eyre::Result<()> {
|
|
use futures::{
|
|
stream,
|
|
prelude::*,
|
|
};
|
|
|
|
reyre!(init(), "Failed to initialise")?;
|
|
reyre!(parallel::main(stream::iter(args_or_out(std::env::args(), 2)
|
|
.skip(1)
|
|
.dedup())
|
|
.filter_map(|file| {
|
|
async move {
|
|
Some(parallel::expand_dir(file).await) //TODO: We gotta in here, too
|
|
}
|
|
})
|
|
.flatten()
|
|
.dedup()).await,
|
|
"Jobs failed")
|
|
}
|
|
|
|
#[cfg(not(feature="parallel"))]
|
|
mod serial;
|
|
|
|
#[cfg(not(feature="parallel"))]
|
|
fn main() -> eyre::Result<()> {
|
|
reyre!(init(), "Failed to initialise")?;
|
|
cfg_if!{
|
|
if #[cfg(feature="recursive")] {
|
|
let args = recurse::walk_dirs(args_or_out(std::env::args(), 2).skip(1)).dedup();
|
|
} else {
|
|
let args = args_or_out(std::env::args(), 2).skip(1).dedup();
|
|
}
|
|
};
|
|
reyre!(serial::main(args),
|
|
"Jobs failed")
|
|
}
|