feat(amazonq): saved prompts (aws#7)

* feat(amazonq): saved prompts

* fix: remove unecessary change

Author: avi-alpert

packages/core/src/amazonq/webview/ui/apps/cwChatConnector.ts

@@ -3,19 +3,27 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import { ChatItemType, MynahUIDataModel } from '@aws/mynah-ui'
+import { ChatItemButton, ChatItemFormItem, ChatItemType, MynahUIDataModel } from '@aws/mynah-ui'
 import { TabType } from '../storages/tabsStorage'
 import { CWCChatItem } from '../connector'
 import { BaseConnector, BaseConnectorProps } from './baseConnector'
 
 export interface ConnectorProps extends BaseConnectorProps {
     onCWCContextCommandMessage: (message: CWCChatItem, command?: string) => string | undefined
     onContextCommandDataReceived: (data: MynahUIDataModel['contextCommands']) => void
+    onShowCustomForm: (
+        tabId: string,
+        formItems?: ChatItemFormItem[],
+        buttons?: ChatItemButton[],
+        title?: string,
+        description?: string
+    ) => void
 }
 
 export class Connector extends BaseConnector {
     private readonly onCWCContextCommandMessage
     private readonly onContextCommandDataReceived
+    private readonly onShowCustomForm
 
     override getTabType(): TabType {
         return 'cwc'
@@ -25,6 +33,7 @@ export class Connector extends BaseConnector {
         super(props)
         this.onCWCContextCommandMessage = props.onCWCContextCommandMessage
         this.onContextCommandDataReceived = props.onContextCommandDataReceived
+        this.onShowCustomForm = props.onShowCustomForm
     }
 
     onSourceLinkClick = (tabID: string, messageId: string, link: string): void => {
@@ -140,6 +149,16 @@ export class Connector extends BaseConnector {
         }
     }
 
+    private showCustomFormMessage = (messageData: any) => {
+        this.onShowCustomForm(
+            messageData.tabID,
+            messageData.formItems,
+            messageData.buttons,
+            messageData.title,
+            messageData.description
+        )
+    }
+
     handleMessageReceive = async (messageData: any): Promise<void> => {
         if (messageData.type === 'chatMessage') {
             await this.processChatMessage(messageData)
@@ -155,7 +174,45 @@ export class Connector extends BaseConnector {
             await this.processContextCommandData(messageData)
             return
         }
+        if (messageData.type === 'showCustomFormMessage') {
+            this.showCustomFormMessage(messageData)
+            return
+        }
+
+        if (messageData.type === 'customFormActionMessage') {
+            this.onCustomFormAction(messageData.tabID, messageData.action)
+            return
+        }
         // For other message types, call the base class handleMessageReceive
         await this.baseHandleMessageReceive(messageData)
     }
+
+    onQuickCommandGroupActionClick = (tabID: string, action: { id: string }) => {
+        this.sendMessageToExtension({
+            command: 'quick-command-group-action-click',
+            actionId: action.id,
+            tabID,
+            tabType: this.getTabType(),
+        })
+    }
+
+    onCustomFormAction(
+        tabId: string,
+        action: {
+            id: string
+            text?: string | undefined
+            formItemValues?: Record<string, string> | undefined
+        }
+    ) {
+        if (action === undefined) {
+            return
+        }
+
+        this.sendMessageToExtension({
+            command: 'form-action-click',
+            action: action,
+            tabType: this.getTabType(),
+            tabID: tabId,
+        })
+    }
 }

packages/core/src/amazonq/webview/ui/commands.ts

@@ -42,5 +42,6 @@ type MessageCommand =
     | 'open-link'
     | 'send-telemetry'
     | 'update-welcome-count'
+    | 'quick-command-group-action-click'
 
 export type ExtensionMessage = Record<string, any> & { command: MessageCommand }

packages/core/src/amazonq/webview/ui/connector.ts

@@ -14,6 +14,8 @@ import {
     ChatPrompt,
     MynahUIDataModel,
     QuickActionCommand,
+    ChatItemFormItem,
+    ChatItemButton,
 } from '@aws/mynah-ui'
 import { Connector as CWChatConnector } from './apps/cwChatConnector'
 import { Connector as FeatureDevChatConnector } from './apps/featureDevChatConnector'
@@ -91,6 +93,13 @@ export interface ConnectorProps {
     handleCommand: (chatPrompt: ChatPrompt, tabId: string) => void
     sendStaticMessages: (tabID: string, messages: ChatItem[]) => void
     onContextCommandDataReceived: (message: MynahUIDataModel['contextCommands']) => void
+    onShowCustomForm: (
+        tabId: string,
+        formItems?: ChatItemFormItem[],
+        buttons?: ChatItemButton[],
+        title?: string,
+        description?: string
+    ) => void
     tabsStorage: TabsStorage
 }
 
@@ -612,6 +621,14 @@ export class Connector {
         }
     }
 
+    onQuickCommandGroupActionClick = (tabId: string, action: { id: string }) => {
+        switch (this.tabsStorage.getTab(tabId)?.type) {
+            case 'cwc':
+                this.cwChatConnector.onQuickCommandGroupActionClick(tabId, action)
+                break
+        }
+    }
+
     onChatItemVoted = (tabId: string, messageId: string, vote: 'upvote' | 'downvote'): void | undefined => {
         switch (this.tabsStorage.getTab(tabId)?.type) {
             case 'cwc':
@@ -655,6 +672,8 @@ export class Connector {
                         type: '',
                         tabType: 'cwc',
                     })
+                } else {
+                    this.cwChatConnector.onCustomFormAction(tabId, action)
                 }
                 break
             case 'agentWalkthrough': {

packages/core/src/amazonq/webview/ui/main.ts

@@ -13,6 +13,8 @@ import {
     NotificationType,
     ReferenceTrackerInformation,
     ProgressField,
+    ChatItemButton,
+    ChatItemFormItem,
 } from '@aws/mynah-ui'
 import { ChatPrompt } from '@aws/mynah-ui/dist/static'
 import { TabsStorage, TabType } from './storages/tabsStorage'
@@ -570,6 +572,15 @@ export const createMynahUI = (
                 }
             }
         },
+        onShowCustomForm(
+            tabId: string,
+            formItems?: ChatItemFormItem[],
+            buttons?: ChatItemButton[],
+            title?: string,
+            description?: string
+        ) {
+            mynahUI.showCustomForm(tabId, formItems, buttons, title, description)
+        },
     })
 
     mynahUI = new MynahUI({
@@ -670,6 +681,7 @@ export const createMynahUI = (
             // handler for the cwc panel
             textMessageHandler.handle(prompt, tabID, eventId as string)
         },
+        onQuickCommandGroupActionClick: connector.onQuickCommandGroupActionClick,
         onVote: connector.onChatItemVoted,
         onInBodyButtonClicked: (tabId, messageId, action, eventId) => {
             switch (action.id) {

packages/core/src/amazonq/webview/ui/tabs/constants.ts

@@ -13,7 +13,7 @@ export type TabTypeData = {
     contextCommands?: QuickActionCommandGroup[]
 }
 
-const workspaceCommand: QuickActionCommandGroup = {
+export const workspaceCommand: QuickActionCommandGroup = {
     groupName: 'Mention code',
     commands: [
         {

packages/core/src/codewhispererChat/app.ts

@@ -26,8 +26,10 @@ import {
     TriggerTabIDReceived,
     UIFocusMessage,
     AcceptDiff,
+    QuickCommandGroupActionClick,
 } from './controllers/chat/model'
 import { EditorContextCommand, registerCommands } from './commands/registerCommands'
+import { CustomFormActionMessage } from './view/connector/connector'
 
 export function init(appContext: AmazonQAppInitContext) {
     const cwChatControllerEventEmitters = {
@@ -49,6 +51,8 @@ export function init(appContext: AmazonQAppInitContext) {
         processResponseBodyLinkClick: new EventEmitter<ResponseBodyLinkClickMessage>(),
         processFooterInfoLinkClick: new EventEmitter<FooterInfoLinkClick>(),
         processContextCommandUpdateMessage: new EventEmitter<any>(),
+        processQuickCommandGroupActionClicked: new EventEmitter<QuickCommandGroupActionClick>(),
+        processCustomFormAction: new EventEmitter<CustomFormActionMessage>(),
     }
 
     const cwChatControllerMessageListeners = {
@@ -100,6 +104,12 @@ export function init(appContext: AmazonQAppInitContext) {
         processContextCommandUpdateMessage: new MessageListener<void>(
             cwChatControllerEventEmitters.processContextCommandUpdateMessage
         ),
+        processQuickCommandGroupActionClicked: new MessageListener<QuickCommandGroupActionClick>(
+            cwChatControllerEventEmitters.processQuickCommandGroupActionClicked
+        ),
+        processCustomFormAction: new MessageListener<CustomFormActionMessage>(
+            cwChatControllerEventEmitters.processCustomFormAction
+        ),
     }
 
     const cwChatControllerMessagePublishers = {
@@ -153,6 +163,12 @@ export function init(appContext: AmazonQAppInitContext) {
         processContextCommandUpdateMessage: new MessagePublisher<void>(
             cwChatControllerEventEmitters.processContextCommandUpdateMessage
         ),
+        processQuickCommandGroupActionClicked: new MessagePublisher<QuickCommandGroupActionClick>(
+            cwChatControllerEventEmitters.processQuickCommandGroupActionClicked
+        ),
+        processCustomFormAction: new MessagePublisher<CustomFormActionMessage>(
+            cwChatControllerEventEmitters.processCustomFormAction
+        ),
     }
 
     new CwChatController(