diff options
Diffstat (limited to 'src/eval/args.rs')
| -rw-r--r-- | src/eval/args.rs | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/src/eval/args.rs b/src/eval/args.rs index 129da463..66ca4a85 100644 --- a/src/eval/args.rs +++ b/src/eval/args.rs @@ -61,6 +61,17 @@ impl Args { Ok(None) } + /// Consume n positional arguments if possible. + pub fn consume(&mut self, n: usize) -> SourceResult<EcoVec<Arg>> { + if n > self.items.len() { + bail!(self.span, "not enough arguments"); + } + let vec = self.items.to_vec(); + let (left, right) = vec.split_at(n); + self.items = right.into(); + return Ok(left.into()); + } + /// Consume and cast the first positional argument. /// /// Returns a `missing argument: {what}` error if no positional argument is |
