fix: use try_function (#598)

* fix: use try_function * Fix: update node version in subgraph test flow * fix: upgrade node version
aragon · May 29, 2024 · a543127 · a543127
1 parent 973bef6
commit a543127
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 9 deletions.
diff --git a/.github/workflows/subgraph-tests.yml b/.github/workflows/subgraph-tests.yml
@@ -34,7 +34,7 @@ jobs:
         uses: actions/setup-node@v3
         with:
           cache: 'yarn'
-          node-version: 16
+          node-version: 18
       - name: Install dependencies
         run: yarn install --pure-lockfile
       - name: Build manifest
@@ -56,7 +56,7 @@ jobs:
         uses: actions/setup-node@v3
         with:
           cache: 'yarn'
-          node-version: 16
+          node-version: 18
       - name: Install dependencies
         run: yarn
       - name: Build contracts

diff --git a/packages/subgraph/src/packages/token/utils.ts b/packages/subgraph/src/packages/token/utils.ts
@@ -15,11 +15,14 @@ export function getDelegation(
   tokenAddress: Address
 ): string | null {
   let contract = GovernanceERC20Contract.bind(tokenAddress);
-  let delegate = contract.delegates(user);
+  let delegate = contract.try_delegates(user);
 
-  return delegate === Address.fromString(ADDRESS_ZERO)
-    ? null
-    : delegate.toHexString();
+  if (!delegate.reverted) {
+    return delegate.value === Address.fromString(ADDRESS_ZERO)
+      ? null
+      : delegate.value.toHexString();
+  }
+  return null;
 }
 
 export function getDelegateeId(
@@ -36,10 +39,16 @@ export function getDelegateeId(
     : null;
 }
 
-export function getVotingPower(user: Address, tokenAddress: Address): BigInt {
+export function getVotingPower(
+  user: Address,
+  tokenAddress: Address
+): BigInt | null {
   let contract = GovernanceERC20Contract.bind(tokenAddress);
-  let votingPower = contract.getVotes(user);
-  return votingPower;
+  let votingPower = contract.try_getVotes(user);
+  if (!votingPower.reverted) {
+    return votingPower.value;
+  }
+  return null;
 }
 
 /**