Function getDecorations

packages/editor/src/extensions/prism-plugin.ts:54–125

({
  doc,
  name,
  defaultLanguage,
  defaultTheme,
  loadingLanguages,
  onLanguageLoaded,
}: {
  doc: ProsemirrorNode;
  name: string;
  defaultLanguage: string | null | undefined;
  defaultTheme: string | null | undefined;
  loadingLanguages: Set<string>;
  onLanguageLoaded: (language: string) => void;
})

}

function getDecorations({
  doc,
  name,
  defaultLanguage,
  defaultTheme,
  loadingLanguages,
  onLanguageLoaded,
}: {
  doc: ProsemirrorNode;
  name: string;
  defaultLanguage: string | null | undefined;
  defaultTheme: string | null | undefined;
  loadingLanguages: Set<string>;
  onLanguageLoaded: (language: string) => void;
}) {
  const decorations: Decoration[] = [];

  findChildren(doc, (node) => node.type.name === name).forEach((block) => {
    let from = block.pos + 1;
    const language = block.node.attrs.language || defaultLanguage;
    const theme = block.node.attrs.theme || defaultTheme;
    let html = '';

    try {
      if (!registeredLang(language) && !loadingLanguages.has(language)) {
        loadingLanguages.add(language);
        import(/* @vite-ignore */ `prismjs/components/prism-${language}`)
          .then(() => {
            loadingLanguages.delete(language);
            onLanguageLoaded(language);
          })
          .catch(() => {
            loadingLanguages.delete(language);
          });
      }

      if (!hasPrismThemeLoaded(theme)) {
        loadPrismTheme(theme);
      }

      html = Prism.highlight(
        block.node.textContent,
        Prism.languages[language],
        language,
      );
    } catch {
      html = Prism.highlight(
        block.node.textContent,
        Prism.languages.javascript,
        'js',
      );
    }

    const nodes = getHighlightNodes(html);

    parseNodes(nodes as RefractorNode[]).forEach((node) => {
      const to = from + node.text.length;