Custom Recipe and test cases for changing exception type

Author: bhavanapidapa

src/main/java/org/openrewrite/java/migrate/ArrayStoreExceptionToTypeNotPresentException.java

@@ -0,0 +1,61 @@
+package org.openrewrite.java.migrate;
+
+import org.openrewrite.ExecutionContext;
+import org.openrewrite.Recipe;
+import org.openrewrite.TreeVisitor;
+import org.openrewrite.java.ChangeType;
+import org.openrewrite.java.JavaVisitor;
+import org.openrewrite.java.MethodMatcher;
+import org.openrewrite.java.tree.Expression;
+import org.openrewrite.java.tree.J;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.regex.Pattern;
+
+public class ArrayStoreExceptionToTypeNotPresentException extends Recipe {
+    @Override
+    public String getDisplayName() {
+        return "Handle TypeNotPresentException instead of ArrayStoreException";
+    }
+
+    @Override
+    public String getDescription() {
+        return "This recipe replaces catch blocks for ArrayStoreException around getAnnotation() with TypeNotPresentException or both exceptions, to ensure compatibility with Java 11+.";
+    }
+
+    @Override
+    public TreeVisitor<?, ExecutionContext> getVisitor() {
+        final MethodMatcher classGetAnnotationMethod = new MethodMatcher("java.lang.Class getAnnotation(java.lang.Class)");
+        return new JavaVisitor<ExecutionContext>() {
+
+            @Override
+            public J visitTry(J.Try tryStmt, ExecutionContext ctx) {
+                boolean flag = false;
+                if (containsGetAnnotation(tryStmt)) {
+                    List<J.Try.Catch> updatedCatches = new ArrayList<>();
+                    for (J.Try.Catch catchClause : tryStmt.getCatches()) {
+                        if (catchClause.getParameter().getType() != null && catchClause.getParameter().getType().isAssignableFrom(Pattern.compile("java.lang.ArrayStoreException"))) {
+                            J.Try.Catch updatedCatch = (J.Try.Catch) new ChangeType("java.lang.ArrayStoreException", "java.lang.TypeNotPresentException", true).getVisitor().visit(catchClause, ctx);
+                            updatedCatches.add(updatedCatch);
+                            flag = true;
+                        } else {
+                            updatedCatches.add(catchClause);
+                        }
+                    }
+                    if (flag)
+                        tryStmt = tryStmt.withCatches(updatedCatches);
+                }
+                return super.visitTry(tryStmt, ctx);
+            }
+
+            private boolean containsGetAnnotation(J.Try tryStmt) {
+                return tryStmt.getBody().getStatements().stream()
+                        .filter(Expression.class::isInstance)
+                        .map(Expression.class::cast)
+                        .anyMatch(classGetAnnotationMethod::matches);
+            }
+        };
+    }
+}
+

src/main/java/org/openrewrite/java/migrate/HandleTypeNotPresentExceptionInsteadOfArrayStoreException.java

@@ -73,7 +73,7 @@ recipeList:
   - org.openrewrite.scala.migrate.UpgradeScala_2_12
   - org.openrewrite.java.migrate.ReplaceComSunAWTUtilitiesMethods
   - org.openrewrite.java.migrate.ReplaceLocalizedStreamMethods
-  - org.openrewrite.java.migrate.HandleTypeNotPresentExceptionInsteadOfArrayStoreException
+  - org.openrewrite.java.migrate.ArrayStoreExceptionToTypeNotPresentException
 
 ---
 type: specs.openrewrite.org/v1beta/recipe

src/main/resources/META-INF/rewrite/java-version-11.yml

@@ -1,29 +1,36 @@
 package org.openrewrite.java.migrate;
 
 import org.junit.jupiter.api.Test;
-import org.openrewrite.java.JavaParser;
 import org.openrewrite.test.RecipeSpec;
 import org.openrewrite.test.RewriteTest;
 
 import static org.openrewrite.java.Assertions.java;
-import static org.openrewrite.java.Assertions.javaVersion;
 
 public class TypeNotPresentExceptionTest implements RewriteTest {
     @Override
     public void defaults(RecipeSpec spec) {
-        spec.recipe(new HandleTypeNotPresentExceptionInsteadOfArrayStoreException())
-          .parser(JavaParser.fromJavaVersion()
-            //language=java
-            .dependsOn(
+        spec.recipe(new ArrayStoreExceptionToTypeNotPresentException());
+    }
+
+    @Test
+    void ArrayStoreExceptionTest() {
+        rewriteRun(
+          //language=java
+          java(
             """
               import java.lang.annotation.*;
               import java.util.*;
-          
+
               public class Test {
                   public void testMethod() {
                       try {
                           Object o = "test";
                           o.getClass().getAnnotation(Override.class);
+                      } catch (ArrayStoreException e) {
+                          System.out.println("Caught Exception");
+                      }
+                      try {
+                          Object.class.getAnnotation(Override.class);
                       } catch (ArrayStoreException e) {
                           System.out.println("Caught ArrayStoreException");
                       }
@@ -33,14 +40,19 @@ public void testMethod() {
             """
               import java.lang.annotation.*;
               import java.util.*;
-              
+
               public class Test {
                   public void testMethod() {
                       try {
                           Object o = "test";
                           o.getClass().getAnnotation(Override.class);
                       } catch (TypeNotPresentException e) {
-                          System.out.println("Caught TypeNotPresentException");
+                          System.out.println("Caught Exception");
+                      }
+                      try {
+                          Object.class.getAnnotation(Override.class);
+                      } catch (TypeNotPresentException e) {
+                          System.out.println("Caught ArrayStoreException");
                       }
                   }
               }
@@ -49,24 +61,23 @@ public void testMethod() {
         );
     }
 
-
     @Test
-    void replaceGetLocalizedOutputStream() {
+    void rOtherExceptionsTest() {
         rewriteRun(
           //language=java
           java(
             """
-          public class Test {
-              public void testMethod() {
-                  try {
-                      Object o = "test";
-                      o.getClass().getAnnotation(Override.class);
-                  } catch (NullPointerException e) {
-                      System.out.println("Caught NullPointerException");
+              public class Test {
+                  public void testMethod() {
+                      try {
+                          Object o = "test";
+                          o.getClass().getAnnotation(Override.class);
+                      } catch (NullPointerException e) {
+                          System.out.println("Caught Exception");
+                      }
                   }
               }
-          }
-          """
+              """
           )
         );
     }