From cb3b7347f61205ea646532056f745ab4dc621367 Mon Sep 17 00:00:00 2001
From: Kun Li <kun@moderne.io>
Date: Mon, 30 Oct 2023 10:31:39 -0700
Subject: [PATCH] Support KTS parsing

---
 .../openrewrite/kotlin/internal/KotlinTreeParserVisitor.java | 5 +++--
 src/test/java/org/openrewrite/kotlin/tree/KTSTest.java       | 1 +
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/main/java/org/openrewrite/kotlin/internal/KotlinTreeParserVisitor.java b/src/main/java/org/openrewrite/kotlin/internal/KotlinTreeParserVisitor.java
index 79427535a..ffcc35810 100644
--- 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
index 112d51133..55ab77ba5 100644
--- 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!")
             """)
         );