From 5c7ec598afa426f949db330c8222f9bf3b4fdcee Mon Sep 17 00:00:00 2001 From: LlamaLad7 Date: Fri, 19 Dec 2025 00:32:55 +0000 Subject: [PATCH] Fix: Support finding the MCP module of files which are currently being edited. `PsiFile#getVirtualFile` seems to be null in these cases, but `FileViewProvider#getVirtualFile` works correctly. --- src/main/kotlin/util/psi-utils.kt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/main/kotlin/util/psi-utils.kt b/src/main/kotlin/util/psi-utils.kt index 0809aab72..3151fc18e 100644 --- a/src/main/kotlin/util/psi-utils.kt +++ b/src/main/kotlin/util/psi-utils.kt @@ -347,7 +347,7 @@ fun LookupElementBuilder.withImportInsertion(toImport: List): LookupEl } fun PsiElement.findMcpModule() = this.cached { - val file = containingFile?.virtualFile ?: return@cached null + val file = containingFile?.viewProvider?.virtualFile ?: return@cached null val index = ProjectFileIndex.getInstance(project) val modules = if (index.isInLibrary(file)) { val library = index.getOrderEntriesForFile(file)