feat: store proposal, vote and follower count on space

scripts/refresh_spaces_counters.ts

@@ -0,0 +1,35 @@
+import 'dotenv/config';
+import db from '../src/helpers/mysql';
+
+// Usage: yarn ts-node scripts/refresh_spaces_counters.ts
+async function main() {
+  const spaces = await db.queryAsync(`SELECT id FROM spaces`);
+
+  console.log(`Found ${spaces.length} spaces`);
+
+  for (const i in spaces) {
+    const stats = await db.queryAsync(
+      `SELECT COUNT(voter) as vote_count, COUNT(DISTINCT(proposal)) as proposal_count FROM votes WHERE space = ? GROUP BY space`,
+      [spaces[i].id]
+    );
+
+    for (const stat of stats) {
+      await db.queryAsync(`UPDATE spaces SET vote_count = ?, proposal_count = ? WHERE id = ?`, [
+        stat.vote_count,
+        stat.proposal_count,
+        spaces[i].id
+      ]);
+      console.log(`${i} / ${spaces.length}`);
+    }
+  }
+}
+
+(async () => {
+  try {
+    await main();
+    process.exit(0);
+  } catch (e) {
+    console.error(e);
+    process.exit(1);
+  }
+})();

src/writer/delete-proposal.ts

@@ -21,6 +21,7 @@ export async function verify(body): Promise<any> {
 export async function action(body): Promise<void> {
   const msg = jsonParse(body.msg);
   const proposal = await getProposal(msg.space, msg.payload.proposal);
+
   const voters = await db.queryAsync(`SELECT voter FROM votes WHERE proposal = ?`, [
     msg.payload.proposal
   ]);
@@ -33,9 +34,12 @@ export async function action(body): Promise<void> {
       SET proposal_count = GREATEST(proposal_count - 1, 0)
       WHERE user = ? AND space = ?
       LIMIT 1;
+    UPDATE spaces
+      SET proposal_count = GREATEST(proposal_count - 1, 0), vote_count = GREATEST(vote_count - ?, 0)
+      WHERE id = ?;
   `;
 
-  const parameters = [id, id, proposal.author, msg.space];
+  const parameters = [id, id, proposal.author, msg.space, voters.length, msg.space];
 
   if (voters.length > 0) {
     queries += `

src/writer/follow.ts

@@ -18,6 +18,11 @@ export const networks =
 export const defaultNetwork = networks[0];
 
 export async function verify(message): Promise<any> {
+  const query = `SELECT * FROM follows WHERE follower = ? AND space = ? LIMIT 1`;
+  const follows = await db.queryAsync(query, [message.from, message.space]);
+
+  if (follows.length !== 0) return Promise.reject('you are already following this space');
+
   const count = await getFollowsCount(message.from);
 
   if (count >= FOLLOWS_LIMIT_PER_USER) {
@@ -31,7 +36,11 @@ export async function verify(message): Promise<any> {
   return true;
 }
 
-export async function action(message, ipfs, receipt, id): Promise<void> {
+export async function action(message, ipfs, _receipt, id): Promise<void> {
+  const query = `
+    INSERT INTO follows SET ?;
+    UPDATE spaces SET follower_count = follower_count + 1 WHERE id = ?;
+  `;
   const params = {
     id,
     ipfs,
@@ -41,5 +50,5 @@ export async function action(message, ipfs, receipt, id): Promise<void> {
     created: message.timestamp
   };
 
-  await db.queryAsync('INSERT INTO follows SET ?', params);
+  await db.queryAsync(query, [params, message.space]);
 }

src/writer/proposal.ts

@@ -253,9 +253,10 @@ export async function action(body, ipfs, receipt, id): Promise<void> {
   const query = `
     INSERT INTO proposals SET ?;
     INSERT INTO leaderboard (space, user, proposal_count)
-    VALUES(?, ?, 1)
-    ON DUPLICATE KEY UPDATE proposal_count = proposal_count + 1
+      VALUES(?, ?, 1)
+      ON DUPLICATE KEY UPDATE proposal_count = proposal_count + 1;
+    UPDATE spaces SET proposal_count = proposal_count + 1 WHERE id = ?;
   `;
 
-  await db.queryAsync(query, [proposal, space, author]);
+  await db.queryAsync(query, [proposal, space, author, space]);
 }

src/writer/unfollow.ts

@@ -1,11 +1,28 @@
 import db from '../helpers/mysql';
 import { defaultNetwork } from './follow';
 
-export async function verify(): Promise<any> {
+export async function verify(message): Promise<any> {
+  const query = `SELECT * FROM follows WHERE follower = ? AND space = ? AND network = ? LIMIT 1`;
+  const follows = await db.queryAsync(query, [
+    message.from,
+    message.space,
+    message.network || defaultNetwork
+  ]);
+
+  if (follows.length === 0) return Promise.reject('you can only unfollow a space you follow');
+
   return true;
 }
 
 export async function action(message): Promise<void> {
-  const query = 'DELETE FROM follows WHERE follower = ? AND space = ? AND network = ? LIMIT 1';
-  await db.queryAsync(query, [message.from, message.space, message.network || defaultNetwork]);
+  const query = `
+    DELETE FROM follows WHERE follower = ? AND space = ? AND network = ? LIMIT 1;
+    UPDATE spaces SET follower_count = GREATEST(follower_count - 1, 0) WHERE id = ?;
+  `;
+  await db.queryAsync(query, [
+    message.from,
+    message.space,
+    message.network || defaultNetwork,
+    message.space
+  ]);
 }

src/writer/vote.ts

@@ -183,9 +183,10 @@ export async function action(body, ipfs, receipt, id, context): Promise<void> {
         INSERT INTO votes SET ?;
         INSERT INTO leaderboard (space, user, vote_count, last_vote)
           VALUES(?, ?, 1, ?)
-          ON DUPLICATE KEY UPDATE vote_count = vote_count + 1, last_vote = ?
+          ON DUPLICATE KEY UPDATE vote_count = vote_count + 1, last_vote = ?;
+        UPDATE spaces SET vote_count = vote_count + 1 WHERE id = ?;
       `,
-      [params, msg.space, voter, created, created]
+      [params, msg.space, voter, created, created, msg.space]
     );
   }