From 30cb4adc6152379fbac15854f1d42ac028c23d58 Mon Sep 17 00:00:00 2001
From: Alexandru Macovei <alexnmaco@gmail.com>
Date: Sat, 30 Jun 2018 15:55:33 +0300
Subject: [PATCH] [ownership] initial struct, parsing, and test framework

---
 src/filter/mod.rs   |  6 +++++
 src/filter/owner.rs | 61 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 67 insertions(+)
 create mode 100644 src/filter/owner.rs

diff --git a/src/filter/mod.rs b/src/filter/mod.rs
index e513c0c..5e45d3b 100644
--- a/src/filter/mod.rs
+++ b/src/filter/mod.rs
@@ -1,5 +1,11 @@
 pub use self::size::SizeFilter;
 pub use self::time::TimeFilter;
 
+#[cfg(unix)]
+pub use self::owner::OwnerFilter;
+
 mod size;
 mod time;
+
+#[cfg(unix)]
+mod owner;
diff --git a/src/filter/owner.rs b/src/filter/owner.rs
new file mode 100644
index 0000000..a4da9b6
--- /dev/null
+++ b/src/filter/owner.rs
@@ -0,0 +1,61 @@
+use anyhow::{anyhow, Result};
+
+#[derive(Clone, Copy, Debug, PartialEq)]
+pub struct OwnerFilter {
+    uid: Option<u32>,
+    gid: Option<u32>,
+}
+
+impl OwnerFilter {
+    pub fn from_string(input: &str) -> Result<Self> {
+        let mut it = input.split(':');
+        let (fst, snd) = (it.next(), it.next());
+
+        let uid = match fst {
+            Some("") | None => None,
+            Some(s) => Some(s.parse()?),
+        };
+        let gid = match snd {
+            Some("") | None => None,
+            Some(s) => Some(s.parse()?),
+        };
+
+        if uid.is_none() && gid.is_none() {
+            Err(anyhow!(
+                "'{}' is not a valid user/group specifier. See 'fd --help'.",
+                input
+            ))
+        } else {
+            Ok(OwnerFilter { uid, gid })
+        }
+    }
+}
+
+#[cfg(test)]
+mod owner_parsing {
+    use super::OwnerFilter;
+
+    macro_rules! owner_tests {
+        ($($name:ident: $value:expr => $result:pat,)*) => {
+            $(
+                #[test]
+                fn $name() {
+                    let o = OwnerFilter::from_string($value);
+                    match o {
+                        $result => {},
+                        _ => panic!("{:?} does not match {}", o, stringify!($result)),
+                    }
+                }
+            )*
+        };
+    }
+
+    owner_tests! {
+        empty:      ""      => Err(_),
+        uid_only:   "5"     => Ok(OwnerFilter { uid: Some(5), gid: None   }),
+        uid_gid:    "9:3"   => Ok(OwnerFilter { uid: Some(9), gid: Some(3)}),
+        gid_only:   ":8"    => Ok(OwnerFilter { uid: None,    gid: Some(8)}),
+        colon_only: ":"     => Err(_),
+        trailing:   "5:"    => Ok(OwnerFilter { uid: Some(5), gid: None   }),
+    }
+}