Function PrismPlugin

packages/editor/src/extensions/prism-plugin.ts:127–242

({
  name,
  defaultLanguage,
  defaultTheme,
}: {
  name: string;
  defaultLanguage: string;
  defaultTheme: string;
})

}

export function PrismPlugin({
  name,
  defaultLanguage,
  defaultTheme,
}: {
  name: string;
  defaultLanguage: string;
  defaultTheme: string;
}) {
  if (!defaultLanguage) {
    throw Error('You must specify the defaultLanguage parameter');
  }

  const loadingLanguages = new Set<string>();
  let pluginView: EditorView | null = null;

  const onLanguageLoaded = (language: string) => {
    if (pluginView) {
      pluginView.dispatch(
        pluginView.state.tr.setMeta(PRISM_LANGUAGE_LOADED_META, language),
      );
    }
  };

  const prismjsPlugin: Plugin<DecorationSet> = new Plugin({
    key: new PluginKey('prism'),

    view(view) {
      pluginView = view;
      return {
        destroy() {
          pluginView = null;
        },
      };
    },

    state: {
      init: (_, { doc }) => {
        return getDecorations({
          doc,
          name,
          defaultLanguage,
          defaultTheme,
          loadingLanguages,
          onLanguageLoaded,
        });
      },
      apply: (transaction, decorationSet, oldState, newState) => {
        const oldNodeName = oldState.selection.$head.parent.type.name;
        const newNodeName = newState.selection.$head.parent.type.name;

        const oldNodes = findChildren(
          oldState.doc,
          (node) => node.type.name === name,
        );
        const newNodes = findChildren(
          newState.doc,
          (node) => node.type.name === name,