app/assets/javascripts/content_editor/extensions/code_block_highlight.js


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

import { CodeBlockHighlight as BaseCodeBlockHighlight } from 'tiptap-extensions';

export default class GlCodeBlockHighlight extends BaseCodeBlockHighlight {
  get schema() {
    const baseSchema = super.schema;

    return {
      ...baseSchema,
      attrs: {
        params: {
          default: null,
        },
      },
      parseDOM: [
        {
          tag: 'pre',
          preserveWhitespace: 'full',
          getAttrs: (node) => {
            const code = node.querySelector('code');

            if (!code) {
              return null;
            }

            return {
              /* `params` is the name of the attribute that
                prosemirror-markdown uses to extract the language
                of a codeblock.
                https://github.com/ProseMirror/prosemirror-markdown/blob/master/src/to_markdown.js#L62
              */
              params: code.getAttribute('lang'),
            };
          },
        },
      ],
    };
  }
}