refactor(223-shire): update file retrieval method in VariableTemplateCompiler and ShireVariableTemplateCompiler #379

Author: phodal

core/src/main/kotlin/cc/unitmesh/devti/custom/compile/VariableTemplateCompiler.kt

@@ -17,6 +17,7 @@ import com.intellij.psi.PsiNameIdentifierOwner
 import cc.unitmesh.devti.intentions.action.getElementToAction
 import com.intellij.openapi.application.ReadAction
 import com.intellij.openapi.application.runReadAction
+import com.intellij.openapi.fileEditor.FileDocumentManager
 import com.intellij.psi.PsiManager
 import kotlinx.coroutines.runBlocking
 import org.apache.velocity.VelocityContext
@@ -137,7 +138,7 @@ class VariableTemplateCompiler(
             ReadAction.compute<PsiElement?, Throwable> {
                 currentEditor?.caretModel?.currentCaret?.offset?.let {
                     val psiFile = currentEditor.let { editor ->
-                        val psiFile = editor.virtualFile?.let { file ->
+                        val psiFile = FileDocumentManager.getInstance().getFile(editor.document)?.let { file ->
                             PsiManager.getInstance(myProject).findFile(file)
                         }
 

exts/devins-lang/src/main/kotlin/cc/unitmesh/devti/language/ast/variable/ShireVariableTemplateCompiler.kt

@@ -7,6 +7,7 @@ import cc.unitmesh.devti.language.ast.variable.resolver.CompositeVariableResolve
 import cc.unitmesh.devti.language.ast.variable.resolver.base.VariableResolverContext
 import com.intellij.openapi.application.runReadAction
 import com.intellij.openapi.editor.Editor
+import com.intellij.openapi.fileEditor.FileDocumentManager
 import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiManager
 
@@ -42,7 +43,7 @@ class ShireVariableTemplateCompiler(
         compiledVariables = additionalMap.mapValues { it.value.toString() }
 
         val file = runReadAction {
-            PsiManager.getInstance(myProject).findFile(editor.virtualFile ?: return@runReadAction null)
+            PsiManager.getInstance(myProject).findFile(FileDocumentManager.getInstance().getFile(editor.document) ?: return@runReadAction null)
         }!!
 
         val templateCompiler = runReadAction { VariableTemplateCompiler(file.language, file, null, editor) }