Started if statement parsing

jmeaster30 · Jun 8, 2023 · bd2142c · bd2142c
1 parent cba5b75
commit bd2142c
Show file tree

Hide file tree

Showing 4 changed files with 142 additions and 8 deletions.
diff --git a/BlastPDF.Template/AST.cs b/BlastPDF.Template/AST.cs
@@ -445,7 +445,7 @@ public bool Is<T>()
 
     public IEnumerable<Diagnostic> GetErrors(string filepath)
     {
-        return new List<Diagnostic>();
+        return Contents.GetErrors(filepath);
     }
 }
 
@@ -462,7 +462,7 @@ public bool Is<T>()
 
     public IEnumerable<Diagnostic> GetErrors(string filepath)
     {
-        return new List<Diagnostic>();
+        return Contents.GetErrors(filepath);
     }
 }
 
@@ -479,7 +479,7 @@ public bool Is<T>()
 
     public IEnumerable<Diagnostic> GetErrors(string filepath)
     {
-        return new List<Diagnostic>();
+        return Contents.GetErrors(filepath);
     }
 }
 
@@ -517,15 +517,46 @@ public IEnumerable<Diagnostic> GetErrors(string filepath)
     }
 }
 
-/*public class BranchNode : IContentNode, IPageNode, IDocumentNode
+public class BranchContentNode : IContentNode
 {
-    
+    public Token IfToken { get; set; } = default!;
+    public IExpressionNode Expression { get; set; } = default!;
+    public Token ThenToken { get; set; } = default!;
+    public List<IContentNode> TrueContents { get; set; } = default!;
+    public Token? ElseToken { get; set; } = default!;
+    public List<IContentNode> FalseContents { get; set; } = default!;
+    public Token EndToken { get; set; } = default!;
+    public bool Is<T>()
+    {
+        return typeof(T) == typeof(BranchContentNode);
+    }
+
+    public IEnumerable<Diagnostic> GetErrors(string filepath)
+    {
+        return Expression.GetErrors(filepath).Concat(TrueContents.GetErrors(filepath))
+            .Concat(FalseContents.GetErrors(filepath));
+    }
 }
 
-public class LoopNode : IContentNode, IPageNode, IDocumentNode
+public class LoopContentNode : IContentNode
 {
-    
-}*/
+    public Token LoopToken { get; set; } = default!;
+    public Token? Identifier { get; set; } = default!;
+    public Token? InToken { get; set; } = default!;
+    public IExpressionNode Expression { get; set; } = default!;
+    public Token ThenToken { get; set; } = default!;
+    public List<IContentNode> Body { get; set; } = default!;
+    public Token EndToken { get; set; } = default!;
+    public bool Is<T>()
+    {
+        return typeof(T) == typeof(LoopContentNode);
+    }
+
+    public IEnumerable<Diagnostic> GetErrors(string filepath)
+    {
+        return Expression.GetErrors(filepath).Concat(Body.GetErrors(filepath));
+    }
+}
 
 public static class AstExtensions
 {

diff --git a/BlastPDF.Template/Parser.cs b/BlastPDF.Template/Parser.cs
@@ -958,6 +958,7 @@ private static (List<IContentNode>, int) ParseContentNodes(IReadOnlyList<Token>
             var (node, idx) = tokens[tokenIndex].Type switch
             {
                 TokenType.Text => ParseTextNode(tokens, tokenIndex),
+                TokenType.If => ParseIfContentStatement(tokens, tokenIndex),
                 _ => (default!, -1)
             };
 
@@ -1049,4 +1050,98 @@ private static (IContentNode, int) ParseTextOutput(IReadOnlyList<Token> tokens,
             Expression = expr
         }, tokenIndex + 1);
     }
+
+    private static (IContentNode, int) ParseIfContentStatement(IReadOnlyList<Token> tokens, int tokenIndex)
+    {
+        var ifToken = tokens[tokenIndex];
+        tokenIndex += 1;
+        if (tokenIndex >= tokens.Count)
+        { 
+            return (new ContentError
+            {
+                ErroredTokens = new List<Token> { ifToken },
+                Message = "Hit end of document while parsing an if statement node",
+                Severity = DiagnosticSeverity.Error,
+            }, tokenIndex);
+        }
+
+        var conditionToken = tokens[tokenIndex];
+        if (conditionToken.Type != TokenType.EmbeddedExpression)
+        {
+            var (errorTokens, nextIndex) = ConsumeUntilNextTokenType(tokens, tokenIndex, IsContentNodeToken);
+            return (new ContentError
+            {
+                ErroredTokens = errorTokens,
+                Message = $"Expected an expression but got a ({conditionToken.Type}, '{conditionToken.Lexeme}').",
+                Severity = DiagnosticSeverity.Error,
+            }, nextIndex);
+        }
+        tokenIndex += 1;
+        if (tokenIndex >= tokens.Count)
+        { 
+            return (new ContentError
+            {
+                ErroredTokens = new List<Token> { ifToken, conditionToken },
+                Message = "Hit end of document while parsing an if statement node",
+                Severity = DiagnosticSeverity.Error,
+            }, tokenIndex);
+        }
+
+        var thenToken = tokens[tokenIndex];
+        if (thenToken.Type != TokenType.Then)
+        {
+            var (errorTokens, nextIndex) = ConsumeUntilNextTokenType(tokens, tokenIndex, IsContentNodeToken);
+            return (new ContentError
+            {
+                ErroredTokens = errorTokens,
+                Message = $"Expected 'then' but got a ({thenToken.Type}, '{thenToken.Lexeme}').",
+                Severity = DiagnosticSeverity.Error,
+            }, nextIndex);
+        }
+        tokenIndex += 1;
+        if (tokenIndex >= tokens.Count)
+        { 
+            return (new ContentError
+            {
+                ErroredTokens = new List<Token> { ifToken, conditionToken, thenToken },
+                Message = "Hit end of document while parsing an if statement node",
+                Severity = DiagnosticSeverity.Error,
+            }, tokenIndex);
+        }
+
+        var (contents, idx) = ParseContentNodes(tokens, tokenIndex);
+        tokenIndex = idx;
+        if (tokenIndex >= tokens.Count)
+        { 
+            return (new ContentError
+            {
+                ErroredTokens = new List<Token> { ifToken, conditionToken, thenToken },
+                Message = "Hit end of document while parsing an if statement node",
+                Severity = DiagnosticSeverity.Error,
+            }, tokenIndex);
+        }
+
+        var endToken = tokens[tokenIndex];
+        if (endToken.Type == TokenType.End)
+        {
+            return (new BranchContentNode
+            {
+                IfToken = ifToken,
+                Expression = new ExpressionValue
+                {
+                    Value = conditionToken
+                },
+                ThenToken = thenToken,
+                TrueContents = contents,
+                EndToken = endToken,
+            }, tokenIndex + 1);
+        }
+
+        return (new ContentError
+        {
+            ErroredTokens = new List<Token> { endToken },
+            Message = "AAAAAAAAAAAAAAAAA",
+            Severity = DiagnosticSeverity.Error,
+        }, tokenIndex);
+    }
 }
diff --git a/BlastPDF/BlastPDF.csproj b/BlastPDF/BlastPDF.csproj
@@ -8,5 +8,9 @@
     <ProjectReference Include="..\BlastIMG\BlastIMG.csproj" />
     <ProjectReference Include="..\BlastSharp\BlastSharp.csproj" />
   </ItemGroup>
+
+  <ItemGroup>
+    <Folder Include="LayoutEngine\" />
+  </ItemGroup>
 
 </Project>
diff --git a/ShowCase/templates/TestLayout.bxpdf b/ShowCase/templates/TestLayout.bxpdf
@@ -26,6 +26,10 @@ page single
         text "The goal of this message is to be a long message that should wrap. I would like to make it so you can write paragraphs as long as you want and the lines will automatically wrap. Perhaps that should be configurable with a command. It probably should be on by default as well. I think newlines and tabs should be ignored honestly."
 
         text "This would show as a wholy new paragraph! My test variable is @{Test}"
+
+        if @{Condition} then
+            text "Conditionally render this"
+        end
     end
     footer
         text "good foot"