language-web/src/main/js/editor/semanticHighlighting.ts


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

import { StateEffect, StateField, TransactionSpec } from '@codemirror/state';
import { EditorView, Decoration, DecorationSet } from '@codemirror/view';

const setSemanticHighlightingEffect = StateEffect.define<DecorationSet>();

export function setSemanticHighlighting(decorations: DecorationSet): TransactionSpec {
  return {
    effects: [
      setSemanticHighlightingEffect.of(decorations),
    ],
  };
}

export const semanticHighlighting = StateField.define<DecorationSet>({
  create() {
    return Decoration.none;
  },
  update(currentDecorations, transaction) {
    let newDecorations: DecorationSet | null = null;
    transaction.effects.forEach((effect) => {
      if (effect.is(setSemanticHighlightingEffect)) {
        newDecorations = effect.value;
      }
    });
    if (newDecorations === null) {
      if (transaction.docChanged) {
        return currentDecorations.map(transaction.changes);
      }
      return currentDecorations;
    }
    return newDecorations;
  },
  provide: (f) => EditorView.decorations.from(f),
});