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

@@ -380,6 +380,15 @@ impl RevsetExpression {
         })
     }
 
+    /// Ancestors of `self`, including `self` up to `generations` back.
+    pub fn ancestors_at(self: &Rc<RevsetExpression>, generations: u64) -> Rc<RevsetExpression> {
+        Rc::new(RevsetExpression::Ancestors {
+            heads: self.clone(),
+            generation: (generations - 1)..generations,
+            is_legacy: false,
+        })
+    }
+
     /// Children of `self`.
     pub fn children(self: &Rc<RevsetExpression>) -> Rc<RevsetExpression> {
         Rc::new(RevsetExpression::Descendants {
@@ -405,6 +414,15 @@ impl RevsetExpression {
         })
     }
 
+    /// Descendants of `self`, including `self` up to `generations` back.
+    pub fn descendants_at(self: &Rc<RevsetExpression>, generations: u64) -> Rc<RevsetExpression> {
+        Rc::new(RevsetExpression::Descendants {
+            roots: self.clone(),
+            generation: (generations - 1)..generations,
+            is_legacy: false,
+        })
+    }
+
     /// Commits that are descendants of `self` and ancestors of `heads`, both
     /// inclusive.
     pub fn dag_range_to(