cherry-doc/lib/walkmd.js

/**
 * @file 遍历所有md文件，生成data文件。
 * @author zhangpeng53
 */
var fs = require('fs');
var path = require('path');
var walk = require('walk');
var yamlFront = require('yaml-front-matter');
var sh = require('child_process').execSync;
var chalk = require('chalk');

module.exports = function walkMD(config) {

    return new Promise(function (resolve, reject) {
        var dist = config.theme + '/tmp';
        var mdData = {};
        var source = {};

        function parseData(arr, category, page, rank, file) {
            var key = category + "_" + page;

            var have = !!arr.length;

            have = arr.some(function (item) {
                if (item.key === key) {
                    item.files[rank] = file;
                    return true;
                }

                return false;
            });

            if (!have) {
                arr.push({
                    page: page,
                    key: key,
                    files: [file]
                });
            }
        }

        function writeFiles(data) {
            for (var k in data) {
                var category = data[k];
                category.forEach(function (item) {
                    var key = item.key;
                    var outFile = '';
                    item.files.forEach(function (f) {
                        outFile += 'require(\'' + f + '\'),\n';
                    });

                    source[key] = './tmp/__' + key;

                    var out = 'module.exports = [\n' + outFile + ']';
                    fs.writeFileSync(dist + ('/__' + key + '.js'), out);
                });
            }
        }

        var walker = walk.walk(config.root);
        walker.on('file', function (root, fileStats, next) {
            var name = fileStats.name;
            var ext = path.extname(name);
            var basename = path.basename(name, config.ext);

            var file = path.resolve('', root + '/' + name);
            if (ext === config.ext) {
                var input = fs.readFileSync(file, 'utf-8');
                var yaml = yamlFront.loadFront(input, 'content');
                // console.log(yaml);
                var page = yaml.page;
                var rank = yaml.rank;
                var category = yaml.category;

                if (!category) category = '__default__';

                if (category !== '__nav__') {
                    if (!mdData[category]) mdData[category] = [];
                    parseData(mdData[category], category, page, rank, file);
                }
            }
            next();
        });

        walker.on('end', function () {
            console.log(chalk.green('=========解析markdown========='));
            console.log(chalk.green(JSON.stringify(mdData, null, 4)));
            console.log(chalk.green('=========end========='));

            writeFiles(mdData);

            //将config 写入 文件
            var data = {
                md: mdData,
                config: config,
                source: source,
                root: process.cwd()
            };
            fs.writeFileSync(dist + '/__md__.json', JSON.stringify(data, null, 4));

            resolve();
        });

        walker.on('error', function (e) {
            console.log(e);
            reject(e);
        });
    });
};
;

var _temp = function () {
    if (typeof __REACT_HOT_LOADER__ === 'undefined') {
        return;
    }

    __REACT_HOT_LOADER__.register(sh, 'sh', 'src/walkmd.js');
}();

;