diff --git a/lib/corona.js b/lib/corona.js index 1520d96..467d042 100644 --- a/lib/corona.js +++ b/lib/corona.js @@ -28,6 +28,8 @@ function getDataByCountry(confirmed, deaths, recovered) { const confirmedMap = _.keyBy(confirmed.locations, (i) => i.country + i.province); const recoveredMap = _.keyBy(recovered.locations, (i) => i.country + i.province); const deathsMap = _.keyBy(deaths.locations, (i) => i.country + i.province); + const countryName = process.argv[2]; + confirmed.locations.forEach(obj => { const countryName = obj.country; const provinceName = obj.province; @@ -64,7 +66,9 @@ function getDataByCountry(confirmed, deaths, recovered) { const countryArr = extraStats( Object.keys(countryMap).map(key => countryMap[key]) ); - return _.sortBy(countryArr, (o) => -o.confirmed) + const countryData = (countryName) ? countryArr.filter(obj => obj.country.toLowerCase() === countryName.toLowerCase()) : _.sortBy(countryArr, (o) => -o.confirmed); + + return countryData; } function extraStats(dataArr) {