Merge pull request #1699 from kleros/feat/add-stats-court-page

feat: add four new stat variables, stake simulator

Author: jaybuidl

subgraph/core/schema.graphql

@@ -120,6 +120,7 @@ type JurorTokensPerCourt @entity {
   id: ID! # user.id-court.id
   juror: User!
   court: Court!
+  effectiveStake: BigInt!
   staked: BigInt!
   locked: BigInt!
   delayed: BigInt!

subgraph/core/src/entities/JurorTokensPerCourt.ts

@@ -23,6 +23,7 @@ export function createJurorTokensPerCourt(jurorAddress: string, courtID: string)
   const jurorTokens = new JurorTokensPerCourt(id);
   jurorTokens.juror = jurorAddress;
   jurorTokens.court = courtID;
+  jurorTokens.effectiveStake = ZERO;
   jurorTokens.staked = ZERO;
   jurorTokens.locked = ZERO;
   jurorTokens.delayed = ZERO;
@@ -31,6 +32,37 @@ export function createJurorTokensPerCourt(jurorAddress: string, courtID: string)
   return jurorTokens;
 }
 
+export function updateJurorEffectiveStake(jurorAddress: string, courtID: string): void {
+  let court = Court.load(courtID);
+  if (!court) {
+    return;
+  }
+
+  while (court) {
+    const jurorTokensPerCourt = ensureJurorTokensPerCourt(jurorAddress, court.id);
+    let totalStake = jurorTokensPerCourt.staked;
+    const childrenCourts = court.children.load();
+
+    for (let i = 0; i < childrenCourts.length; i++) {
+      const childCourtID = childrenCourts[i].id;
+      const childCourt = Court.load(childCourtID);
+      if (childCourt) {
+        const childJurorTokensPerCourt = ensureJurorTokensPerCourt(jurorAddress, childCourt.id);
+        totalStake = totalStake.plus(childJurorTokensPerCourt.effectiveStake);
+      }
+    }
+
+    jurorTokensPerCourt.effectiveStake = totalStake;
+    jurorTokensPerCourt.save();
+
+    if (court.parent && court.parent !== null) {
+      court = Court.load(court.parent as string);
+    } else {
+      break;
+    }
+  }
+}
+
 export function updateJurorStake(
   jurorAddress: string,
   courtID: string,
@@ -61,6 +93,7 @@ export function updateJurorStake(
   juror.save();
   court.save();
   updateEffectiveStake(courtID);
+  updateJurorEffectiveStake(jurorAddress, courtID);
 }
 
 export function updateJurorDelayedStake(jurorAddress: string, courtID: string, amount: BigInt): void {

subgraph/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kleros/kleros-v2-subgraph",
-  "version": "0.8.6",
+  "version": "0.8.8",
   "license": "MIT",
   "scripts": {
     "update:core:arbitrum-sepolia-devnet": "./scripts/update.sh arbitrumSepoliaDevnet arbitrum-sepolia core/subgraph.yaml",