diff --git a/ablescript/src/ast.rs b/ablescript/src/ast.rs index 50f9ceb4..94dd46cc 100644 --- a/ablescript/src/ast.rs +++ b/ablescript/src/ast.rs @@ -160,6 +160,7 @@ pub enum Expr { index: Box>, }, Len(Box>), + Keys(Box>), Variable(String), } diff --git a/ablescript/src/interpret.rs b/ablescript/src/interpret.rs index f666a55e..ffd5bd33 100644 --- a/ablescript/src/interpret.rs +++ b/ablescript/src/interpret.rs @@ -195,7 +195,14 @@ impl ExecEnv { .unwrap_or(Value::Nul) } Len(expr) => Value::Int(self.eval_expr(expr)?.length()), - + Keys(expr) => Value::Cart( + self.eval_expr(expr)? + .into_cart() + .into_keys() + .enumerate() + .map(|(i, k)| (Value::Int(i as isize + 1), ValueRef::new(k))) + .collect(), + ), // TODO: not too happy with constructing an artificial // Ident here. Variable(name) => { diff --git a/ablescript/src/parser.rs b/ablescript/src/parser.rs index 5934e0ab..3b827124 100644 --- a/ablescript/src/parser.rs +++ b/ablescript/src/parser.rs @@ -294,6 +294,10 @@ impl<'source> Parser<'source> { } None => break Expr::Len(Box::new(expr)), }, + Token::GreaterThan if buf.is_none() => { + self.require(Token::RightBracket)?; + break Expr::Keys(Box::new(expr)); + } token => buf = Some(self.parse_expr(token, &mut buf)?), } })