fix(subgraph): fix-stake-handlers

Author: tractorss

subgraph/core/src/SortitionModule.ts

@@ -12,24 +12,11 @@ import { ensureUser } from "./entities/User";
 import { ZERO } from "./utils";
 
 export function handleStakeDelayedAlreadyTransferred(event: StakeDelayedAlreadyTransferred): void {
-  const jurorAddress = event.params._address.toHexString();
-  ensureUser(jurorAddress);
-  const courtID = event.params._courtID.toString();
-
-  updateJurorStake(jurorAddress, courtID.toString(), SortitionModule.bind(event.address), event.block.timestamp);
-
-  //stake is updated instantly so no delayed amount, set delay amount to zero
-  updateJurorDelayedStake(jurorAddress, courtID, ZERO);
+  updateJurorDelayedStake(event.params._address.toHexString(), event.params._courtID.toString(), event.params._amount);
 }
 
 export function handleStakeDelayedAlreadyTransferredWithdrawn(event: StakeDelayedAlreadyTransferredWithdrawn): void {
-  const jurorAddress = event.params._address.toHexString();
-  ensureUser(jurorAddress);
-  const courtID = event.params._courtID.toString();
-
-  updateJurorStake(jurorAddress, courtID.toString(), SortitionModule.bind(event.address), event.block.timestamp);
-
-  updateJurorDelayedStake(jurorAddress, courtID, ZERO);
+  updateJurorDelayedStake(event.params._address.toHexString(), event.params._courtID.toString(), event.params._amount);
 }
 
 export function handleStakeDelayedNotTransferred(event: StakeDelayedNotTransferred): void {

subgraph/core/src/entities/JurorTokensPerCourt.ts

@@ -42,21 +42,21 @@ export function updateJurorStake(
   const jurorTokens = ensureJurorTokensPerCourt(jurorAddress, courtID);
 
   // TODO: index the sortition module and handle these events there
-  // const jurorBalance = contract.getJurorBalance(Address.fromString(jurorAddress), BigInt.fromString(courtID));
-  // const previousStake = jurorTokens.staked;
-  // const previousTotalStake = juror.totalStake;
-  // jurorTokens.staked = jurorBalance.value2;
-  // jurorTokens.locked = jurorBalance.value1;
-  // jurorTokens.save();
-  // const stakeDelta = getDelta(previousStake, jurorTokens.staked);
-  // const newTotalStake = juror.totalStake.plus(stakeDelta);
-  // juror.totalStake = newTotalStake;
-  // court.stake = court.stake.plus(stakeDelta);
-  // updateStakedPNK(stakeDelta, timestamp);
-  // const activeJurorsDelta = getActivityDelta(previousTotalStake, newTotalStake);
-  // const stakedJurorsDelta = getActivityDelta(previousStake, jurorBalance.value2);
-  // court.numberStakedJurors = court.numberStakedJurors.plus(stakedJurorsDelta);
-  // updateActiveJurors(activeJurorsDelta, timestamp);
+  const jurorBalance = contract.getJurorBalance(Address.fromString(jurorAddress), BigInt.fromString(courtID));
+  const previousStake = jurorTokens.staked;
+  const previousTotalStake = juror.totalStake;
+  jurorTokens.staked = jurorBalance.value2;
+  jurorTokens.locked = jurorBalance.value1;
+  jurorTokens.save();
+  const stakeDelta = getDelta(previousStake, jurorTokens.staked);
+  const newTotalStake = juror.totalStake.plus(stakeDelta);
+  juror.totalStake = newTotalStake;
+  court.stake = court.stake.plus(stakeDelta);
+  updateStakedPNK(stakeDelta, timestamp);
+  const activeJurorsDelta = getActivityDelta(previousTotalStake, newTotalStake);
+  const stakedJurorsDelta = getActivityDelta(previousStake, jurorBalance.value2);
+  court.numberStakedJurors = court.numberStakedJurors.plus(stakedJurorsDelta);
+  updateActiveJurors(activeJurorsDelta, timestamp);
   juror.save();
   court.save();
 }

subgraph/core/subgraph.yaml

@@ -25,6 +25,8 @@ dataSources:
         - DisputeKit
         - Counter
       abis:
+        - name: SortitionModule
+          file: ../../contracts/deployments/arbitrumSepoliaDevnet/SortitionModule.json
         - name: DisputeKitClassic
           file: ../../contracts/deployments/arbitrumSepoliaDevnet/DisputeKitClassic.json
         - name: KlerosCore