Filter top contries using ?top=

Author: ssiyad

app.js

@@ -19,6 +19,7 @@ app.use(morgan(':remote-addr :remote-user :method :url :status :res[content-leng
 
 app.get('/', (req, res) => {
   const format = req.query.format ? req.query.format : '';
+  const top = req.query.top ? Number(req.query.top) : 300;
 
   if (format.toLowerCase() === 'json') {
     return getJSONData().then(result => {
@@ -27,7 +28,7 @@ app.get('/', (req, res) => {
     }).catch(error => errorHandler(error, res));
   }
 
-  return getCompleteTable().then(result => {
+  return getCompleteTable(max=top).then(result => {
     res.setHeader('Cache-Control', 's-maxage=900');
     return res.send(result);
   }).catch(error => errorHandler(error, res));

lib/corona.js

@@ -79,7 +79,7 @@ function extraStats(dataArr) {
   );
 }
 
-exports.getCompleteTable = async (emojis = false) => {
+exports.getCompleteTable = async (max = false, emojis = false) => {
   const head = [
     '',
     'Country',
@@ -121,7 +121,7 @@ exports.getCompleteTable = async (emojis = false) => {
     ]
   })
   let rank = 1;
-  countryData.forEach(cd => {
+  countryData.some(cd => {
     const countryEmoji = getEmoji(cd.countryCode);
     const values = [
       getCountry(`${cd.country} (${cd.countryCode})`),
@@ -137,6 +137,7 @@ exports.getCompleteTable = async (emojis = false) => {
       ...(emojis ? [countryEmoji] : [])
     ]
     table.push({ [rank++]: values })
+    return rank == max+1
   });
   const lastUpdated = countryData[0].lastUpdated;
   return table.toString() + footer(lastUpdated);