vmw-cli/lib/streamer.js

CLI interface for my.vmware.com

#!/usr/bin/env node
const stream = require('stream');
const got = require('got');
const ProgressBar = require('progress');
const {promisify} = require('util');
const pipeline = promisify(stream.pipeline);

// colours
const chalk = require('chalk');
const red = chalk.bold.red;
const orange = chalk.keyword('orange');
const green = chalk.green;
const blue = chalk.blueBright;

module.exports = class vmwClient {
	constructor(options) {
                this.options = options;
		// setup stuff
	}
	async download(url, fileName, dstStream) {
		let bar;
		let current = 0;
		let prevTrans = 0;
		let label = 0;
		let labels = [
			'KB',
			'MB',
			'GB'
		];
		return pipeline(
			got.stream(url)
				.on('request', (request) => {
					//console.log('REQUEST START');
				})
				.on('response', (response) => {
					//console.log('FIRST RESPONSE');
				})
				.on('downloadProgress', (progress) => {
					//let rate = Math.round(bar.curr / ((new Date - bar.start) / 1000) / 1000);
					if(progress.total) { // handle no content-header
						if(typeof(bar) == 'undefined') {
							let totalBytes = progress.total;
							let total = parseInt(totalBytes / 1024 * 100) / 100; // get KB
							while(total > 99.99) { // scale up
								label++;
								total = parseInt(total / 1024 * 100) / 100;
							}
							bar = new ProgressBar(blue(fileName) + ' [' + green(':bar') + '] :percent | :etas | :curr/' + pad(total) + ' ' + labels[label], {
								complete: '\u25A0',
								head: '\u25A0',
								incomplete: ' ',
								width: 50,
								renderThrottle: 500,
								total: totalBytes
							});
						} else {
							let chunk = progress.transferred - prevTrans;
							prevTrans = progress.transferred;
							current = progress.transferred;
							for(let i = 0; i <= label; i++) {
								current = parseInt(current / 1024 * 100) / 100;
							}
							bar.tick(chunk, {
								'curr': pad(current)
							});
						}
					}
				}),
			dstStream
				.on('finish', () => {
					if(!bar.complete) {
						bar.update(1);
					}
				})
		);
	}
}

// pad/truncate zeros to left and right of number
function pad(num) {
	let rgx = /^([^.]+)\.([^.]+)/g
	if(m = rgx.exec(num.toFixed(2).toString())) {
		let left = m[1];
		if(left.length < 2) {
			left = ('00' + left).slice(-2);
		}
		return left + '.' + m[2];
	}
};