ane-markdown-loader/compiler.js

avalon 组件库 ane 依赖的文档工具，将 markdown 转为 avalon 组件

var loaderUtils = require('loader-utils');
var MarkdownIt = require('markdown-it');
var hljs = require('highlight.js');
var path = require('path');

module.exports = function (source) {
    this.cacheable();

    var options = loaderUtils.getOptions(this) || {};
    
    var markdownOptions = {
        html: true,
        highlight: function (code) {
            return hljs.highlightAuto(code).value;
        },
        linkify: true
    };
    if (!options.highlight) {
        delete markdownOptions.highlight;
    }

    var md = MarkdownIt(markdownOptions);

    var importScript = [];
    var script = [];
    var newTokens = [];
    var result;

    if (!options.noEval) {
        var tokens = md.parse(source);
        tokens.forEach(function (token) {
            if (token.type === 'fence' && token.tag === 'code') {
                if (token.info.trim() === 'html') {
                    var htmlToken = md.parse('<div></div>')[0];
                    htmlToken.content = token.content;
                    newTokens.push(htmlToken);

                    token.attrSet(':skip', true);
                }
                if (token.info.trim() === 'js') {
                    var tempImportScript = '';
                    token.content = token.content.replace(/import[\s\S]*?(from)?[\s\S]*?\n/g, function ($0) {
                        tempImportScript += $0;
                        if (importScript.every(function (s) { return s.replace(/\s|\n/g, '') !== $0.replace(/\s|\n/g, '') })) {
                            importScript.push($0);
                        }
                        return '';
                    })
                    script.push(token.content);
                    token.content = tempImportScript + token.content;
                    
                    token.attrSet(':skip', true);
                }
            }
            if (token.type === 'table_open') {
                token.attrSet('class', 'table table-bordered');
            }
            newTokens.push(token);
        });
        result = md.renderer.render(newTokens, md.options);
    } else {
        result = md.render(source);
    }

    var componentName = 'component-demo-' + path.relative(__dirname, this.resourcePath)
                                                .replace(new RegExp('\\' + path.sep, 'g'), '/')
                                                .match(/(\.\.\/)*(.*)\.md/)[2]
                                                .replace(/\//g, '-')
                                                .replace(/ms-/g, '');

    var component = [
        importScript.join(''),
        'export const name = \'' + componentName + '\';' +
        'avalon.component(name, {' +
        '    template: `<div>' + result + '</div>`' +
        '});' +
        script.join('\n')
    ];
    return component.join('\n');
}