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

* Fixing bugs IDE-16164 and IDE-16163

Author: LokeshDogga13

.changes/next-release/bugfix-2e1854d3-f326-4552-aa02-8ce19682c312.json

@@ -0,0 +1,4 @@
+{
+    "type" : "bugfix",
+    "description" : "Fix Q: Send to Prompt option to work in split panes on local and remote."
+}

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

@@ -10,6 +10,8 @@ import com.intellij.diff.requests.SimpleDiffRequest
 import com.intellij.ide.BrowserUtil
 import com.intellij.openapi.application.ApplicationManager
 import com.intellij.openapi.application.runInEdt
+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
@@ -95,6 +97,7 @@ import software.aws.toolkits.jetbrains.services.cwc.editor.context.file.FileCont
 import software.aws.toolkits.jetbrains.services.cwc.messages.ChatMessageType
 import software.aws.toolkits.jetbrains.services.telemetry.TelemetryService
 import software.aws.toolkits.jetbrains.ui.feedback.TestGenFeedbackDialog
+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
@@ -1246,9 +1249,22 @@ class CodeTestChatController(
         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.amazonq.webview.FqnWebviewAdapte
 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.clients.chat.model.FullyQual
 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