feat: added protocol for context commands

Author: volodkevych

runtimes/protocol/chat.ts

@@ -41,6 +41,8 @@ import {
     FileClickParams,
     INLINE_CHAT_REQUEST_METHOD,
     InlineChatParams,
+    CONTEXT_COMMAND_NOTIFICATION_METHOD,
+    ContextCommandParams,
 } from './lsp'
 
 export const chatRequestType = new AutoParameterStructuresProtocolRequestType<
@@ -100,3 +102,6 @@ export const chatUpdateNotificationType = new ProtocolNotificationType<ChatUpdat
 export const fileClickNotificationType = new ProtocolNotificationType<FileClickParams, void>(
     FILE_CLICK_NOTIFICATION_METHOD
 )
+export const contextCommandsNotificationType = new ProtocolNotificationType<ContextCommandParams, void>(
+    CONTEXT_COMMAND_NOTIFICATION_METHOD
+)

runtimes/runtimes/auth/auth.test.ts

@@ -48,7 +48,7 @@ function clearHandlers() {
     authHandlers.bearerDeleteHandler = () => {}
 }
 
-const serverLspConnectionMock = <Connection>{
+const serverLspConnectionMock = {
     onRequest: (requestType: any, handler: any) => {
         if (requestType.method === credentialsProtocolMethodNames.iamCredentialsUpdate) {
             authHandlers.iamUpdateHandler = handler
@@ -83,7 +83,7 @@ const serverLspConnectionMock = <Connection>{
     onExecuteCommand: (handler: ServerRequestHandler<ExecuteCommandParams, any, never, void>) => {},
     onInitialize: (handler: ServerRequestHandler<InitializeParams, InitializeResult, never, InitializeError>) => {},
     onInitialized: (handler: NotificationHandler<InitializedParams>) => {},
-}
+} as Connection
 
 const lspRouter = new LspRouter(serverLspConnectionMock, 'name')
 

runtimes/runtimes/base-runtime.ts

@@ -39,6 +39,7 @@ import {
     chatUpdateNotificationType,
     fileClickNotificationType,
     inlineChatRequestType,
+    contextCommandsNotificationType,
 } from '../protocol'
 import { createConnection } from 'vscode-languageserver/browser'
 import {
@@ -144,6 +145,7 @@ export const baseRuntime = (connections: { reader: MessageReader; writer: Messag
         openTab: params => lspConnection.sendRequest(openTabRequestType.method, params),
         sendChatUpdate: params => lspConnection.sendNotification(chatUpdateNotificationType.method, params),
         onFileClicked: handler => lspConnection.onNotification(fileClickNotificationType.method, handler),
+        sendContextCommands: params => lspConnection.sendNotification(contextCommandsNotificationType.method),
     }
 
     const identityManagement: IdentityManagement = {

runtimes/runtimes/chat/baseChat.ts

@@ -38,6 +38,8 @@ import {
     fileClickNotificationType,
     inlineChatRequestType,
     InlineChatParams,
+    ContextCommandParams,
+    contextCommandsNotificationType,
 } from '../../protocol'
 import { Chat } from '../../server-interface'
 
@@ -111,4 +113,8 @@ export class BaseChat implements Chat {
     public onFileClicked(handler: NotificationHandler<FileClickParams>) {
         this.connection.onNotification(fileClickNotificationType.method, handler)
     }
+
+    public sendContextCommands(params: ContextCommandParams) {
+        this.connection.sendNotification(contextCommandsNotificationType.method, params)
+    }
 }

runtimes/runtimes/encoding.test.ts

@@ -3,7 +3,7 @@ import { WebBase64Encoding } from './encoding'
 import sinon, { StubbedInstance, stubInterface } from 'ts-sinon'
 
 describe('WebBase64Encoding', () => {
-    const wdw = <WindowOrWorkerGlobalScope>{}
+    const wdw = {} as WindowOrWorkerGlobalScope
     const encoding = new WebBase64Encoding(wdw)
 
     it('encodes and decodes string', () => {

runtimes/runtimes/lsp/router/lspRouter.test.ts

@@ -28,13 +28,13 @@ describe('LspRouter', () => {
         encode: (value: string) => value,
         decode: (value: string) => value,
     }
-    const logging = <Logging>{
+    const logging = {
         log: sandbox.stub(),
         debug: sandbox.stub(),
         error: sandbox.stub(),
         warn: sandbox.stub(),
         info: sandbox.stub(),
-    }
+    } as Logging
     const lspConnection = stubLspConnection()
 
     let executeCommandHandler: RequestHandler<ExecuteCommandParams, any | undefined | null, void>
@@ -613,7 +613,7 @@ describe('LspRouter', () => {
     })
 
     function stubLspConnection(overrides = {}): Connection {
-        return <Connection>{
+        return {
             console: {
                 info: (message: any) => {},
             },
@@ -627,7 +627,7 @@ describe('LspRouter', () => {
             onNotification: (handler: any) => {},
             onDidChangeConfiguration: (handler: any) => {},
             ...overrides,
-        }
+        } as Connection
     }
 
     function newServer({

runtimes/runtimes/lsp/router/routerByServerName.test.ts

@@ -4,10 +4,10 @@ import { EventIdentifier, FollowupIdentifier } from '../../../protocol'
 import { RouterByServerName } from './routerByServerName'
 
 describe('RouterByServerName', () => {
-    const encoding = <Encoding>{
+    const encoding = {
         encode: value => Buffer.from(value).toString('base64'),
         decode: value => Buffer.from(value, 'base64').toString('utf-8'),
-    }
+    } as Encoding
     const serverName = 'Server_XXX'
 
     let router: RouterByServerName<Partial<EventIdentifier>, FollowupIdentifier>

runtimes/server-interface/chat.ts

@@ -21,6 +21,7 @@ import {
     ChatUpdateParams,
     FileClickParams,
     InlineChatParams,
+    ContextCommandParams,
 } from '../protocol'
 
 /**
@@ -46,4 +47,5 @@ export type Chat = {
     onFollowUpClicked: (handler: NotificationHandler<FollowUpClickParams>) => void
     sendChatUpdate: (params: ChatUpdateParams) => void
     onFileClicked: (handler: NotificationHandler<FileClickParams>) => void
+    sendContextCommands: (params: ContextCommandParams) => void
 }

types/chat.ts

@@ -18,6 +18,7 @@ export const OPEN_TAB_REQUEST_METHOD = 'aws/chat/openTab'
 export const CHAT_UPDATE_NOTIFICATION_METHOD = 'aws/chat/sendChatUpdate'
 export const FILE_CLICK_NOTIFICATION_METHOD = 'aws/chat/fileClick'
 export const INLINE_CHAT_REQUEST_METHOD = 'aws/chat/sendInlineChatPrompt'
+export const CONTEXT_COMMAND_NOTIFICATION_METHOD = 'aws/chat/sendContextCommands'
 
 export interface ChatItemAction {
     pillText: string
@@ -126,6 +127,7 @@ export interface QuickActionCommand {
     command: string
     description?: string
     placeholder?: string
+    icon?: IconType
 }
 
 /**
@@ -256,3 +258,21 @@ export interface FileClickParams {
     filePath: string
     action?: FileAction
 }
+
+export interface ContextCommandGroup {
+    groupName?: string
+    commands: ContextCommand[]
+}
+
+export type IconType = 'file' | 'folder' | 'code-block' | 'list-add' | 'magic' | 'help' | 'trash'
+
+export interface ContextCommand extends QuickActionCommand {
+    id?: string
+    route?: string[]
+    label?: 'file' | 'folder' | 'code'
+    children?: ContextCommandGroup[]
+}
+
+export interface ContextCommandParams {
+    contextCommandGroups: ContextCommandGroup[]
+}