Add Range::bounds (#16)

astral-sh · Mar 6, 2024 · e16f980 · e16f980
1 parent 3c2a410
commit e16f980
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/src/range.rs b/src/range.rs
@@ -124,6 +124,24 @@ impl<V> Range<V> {
     pub fn is_empty(&self) -> bool {
         self.segments.is_empty()
     }
+
+    /// Return all boundary versions of this range.
+    pub fn bounds(&self) -> impl Iterator<Item = &V> {
+        self.segments.iter().flat_map(|segment| {
+            let (v1, v2) = segment;
+            let v1 = match v1 {
+                Included(v) => Some(v),
+                Excluded(v) => Some(v),
+                Unbounded => None,
+            };
+            let v2 = match v2 {
+                Included(v) => Some(v),
+                Excluded(v) => Some(v),
+                Unbounded => None,
+            };
+            v1.into_iter().chain(v2)
+        })
+    }
 }
 
 impl<V: Clone> Range<V> {