90 lines
2.0 KiB
JavaScript
90 lines
2.0 KiB
JavaScript
import { lowlight } from 'lowlight/lib/core';
|
|
import { textblockTypeInputRule } from '@tiptap/core';
|
|
import { base64DecodeUnicode } from '~/lib/utils/text_utility';
|
|
import { PARSE_HTML_PRIORITY_HIGHEST } from '../constants';
|
|
import languageLoader from '../services/code_block_language_loader';
|
|
import CodeBlockHighlight from './code_block_highlight';
|
|
|
|
const backtickInputRegex = /^```(mermaid|plantuml)[\s\n]$/;
|
|
|
|
export default CodeBlockHighlight.extend({
|
|
name: 'diagram',
|
|
|
|
isolating: true,
|
|
|
|
addOptions() {
|
|
return {
|
|
lowlight,
|
|
};
|
|
},
|
|
|
|
addAttributes() {
|
|
return {
|
|
language: {
|
|
default: null,
|
|
parseHTML: (element) => {
|
|
return element.dataset.diagram;
|
|
},
|
|
},
|
|
isDiagram: {
|
|
default: true,
|
|
},
|
|
showPreview: {
|
|
default: true,
|
|
},
|
|
};
|
|
},
|
|
|
|
parseHTML() {
|
|
return [
|
|
{
|
|
priority: PARSE_HTML_PRIORITY_HIGHEST,
|
|
tag: 'pre[lang="mermaid"]',
|
|
getAttrs: () => ({ language: 'mermaid' }),
|
|
},
|
|
{
|
|
priority: PARSE_HTML_PRIORITY_HIGHEST,
|
|
tag: '[data-diagram]',
|
|
getContent(element, schema) {
|
|
const source = base64DecodeUnicode(
|
|
element.dataset.diagramSrc.replace('data:text/plain;base64,', ''),
|
|
);
|
|
const node = schema.node('paragraph', {}, [schema.text(source)]);
|
|
return node.content;
|
|
},
|
|
},
|
|
];
|
|
},
|
|
|
|
renderHTML({ HTMLAttributes: { language, ...HTMLAttributes } }) {
|
|
return [
|
|
'div',
|
|
[
|
|
'pre',
|
|
{
|
|
language,
|
|
class: `content-editor-code-block code highlight`,
|
|
...HTMLAttributes,
|
|
},
|
|
['code', {}, 0],
|
|
],
|
|
];
|
|
},
|
|
|
|
addCommands() {
|
|
return {};
|
|
},
|
|
|
|
addInputRules() {
|
|
const getAttributes = (match) => languageLoader?.loadLanguageFromInputRule(match) || {};
|
|
|
|
return [
|
|
textblockTypeInputRule({
|
|
find: backtickInputRegex,
|
|
type: this.type,
|
|
getAttributes,
|
|
}),
|
|
];
|
|
},
|
|
});
|