feat(subgraph): handle court jump and refactor

Author: shotaronowhere

subgraph/schema.graphql

@@ -119,9 +119,6 @@ type Court @entity {
   id: ID!
   policy: String
   name: String
-  description: String
-  summary: String
-  requiredSkills: String
   parent: Court
   hiddenVotes: Boolean!
   children: [Court!]! @derivedFrom(field: "parent")
@@ -156,14 +153,14 @@ type Dispute @entity {
   overridden: Boolean!
   periodDeadline: BigInt!
   periodNotificationIndex: BigInt!
-  lastPeriodChangeTs: BigInt!
+  lastPeriodChange: BigInt!
   lastPeriodChangeBlockNumber: BigInt!
   rounds: [Round!]! @derivedFrom(field: "dispute")
   currentRound: Round!
   currentRoundIndex: BigInt!
   jurors: [User!]! @derivedFrom(field: "disputes")
   shifts: [TokenAndETHShift!]! @derivedFrom(field: "dispute")
-  disputeKitDispute: DisputeKitDispute @derivedFrom(field: "coreDispute")
+  disputeKitDispute: [DisputeKitDispute!]! @derivedFrom(field: "coreDispute")
 }
 
 type PeriodIndexCounter @entity {
@@ -241,7 +238,6 @@ type ClassicDispute implements DisputeKitDispute @entity {
   currentLocalRoundIndex: BigInt!
 
   numberOfChoices: BigInt!
-  jumped: Boolean!
   extraData: Bytes!
 }
 

subgraph/src/KlerosCore.ts

@@ -10,6 +10,7 @@ import {
   NewPeriod,
   StakeSet,
   TokenAndETHShift as TokenAndETHShiftEvent,
+  CourtJump,
   Ruling,
   StakeDelayed,
   AcceptedFeeToken,
@@ -25,10 +26,11 @@ import { updateJurorDelayedStake, updateJurorStake } from "./entities/JurorToken
 import { createDrawFromEvent } from "./entities/Draw";
 import { updateTokenAndEthShiftFromEvent } from "./entities/TokenAndEthShift";
 import { updateArbitrableCases } from "./entities/Arbitrable";
-import { Court, Dispute, User, PeriodIndexCounter } from "../generated/schema";
+import { Court, Dispute, User } from "../generated/schema";
 import { BigInt } from "@graphprotocol/graph-ts";
 import { updatePenalty } from "./entities/Penalty";
 import { ensureFeeToken } from "./entities/FeeToken";
+import { getAndIncrementCounter } from "./entities/PeriodIndexCounter";
 
 function getPeriodName(index: i32): string {
   const periodArray = ["evidence", "commit", "vote", "appeal", "execution"];
@@ -126,21 +128,11 @@ export function handleNewPeriod(event: NewPeriod): void {
   }
 
   dispute.period = newPeriod;
-  dispute.lastPeriodChangeTs = event.block.timestamp;
-  dispute.lastPeriodChangeBlockNumber = event.block.number;
-  let counter = PeriodIndexCounter.load(newPeriod);
-  if (!counter) {
-    counter = new PeriodIndexCounter(newPeriod);
-    counter.counter = BigInt.fromI32(0);
-  }
-  dispute.periodNotificationIndex = counter.counter;
-  counter.counter = counter.counter.plus(ONE);
-  counter.save();
+  dispute.lastPeriodChange = event.block.timestamp;
   dispute.lastPeriodChangeBlockNumber = event.block.number;
+  dispute.periodNotificationIndex = getAndIncrementCounter(newPeriod);
   if (newPeriod !== "execution") {
-    const contract = KlerosCore.bind(event.address);
-    const courtContractState = contract.getTimesPerPeriod(BigInt.fromString(dispute.court));
-    dispute.periodDeadline = event.block.timestamp.plus(courtContractState[event.params._period]);
+    dispute.periodDeadline = event.block.timestamp.plus(court.timesPerPeriod[event.params._period]);
   } else {
     dispute.periodDeadline = BigInt.fromU64(U64.MAX_VALUE);
   }
@@ -175,6 +167,13 @@ export function handleAppealDecision(event: AppealDecision): void {
   createRoundFromRoundInfo(disputeID, newRoundIndex, roundInfo);
 }
 
+export function handleCourtJump(event: CourtJump): void {
+  const dispute = Dispute.load(event.params._disputeID.toString());
+  if (!dispute) return;
+  dispute.court = event.params._toCourtID.toString();
+  dispute.save();
+}
+
 export function handleDraw(event: DrawEvent): void {
   createDrawFromEvent(event);
   const disputeID = event.params._disputeID.toString();

subgraph/src/PolicyRegistry.ts

@@ -1,6 +1,5 @@
 import { PolicyUpdate } from "../generated/PolicyRegistry/PolicyRegistry";
 import { Court } from "../generated/schema";
-import { ipfs, json, Bytes, TypedMap, JSONValue } from "@graphprotocol/graph-ts";
 
 export function handlePolicyUpdate(event: PolicyUpdate): void {
   const courtID = event.params._courtID.toString();
@@ -9,31 +8,5 @@ export function handlePolicyUpdate(event: PolicyUpdate): void {
     court.policy = event.params._policy;
     court.name = event.params._courtName;
     court.save();
-
-    const jsonObj = getJsonObj(event.params._policy);
-    if (!jsonObj) return;
-
-    court.name = tryGetValue(jsonObj, "name");
-    court.description = tryGetValue(jsonObj, "description");
-    court.requiredSkills = tryGetValue(jsonObj, "requiredSkills");
-    court.save();
   }
 }
-
-function tryGetValue(jsonObj: TypedMap<string, JSONValue>, key: string): string | null {
-  const value = jsonObj.get(key);
-  return value ? value.toString() : null;
-}
-
-function getJsonObj(CID: string): TypedMap<string, JSONValue> | null {
-  const jsonStr = ipfs.cat(CID);
-  if (!jsonStr) return null;
-
-  const jsonObjValueAndSuccess = json.try_fromBytes(jsonStr as Bytes);
-  if (!jsonObjValueAndSuccess.isOk) return null;
-
-  const jsonObj = jsonObjValueAndSuccess.value.toObject();
-  if (!jsonObj) return null;
-
-  return jsonObj;
-}

subgraph/src/entities/ClassicDispute.ts

@@ -8,7 +8,6 @@ export function createClassicDisputeFromEvent(event: DisputeCreation): void {
   classicDispute.coreDispute = coreDisputeID;
   classicDispute.currentLocalRoundIndex = ZERO;
   classicDispute.numberOfChoices = event.params._numberOfChoices;
-  classicDispute.jumped = false;
   classicDispute.extraData = event.params._extraData;
   classicDispute.save();
 }

subgraph/src/entities/Dispute.ts

@@ -16,7 +16,7 @@ export function createDisputeFromEvent(event: DisputeCreation): void {
   dispute.currentRuling = ZERO;
   dispute.tied = true;
   dispute.overridden = false;
-  dispute.lastPeriodChangeTs = event.block.timestamp;
+  dispute.lastPeriodChange = event.block.timestamp;
   dispute.lastPeriodChangeBlockNumber = event.block.number;
   dispute.periodNotificationIndex = getAndIncrementCounter(dispute.period);
   const court = Court.load(courtID);

subgraph/subgraph.yaml

@@ -1,8 +1,6 @@
 specVersion: 0.0.4
 schema:
   file: ./schema.graphql
-features:
-  - ipfsOnEthereumContracts
 dataSources:
   - kind: ethereum
     name: KlerosCore
@@ -58,6 +56,8 @@ dataSources:
           handler: handleRuling
         - event: AcceptedFeeToken(indexed address,indexed bool)
           handler: handleAcceptedFeeToken
+        - event: CourtJump(indexed uint256,indexed uint256,indexed uint96,uint96)
+          handler: handleCourtJump
       file: ./src/KlerosCore.ts
   - kind: ethereum
     name: PolicyRegistry