/**
 * 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 type {LexicalComposerContextWithEditor} from '@lexical/react/LexicalComposerContext';
import type {JSX} from 'react';

/**
 * Props for an `EditorChildrenComponent` (see {@link ReactConfig}): the composer
 * `context`, the `contentEditable` element to render (or `null`), and any
 * additional `children`.
 */
export interface EditorChildrenComponentProps {
  context: LexicalComposerContextWithEditor;
  contentEditable: null | JSX.Element;
  children?: React.ReactNode;
}

export type EditorChildrenComponentType = (
  props: EditorChildrenComponentProps,
) => JSX.Element | null;

/**
 * Props passed to a decorator component registered through {@link ReactConfig}:
 * the composer `context` (the `[editor, context]` tuple) for the editor the
 * decorator is rendered in.
 */
export interface DecoratorComponentProps {
  context: LexicalComposerContextWithEditor;
}
export type DecoratorComponentType =
  | JSX.Element
  | ((props: DecoratorComponentProps) => JSX.Element | null);

export interface EditorComponentProps {
  /**
   * The EditorChildrenComponent from the config
   */
  EditorChildrenComponent: EditorChildrenComponentType;
  /**
   * The children to pass to EditorChildrenComponent
   */
  children?: React.ReactNode;
  /**
   * contentEditable from the config
   */
  contentEditable: JSX.Element | null;
  /**
   * ErrorBoundary from the config
   */
  ErrorBoundary: ErrorBoundaryType;
}

export type EditorComponentType = (
  /** Optional overrides to any of the props, typically only children is used */
  props: Partial<EditorComponentProps>,
) => JSX.Element;

/**
 * Configuration for the {@link ReactExtension}, controlling how the editor is
 * rendered in React: the root `contentEditable` element, the `ErrorBoundary`
 * used for decorators, the `EditorChildrenComponent` that lays out the editor's
 * children, and any additional `decorators` to render.
 */
export interface ReactConfig {
  /**
   * The default root element of the editor as JSX. Uses `<ContentEditable />`
   * from \@lexical/react/ContentEditable by default,
   * but may be null or another component.
   *
   * This component is responsible for calling `editor.setRootElement(elem)`.
   */
  contentEditable: JSX.Element | null;
  /**
   * The ErrorBoundary used for rendering decorators in the editor. By default
   * it is `ErrorBoundary` from \@lexical/react/ErrorBoundary.
   */
  ErrorBoundary: ErrorBoundaryType;
  /**
   * The component that renders the children of the editor context, by default
   * it is {@link DefaultEditorChildrenComponent} which takes the given props
   * and renders them in this order:
   *
   * - contentEditable
   * - children
   */
  EditorChildrenComponent: EditorChildrenComponentType;

  /**
   * An array of JSX or components that return JSX that should be rendered
   * as children of Component. These will be merged by array concatenation.
   */
  decorators: readonly DecoratorComponentType[];
}

/**
 * The output of the {@link ReactExtension}: the editor `Component` used to
 * render the editor, and the composer `context` (equivalent to
 * {@link useLexicalComposerContext}).
 */
export interface ReactOutputs {
  /**
   * The editor component, this can be used by Extensions that depend on this to
   * render the editor such as {@link ReactPluginHostExtension} or internally by
   * {@link LexicalExtensionComposer}.
   *
   * All props have defaults based on the config and editor state, but may be
   * overridden.
   */
  Component: EditorComponentType;
  /**
   * This is equivalent to useLexicalComposerContext() from \@lexical/react/LexicalComposerContext.
   */
  context: LexicalComposerContextWithEditor;
}

export interface ErrorBoundaryProps {
  children: JSX.Element;
  onError: (error: Error) => void;
}

export type ErrorBoundaryType =
  | React.ComponentClass<ErrorBoundaryProps>
  | React.FC<ErrorBoundaryProps>;