Fix: Support finding the MCP module of files which are currently being edited. (#2559)

LlamaLad7 · web-flow · commit d1aa33cf2af3 · 2025-12-19T01:16:23.000Z
`PsiFile#getVirtualFile` seems to be null in these cases, but `FileViewProvider#getVirtualFile` works correctly.
diff --git 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<PsiClass>): 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)

Original file line number	Diff line number	Diff line change
`@@ -347,7 +347,7 @@ fun LookupElementBuilder.withImportInsertion(toImport: List<PsiClass>): LookupEl`
`347`	`347`	`}`
`348`	`348`
`349`	`349`	`fun PsiElement.findMcpModule() = this.cached {`
`350`		`- val file = containingFile?.virtualFile ?: return@cached null`
	`350`	`+ val file = containingFile?.viewProvider?.virtualFile ?: return@cached null`
`351`	`351`	`val index = ProjectFileIndex.getInstance(project)`
`352`	`352`	`val modules = if (index.isInLibrary(file)) {`
`353`	`353`	`val library = index.getOrderEntriesForFile(file)`