More robust highlighting

1 files changed, 8 insertions, 0 deletions

diff --git a/code/src/extension.ts b/code/src/extension.ts
index 554ac8af3..81153b7e6 100644
--- a/code/src/extension.ts
+++ b/code/src/extension.ts
@@ -134,6 +134,14 @@ export function activate(context: vscode.ExtensionContext) {
             textDocumentContentProvider.eventEmitter.fire(uris.syntaxTree)
         })
     }, null, context.subscriptions)
+    vscode.window.onDidChangeActiveTextEditor(async (editor) => {
+        if (!editor || editor.document.languageId != 'rust') return
+        let params: lc.TextDocumentIdentifier = {
+            uri: editor.document.uri.toString()
+        }
+        let decorations = await client.sendRequest<Decoration[]>("m/decorationsRequest", params)
+        setHighlights(editor, decorations)
+    })
 }
 
 // We need to order this after LS updates, but there's no API for that.