side quests for lsp server (#2119)

* all

Signed-off-by: Jess Frazelle <github@jessfraz.com>

* more

Signed-off-by: Jess Frazelle <github@jessfraz.com>

* side quests only

Signed-off-by: Jess Frazelle <github@jessfraz.com>

* updates

Signed-off-by: Jess Frazelle <github@jessfraz.com>

* A snapshot a day keeps the bugs away! 📷🐛 (OS: ubuntu)

* updates

Signed-off-by: Jess Frazelle <github@jessfraz.com>

---------

Signed-off-by: Jess Frazelle <github@jessfraz.com>
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>

src/editor/plugins/lsp/index.ts

@@ -6,6 +6,10 @@ import { CopilotLspCompletionParams } from 'wasm-lib/kcl/bindings/CopilotLspComp
 import { CopilotCompletionResponse } from 'wasm-lib/kcl/bindings/CopilotCompletionResponse'
 import { CopilotAcceptCompletionParams } from 'wasm-lib/kcl/bindings/CopilotAcceptCompletionParams'
 import { CopilotRejectCompletionParams } from 'wasm-lib/kcl/bindings/CopilotRejectCompletionParams'
+import { UpdateUnitsParams } from 'wasm-lib/kcl/bindings/UpdateUnitsParams'
+import { UpdateCanExecuteParams } from 'wasm-lib/kcl/bindings/UpdateCanExecuteParams'
+import { UpdateUnitsResponse } from 'wasm-lib/kcl/bindings/UpdateUnitsResponse'
+import { UpdateCanExecuteResponse } from 'wasm-lib/kcl/bindings/UpdateCanExecuteResponse'
 
 // https://microsoft.github.io/language-server-protocol/specifications/specification-current/
 
@@ -21,12 +25,17 @@ interface LSPRequestMap {
     LSP.SemanticTokensParams,
     LSP.SemanticTokens
   ]
+  'textDocument/formatting': [
+    LSP.DocumentFormattingParams,
+    LSP.TextEdit[] | null
+  ]
+  'textDocument/foldingRange': [LSP.FoldingRangeParams, LSP.FoldingRange[]]
   'copilot/getCompletions': [
     CopilotLspCompletionParams,
     CopilotCompletionResponse
   ]
-  'copilot/notifyAccepted': [CopilotAcceptCompletionParams, any]
-  'copilot/notifyRejected': [CopilotRejectCompletionParams, any]
+  'kcl/updateUnits': [UpdateUnitsParams, UpdateUnitsResponse | null]
+  'kcl/updateCanExecute': [UpdateCanExecuteParams, UpdateCanExecuteResponse]
 }
 
 // Client to server
@@ -39,6 +48,8 @@ interface LSPNotifyMap {
   'workspace/didCreateFiles': LSP.CreateFilesParams
   'workspace/didRenameFiles': LSP.RenameFilesParams
   'workspace/didDeleteFiles': LSP.DeleteFilesParams
+  'copilot/notifyAccepted': CopilotAcceptCompletionParams
+  'copilot/notifyRejected': CopilotRejectCompletionParams
 }
 
 export interface LanguageServerClientOptions {
@@ -56,11 +67,11 @@ export interface LanguageServerOptions {
 
 export class LanguageServerClient {
   private client: Client
-  private name: string
+  readonly name: string
 
   public ready: boolean
 
-  private plugins: LanguageServerPlugin[]
+  readonly plugins: LanguageServerPlugin[]
 
   public initializePromise: Promise<void>
 
@@ -185,6 +196,22 @@ export class LanguageServerClient {
     return await this.request('textDocument/hover', params)
   }
 
+  async textDocumentFormatting(params: LSP.DocumentFormattingParams) {
+    const serverCapabilities = this.getServerCapabilities()
+    if (!serverCapabilities.documentFormattingProvider) {
+      return
+    }
+    return await this.request('textDocument/formatting', params)
+  }
+
+  async textDocumentFoldingRange(params: LSP.FoldingRangeParams) {
+    const serverCapabilities = this.getServerCapabilities()
+    if (!serverCapabilities.foldingRangeProvider) {
+      return
+    }
+    return await this.request('textDocument/foldingRange', params)
+  }
+
   async textDocumentCompletion(params: LSP.CompletionParams) {
     const serverCapabilities = this.getServerCapabilities()
     if (!serverCapabilities.completionProvider) {
@@ -227,22 +254,34 @@ export class LanguageServerClient {
   async accept(uuid: string) {
     const badUids = this.queuedUids.filter((u) => u !== uuid)
     this.queuedUids = []
-    await this.acceptCompletion({ uuid })
-    await this.rejectCompletions({ uuids: badUids })
+    this.acceptCompletion({ uuid })
+    this.rejectCompletions({ uuids: badUids })
   }
 
   async reject() {
     const badUids = this.queuedUids
     this.queuedUids = []
-    return await this.rejectCompletions({ uuids: badUids })
+    this.rejectCompletions({ uuids: badUids })
   }
 
-  async acceptCompletion(params: CopilotAcceptCompletionParams) {
-    return await this.request('copilot/notifyAccepted', params)
+  acceptCompletion(params: CopilotAcceptCompletionParams) {
+    this.notify('copilot/notifyAccepted', params)
   }
 
-  async rejectCompletions(params: CopilotRejectCompletionParams) {
-    return await this.request('copilot/notifyRejected', params)
+  rejectCompletions(params: CopilotRejectCompletionParams) {
+    this.notify('copilot/notifyRejected', params)
+  }
+
+  async updateUnits(
+    params: UpdateUnitsParams
+  ): Promise<UpdateUnitsResponse | null> {
+    return await this.request('kcl/updateUnits', params)
+  }
+
+  async updateCanExecute(
+    params: UpdateCanExecuteParams
+  ): Promise<UpdateCanExecuteResponse> {
+    return await this.request('kcl/updateCanExecute', params)
   }
 
   private processNotifications(notification: LSP.NotificationMessage) {