feat(types): introduce new completion API for edits suggestion (#635)

## Problem

## Solution

<!---
    REMINDER:
    - Read CONTRIBUTING.md first.
    - Add test coverage for your changes.
    - Link to related issues/commits.
    - Testing: how did you test your changes?
    - Screenshots if applicable
-->

## License

By submitting this pull request, I confirm that my contribution is made
under the terms of the Apache 2.0 license.

Author: Will-ShaoHua

runtimes/protocol/editCompletions.ts

@@ -0,0 +1,16 @@
+import {
+    InlineCompletionItem,
+    InlineCompletionList,
+    InlineCompletionParams,
+    InlineCompletionRegistrationOptions,
+} from './lsp'
+
+import { ProtocolRequestType } from 'vscode-languageserver-protocol'
+
+export const editCompletionRequestType = new ProtocolRequestType<
+    InlineCompletionParams,
+    InlineCompletionList | InlineCompletionItem[] | null,
+    InlineCompletionItem[],
+    void,
+    InlineCompletionRegistrationOptions
+>('aws/textDocument/editCompletion')

runtimes/protocol/lsp.ts

@@ -28,6 +28,7 @@ export * from 'vscode-languageserver-protocol'
 // Custom Runtimes LSP extensions
 export * from './inlineCompletionWithReferences'
 export * from './inlineCompletions'
+export * from './editCompletions'
 
 // AutoParameterStructuresProtocolRequestType allows ParameterStructures both by-name and by-position
 export class AutoParameterStructuresProtocolRequestType<P, R, PR, E, RO>

runtimes/runtimes/base-runtime.ts

@@ -104,6 +104,7 @@ import { getClientInitializeParamsHandlerFactory } from './util/lspCacheUtil'
 import { newAgent } from './agent'
 import { ShowSaveFileDialogRequestType } from '../protocol/window'
 import { joinUnixPaths } from './util/pathUtil'
+import { editCompletionRequestType } from '../protocol/editCompletions'
 
 declare const self: WindowOrWorkerGlobalScope
 
@@ -253,6 +254,7 @@ export const baseRuntime = (connections: { reader: MessageReader; writer: Messag
             getClientInitializeParams: getClientInitializeParamsHandlerFactory(lspRouter),
             onCompletion: handler => lspConnection.onCompletion(handler),
             onInlineCompletion: handler => lspConnection.onRequest(inlineCompletionRequestType, handler),
+            onEditCompletion: handler => lspConnection.onRequest(editCompletionRequestType, handler),
             didChangeConfiguration: lspServer.setDidChangeConfigurationHandler,
             onDidFormatDocument: handler => lspConnection.onDocumentFormatting(handler),
             onDidOpenTextDocument: handler => documentsObserver.callbacks.onDidOpenTextDocument(handler),
@@ -289,6 +291,7 @@ export const baseRuntime = (connections: { reader: MessageReader; writer: Messag
                 onGetConfigurationFromServer: lspServer.setServerConfigurationHandler,
                 onInlineCompletionWithReferences: handler =>
                     lspConnection.onRequest(inlineCompletionWithReferencesRequestType, handler),
+                onEditCompletion: handler => lspConnection.onRequest(editCompletionRequestType, handler),
                 onLogInlineCompletionSessionResults: handler => {
                     lspConnection.onNotification(logInlineCompletionSessionResultsNotificationType, handler)
                 },

runtimes/runtimes/standalone.ts

@@ -95,6 +95,7 @@ import { ShowSaveFileDialogRequestType } from '../protocol/window'
 import { getTelemetryReasonDesc } from './util/shared'
 import { writeSync } from 'fs'
 import { format } from 'util'
+import { editCompletionRequestType } from '../protocol/editCompletions'
 
 // Honor shared aws config file
 if (checkAWSConfigFile()) {
@@ -379,6 +380,7 @@ export const standalone = (props: RuntimeProps) => {
                 getClientInitializeParams: getClientInitializeParamsHandlerFactory(lspRouter),
                 onCompletion: handler => lspConnection.onCompletion(handler),
                 onInlineCompletion: handler => lspConnection.onRequest(inlineCompletionRequestType, handler),
+                onEditCompletion: handler => lspConnection.onRequest(editCompletionRequestType, handler),
                 didChangeConfiguration: lspServer.setDidChangeConfigurationHandler,
                 onDidFormatDocument: handler => lspConnection.onDocumentFormatting(handler),
                 onDidOpenTextDocument: handler => documentsObserver.callbacks.onDidOpenTextDocument(handler),
@@ -424,6 +426,7 @@ export const standalone = (props: RuntimeProps) => {
                     onGetConfigurationFromServer: lspServer.setServerConfigurationHandler,
                     onInlineCompletionWithReferences: handler =>
                         lspConnection.onRequest(inlineCompletionWithReferencesRequestType, handler),
+                    onEditCompletion: handler => lspConnection.onRequest(editCompletionRequestType, handler),
                     onLogInlineCompletionSessionResults: handler => {
                         lspConnection.onNotification(logInlineCompletionSessionResultsNotificationType, handler)
                     },

runtimes/server-interface/lsp.ts

@@ -118,6 +118,13 @@ export type Lsp = {
             void
         >
     ) => void
+    onEditCompletion: (
+        handler: RequestHandler<
+            InlineCompletionParams,
+            InlineCompletionItem[] | InlineCompletionList | undefined | null,
+            void
+        >
+    ) => void
     onCompletion: (
         handler: RequestHandler<CompletionParams, CompletionItem[] | CompletionList | undefined | null, void>
     ) => void
@@ -163,6 +170,13 @@ export type Lsp = {
                 void
             >
         ) => void
+        onEditCompletion: (
+            handler: RequestHandler<
+                InlineCompletionWithReferencesParams,
+                InlineCompletionItemWithReferences[] | InlineCompletionListWithReferences | undefined | null,
+                void
+            >
+        ) => void
         onLogInlineCompletionSessionResults: (
             handler: NotificationHandler<LogInlineCompletionSessionResultsParams>
         ) => void