123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236 |
- /*!
- * node-progress
- * Copyright(c) 2011 TJ Holowaychuk <tj@vision-media.ca>
- * MIT Licensed
- */
- /**
- * Expose `ProgressBar`.
- */
- exports = module.exports = ProgressBar;
- /**
- * Initialize a `ProgressBar` with the given `fmt` string and `options` or
- * `total`.
- *
- * Options:
- *
- * - `curr` current completed index
- * - `total` total number of ticks to complete
- * - `width` the displayed width of the progress bar defaulting to total
- * - `stream` the output stream defaulting to stderr
- * - `head` head character defaulting to complete character
- * - `complete` completion character defaulting to "="
- * - `incomplete` incomplete character defaulting to "-"
- * - `renderThrottle` minimum time between updates in milliseconds defaulting to 16
- * - `callback` optional function to call when the progress bar completes
- * - `clear` will clear the progress bar upon termination
- *
- * Tokens:
- *
- * - `:bar` the progress bar itself
- * - `:current` current tick number
- * - `:total` total ticks
- * - `:elapsed` time elapsed in seconds
- * - `:percent` completion percentage
- * - `:eta` eta in seconds
- * - `:rate` rate of ticks per second
- *
- * @param {string} fmt
- * @param {object|number} options or total
- * @api public
- */
- function ProgressBar(fmt, options) {
- this.stream = options.stream || process.stderr;
- if (typeof(options) == 'number') {
- var total = options;
- options = {};
- options.total = total;
- } else {
- options = options || {};
- if ('string' != typeof fmt) throw new Error('format required');
- if ('number' != typeof options.total) throw new Error('total required');
- }
- this.fmt = fmt;
- this.curr = options.curr || 0;
- this.total = options.total;
- this.width = options.width || this.total;
- this.clear = options.clear
- this.chars = {
- complete : options.complete || '=',
- incomplete : options.incomplete || '-',
- head : options.head || (options.complete || '=')
- };
- this.renderThrottle = options.renderThrottle !== 0 ? (options.renderThrottle || 16) : 0;
- this.lastRender = -Infinity;
- this.callback = options.callback || function () {};
- this.tokens = {};
- this.lastDraw = '';
- }
- /**
- * "tick" the progress bar with optional `len` and optional `tokens`.
- *
- * @param {number|object} len or tokens
- * @param {object} tokens
- * @api public
- */
- ProgressBar.prototype.tick = function(len, tokens){
- if (len !== 0)
- len = len || 1;
- // swap tokens
- if ('object' == typeof len) tokens = len, len = 1;
- if (tokens) this.tokens = tokens;
- // start time for eta
- if (0 == this.curr) this.start = new Date;
- this.curr += len
- // try to render
- this.render();
- // progress complete
- if (this.curr >= this.total) {
- this.render(undefined, true);
- this.complete = true;
- this.terminate();
- this.callback(this);
- return;
- }
- };
- /**
- * Method to render the progress bar with optional `tokens` to place in the
- * progress bar's `fmt` field.
- *
- * @param {object} tokens
- * @api public
- */
- ProgressBar.prototype.render = function (tokens, force) {
- force = force !== undefined ? force : false;
- if (tokens) this.tokens = tokens;
- if (!this.stream.isTTY) return;
- var now = Date.now();
- var delta = now - this.lastRender;
- if (!force && (delta < this.renderThrottle)) {
- return;
- } else {
- this.lastRender = now;
- }
- var ratio = this.curr / this.total;
- ratio = Math.min(Math.max(ratio, 0), 1);
- var percent = Math.floor(ratio * 100);
- var incomplete, complete, completeLength;
- var elapsed = new Date - this.start;
- var eta = (percent == 100) ? 0 : elapsed * (this.total / this.curr - 1);
- var rate = this.curr / (elapsed / 1000);
- /* populate the bar template with percentages and timestamps */
- var str = this.fmt
- .replace(':current', this.curr)
- .replace(':total', this.total)
- .replace(':elapsed', isNaN(elapsed) ? '0.0' : (elapsed / 1000).toFixed(1))
- .replace(':eta', (isNaN(eta) || !isFinite(eta)) ? '0.0' : (eta / 1000)
- .toFixed(1))
- .replace(':percent', percent.toFixed(0) + '%')
- .replace(':rate', Math.round(rate));
- /* compute the available space (non-zero) for the bar */
- var availableSpace = Math.max(0, this.stream.columns - str.replace(':bar', '').length);
- if(availableSpace && process.platform === 'win32'){
- availableSpace = availableSpace - 1;
- }
- var width = Math.min(this.width, availableSpace);
- /* TODO: the following assumes the user has one ':bar' token */
- completeLength = Math.round(width * ratio);
- complete = Array(Math.max(0, completeLength + 1)).join(this.chars.complete);
- incomplete = Array(Math.max(0, width - completeLength + 1)).join(this.chars.incomplete);
- /* add head to the complete string */
- if(completeLength > 0)
- complete = complete.slice(0, -1) + this.chars.head;
- /* fill in the actual progress bar */
- str = str.replace(':bar', complete + incomplete);
- /* replace the extra tokens */
- if (this.tokens) for (var key in this.tokens) str = str.replace(':' + key, this.tokens[key]);
- if (this.lastDraw !== str) {
- this.stream.cursorTo(0);
- this.stream.write(str);
- this.stream.clearLine(1);
- this.lastDraw = str;
- }
- };
- /**
- * "update" the progress bar to represent an exact percentage.
- * The ratio (between 0 and 1) specified will be multiplied by `total` and
- * floored, representing the closest available "tick." For example, if a
- * progress bar has a length of 3 and `update(0.5)` is called, the progress
- * will be set to 1.
- *
- * A ratio of 0.5 will attempt to set the progress to halfway.
- *
- * @param {number} ratio The ratio (between 0 and 1 inclusive) to set the
- * overall completion to.
- * @api public
- */
- ProgressBar.prototype.update = function (ratio, tokens) {
- var goal = Math.floor(ratio * this.total);
- var delta = goal - this.curr;
- this.tick(delta, tokens);
- };
- /**
- * "interrupt" the progress bar and write a message above it.
- * @param {string} message The message to write.
- * @api public
- */
- ProgressBar.prototype.interrupt = function (message) {
- // clear the current line
- this.stream.clearLine();
- // move the cursor to the start of the line
- this.stream.cursorTo(0);
- // write the message text
- this.stream.write(message);
- // terminate the line after writing the message
- this.stream.write('\n');
- // re-display the progress bar with its lastDraw
- this.stream.write(this.lastDraw);
- };
- /**
- * Terminates a progress bar.
- *
- * @api public
- */
- ProgressBar.prototype.terminate = function () {
- if (this.clear) {
- if (this.stream.clearLine) {
- this.stream.clearLine();
- this.stream.cursorTo(0);
- }
- } else {
- this.stream.write('\n');
- }
- };
|