Parse dynamic list size into ListDynamicSize node

compiler/include/compiler/ast/node_type.hpp

@@ -60,6 +60,7 @@ enum class NodeType {
     IfStatement,
     IntegerLiteralValue,
     ListAccessor,
+    ListDynamicSize,
     ListStatement,
     PassStatement,
     ProgramRoot,

compiler/include/compiler/frontend/parser/parser_context.hpp

@@ -1,11 +1,10 @@
 #pragma once
 
 #include <functional>
-#include <stack>
 #include <unordered_map>
 
 #include "compiler/ast/node.hpp"
-#include "compiler/ast/types.hpp"
+#include "compiler/ast/node_type.hpp"
 #include "compiler/utils/error_buffer.hpp"
 #include "compiler/utils/source_ref.hpp"
 
@@ -27,21 +26,21 @@ struct ParserContext {
         return *tokenIter;
     }
 
-    static ast::Node::Ptr pushChildNode(ast::Node::Ptr node, const ast::NodeType &nodeType,
+    static ast::Node::Ptr pushChildNode(const ast::Node::Ptr &node, ast::NodeType nodeType,
                                         const utils::SourceRef &ref) {
         auto &childNode = node->children.emplace_back(new ast::Node(nodeType, node));
         childNode->ref = ref;
         return childNode;
     }
 
-    static ast::Node::Ptr unshiftChildNode(ast::Node::Ptr node, const ast::NodeType &nodeType,
+    static ast::Node::Ptr unshiftChildNode(const ast::Node::Ptr &node, ast::NodeType nodeType,
                                            const utils::SourceRef &ref) {
         auto &childNode = node->children.emplace_front(new ast::Node(nodeType, node));
         childNode->ref = ref;
         return childNode;
     }
 
-    ast::Node::Ptr pushChildNode(const ast::NodeType &nodeType) {
+    ast::Node::Ptr pushChildNode(ast::NodeType nodeType) {
         return pushChildNode(node, nodeType, tokenIter->ref);
     }
 

compiler/lib/ast/node.cpp

@@ -200,6 +200,9 @@ void Node::dump(std::ostream &stream, int depth) const {
     case NodeType::ListAccessor:
         stream << "ListAccessor\n";
         break;
+    case NodeType::ListDynamicSize:
+        stream << "ListDynamicSize\n";
+        break;
     case NodeType::ForStatement:
         stream << "ForStatement\n";
         break;
@@ -221,12 +224,12 @@ void Node::dump(std::ostream &stream, int depth) const {
     default:
         stream << "Unknown\n";
     }
-    for (auto child : children)
+    for (const auto &child : children)
         child->dump(stream, depth + 1);
 }
 
 std::string Node::dump(int depth) const {
     std::stringstream str;
-    dump(str);
+    dump(str, depth);
     return str.str();
 }

compiler/lib/frontend/parser/parser.cpp

@@ -774,7 +774,18 @@ void parseListStatement(ParserContext &ctx) {
         ctx.goParentNode();
     }
     ctx.goNextToken();
-    ctx.goParentNode();
+    if (ctx.token().is(Special::EndOfExpression)) {
+        ctx.goParentNode();
+    } else if (ctx.token().is(Operator::Mult)) {
+        ctx.goNextToken();
+        ctx.node = ctx.pushChildNode(NodeType::ListDynamicSize);
+        ctx.node = ctx.pushChildNode(NodeType::Expression);
+        ctx.propagate();
+        ctx.goParentNode();
+        ctx.goParentNode();
+    } else {
+        ctx.pushError("Either end of line or '*' was expected");
+    }
     ctx.goParentNode();
 }