action commands in chat support

Author: leigaol

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/commands/ActionRegistrar.kt

@@ -3,10 +3,17 @@
 
 package software.aws.toolkits.jetbrains.services.cwc.commands
 
+import com.google.gson.Gson
 import com.intellij.openapi.project.Project
 import kotlinx.coroutines.flow.MutableSharedFlow
 import kotlinx.coroutines.flow.asSharedFlow
+import kotlinx.coroutines.runBlocking
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.flareChat.AsyncChatUiListener
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.GenericCommandParams
+import software.aws.toolkits.jetbrains.services.amazonq.lsp.model.aws.chat.TriggerType
 import software.aws.toolkits.jetbrains.services.amazonq.messages.AmazonQMessage
+import software.aws.toolkits.jetbrains.services.cwc.editor.context.ActiveFileContextExtractor
+import software.aws.toolkits.jetbrains.services.cwc.editor.context.ExtractionTriggerType
 
 // Register Editor Actions in the Editor Context Menu
 class ActionRegistrar {
@@ -15,7 +22,29 @@ class ActionRegistrar {
     val flow = _messages.asSharedFlow()
 
     fun reportMessageClick(command: EditorContextCommand, project: Project) {
-        _messages.tryEmit(ContextMenuActionMessage(command, project))
+        if (command == EditorContextCommand.GenerateUnitTests) {
+            // pre-existing old chat code path
+            _messages.tryEmit(ContextMenuActionMessage(command, project))
+        }
+        else {
+            // new agentic chat route
+            runBlocking {
+                val contextExtractor = ActiveFileContextExtractor.create(fqnWebviewAdapter = null, project = project)
+                val fileContext = contextExtractor.extractContextForTrigger(ExtractionTriggerType.ContextMenu)
+                val codeSelection = "\n```\n${fileContext.focusAreaContext?.codeSelection?.trimIndent()?.trim()}\n```\n"
+
+                val params = GenericCommandParams(selection = codeSelection, triggerType = TriggerType.CONTEXT_MENU, genericCommand = command.name)
+
+                val uiMessage = """
+                        {
+                            "command": "genericCommand",
+                            "params": ${Gson().toJson(params)}
+                        }
+                    """.trimIndent()
+                AsyncChatUiListener.notifyPartialMessageUpdate(uiMessage)
+            }
+        }
+
     }
 
     fun reportMessageClick(command: EditorContextCommand, issue: MutableMap<String, String>, project: Project) {
@@ -25,5 +54,13 @@ class ActionRegistrar {
     // provide singleton access
     companion object {
         val instance = ActionRegistrar()
+        
+        /**
+         * Computes the product of 10 * 10
+         * @return The result of 10 * 10
+         */
+        fun computeTenTimesTen(): Int {
+            return 10 * 10
+        }
     }
 }

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/controller/ChatController.kt

@@ -289,28 +289,11 @@ class ChatController private constructor(
         if (message.project != context.project) {
             return
         }
-        val fileContext = contextExtractor.extractContextForTrigger(ExtractionTriggerType.ContextMenu)
-        val triggerId = UUID.randomUUID().toString()
-        val codeSelection = "\n```\n${fileContext.focusAreaContext?.codeSelection?.trimIndent()?.trim()}\n```\n"
-
-        if (message.command == EditorContextCommand.SendToPrompt) {
-            messagePublisher.publish(
-                EditorContextCommandMessage(
-                    message = codeSelection,
-                    command = message.command.actionId,
-                    triggerId = triggerId,
-                ),
-            )
-            return
-        }
+
         if (message.command == EditorContextCommand.GenerateUnitTests) {
             // Publish an event to "codetest" tab with command as "test" and type as "addAnswer"
             val messageToPublish = TestCommandMessage()
             context.messagesFromAppToUi.publish(messageToPublish)
-        } else {
-            // Create prompt
-            val prompt = "${message.command} the following part of my code for me: $codeSelection"
-            processPromptActions(prompt, message, triggerId, fileContext)
         }
     }
 

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

@@ -0,0 +1,18 @@
+// 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
+
+
+//https://github.com/aws/language-server-runtimes/blob/main/chat-client-ui-types/src/uiContracts.ts#L27
+enum class TriggerType(val value: String) {
+    HOTKEYS("hotkeys"),
+    CLICK("click"),
+    CONTEXT_MENU("contextMenu")
+}
+
+data class GenericCommandParams(
+    val selection: String,
+    val triggerType: TriggerType,
+    val genericCommand: String
+)