From dcc709a666d8e60caf5526d444504798e7c1f3ae Mon Sep 17 00:00:00 2001 From: sreedev Date: Mon, 17 Jul 2023 14:06:57 -0400 Subject: [PATCH] added filetype filter --- src/params.rs | 4 ++++ src/scanner.rs | 7 +++++-- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/src/params.rs b/src/params.rs index 71889eb..e37dd1d 100644 --- a/src/params.rs +++ b/src/params.rs @@ -46,4 +46,8 @@ impl Params { let dir = fs::canonicalize(dir_path)?; Ok(dir) } + + pub fn get_types(&self) -> Option { + self.types.clone() + } } diff --git a/src/scanner.rs b/src/scanner.rs index 400827f..c96cdb4 100644 --- a/src/scanner.rs +++ b/src/scanner.rs @@ -36,6 +36,10 @@ impl Scanner { Some(min_size) => scanner.min_size(min_size), None => scanner, }) + .map(|scanner| match app_args.get_types() { + Some(ftypes) => scanner.filetypes(ftypes), + None => scanner, + }) .map(|scanner| match app_args.min_depth { Some(min_depth) => scanner.min_depth(min_depth), None => scanner, @@ -142,8 +146,7 @@ impl Scanner { } pub fn scan(&self) -> Result> { - let progress_style = - ProgressStyle::with_template("[{elapsed_precise}] {pos:>7} {msg}")?; + let progress_style = ProgressStyle::with_template("[{elapsed_precise}] {pos:>7} {msg}")?; let progress_bar = ProgressBar::new_spinner(); progress_bar.set_style(progress_style); progress_bar.enable_steady_tick(Duration::from_millis(50));