Update Formatter.cs

Draco-lang · Oct 23, 2023 · c0508c0 · c0508c0
1 parent ccfd852
commit c0508c0
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/Draco.Compiler/Internal/Syntax/Formatting/Formatter.cs b/src/Draco.Compiler/Internal/Syntax/Formatting/Formatter.cs
@@ -45,6 +45,11 @@ public Formatter(FormatterSettings settings)
         this.Settings = settings;
     }
 
+    public override SyntaxNode VisitCompilationUnit(CompilationUnitSyntax node) => node.Update(this.AppendSequence(
+        node.Declarations,
+        Newline,
+        node.End));
+
     public override SyntaxNode VisitFunctionDeclaration(FunctionDeclarationSyntax node) => node.Update(this.AppendSequence(
         node.VisibilityModifier,
         Space,
@@ -165,7 +170,8 @@ private void AddNormalizedTrailingTrivia(SyntaxList<SyntaxTrivia> trivia)
             if (t.Kind is not TriviaKind.LineComment or TriviaKind.DocumentationComment) continue;
 
             // Indent the trivia
-            if (!first) this.EnsureIndentation(this.indentation);
+            if (first) this.EnsureSpace();
+            else this.EnsureIndentation(this.indentation);
             // Add comment
             this.currentTrivia.Add(t);
             // Add a newline after