1 files changed, 29 insertions, 0 deletions
diff --git a/third_party/rust/smart-default/src/util.rs b/third_party/rust/smart-default/src/util.rs
new file mode 100755
index 0000000000..452ac91f95
--- /dev/null
+++ b/third_party/rust/smart-default/src/util.rs
@@ -0,0 +1,29 @@
+use syn::spanned::Spanned;
+use syn::parse::Error;
+
+/// Return the value that fulfills the predicate if there is one in the slice. Panic if there is
+/// more than one.
+pub fn find_only<T, F>(iter: impl Iterator<Item = T>, pred: F) -> Result<Option<T>, Error>
+where T: Spanned,
+      F: Fn(&T) -> Result<bool, Error>,
+{
+    let mut result = None;
+    for item in iter {
+        if pred(&item)? {
+            if result.is_some() {
+                return Err(Error::new(item.span(), "Multiple defaults"));
+            }
+            result = Some(item);
+        }
+    }
+    Ok(result)
+}
+
+pub fn single_value<T>(mut it: impl Iterator<Item = T>) -> Option<T> {
+    if let Some(result) = it.next() {
+        if it.next().is_none() {
+            return Some(result)
+        }
+    }
+    None
+}