Function addDecorationsToEditors

src/util/addDecorationsToEditor.ts:17–169

(
  hatTokenMap: IndividualHatMap,
  decorations: Decorations
)

}

export function addDecorationsToEditors(
  hatTokenMap: IndividualHatMap,
  decorations: Decorations
) {
  hatTokenMap.clear();

  var editors: vscode.TextEditor[];

  if (vscode.window.activeTextEditor == null) {
    editors = vscode.window.visibleTextEditors;
  } else {
    editors = vscode.window.visibleTextEditors.filter(
      (editor) => editor !== vscode.window.activeTextEditor
    );
    editors.unshift(vscode.window.activeTextEditor);
  }

  const tokens = concat(
    [],
    ...editors.map((editor) => {
      const displayLineMap = getDisplayLineMap(editor);

      const tokens: Token[] = flatten(
        editor.visibleRanges.map((range) =>
          getTokensInRange(editor, range).map((partialToken) => ({
            ...partialToken,
            displayLine: displayLineMap.get(partialToken.range.start.line)!,
            editor,
            expansionBehavior: {
              start: { type: "regex", regex: TOKEN_MATCHER },
              end: { type: "regex", regex: TOKEN_MATCHER },
            },
          }))
        )
      );

      tokens.sort(
        getTokenComparator(
          displayLineMap.get(editor.selection.active.line)!,
          editor.selection.active.character
        )
      );

      return tokens;
    })
  );

  const characterTokens: {
    [key: string]: Map<Token, CharacterTokenInfo>;
  } = {};

  tokens.forEach((token, tokenIdx) => {
    [...token.text].forEach((character, characterIdx) => {
      var characterTokenMap: Map<Token, CharacterTokenInfo>;

      if (character in characterTokens) {
        characterTokenMap = characterTokens[character];
      } else {