From 8dda7e21e007762d2e26fd3776136c8bc80d7625 Mon Sep 17 00:00:00 2001
From: Philip Metzger <philipmetzger@bluewin.ch>
Date: Thu, 6 Jul 2023 23:57:24 +0200
Subject: [PATCH] revsets: Add `descendants_at` and `ancestors_at`

They are needed for `next` and `prev`.
They behave symmetrically for parents and children.
---
 lib/src/revset.rs | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/lib/src/revset.rs b/lib/src/revset.rs
index 669d8f88c..04e267cb6 100644
--- a/lib/src/revset.rs
+++ b/lib/src/revset.rs
@@ -444,6 +444,15 @@ impl RevsetExpression {
         })
     }
 
+    /// Ancestors of `self`, including `self` until `generation` back.
+    pub fn ancestors_at(self: &Rc<RevsetExpression>, generation: u64) -> Rc<RevsetExpression> {
+        Rc::new(RevsetExpression::Ancestors {
+            heads: self.clone(),
+            generation: generation..(generation + 1),
+            is_legacy: false,
+        })
+    }
+
     /// Children of `self`.
     pub fn children(self: &Rc<RevsetExpression>) -> Rc<RevsetExpression> {
         Rc::new(RevsetExpression::Descendants {
@@ -469,6 +478,15 @@ impl RevsetExpression {
         })
     }
 
+    /// Descendants of `self`, including `self` until `generation` ahead.
+    pub fn descendants_at(self: &Rc<RevsetExpression>, generation: u64) -> Rc<RevsetExpression> {
+        Rc::new(RevsetExpression::Descendants {
+            roots: self.clone(),
+            generation: generation..(generation + 1),
+            is_legacy: false,
+        })
+    }
+
     /// Commits that are descendants of `self` and ancestors of `heads`, both
     /// inclusive.
     pub fn dag_range_to(