Robot
/
fd
mirror of https://github.com/sharkdp/fd.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

--filename instead of --full-path

This commit is contained in:
sharkdp 2017-05-12 22:20:14 +02:00
parent e20bd9d3f8
commit 1986fc4373
1 changed files with 25 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
src/main.rs
View File

@ -55,18 +55,22 @@ fn scan(root: &Path, pattern: &Regex, config: &FdOptions) {
.filter(|e| e.path() != root); .filter(|e| e.path() != root);
for entry in walker { for entry in walker {
if let Ok(path_relative) = entry.path().strip_prefix(root) { let path_rel = match entry.path().strip_prefix(root) {
if let Some(path_str) = path_relative.to_str() { Ok(p) => p,
let haystack = if config.search_full_path { Err(_) => continue
path_str };
} else {
path_relative.file_name() if let Some(path_str) = path_rel.to_str() {
.and_then(OsStr::to_str) let res =
.unwrap() if config.search_full_path {
}; pattern.find(path_str)
pattern.find(haystack) } else {
.map(|_| print_entry(&entry, path_str, &config)); path_rel.file_name()
} .and_then(OsStr::to_str)
.and_then(|s| pattern.find(s))
};
res.map(|_| print_entry(&entry, path_str, &config));
} }
} }
} }
@ -83,13 +87,15 @@ fn main() {
let mut opts = Options::new(); let mut opts = Options::new();
opts.optflag("h", "help", "print this help message"); opts.optflag("h", "help", "print this help message");
opts.optflag("s", "sensitive", "case-sensitive search (default: smart case)"); opts.optflag("s", "sensitive",
opts.optflag("f", "full-path", "search full path (default: only filename)"); "case-sensitive search (default: smart case)");
opts.optflag("f", "filename",
"search filenames only (default: full path)");
opts.optflag("F", "follow", "follow symlinks (default: off)"); opts.optflag("F", "follow", "follow symlinks (default: off)");
opts.optflag("n", "no-color", "do not colorize output"); opts.optflag("n", "no-color", "do not colorize output");
let matches = match opts.parse(&args[1..]) { let matches = match opts.parse(&args[1..]) {
Ok(m) => m, Ok(m) => m,
Err(e) => error(e.description()) Err(e) => error(e.description())
}; };
@ -110,11 +116,11 @@ fn main() {
let config = FdOptions { let config = FdOptions {
// The search will be case-sensitive if the command line flag is set or if // The search will be case-sensitive if the command line flag is set or
// the pattern has an uppercase character (smart case). // if the pattern has an uppercase character (smart case).
case_sensitive: matches.opt_present("s") || case_sensitive: matches.opt_present("s") ||
pattern.chars().any(char::is_uppercase), pattern.chars().any(char::is_uppercase),
search_full_path: matches.opt_present("f"), search_full_path: !matches.opt_present("f"),
colored: !matches.opt_present("n"), colored: !matches.opt_present("n"),
follow_links: matches.opt_present("F") follow_links: matches.opt_present("F")
}; };
@ -122,7 +128,7 @@ fn main() {
match RegexBuilder::new(pattern) match RegexBuilder::new(pattern)
.case_insensitive(!config.case_sensitive) .case_insensitive(!config.case_sensitive)
.build() { .build() {
Ok(re) => scan(&current_dir, &re, &config), Ok(re) => scan(&current_dir, &re, &config),
Err(err) => error(err.description()) Err(err) => error(err.description())
} }
} }