wip adds user and group filters

src/cli.rs

@@ -29,6 +29,10 @@ use clap::value_parser;
 use clap::{crate_description, crate_name, crate_version};
 use clap::{Arg, ArgAction, Command};
 
+pub const CONFLICT_FILTER: &str =
+    "the filter options --group and --user are in conflict, clap SHOULD NOT \
+     allow both to be present";
+
 /// Returns command-line parser.
 pub fn build() -> Command {
     let dir = Arg::new("dir")
@@ -80,6 +84,7 @@ pub fn build() -> Command {
         .disable_version_flag(true)
         .arg(dir)
         .args(output())
+        .args(filter())
         .args(mmapplypolicy())
         .arg(max_depth)
         .arg(debug)
@@ -126,6 +131,26 @@ fn output() -> Vec<Arg> {
     vec![block, inodes, both]
 }
 
+fn filter() -> Vec<Arg> {
+    let group = Arg::new("group")
+        .long("group")
+        .conflicts_with("user")
+        .help("filter by group")
+        .long_help("Consider only inodes owned by this group.")
+        .value_name("name|gid")
+        .help_heading("Filtering");
+
+    let user = Arg::new("user")
+        .long("user")
+        .conflicts_with("group")
+        .help("filter by user")
+        .long_help("Consider only inodes owned by this user.")
+        .value_name("name|uid")
+        .help_heading("Filtering");
+
+    vec![group, user]
+}
+
 /// Returns arguments forwarded to `mmapplypolicy`.
 fn mmapplypolicy() -> Vec<Arg> {
     let nodes = Arg::new("nodes")

src/config.rs

@@ -37,6 +37,7 @@ pub fn get() -> Config {
 pub struct Config {
     pub dirs: Option<Vec<PathBuf>>,
     pub debug: bool,
+    pub filter: Option<Filter>,
     pub max_depth: Option<usize>,
     pub mm_nodes: Option<String>,
     pub mm_local_work_dir: Option<PathBuf>,
@@ -53,6 +54,18 @@ impl From<ArgMatches> for Config {
 
         let debug = args.get_one::<bool>("debug").copied().unwrap_or_default();
 
+        let group = args.get_one::<String>("group");
+        let user = args.get_one::<String>("user");
+
+        let filter = match (group, user) {
+            (None, None) => None,
+            (Some(group), None) => Some(Filter::Group(group.clone())),
+            (None, Some(user)) => Some(Filter::User(user.clone())),
+            (Some(_), Some(_)) => {
+                unreachable!("{}", crate::cli::CONFLICT_FILTER)
+            }
+        };
+
         let max_depth = args
             .get_one::<usize>("max-depth")
             .copied()
@@ -71,6 +84,7 @@ impl From<ArgMatches> for Config {
         Self {
             dirs,
             debug,
+            filter,
             max_depth,
             mm_nodes,
             mm_local_work_dir,
@@ -81,6 +95,12 @@ impl From<ArgMatches> for Config {
     }
 }
 
+#[derive(Debug)]
+pub enum Filter {
+    Group(String),
+    User(String),
+}
+
 /// Returns whether to count block usage and/or inode usage.
 fn cbi(args: &ArgMatches) -> (bool, bool) {
     let block = args.get_flag("block");