HasSourceSet should match any SourceFile, not just Java files (#6989)

The visitor used JavaIsoVisitor and checked for JavaSourceFile, which
meant resource files (YAML, properties, etc.) in a source set were
never matched. Change to a generic TreeVisitor that checks the
JavaSourceSet marker on any SourceFile.

Author: Jenson3210

rewrite-java-test/src/test/java/org/openrewrite/java/search/HasSourceSetTest.java

@@ -19,6 +19,7 @@
 import org.openrewrite.test.RewriteTest;
 
 import static org.openrewrite.java.Assertions.*;
+import static org.openrewrite.test.SourceSpecs.text;
 
 class HasSourceSetTest implements RewriteTest {
 
@@ -59,4 +60,45 @@ class Test {
           )
         );
     }
+
+    @Test
+    void mainResources() {
+        rewriteRun(
+          spec -> spec.recipe(new HasSourceSet("main")),
+          srcMainResources(
+            text(
+              "app.name=test",
+              "~~>app.name=test",
+              spec -> spec.path("application.properties")
+            )
+          )
+        );
+    }
+
+    @Test
+    void testResourcesNotMatchedByMain() {
+        rewriteRun(
+          spec -> spec.recipe(new HasSourceSet("main")),
+          srcTestResources(
+            text(
+              "app.name=test",
+              spec -> spec.path("application.properties")
+            )
+          )
+        );
+    }
+
+    @Test
+    void testResources() {
+        rewriteRun(
+          spec -> spec.recipe(new HasSourceSet("test")),
+          srcTestResources(
+            text(
+              "app.name=test",
+              "~~>app.name=test",
+              spec -> spec.path("application.properties")
+            )
+          )
+        );
+    }
 }

rewrite-java/src/main/java/org/openrewrite/java/search/HasSourceSet.java

@@ -19,14 +19,9 @@
 import lombok.Value;
 import org.jspecify.annotations.Nullable;
 import org.openrewrite.*;
-import org.openrewrite.java.JavaIsoVisitor;
 import org.openrewrite.java.marker.JavaSourceSet;
-import org.openrewrite.java.tree.J;
-import org.openrewrite.java.tree.JavaSourceFile;
 import org.openrewrite.marker.SearchResult;
 
-import static java.util.Objects.requireNonNull;
-
 @Value
 @EqualsAndHashCode(callSuper = false)
 public class HasSourceSet extends Recipe {
@@ -43,18 +38,18 @@ public class HasSourceSet extends Recipe {
 
     @Override
     public TreeVisitor<?, ExecutionContext> getVisitor() {
-        return new JavaIsoVisitor<ExecutionContext>() {
+        return new TreeVisitor<Tree, ExecutionContext>() {
             @Override
-            public J visit(@Nullable Tree tree, ExecutionContext ctx) {
-                if (tree instanceof JavaSourceFile) {
-                    JavaSourceFile cu = (JavaSourceFile) requireNonNull(tree);
-                    if (cu.getMarkers().findFirst(JavaSourceSet.class)
+            public @Nullable Tree visit(@Nullable Tree tree, ExecutionContext ctx) {
+                if (tree instanceof SourceFile) {
+                    SourceFile sf = (SourceFile) tree;
+                    if (sf.getMarkers().findFirst(JavaSourceSet.class)
                             .filter(s -> s.getName().equals(sourceSet))
                             .isPresent()) {
-                        return SearchResult.found(cu);
+                        return SearchResult.found(sf);
                     }
                 }
-                return (J) tree;
+                return tree;
             }
         };
     }