Add identity and power operators (#5790)

Author: jevanlingen

rewrite-groovy/src/main/java/org/openrewrite/groovy/GroovyParserVisitor.java

@@ -1021,6 +1021,18 @@ public void visitBinaryExpression(BinaryExpression binary) {
                     case "?=":
                         gBinaryOp = G.Binary.Type.ElvisAssignment;
                         break;
+                    case "**":
+                        gBinaryOp = G.Binary.Type.Power;
+                        break;
+                    case "**=":
+                        gBinaryOp = G.Binary.Type.PowerAssignment;
+                        break;
+                    case "===":
+                        gBinaryOp = G.Binary.Type.IdentityEquals;
+                        break;
+                    case "!==":
+                        gBinaryOp = G.Binary.Type.IdentityNotEquals;
+                        break;
                 }
 
                 cursor += binary.getOperation().getText().length();

rewrite-groovy/src/main/java/org/openrewrite/groovy/GroovyPrinter.java

@@ -172,6 +172,18 @@ public J visitBinary(G.Binary binary, PrintOutputCapture<P> p) {
             case ElvisAssignment:
                 keyword = "?=";
                 break;
+            case Power:
+                keyword = "**";
+                break;
+            case PowerAssignment:
+                keyword = "**=";
+                break;
+            case IdentityEquals:
+                keyword = "===";
+                break;
+            case IdentityNotEquals:
+                keyword = "!==";
+                break;
         }
         beforeSyntax(binary, GSpace.Location.BINARY_PREFIX, p);
         visit(binary.getLeft(), p);

rewrite-groovy/src/main/java/org/openrewrite/groovy/tree/G.java

@@ -796,7 +796,11 @@ public enum Type {
             NotIn,
             Access,
             Spaceship,
-            ElvisAssignment
+            ElvisAssignment,
+            Power,
+            PowerAssignment,
+            IdentityEquals,
+            IdentityNotEquals
         }
 
         public Padding getPadding() {

rewrite-groovy/src/test/java/org/openrewrite/groovy/tree/BinaryTest.java

@@ -281,13 +281,49 @@ void spaceShipOperator() {
         rewriteRun(
           groovy(
             """
-              def justPrint(){
-                  println(1 <=> 2)
-                  println('a' <=> 'z')
-                  def a = 'tiger'
-                  def b = 'cheetah'
-                  println(a <=> b)
-              }
+              println(1 <=> 2)
+              println('a' <=> 'z')
+              def a = 'tiger'
+              def b = 'cheetah'
+              println(a <=> b)
+              """
+          ));
+    }
+
+    @Test
+    void powerOperator() {
+        rewriteRun(
+          groovy(
+            """
+              4 ** 3
+              """
+          ));
+    }
+
+    @Test
+    void powerAssigmentOperator() {
+        rewriteRun(
+          groovy(
+            """
+              def f = 3
+              f **= 2
+              """
+          ));
+    }
+
+    @Test
+    void identicalOperators() {
+        rewriteRun(
+          groovy(
+            """
+              class Creature {}
+              
+              def cat = new Creature()
+              def copyCat = cat
+              def lion = new Creature()
+              
+              assert cat === copyCat
+              assert cat !== lion
               """
           ));
     }