使Intellij Idea插件与Kotlin文件一起使用

Question

I've created this very simple Intellij Idea plugin which folds some reference expressions. It works great for Java files but it doesn't work for Kotlin.

Here is the source: https://github.com/nodes-android/nstack-translation-folding .
I will include here the important parts:
plugin.xml

</idea-plugin>
    <depends>com.intellij.modules.all</depends>

    <application-components>
        <component>
            <implementation-class>com.nodes.folding.TranslationFoldingBuilder</implementation-class>
        </component>
    </application-components>

    <extensions defaultExtensionNs="com.intellij">
        <lang.foldingBuilder language="JAVA" implementationClass="com.nodes.folding.TranslationFoldingBuilder"/>
    </extensions>

</idea-plugin>

TranslationFoldingBuilder.kt

class TranslationFoldingBuilder : FoldingBuilderEx() {

    override fun buildFoldRegions(root: PsiElement, document: Document, quick: Boolean): Array<FoldingDescriptor> {
        if (root !is PsiJavaFile) {
            return FoldingDescriptor.EMPTY
        }

        val descriptors = ArrayList<FoldingDescriptor>()
        // Get all the reference expressions in this Java file
        val referenceExpressions = PsiTreeUtil.findChildrenOfType(root, PsiReferenceExpression::class.java)

        // Some logic

        return descriptors.toTypedArray()
    }
}

My problem is that for Kotlin files the buildFoldRegions() is not called at all.

Answer 1

Of course it will not work for kotlin ,as your

if (root !is PsiJavaFile) { return FoldingDescriptor.EMPTY }

For kotlin files, the file is org.jetbrains.kotlin.psi.KtFile instance, rather than PsiJavaFile

---

Update:

1. You need to add kotlin plugin as the dependency of your plugin in plugin.xml
2. The psi api of Kotlin is not same as Java (They are different languages). You need to write a different class (but some can be same, I just copy your origin code and edit so there is some duplicate in my kt implementation.

You can see my commit here https://github.com/aristotll/nstack-translation-folding/commit/45286e6ec10d3b50defe25d55f8fbd8f122a148b .

Answer 2

就我而言，向项目添加kotlin-compiler库解决了缺少KtFile和其他与Kotlin Psi相关的类的问题。