Adding Parentheses on consecutive equality operators

Author: Janther

src/parser.js

@@ -176,6 +176,10 @@ function parse(text, _parsers, options = _parsers) {
             '&'
           ]);
           break;
+        case '==':
+        case '!=':
+          ctx.left = tryHug(ctx.left, ['==', '!=']);
+          break;
         case '||':
           ctx.left = tryHug(ctx.left, ['&&']);
           ctx.right = tryHug(ctx.right, ['&&']);

tests/format/BinaryOperationHierarchy/__snapshots__/jsfmt.spec.js.snap

@@ -1011,10 +1011,10 @@ contract Group {
             veryVeryVeryLongUint256A ==
             veryVeryVeryLongUint256B ^ veryVeryVeryLongUint256C;
         resultBoolean =
-            veryVeryVeryLongUint256A == veryVeryVeryLongUint256B ==
+            (veryVeryVeryLongUint256A == veryVeryVeryLongUint256B) ==
             veryVeryVeryLongBooleanC;
         resultBoolean =
-            veryVeryVeryLongUint256A == veryVeryVeryLongUint256B !=
+            (veryVeryVeryLongUint256A == veryVeryVeryLongUint256B) !=
             veryVeryVeryLongBooleanC;
         resultBoolean =
             veryVeryVeryLongBooleanA ==
@@ -1065,10 +1065,10 @@ contract Group {
             veryVeryVeryLongUint256A !=
             veryVeryVeryLongUint256B ^ veryVeryVeryLongUint256C;
         resultBoolean =
-            veryVeryVeryLongUint256A != veryVeryVeryLongUint256B ==
+            (veryVeryVeryLongUint256A != veryVeryVeryLongUint256B) ==
             veryVeryVeryLongBooleanC;
         resultBoolean =
-            veryVeryVeryLongUint256A != veryVeryVeryLongUint256B !=
+            (veryVeryVeryLongUint256A != veryVeryVeryLongUint256B) !=
             veryVeryVeryLongBooleanC;
         resultBoolean =
             veryVeryVeryLongBooleanA !=
@@ -2497,12 +2497,12 @@ contract Indent {
             veryVeryVeryExtremelyExtremelyLongUint256B ^
                 veryVeryVeryExtremelyExtremelyLongUint256C;
         resultBoolean =
-            veryVeryVeryExtremelyExtremelyLongUint256A ==
-            veryVeryVeryExtremelyExtremelyLongUint256B ==
+            (veryVeryVeryExtremelyExtremelyLongUint256A ==
+                veryVeryVeryExtremelyExtremelyLongUint256B) ==
             veryVeryVeryExtremelyExtremelyLongBooleanC;
         resultBoolean =
-            veryVeryVeryExtremelyExtremelyLongUint256A ==
-            veryVeryVeryExtremelyExtremelyLongUint256B !=
+            (veryVeryVeryExtremelyExtremelyLongUint256A ==
+                veryVeryVeryExtremelyExtremelyLongUint256B) !=
             veryVeryVeryExtremelyExtremelyLongBooleanC;
         resultBoolean =
             veryVeryVeryExtremelyExtremelyLongBooleanA ==
@@ -2568,12 +2568,12 @@ contract Indent {
             veryVeryVeryExtremelyExtremelyLongUint256B ^
                 veryVeryVeryExtremelyExtremelyLongUint256C;
         resultBoolean =
-            veryVeryVeryExtremelyExtremelyLongUint256A !=
-            veryVeryVeryExtremelyExtremelyLongUint256B ==
+            (veryVeryVeryExtremelyExtremelyLongUint256A !=
+                veryVeryVeryExtremelyExtremelyLongUint256B) ==
             veryVeryVeryExtremelyExtremelyLongBooleanC;
         resultBoolean =
-            veryVeryVeryExtremelyExtremelyLongUint256A !=
-            veryVeryVeryExtremelyExtremelyLongUint256B !=
+            (veryVeryVeryExtremelyExtremelyLongUint256A !=
+                veryVeryVeryExtremelyExtremelyLongUint256B) !=
             veryVeryVeryExtremelyExtremelyLongBooleanC;
         resultBoolean =
             veryVeryVeryExtremelyExtremelyLongBooleanA !=

tests/format/Parentheses/LogicNoParentheses.sol

@@ -17,4 +17,20 @@ contract LogicNoParentheses {
     function andAnd(bool a, bool b, bool c) public pure returns (bool) {
         return a && b && c;
     }
+
+    function equalEqual(bool a, bool b, bool c) public pure returns (bool) {
+        return a == b == c;
+    }
+
+    function equalNotEqual(bool a, bool b, bool c) public pure returns (bool) {
+        return a == b != c;
+    }
+
+    function notEqualEqual(bool a, bool b, bool c) public pure returns (bool) {
+        return a != b == c;
+    }
+
+    function notEqualNotEqual(bool a, bool b, bool c) public pure returns (bool) {
+        return a != b != c;
+    }
 }

tests/format/Parentheses/__snapshots__/jsfmt.spec.js.snap

@@ -1208,6 +1208,22 @@ contract LogicNoParentheses {
     function andAnd(bool a, bool b, bool c) public pure returns (bool) {
         return a && b && c;
     }
+
+    function equalEqual(bool a, bool b, bool c) public pure returns (bool) {
+        return a == b == c;
+    }
+
+    function equalNotEqual(bool a, bool b, bool c) public pure returns (bool) {
+        return a == b != c;
+    }
+
+    function notEqualEqual(bool a, bool b, bool c) public pure returns (bool) {
+        return a != b == c;
+    }
+
+    function notEqualNotEqual(bool a, bool b, bool c) public pure returns (bool) {
+        return a != b != c;
+    }
 }
 
 =====================================output=====================================
@@ -1230,6 +1246,26 @@ contract LogicNoParentheses {
     function andAnd(bool a, bool b, bool c) public pure returns (bool) {
         return a && b && c;
     }
+
+    function equalEqual(bool a, bool b, bool c) public pure returns (bool) {
+        return (a == b) == c;
+    }
+
+    function equalNotEqual(bool a, bool b, bool c) public pure returns (bool) {
+        return (a == b) != c;
+    }
+
+    function notEqualEqual(bool a, bool b, bool c) public pure returns (bool) {
+        return (a != b) == c;
+    }
+
+    function notEqualNotEqual(
+        bool a,
+        bool b,
+        bool c
+    ) public pure returns (bool) {
+        return (a != b) != c;
+    }
 }
 
 ================================================================================