/**
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 *
 */

import {IS_APPLE} from '@lexical/utils';
import {
  defineExtension,
  isExactShortcutMatch,
  isHTMLElement,
  safeCast,
  stopLexicalPropagation,
} from 'lexical';

import {namedSignals} from './namedSignals';
import {effect} from './signals';

function captureKeydown(e: KeyboardEvent) {
  const target = e.target;
  if (
    isExactShortcutMatch(e, 'a', {ctrlKey: !IS_APPLE, metaKey: IS_APPLE}) &&
    isHTMLElement(target) &&
    (target.tagName === 'INPUT' || target.tagName === 'TEXTAREA')
  ) {
    // prevents the event bubbling before it reaches the lexical scope
    target.addEventListener('keydown', stopLexicalPropagation, {once: true});
  }
}

export interface PreventSelectAllConfig {
  disabled: boolean;
}

/**
 * By default, lexical intercepts most events and dispatches the appropriate commands.
 * This extension prevents the keydown event propagating from input/textarea elements,
 * which are typically part of a decorator node, in order to stop dispatching the SELECT_ALL_COMMAND.
 *
 * When used as a dependency of SelectBlockExtension, its disabled state is
 * kept in sync with that extension.
 */
export const PreventSelectAllExtension = /* @__PURE__ */ defineExtension({
  build: (editor, config, state) => namedSignals(config),
  config: /* @__PURE__ */ safeCast<PreventSelectAllConfig>({
    disabled: false,
  }),
  name: '@lexical/extension/PreventSelectAll',
  register: (editor, config, state) => {
    const stores = state.getOutput();
    return effect(() => {
      if (!stores.disabled.value) {
        return editor.registerRootListener(rootElement => {
          if (rootElement) {
            rootElement.addEventListener('keydown', captureKeydown, true);
            return () =>
              rootElement.removeEventListener('keydown', captureKeydown, true);
          }
        });
      }
    });
  },
});