bkui-cli-template-webpack4-saas/template/doc/build/md-highlight.js

bkui-cli-template-webpack4-saas

/**
 * @file highlight.js markdown-it 插件
 *      https://github.com/valeriangalliat/markdown-it-highlightjs/
 */

import hljs from './hljs'

const maybe = f => {
    try {
        return f()
    } catch (e) {
        return false
    }
}

// Highlight with given language.
const highlight = (code, lang) =>
    maybe(() => hljs.highlight(lang, code, true).value) || ''

// Highlight with given language or automatically.
const highlightAuto = (code, lang) =>
    lang
        ? highlight(code, lang)
        : maybe(() => hljs.highlightAuto(code).value) || ''

// Wrap a render function to add `hljs` class to code blocks.
const wrap = render =>
    function (...args) {
        return render.apply(this, args)
            .replace('<code class="', '<code class="hljs ')
            .replace('<code>', '<code class="hljs">')
    }

const highlightjs = (md, opts) => {
    opts = Object.assign({}, highlightjs.defaults, opts)

    md.renderer.rules.table_open = () => '<table class="table">'

    md.options.highlight = opts.auto ? highlightAuto : highlight
    md.renderer.rules.fence = wrap(md.renderer.rules.fence)

    if (opts.code) {
        md.renderer.rules.code_block = wrap(md.renderer.rules.code_block)
    }
}

highlightjs.defaults = {
    auto: true,
    code: true
}

export default highlightjs