From a3df09605ac0761b3964d9a958549b5667bdc3d0 Mon Sep 17 00:00:00 2001
From: orangain <orangain@gmail.com>
Date: Sat, 14 May 2022 14:54:17 +0900
Subject: [PATCH] wip: Try to use kotlinx/ast

kotlinx/ast does not support generating code from AST
https://github.com/kotlinx/ast/issues/43
---
 .idea/jarRepositories.xml                     |  5 ++++
 .idea/misc.xml                                |  2 +-
 app/build.gradle.kts                          |  5 ++--
 .../ktcodeshift/FlattenListTransform.kt       |  8 +++++
 .../main/kotlin/ktcodeshift/KtCodeShift.kt    | 11 +++++++
 app/src/main/kotlin/ktcodeshift/Transform.kt  | 30 +++++++++++++++++++
 app/src/test/kotlin/ktcodeshift/AppTest.kt    | 20 +++++++++++--
 7 files changed, 75 insertions(+), 6 deletions(-)
 create mode 100644 app/src/main/kotlin/ktcodeshift/FlattenListTransform.kt
 create mode 100644 app/src/main/kotlin/ktcodeshift/KtCodeShift.kt
 create mode 100644 app/src/main/kotlin/ktcodeshift/Transform.kt

diff --git a/.idea/jarRepositories.xml b/.idea/jarRepositories.xml
index fdc392f..f5a0c5d 100644
--- a/.idea/jarRepositories.xml
+++ b/.idea/jarRepositories.xml
@@ -16,5 +16,10 @@
       <option name="name" value="MavenRepo" />
       <option name="url" value="https://repo.maven.apache.org/maven2/" />
     </remote-repository>
+    <remote-repository>
+      <option name="id" value="maven" />
+      <option name="name" value="maven" />
+      <option name="url" value="https://jitpack.io" />
+    </remote-repository>
   </component>
 </project>
\ No newline at end of file
diff --git a/.idea/misc.xml b/.idea/misc.xml
index 668048d..49504ef 100644
--- a/.idea/misc.xml
+++ b/.idea/misc.xml
@@ -1,5 +1,5 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
   <component name="ExternalStorageConfigurationManager" enabled="true" />
-  <component name="ProjectRootManager" version="2" languageLevel="JDK_17" project-jdk-name="17" project-jdk-type="JavaSDK" />
+  <component name="ProjectRootManager" version="2" languageLevel="JDK_17" default="true" project-jdk-name="17" project-jdk-type="JavaSDK" />
 </project>
\ No newline at end of file
diff --git a/app/build.gradle.kts b/app/build.gradle.kts
index 51cb359..53272eb 100644
--- a/app/build.gradle.kts
+++ b/app/build.gradle.kts
@@ -18,6 +18,7 @@ plugins {
 repositories {
     // Use Maven Central for resolving dependencies.
     mavenCentral()
+    maven("https://jitpack.io")
 }
 
 dependencies {
@@ -27,8 +28,8 @@ dependencies {
     // Use the Kotlin JDK 8 standard library.
     implementation("org.jetbrains.kotlin:kotlin-stdlib-jdk8")
 
-    // This dependency is used by the application.
-    implementation("com.google.guava:guava:30.1.1-jre")
+    implementation("com.github.kotlinx.ast:common:v0.1.0")
+    implementation("com.github.kotlinx.ast:grammar-kotlin-parser-antlr-kotlin:v0.1.0")
 }
 
 testing {
diff --git a/app/src/main/kotlin/ktcodeshift/FlattenListTransform.kt b/app/src/main/kotlin/ktcodeshift/FlattenListTransform.kt
new file mode 100644
index 0000000..2ef1ccf
--- /dev/null
+++ b/app/src/main/kotlin/ktcodeshift/FlattenListTransform.kt
@@ -0,0 +1,8 @@
+package ktcodeshift
+
+class FlattenListTransform : Transform {
+    override fun transform(fileInfo: FileInfo, api: Api): String {
+        val ast = api.parse(fileInfo.source)
+        return ast.toString()
+    }
+}
\ No newline at end of file
diff --git a/app/src/main/kotlin/ktcodeshift/KtCodeShift.kt b/app/src/main/kotlin/ktcodeshift/KtCodeShift.kt
new file mode 100644
index 0000000..83991c9
--- /dev/null
+++ b/app/src/main/kotlin/ktcodeshift/KtCodeShift.kt
@@ -0,0 +1,11 @@
+package ktcodeshift
+
+import kotlinx.ast.common.AstSource
+import kotlinx.ast.common.ast.Ast
+import kotlinx.ast.grammar.kotlin.target.antlr.kotlin.KotlinGrammarAntlrKotlinParser
+
+fun applyTransform(transform: Transform, fileInfo: FileInfo): String {
+    val api = Api()
+    return transform.transform(fileInfo, api)
+}
+
diff --git a/app/src/main/kotlin/ktcodeshift/Transform.kt b/app/src/main/kotlin/ktcodeshift/Transform.kt
new file mode 100644
index 0000000..41aaab3
--- /dev/null
+++ b/app/src/main/kotlin/ktcodeshift/Transform.kt
@@ -0,0 +1,30 @@
+package ktcodeshift
+
+import kotlinx.ast.common.AstSource
+import kotlinx.ast.common.ast.Ast
+import kotlinx.ast.grammar.kotlin.target.antlr.kotlin.KotlinGrammarAntlrKotlinParser
+
+interface Transform {
+    fun transform(fileInfo: FileInfo, api: Api/*, options: Options*/): String
+}
+
+interface FileInfo {
+    val path: String
+    val source: String
+}
+
+//interface Api {
+//
+//}
+
+class Api {
+    fun parse(source: String): Ast {
+        val astSource = AstSource.String("file", source)
+        val ast = KotlinGrammarAntlrKotlinParser.parseKotlinFile(astSource)
+        return ast
+    }
+}
+
+//interface Options {
+//
+//}
\ No newline at end of file
diff --git a/app/src/test/kotlin/ktcodeshift/AppTest.kt b/app/src/test/kotlin/ktcodeshift/AppTest.kt
index 1a5f8e4..5451439 100644
--- a/app/src/test/kotlin/ktcodeshift/AppTest.kt
+++ b/app/src/test/kotlin/ktcodeshift/AppTest.kt
@@ -4,11 +4,25 @@
 package ktcodeshift
 
 import kotlin.test.Test
+import kotlin.test.assertEquals
 import kotlin.test.assertNotNull
 
 class AppTest {
-    @Test fun appHasAGreeting() {
-        val classUnderTest = App()
-        assertNotNull(classUnderTest.greeting, "app should have a greeting")
+    @Test
+    fun transform() {
+        val changedSource = applyTransform(FlattenListTransform(), object : FileInfo {
+            override val path = "Test.kt"
+            override val source = """
+                package testing
+                
+                val a = listOf(listOf(1, 2), listOf(3))
+            """.trimIndent()
+        })
+
+        assertEquals("""
+            package testing
+                
+            val a = listOf(1, 2, 3)
+        """.trimIndent(), changedSource)
     }
 }