Implement and test ConvertApiResponseCodesToStrings

build.gradle.kts

@@ -18,5 +18,8 @@ dependencies {
 
     testImplementation("org.junit.jupiter:junit-jupiter-engine:latest.release")
 
+    testRuntimeOnly("io.swagger:swagger-annotations:1.6.13")
+    testRuntimeOnly("io.swagger.core.v3:swagger-annotations:2.2.20")
+
     testRuntimeOnly("org.gradle:gradle-tooling-api:latest.release")
 }

src/main/java/org/openrewrite/openapi/swagger/ConvertApiResponseCodesToStrings.java

@@ -16,11 +16,21 @@
 package org.openrewrite.openapi.swagger;
 
 import org.openrewrite.ExecutionContext;
+import org.openrewrite.Preconditions;
 import org.openrewrite.Recipe;
 import org.openrewrite.TreeVisitor;
+import org.openrewrite.internal.ListUtils;
+import org.openrewrite.java.AnnotationMatcher;
 import org.openrewrite.java.JavaIsoVisitor;
+import org.openrewrite.java.search.UsesType;
+import org.openrewrite.java.tree.Expression;
+import org.openrewrite.java.tree.J;
+import org.openrewrite.java.tree.JavaType;
 
 public class ConvertApiResponseCodesToStrings extends Recipe {
+
+    private static final AnnotationMatcher ANNOTATION_MATCHER = new AnnotationMatcher("@io.swagger.v3.oas.annotations.responses.ApiResponse");
+
     @Override
     public String getDisplayName() {
         return "Convert API response codes to strings";
@@ -33,8 +43,40 @@ public String getDescription() {
 
     @Override
     public TreeVisitor<?, ExecutionContext> getVisitor() {
-        return new JavaIsoVisitor<ExecutionContext>(){
-        // TODO: implement
-        };
+        // https://docs.swagger.io/swagger-core/v1.5.0/apidocs/io/swagger/annotations/ApiResponse.html
+        // https://docs.swagger.io/swagger-core/v2.0.0/apidocs/io/swagger/v3/oas/annotations/responses/ApiResponse.html
+        return Preconditions.check(
+                new UsesType<>("io.swagger.v3.oas.annotations.responses.ApiResponse", true),
+                new JavaIsoVisitor<ExecutionContext>() {
+                    @Override
+                    public J.Annotation visitAnnotation(J.Annotation annotation, ExecutionContext ctx) {
+                        J.Annotation an = super.visitAnnotation(annotation, ctx);
+                        if (ANNOTATION_MATCHER.matches(an)) {
+                            return an.withArguments(ListUtils.map(an.getArguments(), this::maybeReplaceResponseCodeTypeAndValue));
+                        }
+                        return an;
+                    }
+
+                    private Expression maybeReplaceResponseCodeTypeAndValue(Expression arg) {
+                        if (arg instanceof J.Assignment) {
+                            J.Assignment assignment = (J.Assignment) arg;
+                            boolean matchesField = assignment.getVariable() instanceof J.Identifier &&
+                                                   "responseCode".equals(((J.Identifier) assignment.getVariable()).getSimpleName());
+                            boolean usesNumberLiteral = assignment.getAssignment() instanceof J.Literal &&
+                                                        ((J.Literal) assignment.getAssignment()).getValue() instanceof Number;
+                            if (matchesField && usesNumberLiteral) {
+                                J.Literal assignedLiteral = (J.Literal) assignment.getAssignment();
+                                return assignment
+                                        .withType(JavaType.Primitive.String)
+                                        .withAssignment(assignedLiteral
+                                                .withValue(String.valueOf(assignedLiteral.getValue()))
+                                                .withValueSource("\"" + assignedLiteral.getValue() + "\"")
+                                                .withType(JavaType.Primitive.String));
+                            }
+                        }
+                        return arg;
+                    }
+                }
+        );
     }
 }

src/test/java/org/openrewrite/openapi/swagger/ConvertApiResponseCodesToStringsTest.java

@@ -0,0 +1,60 @@
+package org.openrewrite.openapi.swagger;
+
+import org.junit.jupiter.api.Test;
+import org.openrewrite.DocumentExample;
+import org.openrewrite.java.JavaParser;
+import org.openrewrite.test.RecipeSpec;
+import org.openrewrite.test.RewriteTest;
+
+import static org.openrewrite.java.Assertions.*;
+
+class ConvertApiResponseCodesToStringsTest implements RewriteTest {
+    @Override
+    public void defaults(RecipeSpec spec) {
+        spec.recipeFromResources("org.openrewrite.openapi.MigrateApiResponsesToApiResponses")
+          .parser(JavaParser.fromJavaVersion().classpath("swagger-annotations-1.+", "swagger-annotations-2.+"));
+    }
+
+    @Test
+    @DocumentExample
+    void convertApiResponseCodesToStrings() {
+        rewriteRun(
+          //language=java
+          java(
+            """
+              import io.swagger.annotations.ApiResponse;
+              
+              class A {
+                  @ApiResponse(code = 200, message = "OK")
+                  void method() {}
+              }
+              """,
+            """
+              import io.swagger.v3.oas.annotations.responses.ApiResponse;
+              
+              class A {
+                  @ApiResponse(responseCode = "200", description = "OK")
+                  void method() {}
+              }
+              """
+          )
+        );
+    }
+
+    @Test
+    void noChangeOnAlreadyConverted() {
+        rewriteRun(
+          //language=java
+          java(
+            """
+              import io.swagger.v3.oas.annotations.responses.ApiResponse;
+              
+              class A {
+                  @ApiResponse(responseCode = "200", description = "OK")
+                  void method() {}
+              }
+              """
+          )
+        );
+    }
+}