typos/src/main.rs

88 lines
2 KiB
Rust
Raw Normal View History

2019-01-23 09:33:51 -05:00
// 2015-edition macros.
#[macro_use]
extern crate clap;
2019-01-22 17:01:33 -05:00
use structopt::StructOpt;
2019-01-23 09:33:51 -05:00
arg_enum!{
#[derive(Debug, Copy, Clone, PartialEq, Eq)]
enum Format {
Silent,
Brief,
Long,
Json,
}
}
impl Format {
2019-01-24 10:24:20 -05:00
fn report(self) -> scorrect::report::Report {
2019-01-23 09:33:51 -05:00
match self {
2019-01-24 10:24:20 -05:00
Format::Silent => scorrect::report::print_silent,
Format::Brief => scorrect::report::print_brief,
Format::Long => scorrect::report::print_long,
Format::Json => scorrect::report::print_json,
2019-01-23 09:33:51 -05:00
}
}
}
impl Default for Format {
fn default() -> Self {
Format::Long
}
}
2019-01-22 17:01:33 -05:00
#[derive(Debug, StructOpt)]
struct Options {
2019-01-23 09:34:05 -05:00
#[structopt(parse(from_os_str), default_value = ".")]
2019-01-22 17:01:33 -05:00
/// Paths to check
path: Vec<std::path::PathBuf>,
2019-01-23 09:33:51 -05:00
#[structopt(long = "format",
raw(possible_values = "&Format::variants()", case_insensitive = "true"),
default_value = "long")]
pub format: Format,
2019-01-22 17:01:33 -05:00
#[structopt(short="j", long="threads", default_value="0")]
/// The approximate number of threads to use.
threads: usize,
}
impl Options {
pub fn infer(mut self) -> Self {
if self.path.len() == 1 {
if self.path[0].is_file() {
self.threads = 1;
}
}
self
}
}
fn run() -> Result<(), failure::Error> {
let options = Options::from_args().infer();
2019-01-24 10:24:20 -05:00
let dictionary = scorrect::Dictionary::new();
2019-01-22 17:01:33 -05:00
let first_path = &options.path.get(0).expect("arg parsing enforces at least one");
let mut walk = ignore::WalkBuilder::new(first_path);
for path in &options.path[1..] {
walk.add(path);
}
walk.threads(options.threads);
// TODO Add build_parallel for options.threads != 1
for entry in walk.build() {
let entry = entry?;
if entry.file_type().map(|t| t.is_file()).unwrap_or(true) {
2019-01-23 09:33:51 -05:00
scorrect::process_file(entry.path(), &dictionary, options.format.report())?;
2019-01-22 17:01:33 -05:00
}
}
Ok(())
}
fn main() {
run().unwrap();
}