fix(amazonq): Bug Fix for "Q: Send to Prompt" doesn't work in split panes on local or remote

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeTest/controller/CodeTestChatController.kt

@@ -9,6 +9,8 @@
 import com.intellij.diff.requests.SimpleDiffRequest
 import com.intellij.ide.BrowserUtil
 import com.intellij.openapi.application.ApplicationManager
+import com.intellij.openapi.editor.EditorFactory
+import com.intellij.openapi.fileEditor.FileDocumentManager
 import com.intellij.openapi.fileEditor.FileEditorManager
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.project.guessProjectDir
@@ -92,6 +94,7 @@
 import software.aws.toolkits.jetbrains.services.cwc.editor.context.file.FileContext
 import software.aws.toolkits.jetbrains.services.cwc.messages.ChatMessageType
 import software.aws.toolkits.jetbrains.services.telemetry.TelemetryService
+import software.aws.toolkits.jetbrains.utils.isRunningOnRemoteBackend
 import software.aws.toolkits.jetbrains.utils.notifyError
 import software.aws.toolkits.resources.message
 import software.aws.toolkits.telemetry.AmazonqTelemetry
@@ -1238,9 +1241,22 @@
         message: IncomingCodeTestMessage.StartTestGen,
     ): ActiveFileInfo? {
         try {
-            val fileEditorManager = FileEditorManager.getInstance(project)
-            val activeEditor = fileEditorManager.selectedEditor
-            val activeFile = fileEditorManager.selectedFiles.firstOrNull()
+            val (activeEditor, activeFile) = when {
+                isRunningOnRemoteBackend() -> {
+                    val editors = EditorFactory.getInstance().allEditors
+                    val editor = editors.firstOrNull()
+                    val file = editor?.document?.let { document ->
+                        FileDocumentManager.getInstance().getFile(document)
+                    }
+                    Pair(editor, file)
+                }
+                else -> {
+                    val fileEditorManager = FileEditorManager.getInstance(project)
+                    val editor = fileEditorManager.selectedEditor
+                    val file = fileEditorManager.selectedFiles.firstOrNull()
+                    Pair(editor, file)
+                }
+            }
             val projectRoot = project.basePath?.let { Path.of(it) }?.toFile()?.toVirtualFile() ?: run {
                 project.guessProjectDir() ?: error("Cannot guess base directory for project ${project.name}")
             }

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/editor/context/file/FileContextExtractor.kt

@@ -13,12 +13,18 @@
 import software.aws.toolkits.jetbrains.services.cwc.editor.context.file.util.LanguageExtractor
 import software.aws.toolkits.jetbrains.services.cwc.editor.context.file.util.MatchPolicyExtractor
 import software.aws.toolkits.jetbrains.utils.computeOnEdt
+import software.aws.toolkits.jetbrains.utils.isRunningOnRemoteBackend
 
 class FileContextExtractor(private val fqnWebviewAdapter: FqnWebviewAdapter?, private val project: Project) {
     private val languageExtractor: LanguageExtractor = LanguageExtractor()
     suspend fun extract(): FileContext? {
+        val editorManager = FileEditorManager.getInstance(project)
+
         val editor = computeOnEdt {
-            FileEditorManager.getInstance(project).selectedTextEditorWithRemotes.firstOrNull()
+            when {
+                isRunningOnRemoteBackend() -> editorManager.selectedTextEditorWithRemotes.firstOrNull()
+                else -> editorManager.selectedTextEditor
+            }
         } ?: return null
 
         val fileLanguage = computeOnEdt {

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/editor/context/focusArea/FocusAreaContextExtractor.kt

@@ -18,15 +18,20 @@
 import software.aws.toolkits.jetbrains.services.cwc.controller.ChatController
 import software.aws.toolkits.jetbrains.services.cwc.editor.context.file.util.LanguageExtractor
 import software.aws.toolkits.jetbrains.utils.computeOnEdt
+import software.aws.toolkits.jetbrains.utils.isRunningOnRemoteBackend
 import java.awt.Point
 import kotlin.math.min
 
 class FocusAreaContextExtractor(private val fqnWebviewAdapter: FqnWebviewAdapter?, private val project: Project) {
 
     private val languageExtractor: LanguageExtractor = LanguageExtractor()
     suspend fun extract(): FocusAreaContext? {
+        val editorManager = FileEditorManager.getInstance(project)
         val editor = computeOnEdt {
-            FileEditorManager.getInstance(project).selectedTextEditorWithRemotes.firstOrNull()
+            when {
+                isRunningOnRemoteBackend() -> editorManager.selectedTextEditorWithRemotes.firstOrNull()
+                else -> editorManager.selectedTextEditor
+            }
         } ?: return null
 
         if (editor.document.text.isBlank()) return null