...

compiler/NubLang/Ast/Node.cs

@@ -29,9 +29,31 @@ public abstract class Node(List<Token> tokens)
     }
 }
 
+public abstract class TopLevelNode(List<Token> tokens) : Node(tokens);
+
+public class ImportNode(List<Token> tokens, IdentifierToken nameToken) : TopLevelNode(tokens)
+{
+    public IdentifierToken NameToken { get; } = nameToken;
+
+    public override IEnumerable<Node> Children()
+    {
+        return [];
+    }
+}
+
+public class ModuleNode(List<Token> tokens, IdentifierToken nameToken) : TopLevelNode(tokens)
+{
+    public IdentifierToken NameToken { get; } = nameToken;
+
+    public override IEnumerable<Node> Children()
+    {
+        return [];
+    }
+}
+
 #region Definitions
 
-public abstract class DefinitionNode(List<Token> tokens, IdentifierToken nameToken) : Node(tokens)
+public abstract class DefinitionNode(List<Token> tokens, IdentifierToken nameToken) : TopLevelNode(tokens)
 {
     public IdentifierToken NameToken { get; } = nameToken;
 }
@@ -75,6 +97,35 @@ public class FuncNode(List<Token> tokens, FuncPrototypeNode prototype, BlockNode
     }
 }
 
+public class StructFieldNode(List<Token> tokens, IdentifierToken nameToken, NubType type, ExpressionNode? value) : Node(tokens)
+{
+    public IdentifierToken NameToken { get; } = nameToken;
+    public NubType Type { get; } = type;
+    public ExpressionNode? Value { get; } = value;
+
+    public override IEnumerable<Node> Children()
+    {
+        if (Value != null)
+        {
+            yield return Value;
+        }
+    }
+}
+
+public class StructNode(List<Token> tokens, IdentifierToken name, NubStructType structType, List<StructFieldNode> fields) : DefinitionNode(tokens, name)
+{
+    public NubStructType StructType { get; } = structType;
+    public List<StructFieldNode> Fields { get; } = fields;
+
+    public override IEnumerable<Node> Children()
+    {
+        foreach (var field in Fields)
+        {
+            yield return field;
+        }
+    }
+}
+
 #endregion
 
 #region Statements