feat: Add alias condition to code-block-lowlight (#3155)

ueberdosis · Sep 14, 2022 · b01ccff · b01ccff
1 parent fa96749
commit b01ccff
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/packages/extension-code-block-lowlight/src/lowlight-plugin.ts b/packages/extension-code-block-lowlight/src/lowlight-plugin.ts
@@ -1,4 +1,6 @@
 import { findChildren } from '@tiptap/core'
+// @ts-ignore
+import highlight from 'highlight.js/lib/core'
 import { Node as ProsemirrorNode } from 'prosemirror-model'
 import { Plugin, PluginKey } from 'prosemirror-state'
 import { Decoration, DecorationSet } from 'prosemirror-view'
@@ -30,6 +32,10 @@ function getHighlightNodes(result: any) {
   return result.value || result.children || []
 }
 
+function registered(aliasOrLanguage: string) {
+  return Boolean(highlight.getLanguage(aliasOrLanguage))
+}
+
 function getDecorations({
   doc,
   name,
@@ -43,7 +49,8 @@ function getDecorations({
       let from = block.pos + 1
       const language = block.node.attrs.language || defaultLanguage
       const languages = lowlight.listLanguages()
-      const nodes = language && languages.includes(language)
+
+      const nodes = language && (languages.includes(language) || registered(language))
         ? getHighlightNodes(lowlight.highlight(language, block.node.textContent))
         : getHighlightNodes(lowlight.highlightAuto(block.node.textContent))