group unchecked blocks (#462)

Author: Janther

src/nodes/UncheckedStatement.js

@@ -1,11 +1,12 @@
 const {
   doc: {
-    builders: { concat }
+    builders: { concat, group }
   }
 } = require('prettier/standalone');
 
 const UncheckedStatement = {
-  print: ({ path, print }) => concat(['unchecked ', path.call(print, 'block')])
+  print: ({ path, print }) =>
+    group(concat(['unchecked ', path.call(print, 'block')]))
 };
 
 module.exports = UncheckedStatement;

tests/AllSolidityFeatures/AllSolidityFeatures.sol

@@ -520,4 +520,13 @@ contract WithUncheckedBlock {
     unchecked   {   x--;   }
     return x;
   }
-}
+  
+  function g() public pure returns (uint) {
+    uint x = 0;
+    unchecked   {   x--; return x; }
+  }
+
+  function mul(uint256 x, uint256 y) internal pure returns (uint256) {
+    unchecked { return x * y; }
+  }
+}

tests/AllSolidityFeatures/__snapshots__/jsfmt.spec.js.snap

@@ -528,8 +528,16 @@ contract WithUncheckedBlock {
     unchecked   {   x--;   }
     return x;
   }
-}
+  
+  function g() public pure returns (uint) {
+    uint x = 0;
+    unchecked   {   x--; return x; }
+  }
 
+  function mul(uint256 x, uint256 y) internal pure returns (uint256) {
+    unchecked { return x * y; }
+  }
+}
 =====================================output=====================================
 // Examples taken from the Solidity documentation online.
 
@@ -1098,11 +1106,21 @@ uint256 constant MULTIPLIER = 2**EXPONENT;
 
 contract WithUncheckedBlock {
     function f() public pure returns (uint256) {
+        uint256 x = 0;
+        unchecked {x--;}
+        return x;
+    }
+
+    function g() public pure returns (uint256) {
         uint256 x = 0;
         unchecked {
             x--;
+            return x;
         }
-        return x;
+    }
+
+    function mul(uint256 x, uint256 y) internal pure returns (uint256) {
+        unchecked {return x * y;}
     }
 }