subprojects/language-web/src/main/js/editor/decorationSetExtension.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
35
36
37
38
39

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

export type TransactionSpecFactory = (decorations: DecorationSet) => TransactionSpec;

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

  function transactionSpecFactory(decorations: DecorationSet) {
    return {
      effects: [
        setEffect.of(decorations),
      ],
    };
  }

  return [transactionSpecFactory, field];
}