1 files changed, 8 insertions, 1 deletions

diff --git a/language-web/src/main/js/editor/EditorStore.ts b/language-web/src/main/js/editor/EditorStore.ts
index 78cf763c..f47f47a0 100644
--- a/language-web/src/main/js/editor/EditorStore.ts
+++ b/language-web/src/main/js/editor/EditorStore.ts
@@ -30,6 +30,7 @@ import {
   TransactionSpec,
 } from '@codemirror/state';
 import {
+  DecorationSet,
   drawSelection,
   EditorView,
   highlightActiveLine,
@@ -43,8 +44,9 @@ import {
 } from 'mobx';
 
 import { problemLanguageSupport } from '../language/problemLanguageSupport';
-import { getLogger } from '../utils/logger';
+import { semanticHighlighting, setSemanticHighlighting } from './semanticHighlighting';
 import type { ThemeStore } from '../theme/ThemeStore';
+import { getLogger } from '../utils/logger';
 import { XtextClient } from '../xtext/XtextClient';
 
 const log = getLogger('editor.EditorStore');
@@ -103,6 +105,7 @@ export class EditorStore {
           top: true,
           matchCase: true,
         }),
+        semanticHighlighting,
         // We add the gutters to `extensions` in the order we want them to appear.
         foldGutter(),
         lineNumbers(),
@@ -201,6 +204,10 @@ export class EditorStore {
     return null;
   }
 
+  updateSemanticHighlighting(decorations: DecorationSet): void {
+    this.dispatch(setSemanticHighlighting(decorations));
+  }
+
   /**
    * @returns `true` if there is history to undo
    */