/**
 * 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 {
  getExtensionDependencyFromEditor,
  getPeerDependencyFromEditor,
} from '@lexical/extension';
import {useLexicalComposerContext} from '@lexical/react/LexicalComposerContext';
import {
  type AnyLexicalExtension,
  type LexicalExtensionDependency,
  type OutputComponentExtension,
} from 'lexical';

export function useExtensionDependency<Extension extends AnyLexicalExtension>(
  extension: Extension,
): LexicalExtensionDependency<Extension> {
  return getExtensionDependencyFromEditor(
    useLexicalComposerContext()[0],
    extension,
  );
}

export function useOptionalExtensionDependency<
  Extension extends AnyLexicalExtension,
>(extension: Extension): undefined | LexicalExtensionDependency<Extension> {
  return usePeerExtensionDependency<typeof extension>(extension.name);
}

export function usePeerExtensionDependency<
  Extension extends AnyLexicalExtension,
>(
  extensionName: Extension['name'],
): undefined | LexicalExtensionDependency<Extension> {
  return getPeerDependencyFromEditor(
    useLexicalComposerContext()[0],
    extensionName,
  );
}

/**
 * Use a Component from the given Extension that uses the ReactExtension convention
 * of exposing a Component property in its output.
 *
 * @param extension - An extension with a Component property in the output
 * @returns `getExtensionConfigFromEditor(useLexicalComposerContext()[0], extension).Component`
 */
export function useExtensionComponent<
  // eslint-disable-next-line @typescript-eslint/no-explicit-any -- contravariant in props
  OutputComponent extends React.ComponentType<any>,
>(extension: OutputComponentExtension<OutputComponent>): OutputComponent {
  return useExtensionDependency(extension).output.Component;
}