fd/src/main.rs

133 lines
4.4 KiB
Rust
Raw Normal View History

#[macro_use]
extern crate clap;
2017-05-12 13:02:20 +02:00
extern crate ansi_term;
2017-06-10 17:30:48 +02:00
extern crate atty;
extern crate regex;
2017-05-15 22:38:34 +02:00
extern crate ignore;
extern crate num_cpus;
2017-06-05 11:56:39 +02:00
pub mod lscolors;
pub mod fshelper;
2017-10-04 14:31:08 +02:00
mod app;
2017-10-10 08:01:17 +02:00
mod internal;
mod output;
mod walk;
2017-05-12 11:50:03 +02:00
use std::env;
2017-05-12 19:34:31 +02:00
use std::error::Error;
2017-10-10 08:01:17 +02:00
use std::path::Path;
use std::sync::Arc;
use std::time;
2017-05-12 11:50:03 +02:00
2017-06-10 17:30:48 +02:00
use atty::Stream;
2017-10-10 08:01:17 +02:00
use regex::RegexBuilder;
2017-05-12 13:02:20 +02:00
2017-10-10 08:01:17 +02:00
use internal::{error, FdOptions, PathDisplay, ROOT_DIR};
2017-06-05 11:56:39 +02:00
use lscolors::LsColors;
2017-10-10 08:01:17 +02:00
use walk::FileType;
2017-05-12 11:50:03 +02:00
fn main() {
2017-10-04 14:31:08 +02:00
let matches = app::build_app().get_matches();
2017-05-12 11:50:03 +02:00
// Get the search pattern
let empty_pattern = String::new();
let pattern = matches.value_of("pattern").unwrap_or(&empty_pattern);
2017-05-12 11:50:03 +02:00
// Get the current working directory
let current_dir_buf = match env::current_dir() {
Ok(cd) => cd,
Err(_) => error("Error: could not get current directory.")
};
2017-05-12 11:50:03 +02:00
let current_dir = current_dir_buf.as_path();
// Get the root directory for the search
let mut root_dir_is_absolute = false;
let root_dir_buf = if let Some(rd) = matches.value_of("path") {
let path = Path::new(rd);
root_dir_is_absolute = path.is_absolute();
fshelper::absolute_path(path).unwrap_or_else(
|_| error(&format!("Error: could not find directory '{}'.", rd))
)
} else {
current_dir_buf.clone()
};
if !root_dir_buf.is_dir() {
error(&format!("Error: '{}' is not a directory.", root_dir_buf.to_string_lossy()));
}
let root_dir = root_dir_buf.as_path();
2017-06-05 14:14:01 +02:00
// The search will be case-sensitive if the command line flag is set or
// if the pattern has an uppercase character (smart case).
let case_sensitive = matches.is_present("case-sensitive") ||
2017-06-05 14:14:01 +02:00
pattern.chars().any(char::is_uppercase);
let colored_output = match matches.value_of("color") {
Some("always") => true,
Some("never") => false,
_ => atty::is(Stream::Stdout)
};
let ls_colors =
2017-06-05 14:14:01 +02:00
if colored_output {
Some(
env::var("LS_COLORS")
.ok()
.map(|val| LsColors::from_string(&val))
.unwrap_or_default()
)
} else {
None
};
let config = FdOptions {
2017-06-05 14:14:01 +02:00
case_sensitive: case_sensitive,
search_full_path: matches.is_present("full-path"),
ignore_hidden: !matches.is_present("hidden"),
read_ignore: !matches.is_present("no-ignore"),
follow_links: matches.is_present("follow"),
null_separator: matches.is_present("null_separator"),
max_depth: matches.value_of("depth")
.and_then(|n| usize::from_str_radix(n, 10).ok()),
threads: std::cmp::max(
matches.value_of("threads")
.and_then(|n| usize::from_str_radix(n, 10).ok())
2017-09-15 21:03:47 +02:00
.unwrap_or_else(num_cpus::get),
1
),
max_buffer_time: matches.value_of("max-buffer-time")
.and_then(|n| u64::from_str_radix(n, 10).ok())
.map(time::Duration::from_millis),
path_display: if matches.is_present("absolute-path") || root_dir_is_absolute {
PathDisplay::Absolute
} else {
PathDisplay::Relative
},
2017-09-17 02:36:07 +02:00
ls_colors: ls_colors,
2017-09-17 09:37:39 +02:00
file_type: match matches.value_of("file-type") {
Some("f") | Some("file") => FileType::RegularFile,
Some("d") | Some("directory") => FileType::Directory,
Some("l") | Some("symlink") => FileType::SymLink,
2017-09-17 09:37:39 +02:00
_ => FileType::Any,
},
2017-09-29 22:33:33 +02:00
extension: matches.value_of("extension")
.map(|e| e.trim_left_matches('.').to_lowercase()),
};
2017-05-12 13:32:30 +02:00
let root = Path::new(ROOT_DIR);
let base = match config.path_display {
PathDisplay::Relative => current_dir,
PathDisplay::Absolute => root
};
match RegexBuilder::new(pattern)
.case_insensitive(!config.case_sensitive)
.build() {
2017-10-10 08:01:17 +02:00
Ok(re) => walk::scan(root_dir, Arc::new(re), base, Arc::new(config)),
Err(err) => error(err.description())
2017-05-12 11:50:03 +02:00
}
}