Support KTS parsing

openrewrite · Oct 30, 2023 · cb3b734 · cb3b734
1 parent b304082
commit cb3b734
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/main/java/org/openrewrite/kotlin/internal/KotlinTreeParserVisitor.java b/src/main/java/org/openrewrite/kotlin/internal/KotlinTreeParserVisitor.java
@@ -1056,12 +1056,13 @@ public J visitSafeQualifiedExpression(KtSafeQualifiedExpression expression, Exec
 
     @Override
     public J visitScript(KtScript script, ExecutionContext data) {
-        throw new UnsupportedOperationException("TODO");
+        return script.getBlockExpression().accept(this, data);
     }
 
     @Override
     public J visitScriptInitializer(KtScriptInitializer initializer, ExecutionContext data) {
-        throw new UnsupportedOperationException("TODO");
+        J j = initializer.getBody().accept(this, data);
+        return j.withPrefix(merge(prefix(initializer), j.getPrefix()));
     }
 
     @Override

diff --git a/src/test/java/org/openrewrite/kotlin/tree/KTSTest.java b/src/test/java/org/openrewrite/kotlin/tree/KTSTest.java
@@ -27,6 +27,7 @@ void helloWorld() {
         rewriteRun(
           kotlinScript("""
             import java.util.List
+
             println("Hello, World!")
             """)
         );