From 14024a57252afdd7008646c5780421f22ee07bf8 Mon Sep 17 00:00:00 2001
From: Prajwal S N <prajwalnadig21@gmail.com>
Date: Wed, 13 Dec 2023 18:40:26 +0530
Subject: [PATCH] feat: add `dfmt_compact_labeled_statements`

Signed-off-by: Prajwal S N <prajwalnadig21@gmail.com>
---
 src/dfmt/ast.d       | 5 ++++-
 src/dfmt/formatter.d | 4 ++--
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/dfmt/ast.d b/src/dfmt/ast.d
index 1d530ae..a2055d1 100644
--- a/src/dfmt/ast.d
+++ b/src/dfmt/ast.d
@@ -1896,7 +1896,10 @@ extern (C++) class FormatVisitor : SemanticTimeTransitiveVisitor
         {
             write(s.ident.toString());
             write(':');
-            newline();
+            if (config.dfmt_compact_labeled_statements)
+                write(' ');
+            else
+                newline();
             if (s.statement)
                 writeStatement(s.statement);
         }
diff --git a/src/dfmt/formatter.d b/src/dfmt/formatter.d
index d17ccfc..7081c03 100644
--- a/src/dfmt/formatter.d
+++ b/src/dfmt/formatter.d
@@ -33,9 +33,9 @@ bool format(string source_desc, ubyte[] buffer, File.LockingTextWriter output,
     Config* formatterConfig)
 {
     initDMD();
-    auto module_ = parseModule(source_desc);
+    auto module_ = parseModule(source_desc)[0];
     scope v = new FormatVisitor(output, formatterConfig);
-    v.visit(module_[0]);
+    v.visit(module_);
 
     return true;
 }