add us states api

Author: sagarkarira

app.js

@@ -12,6 +12,7 @@ const { getCompleteTable, getGraph } = require('./lib/corona');
 const { lookupCountry, htmlTemplate, footer } = require('./lib/helpers');
 const { getLiveUpdates } = require('./lib/reddit.js');
 const { getWorldoMetersTable } = require('./lib/worldoMeters.js');
+const { getUsaStats } = require('./lib/country/us.js');
 const { helpContent, countryNotFound } = require('./lib/constants');
 
 const app = express();
@@ -125,6 +126,26 @@ app.get('/help', (req, res) => {
   return res.send(chalk.green(helpContent));
 });
 
+app.get('/states/:country', (req, res) => {
+  const { country } = req.params;
+  const isCurl = req.isCurl;
+  const format = req.query.format ? req.query.format : '';
+  const minimal = req.query.minimal === 'true';
+  const top = req.query.top ? Number(req.query.top) : 1000;
+
+  const lookupObj = lookupCountry(country);
+
+  if (!lookupObj) {
+    return res.status(404).send(countryNotFound(isCurl));
+  }
+  if (lookupObj.iso2 === 'US') {
+    return getUsaStats({ isCurl, minimal, top, format})
+      .then(result => {
+        return res.send(result);
+      }).catch(error => errorHandler(error, req, res));
+  }
+});
+
 
 app.get('/:country', (req, res) => {
   const { country } = req.params;

bin/index.js

@@ -7,6 +7,7 @@ const { getCompleteTable, getGraph } = require('../lib/corona');
 const { getCountryTable } = require('../lib/byCountry');
 const { getWorldoMetersTable } = require('../lib/worldoMeters');
 const { lookupCountry } = require('../lib/helpers');
+const { getUsaStats } = require('../lib/country/us');
 
 const { argv } = yargs
   .command('$0 [country]', 'Tool to track COVID-19 statistics from terminal', yargs =>
@@ -68,12 +69,27 @@ const { argv } = yargs
       describe: 'Get graph',
       type: 'boolean',
       default: false,
+    },
+    st: {
+      alias: 'states',
+      describe: 'Get state level data of country ',
+      type: 'string',
     }
   })
   .strict()
   .help('help');
 
 argv.countryCode = argv.country;
+if (argv.states === 'US') {
+  getUsaStats(argv).then(result => {
+    console.log(result);
+    process.exit(1);
+  }).catch(error => {
+    console.error(error);
+    process.exit(0);
+  });
+}
+
 if (argv.source === 1) {
   (
     argv.country === 'ALL'

changelog.md

@@ -1,5 +1,9 @@
 # Changelog
 
+## Version 0.9.1
+
+* Added US states api `corona --states=US`
+
 ## Version 0.9.0
 
 * Changed default source to worldoMeters. i.e source 2 is now default

lib/api.js

@@ -75,8 +75,8 @@ exports.getWorldoMetersData = async (countryCode = 'ALL') => {
   });
 
   result.data.forEach(obj => {
-    obj.countryCode = countryNameMap[obj.country];
     obj.confirmed = obj.cases;
+    obj.countryCode =  obj.countryInfo.iso2 || '';
   });
   worldStats.casesPerOneMillion = (worldStats.cases / 7794).toFixed(2);
   worldStats.confirmed = worldStats.cases;
@@ -89,3 +89,18 @@ exports.getWorldoMetersData = async (countryCode = 'ALL') => {
   myCache.set(key, returnObj, 60 * 15);
   return returnObj;
 };
+
+exports.usaStats = async () => {
+  const key = 'usaStats';
+  const cache = myCache.get(key);
+
+  if (cache) {
+    console.log('cache', key);
+    return cache;
+  }
+  const result = await axios('https://corona.lmao.ninja/states');
+  if (!result || !result.data) {
+    throw new Error('usa stats API failure');
+  }
+  return result;
+};

lib/corona.js

@@ -79,7 +79,7 @@ exports.getGraph = async ({ countryCode = 'ALL', isCurl = true}) => {
 
   const confirmedGraph = asciichart.plot(confirmedByDay, graphConfig);
   const body = chalk.greenBright(confirmedGraph)
-  + chalk.cyanBright('\n\n' +padding + '22 Feb' + graphLength + '22 Mar') + '\n';
+  + chalk.cyanBright('\n\n' +padding + '22 Feb' + graphLength + 'Today') + '\n';
 
   if (!isCurl) {
     return htmlTemplate(body);

lib/country/us.js

@@ -0,0 +1,57 @@
+const Table = require('cli-table3');
+const chalk = require('chalk');
+const helpers = require('../helpers');
+const api = require('../api');
+const { cFormatter } = helpers;
+
+
+const getUsaStatsHeaders = (emojis, secondColumnName) => {
+  const head = [
+    'Rank',
+    secondColumnName,
+    `Total Cases ${emojis ? ' ✅' : ''}`,
+    'New Cases ▲',
+    `Total Deaths${emojis ? ' 😞' : ''}`,
+    `New Deaths ▲${emojis ? ' 😞' : ''}`,
+    `Active${emojis ? ' 😷' : ''}`,
+
+  ];
+  return head;
+};
+
+
+exports.getUsaStats = async ({
+  isCurl = true,
+  minimal = false,
+  top = 1000,
+  format,
+}) => {
+  const secondColumnName = 'US States';
+  const table = new Table({
+    head: getUsaStatsHeaders(null, secondColumnName),
+    chars: helpers.getTableBorders(minimal),
+    style: helpers.getTableStyles(minimal),
+  });
+  const { data } = await api.usaStats();
+  if (format === 'json') {
+    return { data };
+  }
+
+  let rank = 1;
+  data.some(cd => {
+    const values = [
+      cFormatter(cd.state , chalk.cyanBright),
+      cFormatter(cd.cases, chalk.green, 'right', true),
+      cFormatter(cd.todayCases, chalk.cyanBright, 'right', true, ' ▲'),
+      cFormatter(cd.deaths, chalk.whiteBright, 'right', true),
+      cFormatter(cd.todayDeaths, chalk.redBright, 'right', true, ' ▲'),
+      cFormatter(cd.active, chalk.blueBright , 'right', true),
+    ];
+    table.push({ [rank++]: values });
+    return rank === top + 1;
+  });
+
+  const lastUpdated = new Date();
+  const ret = table.toString() + helpers.footer(lastUpdated);
+  return isCurl ? ret : helpers.htmlTemplate(ret);
+};

lib/helpers.js

@@ -171,8 +171,7 @@ ${chalk.greenBright('Twitter')}: ${chalk.blueBright('https://twitter.com/ekrysis
 
 ${chalk.magentaBright('Last Updated on:')} ${moment(lastUpdated).utc().format('DD-MMM-YYYY HH:MM')} UTC
 
-${chalk.red.bold('UPDATE')}: ${chalk.blueBright('Source 2 is now default source')}
-${chalk.red.bold('JHU Source 1 table')}: ${chalk.blueBright('https://corona-stats.online?source=1')}
+${chalk.red.bold('US STATES API')}: ${chalk.blueBright('https://corona-stats.online/states/us')}
 ${chalk.red.bold('HELP')}: ${chalk.blueBright('https://corona-stats.online/help')}
 `;
 
@@ -230,7 +229,6 @@ e.getTableHeadersV2 = (emojis, secondColumnName) => {
   ];
   return head;
 };
-
 e.extraStats = (dataArr) => {
   return dataArr.map(obj => {
     return {

package.json

@@ -1,6 +1,6 @@
 {
   "name": "coronavirus-tracker-cli",
-  "version": "0.9.0",
+  "version": "0.9.1",
   "description": "track conronavirus cases from cli",
   "repository": {
     "type": "git",