wip: enums

compiler/NubLang/Parsing/Parser.cs

@@ -8,10 +8,10 @@ namespace NubLang.Parsing;
 public sealed class Parser
 {
     private readonly List<Diagnostic> _diagnostics = [];
+    private readonly HashSet<string> _templateArguments = [];
     private List<Token> _tokens = [];
     private int _tokenIndex;
     private string _moduleName = string.Empty;
-    private HashSet<string> _templateArguments = [];
 
     private Token? CurrentToken => _tokenIndex < _tokens.Count ? _tokens[_tokenIndex] : null;
     private bool HasToken => CurrentToken != null;
@@ -86,10 +86,11 @@ public sealed class Parser
                 }
 
                 var keyword = ExpectSymbol();
-                DefinitionSyntax definition = keyword.Symbol switch
+                var definition = keyword.Symbol switch
                 {
                     Symbol.Func => ParseFunc(startIndex, exported, null),
                     Symbol.Struct => ParseStruct(startIndex, exported),
+                    Symbol.Enum => ParseEnum(startIndex, exported),
                     _ => throw new ParseException(Diagnostic
                         .Error($"Expected 'func' or 'struct' but found '{keyword.Symbol}'")
                         .WithHelp("Valid definition keywords are 'func' and 'struct'")
@@ -228,6 +229,36 @@ public sealed class Parser
         return new StructSyntax(GetTokens(startIndex), name.Value, exported, fields, funcs);
     }
 
+    private EnumSyntax ParseEnum(int startIndex, bool exported)
+    {
+        var name = ExpectIdentifier();
+
+        TypeSyntax? type = null;
+        if (TryExpectSymbol(Symbol.Colon))
+        {
+            type = ParseType();
+        }
+
+        var values = new List<EnumValueSyntax>();
+
+        ExpectSymbol(Symbol.OpenBrace);
+        while (!TryExpectSymbol(Symbol.CloseBrace))
+        {
+            var valueStartIndex = _tokenIndex;
+            var valueName = ExpectIdentifier().Value;
+
+            ExpressionSyntax? valueValue = null;
+            if (TryExpectSymbol(Symbol.Assign))
+            {
+                valueValue = ParseExpression();
+            }
+
+            values.Add(new EnumValueSyntax(GetTokens(valueStartIndex), valueName, valueValue));
+        }
+
+        return new EnumSyntax(GetTokens(startIndex), name.Value, exported, type, values);
+    }
+
     private StatementSyntax ParseStatement()
     {
         var startIndex = _tokenIndex;

compiler/NubLang/Parsing/Syntax/DefinitionSyntax.cs

@@ -16,4 +16,8 @@ public record StructFuncSyntax(List<Token> Tokens, string Name, string? Hook, Fu
 
 public record StructSyntax(List<Token> Tokens, string Name, bool Exported, List<StructFieldSyntax> Fields, List<StructFuncSyntax> Functions) : DefinitionSyntax(Tokens, Name, Exported);
 
-public record StructTemplateSyntax(List<Token> Tokens, List<string> TemplateArguments, string Name, bool Exported, List<StructFieldSyntax> Fields, List<StructFuncSyntax> Functions) : DefinitionSyntax(Tokens, Name, Exported);
+public record StructTemplateSyntax(List<Token> Tokens, List<string> TemplateArguments, string Name, bool Exported, List<StructFieldSyntax> Fields, List<StructFuncSyntax> Functions) : DefinitionSyntax(Tokens, Name, Exported);
+
+public record EnumValueSyntax(List<Token> Tokens, string Name, ExpressionSyntax? Value) : SyntaxNode(Tokens);
+
+public record EnumSyntax(List<Token> Tokens, string Name, bool Exported, TypeSyntax? Type, List<EnumValueSyntax> Values) : DefinitionSyntax(Tokens, Name, Exported);

compiler/NubLang/Tokenization/Token.cs

@@ -82,4 +82,5 @@ public enum Symbol
     Export,
     Defer,
     At,
+    Enum,
 }

compiler/NubLang/Tokenization/Tokenizer.cs

@@ -24,6 +24,7 @@ public sealed class Tokenizer
         ["export"] = Symbol.Export,
         ["import"] = Symbol.Import,
         ["defer"] = Symbol.Defer,
+        ["enum"] = Symbol.Enum,
     };
 
     private static readonly Dictionary<char[], Symbol> Symbols = new()