IntelliJ插件开发：如何解析Kotlin引用？

我明白你遇到的问题了——IntelliJ平台对Java和Kotlin的PSI（Program Structure Interface）实现是分开的，所以Java里直接用psi.reference.resolve()的方式在Kotlin里行不通。下面我来一步步帮你解决这两个问题：

问题1：为什么Kotlin文件里node.psi.reference返回null？怎么解析Kotlin引用？ #

Java的PSI元素大多实现了PsiReferenceOwner接口，所以能直接通过reference属性获取引用，但Kotlin的引用体系是独立的，它的引用表达式（比如变量、属性引用）属于KtReferenceExpression类型，自带resolve()方法，不需要通过reference属性。

你需要修改getPlaceholderText方法，做语言类型的分支处理，适配Java和Kotlin两种情况：

import org.jetbrains.kotlin.psi.KtReferenceExpression
import com.intellij.psi.PsiReferenceOwner

class KotlinFoldingBuilder : FoldingBuilderEx() {
    override fun buildFoldRegions(root: PsiElement, document: Document, quick: Boolean): Array<FoldingDescriptor> { 
        // 你的现有逻辑
    }

    override fun getPlaceholderText(node: ASTNode): String? {
        val psiElement = node.psi
        
        return when (psiElement) {
            // 处理Kotlin引用
            is KtReferenceExpression -> {
                val resolvedElement = psiElement.resolve()
                resolvedElement?.text // 返回解析后的元素文本，比如属性值或变量名
            }
            // 处理Java及其他支持PsiReference的元素
            is PsiReferenceOwner -> {
                psiElement.reference?.resolve()?.text
            }
            // 其他类型元素返回null或默认占位符
            else -> null
        }
    }

    override fun isCollapsedByDefault(node: ASTNode): Boolean {
        return true
    }
}

问题2：解析PsiElement类型的Kotlin引用的值 #

如果你的PsiElement是一个Kotlin引用，首先要把它强转成对应的Kotlin PSI类型，再根据类型来提取值：

情况1：引用指向Kotlin类属性（KtProperty）

import org.jetbrains.kotlin.psi.KtProperty
import org.jetbrains.kotlin.psi.KtLiteralExpression

fun getKotlinPropertyValue(property: KtProperty): String? {
    return property.initializer?.let { initializer ->
        // 如果是字面量（字符串、数字等），直接取文本
        if (initializer is KtLiteralExpression) {
            initializer.text
        } else {
            // 复杂表达式（比如函数调用、计算式）需要额外处理
            // 可以考虑使用IntelliJ的表达式求值API，或者解析PSI结构
            "..."
        }
    }
}

情况2：通用的Kotlin引用解析

import org.jetbrains.kotlin.psi.KtReferenceExpression
import org.jetbrains.kotlin.psi.KtVariableDeclaration

fun resolveKotlinReferenceValue(referenceElement: PsiElement): String? {
    return if (referenceElement is KtReferenceExpression) {
        val resolvedElement = referenceElement.resolve()
        when (resolvedElement) {
            is KtProperty -> getKotlinPropertyValue(resolvedElement)
            is KtVariableDeclaration -> resolvedElement.initializer?.text
            else -> resolvedElement?.text // 比如指向函数、类等，返回元素文本
        }
    } else {
        null
    }
}

额外注意事项 #

• 确保你的插件依赖中包含Kotlin插件：在build.gradle.kts的intellij配置里添加"Kotlin"插件，这样才能导入Kotlin的PSI类：

intellij {
    version.set("2023.2") // 对应你的IntelliJ版本
    plugins.set(listOf("Kotlin", "java"))
}

• 处理PSI元素时一定要做类型判断，避免类型转换异常——不同语言的PSI结构差异很大，不能一概而论。
• 对于复杂的表达式求值，可能需要用到Kotlin插件的ExpressionEvaluator或IntelliJ平台的求值API，这部分可以参考IntelliJ SDK文档里的表达式求值相关内容。

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