1

我想制作一个插件，为某些特定的 kotlin 文件生成一些模板代码。现在我需要从 PsiElement 或 VirtualFile 获取 KtClass。我尝试了一些如下代码，但它不起作用。

val psiFile = e.getData(CommonDataKeys.PSI_FILE)
        psiFile ?: return
        val editor = e.getData(CommonDataKeys.EDITOR)
        editor ?: return
        println(psiFile.name)
        val pe = psiFile.findElementAt(editor.caretModel.offset)
        if (pe != null) {
            println(pe.text)
        } else {
            println("PE is null")
        }
        val kt = PsiTreeUtil.getParentOfType(pe, KtClass::class.java)
        kt?.run {
            println(this.text)
        }

PS：我发现有2个PsiTreeUtils，一个 from org.jetbrains.kotlin.com.intellij.psi.utilpackage ，一个 from com.intellij.psi.util，不知道用哪个

4

1 回答 1

0

一个 PsiFile 可能包含零个或多个KtClasss。你可以通过这样做得到它们：

val classes = psiFile.getChildrenOfType<KtClass>().asList()

于 2020-06-20T13:16:25.043 回答