feat: stake set

this code was not pushed to the repo

Author: greenlucid

contracts/kleros.json

@@ -518,6 +518,137 @@ export class TokenAndETHShift extends Entity {
   }
 }
 
+export class StakeSet extends Entity {
+  constructor(id: string) {
+    super();
+    this.set("id", Value.fromString(id));
+  }
+
+  save(): void {
+    let id = this.get("id");
+    assert(id != null, "Cannot save StakeSet entity without an ID");
+    if (id) {
+      assert(
+        id.kind == ValueKind.STRING,
+        `Entities of type StakeSet must have an ID of type String but the id '${id.displayData()}' is of type ${id.displayKind()}`
+      );
+      store.set("StakeSet", id.toString(), this);
+    }
+  }
+
+  static loadInBlock(id: string): StakeSet | null {
+    return changetype<StakeSet | null>(store.get_in_block("StakeSet", id));
+  }
+
+  static load(id: string): StakeSet | null {
+    return changetype<StakeSet | null>(store.get("StakeSet", id));
+  }
+
+  get id(): string {
+    let value = this.get("id");
+    if (!value || value.kind == ValueKind.NULL) {
+      throw new Error("Cannot return null for a required field.");
+    } else {
+      return value.toString();
+    }
+  }
+
+  set id(value: string) {
+    this.set("id", Value.fromString(value));
+  }
+
+  get address(): Bytes {
+    let value = this.get("address");
+    if (!value || value.kind == ValueKind.NULL) {
+      throw new Error("Cannot return null for a required field.");
+    } else {
+      return value.toBytes();
+    }
+  }
+
+  set address(value: Bytes) {
+    this.set("address", Value.fromBytes(value));
+  }
+
+  get subcourtID(): BigInt {
+    let value = this.get("subcourtID");
+    if (!value || value.kind == ValueKind.NULL) {
+      throw new Error("Cannot return null for a required field.");
+    } else {
+      return value.toBigInt();
+    }
+  }
+
+  set subcourtID(value: BigInt) {
+    this.set("subcourtID", Value.fromBigInt(value));
+  }
+
+  get stake(): BigInt {
+    let value = this.get("stake");
+    if (!value || value.kind == ValueKind.NULL) {
+      throw new Error("Cannot return null for a required field.");
+    } else {
+      return value.toBigInt();
+    }
+  }
+
+  set stake(value: BigInt) {
+    this.set("stake", Value.fromBigInt(value));
+  }
+
+  get newTotalStake(): BigInt {
+    let value = this.get("newTotalStake");
+    if (!value || value.kind == ValueKind.NULL) {
+      throw new Error("Cannot return null for a required field.");
+    } else {
+      return value.toBigInt();
+    }
+  }
+
+  set newTotalStake(value: BigInt) {
+    this.set("newTotalStake", Value.fromBigInt(value));
+  }
+
+  get timestamp(): BigInt {
+    let value = this.get("timestamp");
+    if (!value || value.kind == ValueKind.NULL) {
+      throw new Error("Cannot return null for a required field.");
+    } else {
+      return value.toBigInt();
+    }
+  }
+
+  set timestamp(value: BigInt) {
+    this.set("timestamp", Value.fromBigInt(value));
+  }
+
+  get blocknumber(): BigInt {
+    let value = this.get("blocknumber");
+    if (!value || value.kind == ValueKind.NULL) {
+      throw new Error("Cannot return null for a required field.");
+    } else {
+      return value.toBigInt();
+    }
+  }
+
+  set blocknumber(value: BigInt) {
+    this.set("blocknumber", Value.fromBigInt(value));
+  }
+
+  get logIndex(): BigInt {
+    let value = this.get("logIndex");
+    if (!value || value.kind == ValueKind.NULL) {
+      throw new Error("Cannot return null for a required field.");
+    } else {
+      return value.toBigInt();
+    }
+  }
+
+  set logIndex(value: BigInt) {
+    this.set("logIndex", Value.fromBigInt(value));
+  }
+}
+
 export class EvidenceGroup extends Entity {
   constructor(id: Bytes) {
     super();

generated/Kleros/Kleros.ts

@@ -5,6 +5,7 @@ import {
   NewPeriod as NewPeriodEv,
   Draw as DrawEv,
   TokenAndETHShift as TokenAndETHShiftEv,
+  StakeSet as StakeSetEv,
   Kleros,
 } from "../generated/Kleros/Kleros";
 import {
@@ -21,6 +22,7 @@ import {
   Evidence,
   EvidenceGroup,
   Round,
+  StakeSet,
   TokenAndETHShift,
 } from "../generated/schema";
 import { ADDRESS, ONE, ZERO, ZERO_B } from "./const";
@@ -164,6 +166,31 @@ export function handleTokenAndETHShift(ev: TokenAndETHShiftEv): void {
   shift.save();
 }
 
+export function handleStakeSet(ev: StakeSetEv): void {
+  // Keep rolling until we find a free ID
+  let i = 0;
+  while (true) {
+    const stakeSet = StakeSet.load(
+      `${ev.params._address.toHexString()}-${ev.params._subcourtID}-${i}`
+    );
+    if (stakeSet == null) break;
+    i++;
+  }
+
+  const stakeSet = new StakeSet(
+    `${ev.params._address.toHexString()}-${ev.params._subcourtID}-${i}`
+  );
+
+  stakeSet.address = ev.params._address;
+  stakeSet.subcourtID = ev.params._subcourtID;
+  stakeSet.stake = ev.params._stake;
+  stakeSet.newTotalStake = ev.params._newTotalStake;
+  stakeSet.timestamp = ev.block.timestamp;
+  stakeSet.blocknumber = ev.block.number;
+  stakeSet.logIndex = ev.logIndex;
+  stakeSet.save();
+}
+
 export function handleMetaEvidence(ev: MetaEvidenceEv): void {
   const arbitrableHistory = new ArbitrableHistory(
     Bytes.fromByteArray(

generated/schema.ts

@@ -7,7 +7,7 @@
   "scripts": {
     "build": "graph build",
     "gen": "graph codegen --output-dir generated",
-    "deploy-xdai": "mustache config/xdai.json subgraph.template.yaml > subgraph.yaml && mustache config/xdai.json mappings/const.template.ts > mappings/const.ts && yarn gen && graph deploy --product hosted-service andreimvp/kleros-display",
-    "deploy-mainnet": "mustache config/mainnet.json subgraph.template.yaml > subgraph.yaml && mustache config/mainnet.json mappings/const.template.ts > mappings/const.ts && yarn gen && graph deploy --product hosted-service andreimvp/kleros-display-mainnet"
+    "deploy-xdai": "mustache config/xdai.json subgraph.template.yaml > subgraph.yaml && mustache config/xdai.json mappings/const.template.ts > mappings/const.ts && yarn gen && graph deploy --product hosted-service greenlucid/kleros-display",
+    "deploy-mainnet": "mustache config/mainnet.json subgraph.template.yaml > subgraph.yaml && mustache config/mainnet.json mappings/const.template.ts > mappings/const.ts && yarn gen && graph deploy --product hosted-service greenlucid/kleros-display-mainnet"
   }
 }

mappings/index.ts

@@ -56,6 +56,18 @@ type TokenAndETHShift @entity {
   tokenAmount: BigInt!
 }
 
+type StakeSet @entity {
+  "address-subcourtId-(discriminator)"
+  id: ID!
+  address: Bytes!
+  subcourtID: BigInt!
+  stake: BigInt!
+  newTotalStake: BigInt!
+  timestamp: BigInt!
+  blocknumber: BigInt!
+  logIndex: BigInt!
+}
+
 type EvidenceGroup @entity {
   id: Bytes!
   dispute: Dispute

package.json

@@ -33,6 +33,8 @@ dataSources:
           handler: handleDraw
         - event: TokenAndETHShift(indexed address,indexed uint256,int256,int256)
           handler: handleTokenAndETHShift
+        - event: StakeSet(indexed address,uint256,uint128,uint256)
+          handler: handleStakeSet
 templates:
   - name: Arbitrable
     kind: ethereum/contract

schema.graphql

@@ -5,11 +5,11 @@ schema:
 dataSources:
   - name: Kleros
     kind: ethereum
-    network: xdai
+    network: mainnet
     source:
-      address: "0x9c1da9a04925bdfdedf0f6421bc7eea8305f9002"
+      address: "0x988b3a538b618c7a603e1c11ab82cd16dbe28069"
       abi: Kleros
-      startBlock: 16895601
+      startBlock: 7303699
     mapping:
       kind: ethereum/events
       apiVersion: 0.0.7
@@ -33,10 +33,12 @@ dataSources:
           handler: handleDraw
         - event: TokenAndETHShift(indexed address,indexed uint256,int256,int256)
           handler: handleTokenAndETHShift
+        - event: StakeSet(indexed address,uint256,uint128,uint256)
+          handler: handleStakeSet
 templates:
   - name: Arbitrable
     kind: ethereum/contract
-    network: xdai
+    network: mainnet
     source:
       abi: Arbitrable
     mapping: