feat: set proposal vp value on proposal creation

chore: fix types

Author: wa0x6e

src/helpers/vpValue.ts

@@ -0,0 +1,58 @@
+import hubDB from './mysql';
+import { fetchWithKeepAlive } from './utils';
+
+type Proposal = {
+  id: string;
+  network: string;
+  strategies: any[];
+  start: number;
+};
+
+const OVERLORD_URL = 'https://overlord.snapshot.org';
+const CB_LAST = 5;
+const CB_ERROR = 0;
+
+export async function setProposalVpValue(proposal: Proposal) {
+  if (proposal.start > Date.now()) {
+    return;
+  }
+
+  try {
+    const vpValue = await getVpValue(proposal);
+
+    await hubDB.queryAsync('UPDATE proposals SET vp_value = ?, cb = ? WHERE id = ? LIMIT 1', [
+      vpValue,
+      CB_LAST,
+      proposal.id
+    ]);
+  } catch {
+    await hubDB.queryAsync('UPDATE proposals SET cb = ? WHERE id = ? LIMIT 1', [
+      CB_ERROR,
+      proposal.id
+    ]);
+    return;
+  }
+}
+
+async function getVpValue(proposal: Proposal) {
+  const init = {
+    method: 'POST',
+    headers: {
+      Accept: 'application/json',
+      'Content-Type': 'application/json'
+    },
+    body: JSON.stringify({
+      jsonrpc: '2.0',
+      method: 'get_vp_value_by_strategy',
+      params: {
+        network: proposal.network,
+        strategies: proposal.strategies,
+        snapshot: proposal.start
+      },
+      id: Math.random().toString(36).substring(7)
+    })
+  };
+  const res = await fetchWithKeepAlive(OVERLORD_URL, init);
+  const { result } = await res.json();
+  return result;
+}

src/writer/proposal.ts

@@ -10,6 +10,7 @@ import { isMalicious } from '../helpers/monitoring';
 import db from '../helpers/mysql';
 import { getLimits, getSpaceType } from '../helpers/options';
 import { captureError, getQuorum, jsonParse, validateChoices } from '../helpers/utils';
+import { setProposalVpValue } from '../helpers/vpValue';
 
 const scoreAPIUrl = process.env.SCORE_API_URL || 'https://score.snapshot.org';
 const broviderUrl = process.env.BROVIDER_URL || 'https://rpc.snapshot.org';
@@ -316,4 +317,6 @@ export async function action(body, ipfs, receipt, id): Promise<void> {
   `;
 
   await db.queryAsync(query, [proposal, space, author, space]);
+
+  setProposalVpValue({ ...proposal, strategies: spaceSettings.strategies });
 }