diff --git a/lang-parser b/lang-parser index 910d1b4..d7c58c4 160000 --- a/lang-parser +++ b/lang-parser @@ -1 +1 @@ -Subproject commit 910d1b4d312002a769869064e83250513257b63e +Subproject commit d7c58c48cc7afc031e5dc350eb2a497c1bac5511 diff --git a/src/build_visitor.cpp b/src/build_visitor.cpp index fc4bec4..ae5526c 100644 --- a/src/build_visitor.cpp +++ b/src/build_visitor.cpp @@ -633,10 +633,9 @@ void BuildVisitor::Visit(ForLoop* node) { auto parse_node = current_node_; - std::string modifier = parse_node.NthChild(1).GetValue(); - if (modifier == "const") { + if (parse_node.ChildCount() < 7) { // no "var" node node->variable_modifier = utils::IsConstModifier::Const; - } else if (modifier == "var") { + } else { node->variable_modifier = utils::IsConstModifier::Var; } diff --git a/src/print_visitor.cpp b/src/print_visitor.cpp index 91b4a1a..e39c54e 100644 --- a/src/print_visitor.cpp +++ b/src/print_visitor.cpp @@ -334,7 +334,6 @@ void PrintVisitor::Visit(ForLoop* node) { out_ << "[For "; switch (node->variable_modifier) { case utils::IsConstModifier::Const: - out_ << "const"; break; case utils::IsConstModifier::Var: out_ << "var";