From f9b199728354b075dc0ada2e0cf75f94a78c1568 Mon Sep 17 00:00:00 2001
From: Hackerpilot <briancschott@gmail.com>
Date: Fri, 9 Oct 2015 14:21:30 -0700
Subject: [PATCH] Fix #190

---
 src/dfmt/formatter.d         | 58 +++++++++++++++++++++++++-----------
 tests/allman/issue0190.d.ref |  8 +++++
 tests/issue0190.d            |  8 +++++
 tests/otbs/issue0190.d.ref   |  6 ++++
 4 files changed, 63 insertions(+), 17 deletions(-)
 create mode 100644 tests/allman/issue0190.d.ref
 create mode 100644 tests/issue0190.d
 create mode 100644 tests/otbs/issue0190.d.ref

diff --git a/src/dfmt/formatter.d b/src/dfmt/formatter.d
index c0e5f7a..e2252ce 100644
--- a/src/dfmt/formatter.d
+++ b/src/dfmt/formatter.d
@@ -143,6 +143,9 @@ private:
     /// True if a space should be placed when parenDepth reaches zero
     bool spaceAfterParens;
 
+    /// True if we're in an ASM block
+    bool inAsm;
+
     void formatStep()
     {
         assert(index < tokens.length);
@@ -202,14 +205,35 @@ private:
         {
             formatBlockHeader();
         }
-		else if (currentIs(tok!"do"))
-		{
-			formatBlockHeader();
-		}
+        else if (currentIs(tok!"do"))
+        {
+            formatBlockHeader();
+        }
         else if (currentIs(tok!"else"))
         {
             formatElse();
         }
+        else if (currentIs(tok!"asm"))
+        {
+            formatKeyword();
+            while (index < tokens.length && !currentIs(tok!"{"))
+                formatStep();
+            if (index < tokens.length)
+            {
+                int depth = 1;
+                formatStep();
+                inAsm = true;
+                while (index < tokens.length)
+                {
+                    if (currentIs(tok!"{"))
+                        ++depth;
+                    else if (currentIs(tok!"}"))
+                        --depth;
+                    formatStep();
+                }
+                inAsm = false;
+            }
+        }
         else if (isKeyword(current.type))
         {
             formatKeyword();
@@ -228,7 +252,9 @@ private:
         {
             writeToken();
             if (index < tokens.length && (currentIs(tok!"identifier")
-                    || isBasicType(current.type) || currentIs(tok!"@") || currentIs(tok!"if")))
+                    || isBasicType(current.type) || currentIs(tok!"@")
+                    || currentIs(tok!"if") || isNumberLiteral(tokens[index].type)
+                    || (inAsm && peekBack2Is(tok!";") && currentIs(tok!"["))))
             {
                 write(" ");
             }
@@ -699,8 +725,6 @@ private:
 
     void formatBlockHeader()
     {
-		//import std.stdio:stderr;
-		//stderr.writeln(__FUNCTION__);
         immutable bool a = !currentIs(tok!"version") && !currentIs(tok!"debug");
         immutable bool b = a
             || astInformation.conditionalWithElseLocations.canFindIndex(current.index);
@@ -712,11 +736,11 @@ private:
         if (shouldPushIndent)
             indents.push(current.type);
         writeToken();
-		if (currentIs(tok!"("))
-		{
-			write(" ");
-			writeParens(false);
-		}
+        if (currentIs(tok!"("))
+        {
+            write(" ");
+            writeParens(false);
+        }
         if (currentIs(tok!"switch") || (currentIs(tok!"final") && peekIs(tok!"switch")))
             write(" ");
         else if (currentIs(tok!"comment"))
@@ -1399,9 +1423,9 @@ const pure @safe @nogc:
         return peekImplementation(tokenType, -1, ignoreComments);
     }
 
-	bool peekBackIsKeyword(bool ignoreComments = true)
-	{
-		if (index == 0)
+    bool peekBackIsKeyword(bool ignoreComments = true)
+    {
+        if (index == 0)
             return false;
         auto i = index - 1;
         if (ignoreComments)
@@ -1411,8 +1435,8 @@ const pure @safe @nogc:
                     return false;
                 i--;
             }
-		return isKeyword(tokens[i].type);
-	}
+        return isKeyword(tokens[i].type);
+    }
 
     bool peekBackIsOneOf(bool ignoreComments, IdType[] tokenTypes...)
     {
diff --git a/tests/allman/issue0190.d.ref b/tests/allman/issue0190.d.ref
new file mode 100644
index 0000000..038b97f
--- /dev/null
+++ b/tests/allman/issue0190.d.ref
@@ -0,0 +1,8 @@
+unittest
+{
+    asm
+    {
+        dl 12345;
+        movdqu [R8], XMM0;
+    }
+}
diff --git a/tests/issue0190.d b/tests/issue0190.d
new file mode 100644
index 0000000..c0ee5ce
--- /dev/null
+++ b/tests/issue0190.d
@@ -0,0 +1,8 @@
+unittest
+{
+	asm
+	{
+		dl 12345;
+		movdqu [R8], XMM0;
+	}
+}
diff --git a/tests/otbs/issue0190.d.ref b/tests/otbs/issue0190.d.ref
new file mode 100644
index 0000000..ed0d398
--- /dev/null
+++ b/tests/otbs/issue0190.d.ref
@@ -0,0 +1,6 @@
+unittest {
+    asm {
+        dl 12345;
+        movdqu [R8], XMM0;
+    }
+}