refactor: use getScoresDirect to get scores in getVp

src/utils.ts

@@ -2,9 +2,8 @@ import fetch from 'cross-fetch';
 import _strategies from './strategies';
 import snapshot from '@snapshot-labs/snapshot.js';
 import { getDelegations } from './utils/delegation';
-import { getVp } from './utils/vp';
 import { createHash } from 'crypto';
-import { Protocol, Score, Snapshot } from './types';
+import { Protocol, Score, Snapshot, VotingPower } from './types';
 import { VALID_PROTOCOLS } from './constants';
 
 export function sha256(str) {
@@ -75,6 +74,31 @@ export async function getScoresDirect(
   }
 }
 
+export async function getVp(
+  address: string,
+  network: string,
+  strategies: any[],
+  snapshot: Snapshot,
+  space: string
+): Promise<VotingPower> {
+  const scores: Score[] = await getScoresDirect(
+    space,
+    strategies,
+    network,
+    getProvider(network),
+    [address],
+    snapshot
+  );
+
+  const vpByStrategy = scores.map((score) => score[address] || 0);
+
+  return {
+    vp: vpByStrategy.reduce((a, b) => a + b, 0),
+    vp_by_strategy: vpByStrategy,
+    vp_state: snapshot === 'latest' ? 'pending' : 'final'
+  };
+}
+
 export function customFetch(
   url: RequestInfo | URL,
   options: RequestInit = {},

test/__snapshots__/vp.test.ts.snap → test/integration/__snapshots__/utils.test.ts.snap

@@ -1,6 +1,6 @@
 // Jest Snapshot v1, https://goo.gl/fbAQLP
 
-exports[` getVp 1`] = `
+exports[`utils getVp 1`] = `
 {
   "vp": 21.55404462002206,
   "vp_by_strategy": [

test/vp.test.ts → test/integration/utils.test.ts

@@ -1,4 +1,4 @@
-import { getVp } from '../src/utils/vp';
+import { getVp } from '../../src/utils';
 
 const address = '0xeF8305E140ac520225DAf050e2f71d5fBcC543e7';
 const space = 'cvx.eth';
@@ -30,10 +30,9 @@ const strategies = [
   }
 ];
 
-describe('', () => {
+describe('utils', () => {
   it('getVp', async () => {
     const scores = await getVp(address, network, strategies, snapshot, space);
     expect(scores).toMatchSnapshot();
-    console.log(scores);
   }, 20e3);
 });