能否开发Kotlin编译器插件直接修改源码而非生成独立代码？

Kotlin编译器插件的核心能力之一就是在编译流程中直接修改代码的抽象语法树（AST），不管是调整函数逻辑、替换符号引用还是修改导入语句，都可以实现，无需生成独立的代码文件。下面针对你提到的两个场景具体说明：

场景1：将Button点击逻辑包裹进track函数 #

假设原代码是：

@Composable
fun MyScreen() {
    Button(onClick = { 
        navigateToSettings()
    }) {
        Text("Go to Settings")
    }
}

实现思路 #

在编译器的IR（Intermediate Representation）阶段（Kotlin 1.4+推荐的编译扩展阶段），通过继承AbstractIrElementTransformerVoid遍历AST：

1. 识别带有@Composable注解的函数节点；
2. 在函数内部找到Button函数的调用节点；
3. 提取onClick参数对应的Lambda表达式；
4. 将原Lambda的内容包裹进track函数的Lambda参数中，替换原onClick的取值。

修改后编译生成的等效代码为：

@Composable
fun MyScreen() {
    Button(onClick = { 
        track("my_screen_settings_button_click") {
            navigateToSettings()
        }
    }) {
        Text("Go to Settings")
    }
}

如果需要在IDE中实时看到修改后的代码（而非仅编译阶段生效），可以结合Kotlin IDE插件的PSI（Program Structure Interface）API，在编辑器的代码分析阶段修改AST，让用户编写代码时就能看到实时的逻辑包裹效果。

场景2：替换Button的导入路径 #

假设原导入语句是：

import androidx.compose.material.Button

实现思路 #

可以通过两种方式实现：

1. 导入解析阶段拦截：实现ImportProcessor接口，在编译器处理导入语句时，匹配androidx.compose.material.Button并替换为com.mycompany.project.wrappers.Button；
2. IR阶段符号替换：遍历AST中的Button符号引用，将其指向com.mycompany.project.wrappers.Button的符号，编译器会自动调整导入语句以匹配新的符号引用。

最终编译时会自动使用你的自定义包装类，无需手动修改源代码中的导入或函数调用。

内容的提问来源于stack exchange，提问作者Samuel Neff