From 995c21ab87e3bfe1b56ce80a36bc499a7485b83e Mon Sep 17 00:00:00 2001
From: Eugen Wissner <belka@caraus.de>
Date: Sun, 13 Jun 2021 06:42:08 +0200
Subject: [PATCH] Format do like body after function attributes

---
 src/dfmt/formatter.d         | 8 ++++----
 tests/allman/issue0528.d.ref | 4 ++++
 tests/issue0528.d            | 4 ++++
 tests/otbs/issue0528.d.ref   | 3 +++
 4 files changed, 15 insertions(+), 4 deletions(-)
 create mode 100644 tests/allman/issue0528.d.ref
 create mode 100644 tests/issue0528.d
 create mode 100644 tests/otbs/issue0528.d.ref

diff --git a/src/dfmt/formatter.d b/src/dfmt/formatter.d
index 0a355db..1165fcb 100644
--- a/src/dfmt/formatter.d
+++ b/src/dfmt/formatter.d
@@ -274,6 +274,10 @@ private:
             else
                 formatBlockHeader();
         }
+        else if ((current.text == "body" || current == tok!"do") && peekBackIsFunctionDeclarationEnding())
+        {
+            formatKeyword();
+        }
         else if (currentIs(tok!"do"))
         {
             formatBlockHeader();
@@ -321,10 +325,6 @@ private:
                 inlineElse = true;
             formatKeyword();
         }
-        else if (current.text == "body" && peekBackIsFunctionDeclarationEnding())
-        {
-            formatKeyword();
-        }
         else if (isBasicType(current.type))
         {
             writeToken();
diff --git a/tests/allman/issue0528.d.ref b/tests/allman/issue0528.d.ref
new file mode 100644
index 0000000..3592591
--- /dev/null
+++ b/tests/allman/issue0528.d.ref
@@ -0,0 +1,4 @@
+void f() return
+do
+{
+}
diff --git a/tests/issue0528.d b/tests/issue0528.d
new file mode 100644
index 0000000..3592591
--- /dev/null
+++ b/tests/issue0528.d
@@ -0,0 +1,4 @@
+void f() return
+do
+{
+}
diff --git a/tests/otbs/issue0528.d.ref b/tests/otbs/issue0528.d.ref
new file mode 100644
index 0000000..304b696
--- /dev/null
+++ b/tests/otbs/issue0528.d.ref
@@ -0,0 +1,3 @@
+void f() return
+do {
+}