如何从自研的Intellij插件修改Kotlin代码？

Question

我计划为 Intellij（更具体地说是 Android Studio）创建一个插件，其中除其他外，还应修改现有代码，例如我想向现有 Kotlin 文件的给定函数中的 when 语句添加一个新的 case：

fun test() {
    val num = 1

    when(num) {
        1 -> NumberOne()
        2 -> NumberTwo()
        // want to add 3 -> NumberThree() here
    }
}

我已经做了一些研究并偶然发现了 Intellij PSI。然而链接的文档只提到了 Java，我不清楚是否可以使用 PSI 修改 Kotlin 代码。在这个主题上有更多经验的人可以分享他在这方面的知识吗？顺便说一句，我只需要修改 Kotlin 代码，不需要能够修改 Java 和 Kotlin 代码。

Answer 1

我不清楚是否可以使用 PSI 修改 Kotlin 代码

是的，你可以实现这一目标。

我无法为您提供相关代码，但如果您检查 Kotlin 函数参数帮助器插件的源代码，它会实现类似的功能：

https://github.com/DeveloperUtils/kotlin-function-arguments-helper