Fix #7

UnderminersTeam · Aug 12, 2024 · 409138d · 409138d
1 parent afac234
commit 409138d
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/Underanalyzer/Decompiler/AST/Nodes/EnumDeclNode.cs b/Underanalyzer/Decompiler/AST/Nodes/EnumDeclNode.cs
@@ -34,6 +34,10 @@ public void Print(ASTPrinter printer)
     {
         printer.Write("enum ");
         printer.Write(Enum.Name);
+        if (printer.Context.Settings.OpenBlockBraceOnSameLine)
+        {
+            printer.Write(' ');
+        }
         printer.OpenBlock();
 
         // Sort values of enum by value

diff --git a/UnderanalyzerTest/DecompileContext.DecompileToString.Settings.cs b/UnderanalyzerTest/DecompileContext.DecompileToString.Settings.cs
@@ -1119,4 +1119,27 @@ pop.v.i self.d
             }
         );
     }
+
+    [Fact]
+    public void TestEnumOpenBraceSameLine()
+    {
+        TestUtil.VerifyDecompileResult(
+            """
+            push.l 0
+            pop.v.l self.a
+            """,
+            """
+            a = UnknownEnum.Value_0;
+
+            enum UnknownEnum {
+                Value_0
+            }
+            """,
+            null,
+            new DecompileSettings()
+            {
+                OpenBlockBraceOnSameLine = true
+            }
+        );
+    }
 }