/*
 * SPDX-FileCopyrightText: 2021-2023 The Refinery Authors <https://refinery.tools/>
 *
 * SPDX-License-Identifier: EPL-2.0
 */

import {
  closeBrackets,
  closeBracketsKeymap,
  autocompletion,
  completionKeymap,
} from '@codemirror/autocomplete';
import {
  defaultKeymap,
  history,
  historyKeymap,
  indentWithTab,
} from '@codemirror/commands';
import {
  bracketMatching,
  codeFolding,
  foldGutter,
  foldKeymap,
  indentOnInput,
  syntaxHighlighting,
} from '@codemirror/language';
import { lintKeymap, lintGutter } from '@codemirror/lint';
import { search, searchKeymap } from '@codemirror/search';
import { EditorState } from '@codemirror/state';
import {
  drawSelection,
  highlightActiveLine,
  highlightActiveLineGutter,
  highlightSpecialChars,
  keymap,
  lineNumbers,
  rectangularSelection,
} from '@codemirror/view';
import { classHighlighter } from '@lezer/highlight';

import problemLanguageSupport from '../language/problemLanguageSupport';

import type EditorStore from './EditorStore';
import SearchPanel from './SearchPanel';
import exposeDiagnostics from './exposeDiagnostics';
import findOccurrences from './findOccurrences';
import semanticHighlighting from './semanticHighlighting';

export default function createEditorState(
  initialValue: string,
  store: EditorStore,
): EditorState {
  return EditorState.create({
    doc: initialValue,
    extensions: [
      autocompletion({
        activateOnTyping: true,
        override: [(context) => store.contentAssist(context)],
      }),
      closeBrackets(),
      bracketMatching(),
      drawSelection(),
      EditorState.allowMultipleSelections.of(true),
      exposeDiagnostics,
      findOccurrences,
      highlightActiveLine(),
      highlightActiveLineGutter(),
      highlightSpecialChars(),
      history(),
      indentOnInput(),
      rectangularSelection(),
      search({
        createPanel(view) {
          return new SearchPanel(view, store.searchPanel);
        },
      }),
      syntaxHighlighting(classHighlighter),
      semanticHighlighting,
      // We add the gutters to `extensions` in the order we want them to appear.
      lintGutter(),
      lineNumbers(),
      codeFolding({
        placeholderDOM(_view, onClick) {
          const button = document.createElement('button');
          button.className = 'problem-editor-foldPlaceholder';
          button.ariaLabel = 'Unfold lines';
          const span = document.createElement('span');
          span.innerText = '...';
          button.appendChild(span);
          button.addEventListener('click', onClick);
          return button;
        },
      }),
      foldGutter({
        markerDOM(open) {
          const div = document.createElement('div');
          div.className = [
            'problem-editor-foldMarker',
            `problem-editor-foldMarker-${open ? 'open' : 'closed'}`,
          ].join(' ');
          return div;
        },
      }),
      keymap.of([
        { key: 'Mod-Shift-f', run: () => store.formatText() },
        ...closeBracketsKeymap,
        ...completionKeymap,
        ...foldKeymap,
        ...historyKeymap,
        indentWithTab,
        // Override keys in `lintKeymap` to go through the `EditorStore`.
        { key: 'Mod-Shift-m', run: () => store.lintPanel.open() },
        ...lintKeymap,
        // Override keys in `searchKeymap` to go through the `EditorStore`.
        {
          key: 'Mod-f',
          run: () => store.searchPanel.open(),
          scope: 'editor search-panel',
        },
        {
          key: 'Escape',
          run: () => store.searchPanel.close(),
          scope: 'editor search-panel',
        },
        ...searchKeymap,
        ...defaultKeymap,
      ]),
      problemLanguageSupport(),
    ],
  });
}