Refresh CodeLens using LSP request instead of document document modif…

…y trick

extension/src/emulator/server/language-server.ts

@@ -9,6 +9,7 @@ import { Mutex } from 'async-mutex'
 import { exec } from 'child_process'
 import { verifyEmulator } from '../local/emulatorScanner'
 import { delay } from '../../utils/utils'
+import { CodeLensParams, CodeLensRequest, ExecuteCommandRequest, TextDocumentIdentifier } from 'vscode-languageclient'
 
 // Identities for commands handled by the Language server
 const CREATE_ACCOUNT_SERVER = 'cadence.server.flow.createAccount'
@@ -155,12 +156,18 @@ export class LanguageServerAPI {
   }
 
   async #sendRequest (cmd: string, args: any[] = []): Promise<any> {
-    return await this.client?.sendRequest('workspace/executeCommand', {
+    return await this.client?.sendRequest(ExecuteCommandRequest.type, {
       command: cmd,
       arguments: args
     })
   }
 
+  async refreshCodeLens () {
+    const textDocumentUri = window.activeTextEditor?.document.uri
+    if (textDocumentUri == null) return
+    return await this.client?.sendRequest(CodeLensRequest.type, <CodeLensParams>{ textDocument: TextDocumentIdentifier.create(textDocumentUri.path) })
+  }
+
   async reset (): Promise<void> {
     const enableFlow = await verifyEmulator()
     await this.restart(enableFlow)

extension/src/extension.ts

@@ -1,7 +1,6 @@
 /* The extension */
 import { EmulatorState, EmulatorController } from './emulator/emulator-controller'
 import { CommandController } from './commands/command-controller'
-import { refreshCodeLenses } from './utils/codelens'
 import { Account } from './emulator/account'
 import { UIController } from './ui/ui-controller'
 import { ExtensionContext } from 'vscode'
@@ -69,7 +68,7 @@ export class Extension {
 
     // Update UI
     this.#uiCtrl.emulatorStateChanged()
-    refreshCodeLenses()
+    this.emulatorCtrl.api.refreshCodeLens()
   }
 
   checkDependencies (): void {