Add section directives

2026-07-27 13:11:59 -07:00 · 2024-02-05 09:23:46 -06:00
parent 804bca823b
commit 5a0473e499
8 changed files with 192 additions and 27 deletions
@@ -0,0 +1,52 @@
+using Microsoft.Iris.Asm.Models;
+using Sprache;
+
+namespace Microsoft.Iris.Asm;
+
+partial class Lexer
+{
+    private static IResult<IDirective> ParseDirective(IInput input)
+    {
+        input = ConsumeWhitespace(input);
+        var line = input.Line;
+        var column = input.Column;
+
+        var directiveBeginResult = Parse.String(".")(input);
+        input = directiveBeginResult.Remainder;
+        if (!directiveBeginResult.WasSuccessful)
+            return Result.Failure<IDirective>(input, "Invalid directive", ["Expected '.', followed by an identifier"]);
+
+        var directiveIdResult = Parse.Letter.AtLeastOnce().Text()(input);
+        input = directiveIdResult.Remainder;
+        if (!directiveIdResult.WasSuccessful)
+            return Result.Failure<IDirective>(input, "Invalid directive", ["Expected an identifier"]);
+
+        IDirective directive;
+        switch (directiveIdResult.Value.ToUpperInvariant())
+        {
+            case "IMPORT":
+                return ParseImportAsDirective(input);
+
+            case "SECTION":
+                if (StatementEnd(input).WasSuccessful)
+                    return Result.Failure<IImport>(input, "Invalid section directive", ["Expected a section name"]);
+
+                var sectionNameResult = Parse.Letter.AtLeastOnce().Token().Text()(input);
+                input = sectionNameResult.Remainder;
+                if (!sectionNameResult.WasSuccessful)
+                    return Result.Failure<IImport>(input, "Invalid section name", ["Expected a section name containing only letters"]);
+
+                directive = new SectionDirective(sectionNameResult.Value)
+                {
+                    Line = line,
+                    Column = column,
+                };
+                break;
+
+            default:
+                return Result.Failure<IImport>(input, $"Unknown import type '{directiveIdResult.Value}'", ["Expected 'import' or 'section'"]);
+        }
+
+        return Result.Success(directive, input);
+    }
+}