👌 IMPROVE: Single or Stacked bars

ahmadawais · Apr 23, 2020 · 757af7a · 757af7a
1 parent 0a1c934
commit 757af7a
Showing 1 changed file with 13 additions and 7 deletions.
diff --git a/utils/getBar.js b/utils/getBar.js
@@ -38,13 +38,19 @@ module.exports = async (spinner, countryName, states, { bar, log, sortBy, limit,
 		}
         // Format Stack Data
         barCountries = {};
-        allCountries.map((country, count)=>{
-            barCountries[country.country]=[
-                logScale(country.cases),
-                logScale(country.deaths),
-                logScale(country.recovered)
-            ];
-        });
+		allCountries.map(country => {
+			if (customSort) {
+				barCountries[country.country] = [
+					logScale(country[sortingKeys[sortBy]])
+				];
+			} else {
+				barCountries[country.country] = [
+					logScale(country.cases),
+					logScale(country.deaths),
+					logScale(country.recovered)
+				];
+			}
+		});
 
         const names = Object.keys(barCountries);
         const data = Object.values(barCountries);