feat: add Range::as_singleton (#186)

Co-authored-by: Zanie Blue <[email protected]> Co-authored-by: Andrew Gallant <[email protected]> We use this method for better error messages: https://github.com/astral-sh/uv/blob/8d721830db8ad75b8b7ef38edc0e346696c52e3d/crates/uv-resolver/src/pubgrub/report.rs#L92-L96 https://github.com/astral-sh/uv/blob/8d721830db8ad75b8b7ef38edc0e346696c52e3d/crates/uv-resolver/src/pubgrub/report.rs#L509-L520
pubgrub-rs · Mar 11, 2024 · a20c414 · a20c414
1 parent fdd9cc9
commit a20c414
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/range.rs b/src/range.rs
@@ -186,6 +186,21 @@ impl<V: Clone> Range<V> {
 }
 
 impl<V: Ord> Range<V> {
+    /// If the range includes a single version, return it.
+    /// Otherwise, returns [None].
+    pub fn as_singleton(&self) -> Option<&V> {
+        match self.segments.as_slice() {
+            [(Included(v1), Included(v2))] => {
+                if v1 == v2 {
+                    Some(v1)
+                } else {
+                    None
+                }
+            }
+            _ => None,
+        }
+    }
+
     /// Convert to something that can be used with
     /// [BTreeMap::range](std::collections::BTreeMap::range).
     /// All versions contained in self, will be in the output,