add kotlin parser

openrewrite · Oct 9, 2023 · 466755f · 466755f
1 parent f41f599
commit 466755f
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/pom.xml b/pom.xml
@@ -82,6 +82,7 @@
         <!-- Pinned versions, as RELEASE would make it into the published pom.xml -->
         <rewrite.version>8.8.0-SNAPSHOT</rewrite.version>
         <rewrite.python.version>1.2.0-SNAPSHOT</rewrite.python.version>
+        <rewrite.kotlin.version>1.5.0-SNAPSHOT</rewrite.kotlin.version>
 
         <!-- using 'ssh' url scheme by default, which assumes a human is performing git operations leveraging an ssh key -->
         <developerConnectionUrl>scm:git:ssh://[email protected]/openrewrite/rewrite-maven-plugin.git
@@ -208,7 +209,11 @@
             <artifactId>rewrite-python</artifactId>
             <version>${rewrite.python.version}</version>
         </dependency>
-
+        <dependency>
+            <groupId>org.openrewrite</groupId>
+            <artifactId>rewrite-kotlin</artifactId>
+            <version>${rewrite.kotlin.version}</version>
+        </dependency>
         <dependency>
             <groupId>com.puppycrawl.tools</groupId>
             <artifactId>checkstyle</artifactId>

diff --git a/src/main/java/org/openrewrite/maven/ResourceParser.java b/src/main/java/org/openrewrite/maven/ResourceParser.java
@@ -22,6 +22,7 @@
 import org.openrewrite.hcl.HclParser;
 import org.openrewrite.java.JavaParser;
 import org.openrewrite.json.JsonParser;
+import org.openrewrite.kotlin.KotlinParser;
 import org.openrewrite.properties.PropertiesParser;
 import org.openrewrite.protobuf.ProtoParser;
 import org.openrewrite.python.PythonParser;
@@ -148,6 +149,9 @@ public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) {
         PythonParser pythonParser = PythonParser.builder().build();
         List<Path> pythonPaths = new ArrayList<>();
 
+        KotlinParser kotlinParser = KotlinParser.builder().build();
+        List<Path> kotlinPaths = new ArrayList<>();
+
         HclParser hclParser = HclParser.builder().build();
         List<Path> hclPaths = new ArrayList<>();
 
@@ -172,6 +176,8 @@ public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) {
                 protoPaths.add(path);
             } else if (pythonParser.accept(path)) {
                 pythonPaths.add(path);
+            } else if (kotlinParser.accept(path)) {
+                kotlinPaths.add(path);
             } else if (hclParser.accept(path)) {
                 hclPaths.add(path);
             } else if (quarkParser.accept(path)) {
@@ -214,6 +220,11 @@ public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) {
             alreadyParsed.addAll(pythonPaths);
         }
 
+        if (!kotlinPaths.isEmpty()) {
+            sourceFiles = Stream.concat(sourceFiles, (Stream<S>) kotlinParser.parse(kotlinPaths, baseDir, ctx));
+            alreadyParsed.addAll(kotlinPaths);
+        }
+
         if (!hclPaths.isEmpty()) {
             sourceFiles = Stream.concat(sourceFiles, (Stream<S>) hclParser.parse(hclPaths, baseDir, ctx));
             alreadyParsed.addAll(hclPaths);