feat(amazonq): client emits messages on TabEvents (#5574)

* Set up Flare chat connection

* Partial chat results

* feedback

* detekt

* syntax error

* detekt

* detekt

* Add quick actions to chat

* adding commands from initialize result

* add tabEvent message handling

* added trace logging

* configure param deserialization

* feedback

---------

Co-authored-by: Manodnya Bhoite <[email protected]>
Co-authored-by: manodnyab <[email protected]>

Author: samgst-amazon

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/webview/BrowserConnector.kt

@@ -35,6 +35,9 @@ import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_QUICK_ACTION
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_READY
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_SOURCE_LINK_CLICK
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_TAB_ADD
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_TAB_CHANGE
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_TAB_REMOVE
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.ChatNotification
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.ChatParams
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.ChatPrompt
@@ -55,6 +58,8 @@ import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.SEND_
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.SendChatPromptRequest
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.SourceLinkClickNotification
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.SourceLinkClickParams
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.TabEventParams
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.TabEventRequest
 import software.aws.toolkits.jetbrains.services.amazonq.util.command
 import software.aws.toolkits.jetbrains.services.amazonq.util.tabType
 import software.aws.toolkits.jetbrains.services.amazonq.webview.theme.AmazonQTheme
@@ -233,6 +238,22 @@ class BrowserConnector(
                     server.chatReady()
                 }
             }
+            CHAT_TAB_ADD -> {
+                handleChatNotification<TabEventRequest, TabEventParams>(node) { server, params ->
+                    server.tabAdd(params)
+                }
+            }
+            CHAT_TAB_REMOVE -> {
+                handleChatNotification<TabEventRequest, TabEventParams>(node) { server, params ->
+                    chatCommunicationManager.removePartialChatMessage(params.tabId)
+                    server.tabRemove(params)
+                }
+            }
+            CHAT_TAB_CHANGE -> {
+                handleChatNotification<TabEventRequest, TabEventParams>(node) { server, params ->
+                    server.tabChange(params)
+                }
+            }
             CHAT_LINK_CLICK -> {
                 handleChatNotification<LinkClickNotification, LinkClickParams>(node) { server, params ->
                     server.linkClick(params)

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/lsp/AmazonQLanguageServer.kt

@@ -17,6 +17,9 @@ import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_QUICK_ACTION
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_READY
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_SOURCE_LINK_CLICK
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_TAB_ADD
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_TAB_CHANGE
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.CHAT_TAB_REMOVE
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.EncryptedChatParams
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.EncryptedQuickActionChatParams
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.FeedbackParams
@@ -25,6 +28,7 @@ import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.InfoL
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.LinkClickParams
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.SEND_CHAT_COMMAND_PROMPT
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.SourceLinkClickParams
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.TabEventParams
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.credentials.UpdateCredentialsPayload
 import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.dependencies.DidChangeDependencyPathsParams
 import java.util.concurrent.CompletableFuture
@@ -52,6 +56,15 @@ interface AmazonQLanguageServer : LanguageServer {
     @JsonRequest(SEND_CHAT_COMMAND_PROMPT)
     fun sendChatPrompt(params: EncryptedChatParams): CompletableFuture<String>
 
+    @JsonNotification(CHAT_TAB_ADD)
+    fun tabAdd(params: TabEventParams): CompletableFuture<Unit>
+
+    @JsonNotification(CHAT_TAB_REMOVE)
+    fun tabRemove(params: TabEventParams): CompletableFuture<Unit>
+
+    @JsonNotification(CHAT_TAB_CHANGE)
+    fun tabChange(params: TabEventParams): CompletableFuture<Unit>
+
     @JsonRequest(CHAT_QUICK_ACTION)
     fun sendQuickAction(params: EncryptedQuickActionChatParams): CompletableFuture<String>
 

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/lsp/model/aws/chat/FlareChatCommands.kt

@@ -11,3 +11,6 @@ const val CHAT_INFO_LINK_CLICK = "aws/chat/infoLinkClick"
 const val CHAT_SOURCE_LINK_CLICK = "aws/chat/sourceLinkClick"
 const val CHAT_FEEDBACK = "aws/chat/feedback"
 const val CHAT_FOLLOW_UP_CLICK = "aws/chat/followUpClick"
+const val CHAT_TAB_ADD = "aws/chat/tabAdd"
+const val CHAT_TAB_CHANGE = "aws/chat/tabChange"
+const val CHAT_TAB_REMOVE = "aws/chat/tabRemove"

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/lsp/model/aws/chat/TabEvent.kt

@@ -0,0 +1,13 @@
+// Copyright 2025 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+package software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat
+
+data class TabEventRequest(
+    override val command: String,
+    override val params: TabEventParams,
+) : ChatNotification<TabEventParams>
+
+data class TabEventParams(
+    val tabId: String,
+)