Don't write a space after the return attribute

Fixes #521.

src/dfmt/formatter.d

@@ -224,7 +224,9 @@ private:
         else if (currentIs(tok!"return"))
         {
             writeToken();
-            if (!currentIs(tok!";") && !currentIs(tok!")") && !currentIs(tok!"{"))
+            if (!currentIs(tok!";") && !currentIs(tok!")") && !currentIs(tok!"{")
+                    && !currentIs(tok!"in") && !currentIs(tok!"out") && !currentIs(tok!"do")
+                    && (hasCurrent && tokens[index].text != "body"))
                 write(" ");
         }
         else if (currentIs(tok!"with"))

tests/allman/issue0521.d.ref

@@ -0,0 +1,17 @@
+public int f() return
+in (true)
+{
+    return 0;
+}
+
+public int g() return
+out (; true)
+{
+    return 0;
+}
+
+public int h() return
+body
+{
+    return 0;
+}

tests/issue0521.d

@@ -0,0 +1,17 @@
+public int f() return
+in (true)
+{
+    return 0;
+}
+
+public int g() return
+out (; true)
+{
+    return 0;
+}
+
+public int h() return
+body
+{
+    return 0;
+}

tests/otbs/issue0521.d.ref

@@ -0,0 +1,14 @@
+public int f() return
+in (true) {
+    return 0;
+}
+
+public int g() return
+out (; true) {
+    return 0;
+}
+
+public int h() return
+body {
+    return 0;
+}