add: dfs test, and code fixes
This commit is contained in:
15
src/lower.rs
15
src/lower.rs
@@ -1352,9 +1352,18 @@ impl Parser {
|
||||
self.advance(); // KwDim
|
||||
if self.peek() == &Token::LParen {
|
||||
self.advance(); // (
|
||||
if let Ok(expr) = self.parse_expression() {
|
||||
self.eat(&Token::RParen);
|
||||
return VarKeyword::Dim(expr);
|
||||
if let Ok(rows) = self.parse_expression() {
|
||||
if self.eat(&Token::Colon) {
|
||||
// DIM(rows: cols) — 2-D array
|
||||
if let Ok(cols) = self.parse_expression() {
|
||||
self.eat(&Token::RParen);
|
||||
return VarKeyword::Dim2(rows, cols);
|
||||
}
|
||||
} else {
|
||||
// DIM(n) — 1-D array
|
||||
self.eat(&Token::RParen);
|
||||
return VarKeyword::Dim(rows);
|
||||
}
|
||||
}
|
||||
self.eat(&Token::RParen);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user