add visitor walk for type in const and var declaration

pmqtt · Jan 7, 2025 · b7ee5f5 · b7ee5f5
1 parent 140b658
commit b7ee5f5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/ast/ConstantDeclarationNode.cpp b/src/ast/ConstantDeclarationNode.cpp
@@ -10,6 +10,7 @@ ConstantDeclarationNode::ConstantDeclarationNode(const std::string& var_name, co
 void ConstantDeclarationNode::accept(const std::shared_ptr<Visitor>& v) {
   v->begin(shared_from_this());
   auto v_next = v->visit(shared_from_this());
+  _type->accept(v_next);
   _expression->accept(v_next);
   v->end(shared_from_this());
 }
diff --git a/src/ast/VariableDeclarationNode.cpp b/src/ast/VariableDeclarationNode.cpp
@@ -11,6 +11,7 @@ VariableDeclarationNode::VariableDeclarationNode(const std::string& var_name, co
 void VariableDeclarationNode::accept(const std::shared_ptr<Visitor>& v) {
   v->begin(shared_from_this());
   auto v_next = v->visit(shared_from_this());
+  _type->accept(v_next);
   _expression->accept(v_next);
   v->end(shared_from_this());
 }