feat: add ignore flag for ignoring directories

Author: elliottcrif

src/main.rs

@@ -22,6 +22,7 @@ use ::std::sync::Arc;
 use ::std::thread::park_timeout;
 use ::std::{thread, time};
 use ::structopt::StructOpt;
+use nix::dir;
 
 use ::tui::backend::Backend;
 use crossterm::event::KeyModifiers;
@@ -53,6 +54,9 @@ pub struct Opt {
     /// The folder to scan
     folder: Option<PathBuf>,
     #[structopt(short, long)]
+    // The directories to ignore
+    ignore: Vec<PathBuf>,
+    #[structopt(short, long)]
     /// Show file sizes rather than their block usage on disk
     apparent_size: bool,
     #[structopt(short, long)]
@@ -85,10 +89,13 @@ fn try_main() -> Result<(), failure::Error> {
             if !folder.as_path().is_dir() {
                 failure::bail!("Folder '{}' does not exist", folder.to_string_lossy())
             }
+            // TODO: maybe validate the ignore flag
+
             start(
                 terminal_backend,
                 Box::new(terminal_events),
                 folder,
+                opts.ignore,
                 opts.apparent_size,
                 opts.disable_delete_confirmation,
             );
@@ -103,6 +110,7 @@ pub fn start<B>(
     terminal_backend: B,
     terminal_events: Box<dyn Iterator<Item = BackEvent> + Send>,
     path: PathBuf,
+    ignore: Vec<PathBuf>,
     show_apparent_size: bool,
     disable_delete_confirmation: bool,
 ) where
@@ -188,6 +196,30 @@ pub fn start<B>(
                 let loaded = loaded.clone();
                 move || {
                     'scanning: for entry in WalkDir::new(&path)
+                        .process_read_dir({
+                            let ignore = ignore.clone();
+                            move |_read_dir_state, children| {
+                                if !ignore.is_empty() {
+                                    children.retain(|dir_entry_result| {
+                                        dir_entry_result
+                                            .as_ref()
+                                            .map(|dir_entry| {
+                                                let entry_path = dir_entry.path();
+                                                let entry_absolute =
+                                                    entry_path.canonicalize().unwrap_or(entry_path);
+                                                !ignore.iter().any(|ignore_path| {
+                                                    let ignore_absolute =
+                                                        ignore_path.canonicalize().unwrap_or_else(
+                                                            |_| ignore_path.to_path_buf(),
+                                                        );
+                                                    entry_absolute.starts_with(&ignore_absolute)
+                                                })
+                                            })
+                                            .unwrap_or(false)
+                                    });
+                                }
+                            }
+                        })
                         .parallelism(if SHOULD_SCAN_HD_FILES_IN_MULTIPLE_THREADS {
                             RayonDefaultPool
                         } else {