fix(codewhisperer): inlay not supported error in injected editior (#3710)

Author: zixlin7

.changes/next-release/bugfix-4368a4bf-b49c-4ce9-8a74-f2b6b4a31c26.json

@@ -0,0 +1,4 @@
+{
+  "type" : "bugfix",
+  "description" : "Inlay not supported exception in injected editor"
+}

jetbrains-core/src/software/aws/toolkits/jetbrains/services/codewhisperer/service/CodeWhispererService.kt

@@ -71,6 +71,7 @@ import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhisperer
 import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererUtil.notifyErrorCodeWhispererUsageLimit
 import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererUtil.promptReAuth
 import software.aws.toolkits.jetbrains.services.codewhisperer.util.FileContextProvider
+import software.aws.toolkits.jetbrains.utils.isInjectedText
 import software.aws.toolkits.resources.message
 import software.aws.toolkits.telemetry.CodewhispererCompletionType
 import software.aws.toolkits.telemetry.CodewhispererSuggestionState
@@ -100,6 +101,8 @@ class CodeWhispererService {
             }
             return
         }
+        val isInjectedFile = runReadAction { psiFile.isInjectedText() }
+        if (isInjectedFile) return
 
         val requestContext = try {
             getRequestContext(triggerTypeInfo, editor, project, psiFile, latencyContext)

jetbrains-core/src/software/aws/toolkits/jetbrains/utils/PsiUtils.kt

@@ -10,11 +10,16 @@ import com.intellij.psi.PsiElement
 fun PsiElement.isTestOrInjectedText(): Boolean {
     val project = this.project
     val virtualFile = this.containingFile.virtualFile ?: return false
-    if (virtualFile is VirtualFileWindow) {
+    if (this.isInjectedText() || ProjectRootManager.getInstance(project).fileIndex.isInTestSourceContent(virtualFile)) {
         return true
     }
 
-    if (ProjectRootManager.getInstance(project).fileIndex.isInTestSourceContent(virtualFile)) {
+    return false
+}
+
+fun PsiElement.isInjectedText(): Boolean {
+    val virtualFile = this.containingFile.virtualFile ?: return false
+    if (virtualFile is VirtualFileWindow) {
         return true
     }