命令行进度条实现

对于很多包管理工具而言,拥有一个进度条基本上来说是必备的,可以轻松的让用户知道当前程序的进度。


命令行进度条实现_第1张图片
VSCode命令行进度条美化版

命令行下其实也可以活得很精彩。


今天来学一下如何制作一个命令行下的进度条工具。回想之前用Python写过的getpass2,用了标准输出来实现的。和这次的进度条需求相比,感觉应该是一回事。应该就是清空上次的打印内容,然后write本行数据。

单行输出

果不其然,在网上大致搜了一下,发现很多都是使用类似的思路。那既然如此,就没必要重复造轮子了。干脆用人家的好了。

single-line-log

关于这个模块的使用,可以通过npm进行安装,也可以下载源代码,放到本地使用。其核心代码超少,我就直接贴出代码好了。

/**
 * 单行内容输出效果。
 */
const MOVE_LEFT = new Buffer('1b5b3130303044', 'hex').toString();
const MOVE_UP = new Buffer('1b5b3141', 'hex').toString();
const CLEAR_LINE = new Buffer('1b5b304b', 'hex').toString();

/**
 * 借助一个第三方的计算字符串长度的模块。
 */
const stringwidth = require('string-width');

module.exports = function(stream) {
    var write = stream.write;
    var str;

    stream.write = function(data) {
        if(str && data!== str) str = null;
        return write.apply(this, arguments);
    };

    if(stream === process.stderr || stream === process.stdout) {
        process.on('exit', ()=>{
            if(str!==null) stream.write('');
        });
    }

    var prevLineCount = 0;
    var log = function(){
        str = '';
        var nextstr = Array.prototype.join.call(arguments, ' ');

        // 清屏
        for(var i =0; i < prevLineCount; i++) {
            str += MOVE_LEFT + CLEAR_LINE + (i < prevLineCount-1?MOVE_UP:'');
        }

        // 更新实际内容
        str += nextstr;
        stream.write(str);

        // 下行要删除多少行内容
        var prevLines = nextstr.split('\n');
        prevLineCount = 0;
        for(var i = 0; i < prevLines.length; i++) {
            prevLineCount += Math.ceil(stringwidth(prevLines[i]) / stream.columns) || 1;
        }
    };

    log.clear = function() {
        stream.write('');
    };

    return log;
};

/**
 * 对外提供接口。
 */
module.exports.stdout = module.exports(process.stdout);
module.exports.stderr = module.exports(process.stderr);

是的,就只有这么几行的内容。阅读代码就不难发现,就是我上面说的那个思路。

测试输出

接下来简单的进行测试一下。因为对于小文件来说,的一下就完事了,所以干脆找个大文件,比如一个电影来进行测试。

var log = require('./singleline').stdout;
var fs = require('fs');
var strformat = require('str-format');
var read = 0;
var filepath = "E:\\Video\\电影\\大内密探零零发.mkv";
var size = fs.statSync(filepath).size;

var rs = fs.createReadStream(filepath);
rs.on('data', function(data){
    read += data.length;
    var percentage = Math.floor(100*read/size);
    // log("Writing to super large file\n["+percentage+'%]', read, 'bytes readed.')
    log(strformat.format('Writing to super large file.\n[ {}% ] bytes readed.', [percentage]))
});

里面用到了我之前写的一个Python风格的字符串格式化模块。有需要的可以使用

npm install str-format

来进行安装和使用。

下面来看下运行的效果。


命令行进度条实现_第2张图片
单行输出测试,大电影形式

其实这样就实现了一个简单的进度条了。但是为了使其更加通用,接下来对其进行了一个封装。


progressbar

虽然说是进行了封装, 其实重要的还只是percentage那部分的内容了。

封装代码

/**
 * 进度条实现。
 */
const log = require('./singleline').stdout;
const strformat = require('str-format').format;

/**
 * 封装一个进度条工具。
 */
function ProgressBar(description, bar_length) {
    this.description = description || "PROGRESS";
    this.length = bar_length.length || 28;

    // 刷新进度条图案,文字的方法
    this.render = function(opts) {
        var percentage = (opts.completed/opts.total).toFixed(4);
        var cell_num = Math.floor(percentage * this.length);
        // 拼接黑色条
        var cell = '';
        for(var i = 0; i < cell_num; i++) {
            cell += '█';
        }
        // 拼接灰色条
        var empty = '';
        for(var i = 0; i < this.length - cell_num; i++) {
            empty += '░';
        }
        // 拼接最终文本
        var cmdtext = strformat("<{}:{}%> {}{}  [ {}/{} `{}`]", [this.description, (100*percentage).toFixed(2),
        cell, empty, opts.completed, opts.total, opts.status]);
        log(cmdtext);
    };
}

/**
 * 模块导出。
 */
module.exports = ProgressBar;

测试进度条效果

var ProgressBar = require('./progressbar');
var pb = new ProgressBar('下载进度', 50);
var num = 0, total = 100;
function downloading() {
    if (num < total) {
        pb.render({ completed: num, total: total, status: '下载中...' });
        num++;
        setTimeout(function () {
            downloading();
        }, 50);
    }else{
        pb.render({completed: num, total: total, status: "下载完毕."});
            process.exit(0);
    }
}

downloading()

运行效果,如下:


命令行进度条实现_第3张图片
进度条本地测试

至此,一个还算是比较通用的进度条就算是完成了。每次需要手动更新进度的时候,只需要调用render方法即可。

美化

单纯的黑白字符,看起来挺枯燥的,下面使用一个终端彩色模块来对其进行一下美化。

美化版进度条包装

/**
 * 进度条实现。
 */
const log = require('./singleline').stdout;
const strformat = require('str-format').format;
const clicolor = require('cli-color');

/**
 * 封装一个进度条工具。
 */
function ProgressBar(description, bar_length) {
    this.description = description || "PROGRESS";
    this.length = bar_length.length || 28;

    // 刷新进度条图案,文字的方法
    this.render = function(opts) {
        var percentage = (opts.completed/opts.total).toFixed(4);
        var cell_num = Math.floor(percentage * this.length);
        // 拼接黑色条
        var cell = '';
        for(var i = 0; i < cell_num; i++) {
            cell += '█';
        }
        // 拼接灰色条
        var empty = '';
        for(var i = 0; i < this.length - cell_num; i++) {
            empty += '░';
        }
        
        var percent = (100*percentage).toFixed(2);
        /**
         * 使用cli-color进行包装美化。
         */
        this.description = clicolor.blue.bold(this.description);
        cell = clicolor.green.bgBlack.bold(cell);
        opts.completed = clicolor.yellow.bold(opts.completed);
        opts.total = clicolor.blue.bold(opts.total);
        opts.status = percent==100.00?clicolor.green.bold(opts.status):clicolor.red.bold(opts.status);


        // 拼接最终文本
        var cmdtext = strformat("<{}:{}%> {}{}  [ {}/{} `{}`]", [this.description, percent,
        cell, empty, opts.completed, opts.total, opts.status]);
        log(cmdtext);
    };
}

/**
 * 模块导出。
 */
module.exports = ProgressBar;

美化版进度条测试

var ProgressBar = require('./progressbar');
var pb = new ProgressBar('下载进度', 72);
var num = 0, total = 100;
function downloading() {
    if (num < total) {
        pb.render({ completed: num, total: total, status: '下载中...' });
        num++;
        setTimeout(function () {
            downloading();
        }, 20);
    }else{
        pb.render({completed: num, total: total, status: "下载完毕."});
            process.exit(0);
    }
}

downloading()

下面看下执行结果。

  • 先看下在VSCode下的执行效果。


    命令行进度条实现_第4张图片
    VSCode命令行进度条美化版
  • 然后是常规的CMD命令行下效果。


    命令行进度条实现_第5张图片
    常规CMD命令行进度条美化版

好了,大致就是这么个内容了。在Linux上还有很多更加优秀的此类实现。有机会再去借鉴借鉴好了。

你可能感兴趣的:(命令行进度条实现)