wx2/src/util/progress.js

微信小程序 转换 百度小程序

const path = require('path');
const fs = require('fs');
const EventEmitter = require('events');
const emitter = new EventEmitter.EventEmitter();
const slog = require('single-line-log').stdout;

global.emitter = emitter;

exports.getAllFile = dir => {
    let transformFilesLength = 0;
    let transFileEnd = 0;
    const eachFile = path.resolve(__dirname, dir);
    const getFile = childDir => {
        fs.readdir(childDir, (err, files) => {
            files.forEach(item => {
                const filePath = path.resolve(childDir, item);
                // 判断是否是文件夹，如果是，递归调用
                if (fs.statSync(filePath).isDirectory()) {
                    getFile(filePath);
                }
                // 判断是否是文件，如果是，总文件数加一
                if (fs.statSync(filePath).isFile()) {
                    transformFilesLength += 1;
                }
            });
        });
    };
    getFile(eachFile);

    //  监听打印进度
    emitter.on('event', args => {
        if (args) {
            transFileEnd = transformFilesLength;
            emitter.off('event', () => {});
        } else {
            transFileEnd += 1;
        }
        global.transformFilesLength = transformFilesLength;
        new ProgressBar().render({completed: transFileEnd, total: transformFilesLength});
    });
};

function ProgressBar(description, length) {
    this.description = description || '文件转换进度';
    this.length = length || 30;

    this.render = function (opts) {
        const percent = (opts.completed / opts.total).toFixed(4);
        const cellNum = Math.floor(percent * this.length);
        // 拼接黑色条
        let cell = '';
        for (let i = 0; i < cellNum; i++) {
            cell += '█';
        }
        // 拼接灰色条
        let empty = '';
        for (let i = 0; i < this.length - cellNum; i++) {
            empty += '░';
        }

        // 进度条 & 比例
        const shadow = `${cell}${empty} ${opts.completed}/${opts.total}`;
        // 拼接最终文本
        const cmdText = `🎉  ${this.description} : ${(100 * percent).toFixed(2)}% ${shadow}\n`;
        // 在单行输出文本
        slog(cmdText);
    };
}