Better enum formatting and whitespace cleanup

src/dfmt.d

@@ -36,14 +36,26 @@ import std.array;
 
 int main(string[] args)
 {
-    if (args.length < 2)
+    ubyte[] buffer;
+    if (args.length == 1)
     {
-        writeln("File name is a required argument");
+        ubyte[4096] inputBuffer;
-        return 1;
+        ubyte[] b;
+        while (true)
+        {
+            b = stdin.rawRead(inputBuffer);
+            if (b.length)
+                buffer ~= b;
+            else
+                break;
         }
+    }
+    else
+    {
         File f = File(args[1]);
-    ubyte[] buffer = new ubyte[](f.size);
+        buffer = new ubyte[](f.size);
         f.rawRead(buffer);
+    }
     LexerConfig config;
     config.stringBehavior = StringBehavior.source;
     config.whitespaceBehavior = WhitespaceBehavior.skip;
@@ -54,7 +66,7 @@ int main(string[] args)
     ASTInformation astInformation;
     FormatterConfig formatterConfig;
     auto parseTokens = getTokensForParser(buffer, parseConfig, &cache);
-    auto mod = parseModule(parseTokens, args[1]);
+    auto mod = parseModule(parseTokens, args.length > 1 ? args[1] : "stdin");
     auto visitor = new FormatVisitor(&astInformation);
     visitor.visit(mod);
     astInformation.cleanup();
@@ -87,6 +99,8 @@ struct TokenFormatter
         assert (indentLevel >= 0);
     }
 
+private:
+
     void formatStep()
     {
         import std.range:assumeSorted;
@@ -312,18 +326,21 @@ struct TokenFormatter
             assert (false, str(current.type));
     }
 
+	/// Pushes a temporary indent level
     void pushIndent()
     {
         if (tempIndent == 0)
             tempIndent++;
     }
 
+	/// Pops a temporary indent level
     void popIndent()
     {
         if (tempIndent > 0)
             tempIndent--;
     }
 
+	/// Writes balanced braces
     void writeBraces()
     {
         import std.range : assumeSorted;
@@ -349,6 +366,9 @@ struct TokenFormatter
             }
             else if (current.type == tok!"}")
             {
+				// Silly hack to format enums better.
+                if (peekBackIs(tok!"identifier"))
+                    newline();
                 write("}");
                 depth--;
                 if (index < tokens.length &&
@@ -653,7 +673,7 @@ struct FormatterConfig
     uint columnHardLimit = 120;
 
     /// Use the One True Brace Style
-    BraceStyle braceStyle = BraceStyle.otbs;
+    BraceStyle braceStyle = BraceStyle.allman;
 }
 
 ///
@@ -700,6 +720,12 @@ final class FormatVisitor : ASTVisitor
         functionBody.accept(this);
     }
 
+    override void visit(const EnumBody enumBody)
+    {
+        astInformation.doubleNewlineLocations ~= enumBody.endLocation;
+        enumBody.accept(this);
+    }
+
     override void visit(const Unittest unittest_)
     {
         astInformation.doubleNewlineLocations ~= unittest_.blockStatement.endLocation;