1
Fork 0
mirror of https://github.com/RGBCube/uutils-coreutils synced 2025-07-29 20:17:45 +00:00

du: refactor pattern building in du

This commit is contained in:
Ackerley Tng 2022-07-28 20:45:43 -07:00 committed by Sylvestre Ledru
parent defde8c91e
commit 898689d924

View file

@ -488,55 +488,28 @@ fn file_as_vec(filename: impl AsRef<Path>) -> Vec<String> {
// Given the --exclude-from and/or --exclude arguments, returns the globset lists // Given the --exclude-from and/or --exclude arguments, returns the globset lists
// to ignore the files // to ignore the files
fn get_glob_ignore(matches: &ArgMatches) -> UResult<Vec<Pattern>> { fn build_exclude_patterns(matches: &ArgMatches) -> UResult<Vec<Pattern>> {
let mut excludes_from = if matches.contains_id(options::EXCLUDE_FROM) { let exclude_from_iterator = matches
match matches.values_of(options::EXCLUDE_FROM) { .values_of(options::EXCLUDE_FROM)
Some(all_files) => { .unwrap_or_default()
let mut exclusion = Vec::<String>::new(); .flat_map(|f| file_as_vec(&f));
// Read the exclude lists from all the files
// and add them into a vector of string
let files: Vec<String> = all_files.clone().map(|v| v.to_owned()).collect();
for f in files {
exclusion.extend(file_as_vec(&f));
}
exclusion
}
None => Vec::<String>::new(),
}
} else {
Vec::<String>::new()
};
let mut excludes = if matches.contains_id(options::EXCLUDE) { let excludes_iterator = matches
match matches.values_of(options::EXCLUDE) { .values_of(options::EXCLUDE)
Some(v) => { .unwrap_or_default()
// Read the various arguments .map(|v| v.to_owned());
v.clone().map(|v| v.to_owned()).collect()
}
None => Vec::<String>::new(),
}
} else {
Vec::<String>::new()
};
// Merge the two lines let mut exclude_patterns = Vec::new();
excludes.append(&mut excludes_from); for f in excludes_iterator.chain(exclude_from_iterator) {
if !&excludes.is_empty() { if matches.is_present(options::VERBOSE) {
let mut builder = Vec::new();
// Create the `Vec` of excludes
for f in excludes {
if matches.contains_id(options::VERBOSE) {
println!("adding {:?} to the exclude list ", &f); println!("adding {:?} to the exclude list ", &f);
} }
match Pattern::new(&f) { match Pattern::new(&f) {
Ok(glob) => builder.push(glob), Ok(glob) => exclude_patterns.push(glob),
Err(err) => return Err(DuError::InvalidGlob(err.to_string()).into()), Err(err) => return Err(DuError::InvalidGlob(err.to_string()).into()),
};
} }
Ok(builder)
} else {
Ok(Vec::new())
} }
Ok(exclude_patterns)
} }
#[uucore::main] #[uucore::main]
@ -615,14 +588,13 @@ pub fn uumain(args: impl uucore::Args) -> UResult<()> {
"\n" "\n"
}; };
let excludes = get_glob_ignore(&matches)?; let excludes = build_exclude_patterns(&matches)?;
let mut grand_total = 0; let mut grand_total = 0;
'loop_file: for path_string in files { 'loop_file: for path_string in files {
// Skip if we don't want to ignore anything // Skip if we don't want to ignore anything
if !&excludes.is_empty() { if !&excludes.is_empty() {
for pattern in &excludes { for pattern in &excludes {
{
if pattern.matches(path_string) { if pattern.matches(path_string) {
// if the directory is ignored, leave early // if the directory is ignored, leave early
if options.verbose { if options.verbose {
@ -632,11 +604,9 @@ pub fn uumain(args: impl uucore::Args) -> UResult<()> {
} }
} }
} }
}
let path = PathBuf::from(&path_string); let path = PathBuf::from(&path_string);
match Stat::new(path, &options) { if let Ok(stat) = Stat::new(path, &options) {
Ok(stat) => {
let mut inodes: HashSet<FileInfo> = HashSet::new(); let mut inodes: HashSet<FileInfo> = HashSet::new();
if let Some(inode) = stat.inode { if let Some(inode) = stat.inode {
inodes.insert(inode); inodes.insert(inode);
@ -651,7 +621,7 @@ pub fn uumain(args: impl uucore::Args) -> UResult<()> {
continue; continue;
} }
if matches.contains_id(options::TIME) { if matches.is_present(options::TIME) {
let tm = { let tm = {
let secs = { let secs = {
match matches.value_of(options::TIME) { match matches.value_of(options::TIME) {
@ -686,8 +656,7 @@ pub fn uumain(args: impl uucore::Args) -> UResult<()> {
grand_total += size; grand_total += size;
} }
} }
} } else {
Err(_) => {
show_error!( show_error!(
"{}: {}", "{}: {}",
path_string.maybe_quote(), path_string.maybe_quote(),
@ -695,7 +664,6 @@ pub fn uumain(args: impl uucore::Args) -> UResult<()> {
); );
} }
} }
}
if options.total { if options.total {
print!("{}\ttotal", convert_size(grand_total)); print!("{}\ttotal", convert_size(grand_total));