Add variant for inspecting None

Signed-off-by: Matthias Beyer <mail@beyermatthias.de>

1 files changed, 35 insertions, 0 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 8c2fe74..9dfe3cf 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -38,3 +38,38 @@ impl<F, T> OptionInspectRef<F, T> for Option<T>
         }
     }
 }
+
+
+pub trait OptionInspectNone<F>
+    where F: FnOnce(),
+{
+    fn inspect_none(self, f: F) -> Self;
+}
+
+pub trait OptionInspectNoneRef<F>
+    where F: FnOnce(),
+{
+    fn inspect_none(&self, f: F);
+}
+
+impl<F, T> OptionInspectNone<F> for Option<T>
+    where F: FnOnce()
+{
+    fn inspect_none(self, f: F) -> Self {
+        if let None = self.as_ref() {
+            (f)();
+        }
+
+        self
+    }
+}
+
+impl<F, T> OptionInspectNoneRef<F> for Option<T>
+    where F: FnOnce()
+{
+    fn inspect_none(&self, f: F) {
+        if let None = self {
+            (f)();
+        }
+    }
+}