amWiki/lib/export.github.js

/**
 * @desc amWiki 工作端·导出项目(github)模块
 * @author Tevin
 */

var electronRemote = require('electron').remote,
    dialog = electronRemote.dialog;
var fs = require("fs");
var directories = require('./directories');

module.exports = {
    //拷贝一篇文档
    _copyMd: function (fileName, pathTo) {
        var file = fs.readFileSync(fileName[0], 'utf-8');
        //相对路径图片地址转换
        file = file.replace(/!\[(.*?)]\(assets(.*?)\)/g, '![$1](' + this._githubUrl + 'wiki/images$2)');
        fs.writeFileSync(pathTo + '/' + fileName[1], file, 'utf-8');
    },
    //拷贝一张图片
    _copyImg: function (from, to) {
        fs.createReadStream(from).pipe(fs.createWriteStream(to));
    },
    //导出导航、底部签名
    _exportNavigation: function (fileList, duplicate, pathFrom, pathTo) {
    },
    //导出首页
    _exportHome: function (pathFrom, pathTo) {
        var that = this;
        var file = fs.readFileSync(pathFrom + '/首页.md', 'utf-8');
        file = file
            //相对路径图片地址转换
            .replace(/!\[(.*?)]\(assets(.*?)\)/g, '![$1](' + this._githubUrl + 'wiki/images$2)')
            //logo复制与图片引用地址转换
            .replace('amWiki/images/logo.png', function () {
                that._copyImg(pathFrom + '../amWiki/images/logo.png', pathTo + '\\images\\amWiki-logo.png');
                return that._githubUrl + 'wiki/images/amWiki-logo.png';
            });
        fs.writeFileSync(pathTo + '/Home.md', file, 'utf-8');
    },
    //导出图片
    _exportImage: function (pathFrom, pathTo) {
        var that = this;
        if (!fs.existsSync(pathTo + '/images/')) {
            fs.mkdirSync(pathTo + '/images/', 0777);
        }
        //文件夹拷贝
        var copyFolder = function(from, to){
            var list = fs.readdirSync(from);
            var path, to2;
            for (var i = 0, item; item = list[i]; i++) {
                path = from + '/' + item;
                to2 = to + '/' + item;
                if (fs.statSync(path).isDirectory(path)) {
                    fs.mkdirSync(to2, 0777);
                    copyFolder(path, to + '/' + item);
                } else {
                    that._copyImg(path, to + '/' + item);
                }
            }
        };
        copyFolder(pathFrom + '../assets/', pathTo + '/images/')
    },
    //检查重名
    _checkDuplicate: function (fileList) {
        var dup = [];
        for (var i = 0, item; item = fileList[i]; i++) {
            for (var j = i + 1, item2; item2 = fileList[j]; j++) {
                if (item[1] == item2[1]) {
                    dup.push([item, item2]);
                }
            }
        }
        return dup;
    },
    //导出重复文档
    _exportDuplicate: function (list, pathTo) {
        if (list.length == 0) {
            return;
        }
        var that = this;
        var file;
        var checkExist = function (fileName) {
            //如果已存在，增加空格
            if (fs.existsSync(pathTo + '/' + fileName[1])) {
                fileName[1] = fileName[1].replace(/\.md$/, ' .md');
                checkExist(fileName);
            }
            //不存在正常复制
            else {
                that._copyMd(fileName, pathTo);
            }
        };
        for (var i = 0, fileName; fileName = list[i]; i++) {
            checkExist(fileName);
        }
    },
    //导出普通文档
    _exportNormal: function (list, pathTo) {
        if (list.length == 0) {
            return;
        }
        for (var i = 0, fileName; fileName = list[i]; i++) {
            this._copyMd(fileName, pathTo);
        }
    },
    //开始导出
    _toExport: function (pathFrom, pathTo, fileList, duplicates) {
        if (fs.readdirSync(pathTo).length > 0 && confirm('所选文件夹不为空，是否需要清空？')) {
            directories.cleanDir(pathTo);
        }
        var fileList2 = [];
        var duplicate2 = [];
        if (typeof duplicates != 'undefined') {
            for (var i = 0, file; file = fileList[i]; i++) {
                var fileDup = false;
                for (var j = 0, dup; dup = duplicates[j]; j++) {
                    if (file[0] == dup) {
                        fileDup = true;
                        duplicate2.push(file);
                        break;
                    }
                }
                if (!fileDup) {
                    fileList2.push(file);
                }
            }
        } else {
            fileList2 = fileList;
        }
        //开始拷贝
        this._exportNormal(fileList2, pathTo);
        this._exportDuplicate(duplicate2, pathTo);
        this._exportImage(pathFrom, pathTo);
        this._exportHome(pathFrom, pathTo);
        this._exportNavigation(fileList2, duplicate2, pathFrom, pathTo);
    },
    //导出准备
    _toPrepare: function (pathFrom, pathTo) {
        var that = this;
        var fileList = [];
        pathFrom += pathFrom.substr(pathFrom.length - 1, 1) == '\\' ? 'library\\' : '\\library\\';
        //读取文件夹
        directories.readLibraryDir(pathFrom, function (err, tree, files) {
            if (err) {
                console.warn(err);
            } else {
                //console.log(tree, files);
                var fileName = '';
                //提取文件名
                for (var i = 0, item; item = files[i]; i++) {
                    fileName = item.split(/\\|\//);
                    fileList.push([
                        item,
                        fileName[fileName.length - 1].replace(/^\d+(\.\d+)?[-_](.*?)$/, '$2')
                    ]);
                }
                //重名检查
                var duplicate = that._checkDuplicate(fileList);
                //重名文件单独处理
                if (duplicate.length > 0) {
                    var dups = [];
                    var message = '以下文件脱离文件夹偏平化时将会重名：\n\n';
                    for (var j = 0, dup; dup = duplicate[j]; j++) {
                        message += dup[0][0].replace(/\//g, '\\') + '\n' +
                            dup[1][0].replace(/\//g, '\\') + '\n\n';
                        dups.push(dup[0][0], dup[1][0]);
                    }
                    message += '点击确认将自动处理（追加额外空格）并继续导出；\n' +
                        '点击取消将退出导出，您可以在改名后再次重新导出。';
                    if (confirm(message)) {
                        that._toExport(pathFrom, pathTo, fileList, dups);
                    }
                } else {
                    that._toExport(pathFrom, pathTo, fileList);
                }
            }
        });
    },
    //GitHub地址
    _githubUrl: '',
    //解析 GitHub url
    _parseGithubUrl: function(path) {
        var config = JSON.parse(fs.readFileSync(path + '/config.json', 'utf-8'));
        if (typeof config['github-url'] == 'undefined') {
            alert('导出失败！\n未检测到 “github-url” 配置，请在 config.json 中配置您项目的 github-url');
            return false;
        }
        var url = config['github-url'];
        if (url.indexOf('github.com') == 0 || url.split('github.com')[1] == '') {
            alert('导出失败！\n请配置完整 github 项目地址！');
            return false;
        }
        var urlArr = url.split('github.com')[1].split('/');
        if (urlArr.length < 3) {
            alert('导出失败！\n请配置完整 github 项目地址！');
            return false;
        }
        return '/' + urlArr[1] + '/' + urlArr[2] + '/';
    },
    //导出
    export: function () {
        var that = this;
        var editor = atom.workspace.getActiveTextEditor();
        if (!editor) {
            return;
        }
        var path = editor.getPath();
        //检测是否为 amWiki 项目
        path = directories.isAmWiki(path);
        if (!path) {
            alert('导出失败！\n只能对一个 amWiki 项目进行导出！');
            return;
        }
        //检测 GitHub 项目地址
        this._githubUrl = this._parseGithubUrl(path);
        if (!this._githubUrl) {
            return;
        }
        dialog.showOpenDialog({properties: ['openDirectory']}, function (data) {
            if (data && data.length) {
                that._toPrepare(path, data[0]);
            }
        });
    }
};