Refresh leaderboard for all spaces (#399)

Author: ChaituVR

scripts/refresh-space-leaderboard.ts

@@ -5,29 +5,33 @@ import db from '../src/helpers/mysql';
 // This script assumes there are no active proposals in the space
 
 async function getFirstAndLastVote(space: string) {
-  console.log('Getting first and last vote for space', space);
-  const firstVoted = await db.queryAsync(
-    'SELECT created FROM votes WHERE space = ? ORDER BY created ASC LIMIT 1',
-    [space]
-  );
-
-  if (!firstVoted.length) throw new Error('No votes found in the database');
-  const firstVote = firstVoted[0].created as number;
-
-  const lastVote = (
-    await db.queryAsync('SELECT created FROM votes WHERE space = ? ORDER BY created DESC LIMIT 1', [
-      space
-    ])
-  )[0].created;
+  const startArg = process.argv.indexOf('--start');
+  const endArg = process.argv.indexOf('--end');
+  let start = startArg !== -1 ? parseInt(process.argv[startArg + 1]) : null;
+  let end = endArg !== -1 ? parseInt(process.argv[endArg + 1]) : null;
 
-  console.log(
-    'Will process votes from',
-    new Date(firstVote * 1000),
-    'to',
-    new Date(lastVote * 1000)
-  );
+  console.log('Getting first and last vote for space', space);
+  if (!start) {
+    console.log('Start date is missing');
+    const firstVoted = await db.queryAsync(
+      `SELECT created FROM votes ${space ? 'WHERE space = ?' : ''} ORDER BY created ASC LIMIT 1`,
+      space ? [space] : []
+    );
+    if (!firstVoted.length) throw new Error('No votes found in the database');
+    start = firstVoted[0].created as number;
+  }
+  if (!end) {
+    end = (
+      await db.queryAsync(
+        `SELECT created FROM votes ${space ? 'WHERE space = ?' : ''} ORDER BY created DESC LIMIT 1`,
+        space ? [space] : []
+      )
+    )[0].created;
+  }
+  // @ts-ignore
+  console.log('Will process votes from', new Date(start * 1000), 'to', new Date(end * 1000));
 
-  return { firstVote, lastVote };
+  return { firstVote: start, lastVote: end };
 }
 
 async function deleteLeaderboard(space: string) {
@@ -37,17 +41,17 @@ async function deleteLeaderboard(space: string) {
 }
 
 async function processVotes(space: string, start: number, end: number) {
-  console.log('Get votes from', new Date(start * 1000), 'to', new Date(end * 1000));
+  console.log('Get votes from', new Date(start * 1000), start, 'to', new Date(end * 1000), end);
 
   await db.queryAsync(
     `INSERT INTO leaderboard (space, user, vote_count, last_vote)
       (SELECT space, voter AS user, COUNT(*) AS vote_count, MAX(created) AS last_vote
       FROM votes
-      WHERE space = ? AND created >= ? AND created < ?
-      GROUP BY voter)
+      WHERE ${space ? 'WHERE space = ? AND' : ''} created >= ? AND created < ?
+      GROUP BY space, voter)
     ON DUPLICATE KEY UPDATE vote_count = vote_count + VALUES(vote_count), last_vote = VALUES(last_vote)
   `,
-    [space, start, end]
+    space ? [space, start, end] : [start, end]
   );
 }
 
@@ -58,17 +62,17 @@ async function processProposalsCount(space: string) {
     INSERT INTO leaderboard (space, user, proposal_count)
       (SELECT space, author AS user, COUNT(*) AS proposal_count
       FROM proposals
-      WHERE space = ?
-      GROUP BY author)
+      ${space ? 'WHERE space = ?' : ''}
+      GROUP BY space, author)
     ON DUPLICATE KEY UPDATE proposal_count = proposal_count + VALUES(proposal_count)
   `,
-    [space]
+    space ? [space] : []
   );
   console.log('Proposals count for space', space, 'has been processed');
 }
 
 async function main(space) {
-  await deleteLeaderboard(space);
+  if (space) await deleteLeaderboard(space);
   const { firstVote, lastVote } = await getFirstAndLastVote(space);
 
   // Process votes in chunks of 24 hours
@@ -77,6 +81,7 @@ async function main(space) {
     const end = start + 86400; // 24 hours
     await processVotes(space, start, end);
 
+    // @ts-ignore
     if (end > lastVote) break;
     start = end;
   }
@@ -85,6 +90,7 @@ async function main(space) {
   process.exit(0);
 }
 
-const space = process.argv[process.argv.indexOf('--space') + 1];
-if (!space) throw new Error('Space ID is missing');
+const spaceArg = process.argv.indexOf('--space');
+const space = spaceArg !== -1 ? process.argv[spaceArg + 1] : null;
+if (spaceArg !== -1 && !space) throw new Error('Space ID is missing');
 main(space);