Measure upload speed

api.js

@@ -2,37 +2,56 @@
 /* eslint-env browser */
 const puppeteer = require('puppeteer');
 const Observable = require('zen-observable');
+const equals = require('deep-equal');
 
-async function init(browser, page, observer, prevSpeed) {
-	const result = await page.evaluate(() => {
-		const $ = document.querySelector.bind(document);
+function delay(ms) {
+	return new Promise(resolve => setTimeout(resolve, ms));
+}
 
-		return {
-			speed: Number($('#speed-value').textContent),
-			unit: $('#speed-units').textContent.trim(),
-			isDone: Boolean($('#speed-value.succeeded'))
-		};
-	});
+async function init(browser, page, observer) {
+	let prevResult;
 
-	if (result.speed > 0 && result.speed !== prevSpeed) {
-		observer.next(result);
-	}
+	/* eslint-disable no-constant-condition, no-await-in-loop */
+
+	while (true) {
+		const result = await page.evaluate(() => {
+			const $ = document.querySelector.bind(document);
+
+			return {
+				downloadSpeed: Number($('#speed-value').textContent),
+				uploadSpeed: Number($('#upload-value').textContent),
+				downloadUnit: $('#speed-units').textContent.trim(),
+				uploadUnit: $('#upload-units').textContent.trim(),
+				isDone: Boolean(
+					$('#speed-value.succeeded') && $('#upload-value.succeeded'),
+				)
+			};
+		});
 
-	if (result.isDone) {
-		browser.close();
-		observer.complete();
-	} else {
-		setTimeout(init, 100, browser, page, observer, result.speed);
+		if (result.downloadSpeed > 0 && !equals(result, prevResult)) {
+			observer.next(result);
+		}
+
+		if (result.isDone) {
+			browser.close();
+			observer.complete();
+			return;
+		}
+
+		await delay(100);
 	}
+
+	/* eslint-enable no-constant-condition, no-await-in-loop */
 }
 
-module.exports = () => new Observable(observer => {
-	// Wrapped in async IIFE as `new Observable` can't handle async function
-	(async () => {
-		const browser = await puppeteer.launch({args: ['--no-sandbox']});
-		const page = await browser.newPage();
+module.exports = () =>
+	new Observable(observer => {
+		// Wrapped in async IIFE as `new Observable` can't handle async function
+		(async () => {
+			const browser = await puppeteer.launch({args: ['--no-sandbox']});
+			const page = await browser.newPage();
 
-		await page.goto('https://fast.com');
-		await init(browser, page, observer);
-	})().catch(observer.error.bind(observer));
-});
+			await page.goto('https://fast.com');
+			await init(browser, page, observer);
+		})().catch(observer.error.bind(observer));
+	});

cli.js

@@ -7,10 +7,11 @@ const logUpdate = require('log-update');
 const ora = require('ora');
 const api = require('./api');
 
-meow(`
+const cli = meow(`
 	Usage
 	  $ fast
-	  $ fast > file
+		$ fast > file
+		$ fast --verbose
 `);
 
 // Check connection
@@ -24,13 +25,30 @@ dns.lookup('fast.com', err => {
 let data = {};
 const spinner = ora();
 
-const speed = () => chalk[data.isDone ? 'green' : 'cyan'](data.speed + ' ' + chalk.dim(data.unit)) + '\n\n';
+const speed = () =>
+	chalk[data.isDone ? 'green' : 'cyan'](
+		data.downloadSpeed +
+			' ' +
+			chalk.dim(data.downloadUnit) +
+			(cli.flags.verbose ? (
+				chalk.gray(' / ') +
+				(data.uploadSpeed || '-') +
+				' ' +
+				chalk.dim(data.uploadUnit)
+			) : '')
+	) + '\n\n';
 
 function exit() {
 	if (process.stdout.isTTY) {
 		logUpdate(`\n\n    ${speed()}`);
+	} else if (cli.flags.verbose) {
+		console.log(
+			`${data.downloadSpeed} ${data.downloadUnit} / ${data.uploadSpeed} ${data.uploadUnit}`,
+		);
 	} else {
-		console.log(`${data.speed} ${data.unit}`);
+		console.log(
+			`${data.downloadSpeed} ${data.downloadUnit}`,
+		);
 	}
 
 	process.exit();
@@ -40,12 +58,17 @@ if (process.stdout.isTTY) {
 	setInterval(() => {
 		const pre = '\n\n  ' + chalk.gray.dim(spinner.frame());
 
-		if (!data.speed) {
+		if (!data.downloadSpeed) {
 			logUpdate(pre + '\n\n');
 			return;
 		}
 
 		logUpdate(pre + speed());
+
+		if (!cli.flags.verbose && data.uploadSpeed) {
+			data.isDone = true;
+			exit();
+		}
 	}, 50);
 }
 

package.json

@@ -42,6 +42,7 @@
 	],
 	"dependencies": {
 		"chalk": "^2.1.0",
+		"deep-equal": "^1.0.1",
 		"log-update": "^2.1.0",
 		"meow": "^5.0.0",
 		"ora": "^2.0.0",