如何通过KtCallExpression获取声明的KtFile/KtClass？

要从KtCallExpression拿到方法的定义位置（所在文件、类），核心是通过Kotlin编译器的语义分析上下文（BindingContext）解析调用的引用目标，再从目标描述符（Descriptor）中提取位置信息。以下是具体实现步骤和针对你示例的说明：

核心步骤 #

1. 获取调用的引用目标描述符
   从KtCallExpression中提取调用的名称节点，再通过BindingContext获取该引用对应的FunctionDescriptor（方法的语义描述符）。
2. 从描述符中提取文件信息
   通过描述符的sourceElement关联到源文件节点，进而获取文件路径或名称。
3. 从描述符中提取类信息
   通过描述符的containingDeclaration向上遍历，找到包含该方法的类描述符（如果是类成员方法或伴生对象方法）。

代码示例 #

假设你已经在语义分析阶段拿到了KtCallExpression和对应的BindingContext：

import org.jetbrains.kotlin.descriptors.ClassDescriptor
import org.jetbrains.kotlin.descriptors.FunctionDescriptor
import org.jetbrains.kotlin.psi.KtCallExpression
import org.jetbrains.kotlin.resolve.BindingContext
import org.jetbrains.kotlin.resolve.descriptorutil.DescriptorUtils

fun getMethodDeclarationInfo(callExpr: KtCallExpression, bindingContext: BindingContext) {
    // 1. 获取调用的名称节点
    val calleeExpr = callExpr.calleeExpression ?: return
    
    // 2. 获取对应的方法描述符
    val functionDescriptor = bindingContext.get(BindingContext.REFERENCE_TARGET, calleeExpr) as? FunctionDescriptor
        ?: return

    // 3. 获取方法所在的源文件
    val sourceFile = functionDescriptor.sourceElement.containingFile
    println("方法所在文件：${sourceFile.name}") // 示例中f1()会输出Main.kt

    // 4. 获取方法所在的类（含伴生对象的外部类）
    var containingClass = DescriptorUtils.getParentOfType(functionDescriptor, ClassDescriptor::class.java)
    // 如果是伴生对象中的方法，需要再往上找外部类
    if (containingClass != null && DescriptorUtils.isCompanionObject(containingClass)) {
        containingClass = DescriptorUtils.getParentOfType(containingClass, ClassDescriptor::class.java)
    }
    containingClass?.let {
        println("方法所在类：${it.name.asString()}") // 示例中f2()会输出Test
    }
}

针对你示例的说明 #

• 对于f1()调用：
  functionDescriptor.containingDeclaration是FileDescriptor，对应Main.kt文件，通过sourceElement可直接关联到该文件。
• 对于Test.f2()调用：
  functionDescriptor.containingDeclaration是伴生对象的CompanionObjectDescriptor，通过DescriptorUtils.getParentOfType向上遍历就能拿到外部的Test类描述符。

注意事项 #

• 必须在语义分析阶段之后执行这段逻辑，只有此时BindingContext才完成了引用与声明的绑定，否则无法获取到有效的FunctionDescriptor。
• 如果方法是来自外部依赖库（非本地源码），sourceElement可能无法关联到本地文件，需要额外处理这种场景。

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