//
// Copyright 2024 DXOS.org
//

import { type Extension } from '@codemirror/state';
import { keymap } from '@codemirror/view';
import { vim } from '@replit/codemirror-vim';
import { vscodeKeymap } from '@replit/codemirror-vscode-keymap';

import { singleValueFacet } from '../../util/index.ts';

export type EditorInputConfig = {
  type?: string;
  ignoreEscape?: boolean;
};

export const editorInputMode = singleValueFacet<EditorInputConfig>({});

/**
 * Map of input-mode keymap bundles (default, vscode, vim).
 */
export const InputModeExtensions: { [mode: string]: Extension } = {
  default: [],
  vscode: [
    // https://github.com/replit/codemirror-vscode-keymap
    editorInputMode.of({ type: 'vscode' }),
    keymap.of(vscodeKeymap),
  ],
  vim: [
    // https://github.com/replit/codemirror-vim
    vim(),
    editorInputMode.of({ type: 'vim', ignoreEscape: true }),
    keymap.of([
      {
        key: 'Alt-Escape',
        run: (view) => {
          // Focus container for tab navigation.
          view.dom.parentElement?.focus();
          return true;
        },
      },
    ]),
  ],
};