fix: newtotalstake is juror specific

Author: kemuru

subgraph/core-university/src/SortitionModule.ts

@@ -1,18 +1,32 @@
-import { SortitionModule, StakeLocked, StakeSet } from "../generated/SortitionModule/SortitionModule";
+import { SortitionModule, StakeLocked, StakeSet as StakeSetEvent } from "../generated/SortitionModule/SortitionModule";
+import { StakeSet as StakeSetEntity, User } from "../generated/schema";
 
 import { updateJurorDelayedStake, updateJurorStake } from "./entities/JurorTokensPerCourt";
 import { ensureUser } from "./entities/User";
 import { ZERO } from "./utils";
 
-export function handleStakeSet(event: StakeSet): void {
+export function handleStakeSet(event: StakeSetEvent): 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);
+
+  const juror = User.load(jurorAddress);
+  if (!juror) return;
+  const stakeSet = new StakeSetEntity(event.transaction.hash.toHex() + "-" + event.logIndex.toString());
+  stakeSet.address = jurorAddress;
+  stakeSet.courtID = event.params._courtID;
+  stakeSet.stake = event.params._amount;
+  stakeSet.newTotalStake = juror.totalStake;
+  stakeSet.blocknumber = event.block.number;
+  stakeSet.timestamp = event.block.timestamp;
+  stakeSet.logIndex = event.logIndex;
+  stakeSet.save();
 }
+
 export function handleStakeLocked(event: StakeLocked): void {
   // NOP
 }

subgraph/core/schema.graphql

@@ -356,7 +356,7 @@ type ClassicContribution implements Contribution @entity {
 
 type StakeSet @entity {
   id: ID! # event.transaction.hash.toHex() + - + event.logIndex.toString()
-  address: Bytes!
+  address: String!
   courtID: BigInt!
   stake: BigInt!
   newTotalStake: BigInt!

subgraph/core/src/SortitionModule.ts

@@ -6,7 +6,7 @@ import {
   StakeLocked,
   StakeSet as StakeSetEvent,
 } from "../generated/SortitionModule/SortitionModule";
-import { Court, StakeSet as StakeSetEntity } from "../generated/schema";
+import { StakeSet as StakeSetEntity, User } from "../generated/schema";
 
 import { updateJurorDelayedStake, updateJurorStake } from "./entities/JurorTokensPerCourt";
 import { ensureUser } from "./entities/User";
@@ -33,13 +33,13 @@ export function handleStakeSet(event: StakeSetEvent): void {
   //stake is updated instantly so no delayed amount, set delay amount to zero
   updateJurorDelayedStake(jurorAddress, courtID, ZERO);
 
-  const generalCourt = Court.load("1");
-  if (!generalCourt) return;
+  const juror = User.load(jurorAddress);
+  if (!juror) return;
   const stakeSet = new StakeSetEntity(event.transaction.hash.toHex() + "-" + event.logIndex.toString());
-  stakeSet.address = event.params._address;
+  stakeSet.address = jurorAddress;
   stakeSet.courtID = event.params._courtID;
   stakeSet.stake = event.params._amount;
-  stakeSet.newTotalStake = generalCourt.effectiveStake;
+  stakeSet.newTotalStake = juror.totalStake;
   stakeSet.blocknumber = event.block.number;
   stakeSet.timestamp = event.block.timestamp;
   stakeSet.logIndex = event.logIndex;