Adds limit functionality to cli

index.js

@@ -28,6 +28,7 @@ const {
 } = require("./utils/table.js");
 const xcolor = cli.flags.xcolor;
 const sortBy = cli.flags.sort;
+const limit = cli.flags.limit;
 
 (async () => {
 	// Init.
@@ -48,8 +49,8 @@ const sortBy = cli.flags.sort;
 	spinner.start();
 	await getWorldwide(table, states);
 	await getCountry(spinner, table, states, country);
-	await getStates(spinner, table, states, { sort: sortBy });
-	await getAll(spinner, table, states, country, { sort: sortBy });
+	await getStates(spinner, table, states, { sort: sortBy, limit });
+	await getAll(spinner, table, states, country, { sort: sortBy, limit });
 
 	theEnd(states);
 })();

readme.md

@@ -108,6 +108,14 @@ corona -s per-million
 
 [![📟](./.github/sort.gif)](./../../)
 
+### Limit output
+
+Only output the top N results (depending on your sort criteria)
+
+````sh
+corona --sort cases -n 10
+````
+
 #### CLI Help
 
 ```sh

utils/cli.js

@@ -16,6 +16,7 @@ module.exports = meow(
 	Options
 	  ${yellow(`--xcolor`)}, ${yellow(`-x`)}         Single colored output
 	  ${yellow(`--sort`)}, ${yellow(`-s`)}           Sort data by type
+	  ${yellow(`--limit`)}, ${yellow(`-n`)}          Limit output to N entries
 
 	Examples
 		${green(`corona`)} ${cyan(`china`)}
@@ -41,6 +42,11 @@ module.exports = meow(
 				type: 'string',
 				default: 'cases-today',
 				alias: 's'
+			},
+			limit: {
+				type: 'number',
+				default: Number.MAX_SAFE_INTEGER,
+				alias: 'n'
 			}
 		}
 	}

utils/getAll.js

@@ -6,7 +6,7 @@ const { sortKeys, sortOrders } = require("./table.js");
 module.exports = async (spinner, table, states, country, options) => {
 	if (!country && !states) {
 		const api = await axios.get(`https://corona.lmao.ninja/countries`);
-		let all = api.data.map(one => Object.values(one));
+		let all = api.data.map(one => Object.values(one)).slice(0, options.limit);
 
 		const sortIndex = sortKeys.indexOf(options.sort);
 
@@ -15,6 +15,8 @@ module.exports = async (spinner, table, states, country, options) => {
 			all = all.sort((a, b) => (a[sortIndex] > b[sortIndex] ? dir : -dir));
 		}
 
+		all = all.slice(0, options.limit);
+
 		all.map(one => {
 			one = one.map(d => comma(d));
 			return table.push(one);

utils/getStates.js

@@ -15,6 +15,8 @@ module.exports = async (spinner, table, states, options) => {
 			all = all.sort((a, b) => (a[sortIndex] > b[sortIndex] ? dir : -dir));
 		}
 
+		all = all.slice(0, options.limit);
+
 		all.map(one => {
 			one = one.map(d => comma(d));
 			return table.push(one);