feat: support for shutter disputekit in devnet

kemuru · kemuru · commit 317aed6c6f61 · 2025-05-07T00:10:02.000+02:00
diff --git a/subgraph/core/src/DisputeKitClassic.ts b/subgraph/core/src/DisputeKitClassic.ts
@@ -9,7 +9,15 @@ import {
   CommitCast,
 } from "../generated/DisputeKitClassic/DisputeKitClassic";
 import { KlerosCore } from "../generated/KlerosCore/KlerosCore";
-import { ClassicDispute, ClassicJustification, ClassicRound, ClassicVote, Dispute } from "../generated/schema";
+import {
+  ClassicDispute,
+  ClassicJustification,
+  ClassicRound,
+  ClassicVote,
+  Dispute,
+  DisputeKit,
+  Round,
+} from "../generated/schema";
 import { ensureClassicContributionFromEvent } from "./entities/ClassicContribution";
 import { createClassicDisputeFromEvent } from "./entities/ClassicDispute";
 import {
@@ -19,23 +27,31 @@ import {
   updateCountsAndGetCurrentRuling,
 } from "./entities/ClassicRound";
 import { ensureClassicVote } from "./entities/ClassicVote";
-import { ONE, ZERO } from "./utils";
-
-export const DISPUTEKIT_ID = "1";
+import { ONE, ZERO, extractDisputeKitIDFromExtraData } from "./utils";
 
 export function handleDisputeCreation(event: DisputeCreation): void {
   const disputeID = event.params._coreDisputeID.toString();
-  createClassicDisputeFromEvent(event);
+  const disputeKitID = extractDisputeKitIDFromExtraData(event.params._extraData);
+  createClassicDisputeFromEvent(event, disputeKitID);
   const numberOfChoices = event.params._numberOfChoices;
-  createClassicRound(disputeID, numberOfChoices, ZERO);
+  createClassicRound(disputeID, numberOfChoices, ZERO, disputeKitID);
 }
 
 export function handleCommitCast(event: CommitCast): void {
-  const coreDisputeID = event.params._coreDisputeID;
-  const coreDispute = Dispute.load(coreDisputeID.toString());
-  const classicDisputeID = `${DISPUTEKIT_ID}-${coreDisputeID}`;
+  const coreDisputeID = event.params._coreDisputeID.toString();
+  const coreDispute = Dispute.load(coreDisputeID);
+  if (!coreDispute) return;
+
+  const coreCurrentRound = Round.load(coreDispute.currentRound);
+  if (!coreCurrentRound) return;
+
+  const disputeKitID = coreCurrentRound.disputeKit;
+
+  const classicDisputeID = `${disputeKitID}-${coreDisputeID}`;
+
   const classicDispute = ClassicDispute.load(classicDisputeID);
-  if (!classicDispute || !coreDispute) return;
+  if (!classicDispute) return;
+
   const currentLocalRoundID = classicDispute.id + "-" + classicDispute.currentLocalRoundIndex.toString();
   const voteIDs = event.params._voteIDs;
   for (let i = 0; i < voteIDs.length; i++) {
@@ -55,9 +71,18 @@ export function handleVoteCast(event: VoteCast): void {
   const juror = event.params._juror.toHexString();
   const coreDisputeID = event.params._coreDisputeID.toString();
   const coreDispute = Dispute.load(coreDisputeID);
-  const classicDisputeID = `${DISPUTEKIT_ID}-${coreDisputeID}`;
+  if (!coreDispute) return;
+
+  const coreCurrentRound = Round.load(coreDispute.currentRound);
+  if (!coreCurrentRound) return;
+
+  const disputeKitID = coreCurrentRound.disputeKit;
+
+  const classicDisputeID = `${disputeKitID}-${coreDisputeID}`;
+
   const classicDispute = ClassicDispute.load(classicDisputeID);
-  if (!classicDispute || !coreDispute) return;
+  if (!classicDispute) return;
+
   const choice = event.params._choice;
   const currentLocalRoundID = classicDispute.id + "-" + classicDispute.currentLocalRoundIndex.toString();
   const voteIDs = event.params._voteIDs;
@@ -70,6 +95,7 @@ export function handleVoteCast(event: VoteCast): void {
   justification.transactionHash = event.transaction.hash.toHexString();
   justification.timestamp = event.block.timestamp;
   justification.save();
+
   const currentRulingInfo = updateCountsAndGetCurrentRuling(
     currentLocalRoundID,
     choice,
@@ -78,6 +104,7 @@ export function handleVoteCast(event: VoteCast): void {
   coreDispute.currentRuling = currentRulingInfo.ruling;
   coreDispute.tied = currentRulingInfo.tied;
   coreDispute.save();
+
   let classicVote: ClassicVote;
   for (let i = 0; i < voteIDs.length; i++) {
     classicVote = ensureClassicVote(currentLocalRoundID, juror, voteIDs[i], coreDispute);
@@ -97,7 +124,16 @@ export function handleChoiceFunded(event: ChoiceFunded): void {
   const coreDisputeID = event.params._coreDisputeID.toString();
   const coreRoundIndex = event.params._coreRoundID.toString();
   const choice = event.params._choice;
-  const roundID = `${DISPUTEKIT_ID}-${coreDisputeID}-${coreRoundIndex}`;
+
+  const coreDispute = Dispute.load(coreDisputeID);
+  if (!coreDispute) return;
+
+  const roundId = `${coreDisputeID}-${coreRoundIndex}`;
+  const coreRound = Round.load(roundId);
+  if (!coreRound) return;
+  const disputeKitID = coreRound.disputeKit;
+
+  const roundID = `${disputeKitID}-${coreDisputeID}-${coreRoundIndex}`;
 
   const localRound = ClassicRound.load(roundID);
   if (!localRound) return;
@@ -123,13 +159,13 @@ export function handleChoiceFunded(event: ChoiceFunded): void {
 
     localRound.feeRewards = localRound.feeRewards.minus(appealCost);
 
-    const localDispute = ClassicDispute.load(`${DISPUTEKIT_ID}-${coreDisputeID}`);
+    const localDispute = ClassicDispute.load(`${disputeKitID}-${coreDisputeID}`);
     if (!localDispute) return;
     const newRoundIndex = localDispute.currentLocalRoundIndex.plus(ONE);
     const numberOfChoices = localDispute.numberOfChoices;
     localDispute.currentLocalRoundIndex = newRoundIndex;
     localDispute.save();
-    createClassicRound(coreDisputeID, numberOfChoices, newRoundIndex);
+    createClassicRound(coreDisputeID, numberOfChoices, newRoundIndex, disputeKitID);
   }
 
   localRound.save();
@@ -144,7 +180,16 @@ export function handleWithdrawal(event: Withdrawal): void {
   // check if all appeal fees have been withdrawn
   const coreDisputeID = event.params._coreDisputeID.toString();
   const coreRoundIndex = event.params._coreRoundID.toString();
-  const roundID = `${DISPUTEKIT_ID}-${coreDisputeID}-${coreRoundIndex}`;
+
+  const coreDispute = Dispute.load(coreDisputeID);
+  if (!coreDispute) return;
+
+  const roundId = `${coreDisputeID}-${coreRoundIndex}`;
+  const coreRound = Round.load(roundId);
+  if (!coreRound) return;
+  const disputeKitID = coreRound.disputeKit;
+
+  const roundID = `${disputeKitID}-${coreDisputeID}-${coreRoundIndex}`;
 
   const localRound = ClassicRound.load(roundID);
   if (!localRound) return;
diff --git a/subgraph/core/src/KlerosCore.ts b/subgraph/core/src/KlerosCore.ts
@@ -156,7 +156,7 @@ export function handleNewPeriod(event: NewPeriod): void {
           updateTotalLeaderboardJurors(ONE, event.block.timestamp);
         }
 
-        // Since this is a ClassicVote entity, this will only work for the Classic DisputeKit (which has ID "1").
+        // Since this is a ClassicVote entity, this will only work for the ClassicDisputeKit and ShutterDisputeKit.
         const vote = ClassicVote.load(`${round.disputeKit}-${draw.id}`);
 
         if (!vote) {
diff --git a/subgraph/core/src/entities/ClassicContribution.ts b/subgraph/core/src/entities/ClassicContribution.ts
@@ -1,13 +1,23 @@
-import { ClassicContribution } from "../../generated/schema";
+import { ClassicContribution, Dispute, DisputeKit, Round } from "../../generated/schema";
 import { Contribution as ContributionEvent, Withdrawal } from "../../generated/DisputeKitClassic/DisputeKitClassic";
-import { DISPUTEKIT_ID } from "../DisputeKitClassic";
 import { ensureUser } from "./User";
 
 export function ensureClassicContributionFromEvent<T>(event: T): ClassicContribution | null {
   if (!(event instanceof ContributionEvent) && !(event instanceof Withdrawal)) return null;
   const coreDisputeID = event.params._coreDisputeID.toString();
   const coreRoundIndex = event.params._coreRoundID.toString();
-  const roundID = `${DISPUTEKIT_ID}-${coreDisputeID}-${coreRoundIndex}`;
+
+  const coreDispute = Dispute.load(coreDisputeID);
+  if (!coreDispute) return null;
+
+  const roundId = `${coreDisputeID}-${coreRoundIndex}`;
+  const coreRound = Round.load(roundId);
+  if (!coreRound) return null;
+
+  const disputeKitID = coreRound.disputeKit;
+
+  const roundID = `${disputeKitID}-${coreDisputeID}-${coreRoundIndex}`;
+
   ensureUser(event.params._contributor.toHexString());
   const contributor = event.params._contributor.toHexString();
   const choice = event.params._choice;
diff --git a/subgraph/core/src/entities/ClassicDispute.ts b/subgraph/core/src/entities/ClassicDispute.ts
@@ -1,10 +1,11 @@
+import { Bytes } from "@graphprotocol/graph-ts";
 import { DisputeCreation } from "../../generated/DisputeKitClassic/DisputeKitClassic";
 import { ClassicDispute } from "../../generated/schema";
 import { ZERO } from "../utils";
 
-export function createClassicDisputeFromEvent(event: DisputeCreation): void {
+export function createClassicDisputeFromEvent(event: DisputeCreation, disputeKitID: string): void {
   const coreDisputeID = event.params._coreDisputeID.toString();
-  const classicDispute = new ClassicDispute(`1-${coreDisputeID}`);
+  const classicDispute = new ClassicDispute(`${disputeKitID}-${coreDisputeID}`);
   classicDispute.coreDispute = coreDisputeID;
   classicDispute.currentLocalRoundIndex = ZERO;
   classicDispute.numberOfChoices = event.params._numberOfChoices;
diff --git a/subgraph/core/src/entities/ClassicRound.ts b/subgraph/core/src/entities/ClassicRound.ts
@@ -1,10 +1,15 @@
 import { BigInt } from "@graphprotocol/graph-ts";
 import { Contribution } from "../../generated/DisputeKitClassic/DisputeKitClassic";
-import { Answer, ClassicRound } from "../../generated/schema";
+import { Answer, ClassicRound, Dispute, Round } from "../../generated/schema";
 import { ZERO } from "../utils";
 
-export function createClassicRound(disputeID: string, numberOfChoices: BigInt, roundIndex: BigInt): void {
-  const localDisputeID = `1-${disputeID}`;
+export function createClassicRound(
+  disputeID: string,
+  numberOfChoices: BigInt,
+  roundIndex: BigInt,
+  disputeKitID: string
+): void {
+  const localDisputeID = `${disputeKitID}-${disputeID}`;
   const id = `${localDisputeID}-${roundIndex.toString()}`;
   const classicRound = new ClassicRound(id);
   classicRound.localDispute = localDisputeID;
@@ -67,9 +72,16 @@ export function updateCountsAndGetCurrentRuling(id: string, choice: BigInt, delt
 }
 
 export function updateChoiceFundingFromContributionEvent(event: Contribution): void {
-  const disputeKitID = "1";
   const coreDisputeID = event.params._coreDisputeID.toString();
   const coreRoundIndex = event.params._coreRoundID.toString();
+  const coreDispute = Dispute.load(coreDisputeID);
+  if (!coreDispute) return;
+
+  const roundId = `${coreDisputeID}-${coreRoundIndex}`;
+  const coreRound = Round.load(roundId);
+  if (!coreRound) return;
+  const disputeKitID = coreRound.disputeKit;
+
   const roundID = `${disputeKitID}-${coreDisputeID}-${coreRoundIndex}`;
 
   const classicRound = ClassicRound.load(roundID);
diff --git a/subgraph/core/src/utils.ts b/subgraph/core/src/utils.ts
@@ -1,4 +1,10 @@
-import { BigInt } from "@graphprotocol/graph-ts";
+import { BigInt, Bytes } from "@graphprotocol/graph-ts";
 
 export const ZERO = BigInt.fromI32(0);
 export const ONE = BigInt.fromI32(1);
+
+export function extractDisputeKitIDFromExtraData(extraData: Bytes): string {
+  const start = extraData.length - 32;
+  const littleEndian = extraData.subarray(start, extraData.length).reverse();
+  return BigInt.fromUnsignedBytes(Bytes.fromUint8Array(littleEndian)).toString();
+}
diff --git a/subgraph/core/subgraph.yaml b/subgraph/core/subgraph.yaml
@@ -120,6 +120,42 @@ dataSources:
         - event: CommitCast(indexed uint256,indexed address,uint256[],bytes32)
           handler: handleCommitCast
       file: ./src/DisputeKitClassic.ts
+  - kind: ethereum
+    name: DisputeKitShutter
+    network: arbitrum-sepolia
+    source:
+      address: "0x09F3d00B995186D76Af9AA8627D06351d0d9f950"
+      abi: DisputeKitShutter
+      startBlock: 148194178
+    mapping:
+      kind: ethereum/events
+      apiVersion: 0.0.7
+      language: wasm/assemblyscript
+      entities:
+        - ClassicDispute
+        - ClassicRound
+        - ClassicVote
+        - ClassicContribution
+      abis:
+        - name: DisputeKitShutter
+          file: ../../contracts/deployments/arbitrumSepoliaDevnet/DisputeKitShutter.json
+        - name: KlerosCore
+          # FIX: temporarily point to abi with event addition
+          file: ./abi-migrations/KlerosCore.json
+      eventHandlers:
+        - event: DisputeCreation(indexed uint256,uint256,bytes)
+          handler: handleDisputeCreation
+        - event: Contribution(indexed uint256,indexed uint256,uint256,indexed address,uint256)
+          handler: handleContributionEvent
+        - event: Withdrawal(indexed uint256,indexed uint256,uint256,indexed address,uint256)
+          handler: handleWithdrawal
+        - event: ChoiceFunded(indexed uint256,indexed uint256,indexed uint256)
+          handler: handleChoiceFunded
+        - event: VoteCast(indexed uint256,indexed address,uint256[],indexed uint256,string)
+          handler: handleVoteCast
+        - event: CommitCast(indexed uint256,indexed address,uint256[],bytes32)
+          handler: handleCommitCast
+      file: ./src/DisputeKitClassic.ts
   - kind: ethereum
     name: EvidenceModule
     network: arbitrum-sepolia
diff --git a/subgraph/package.json b/subgraph/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@kleros/kleros-v2-subgraph",
-  "version": "0.15.2",
+  "version": "0.16.0",
   "drtVersion": "0.12.0",
   "license": "MIT",
   "scripts": {

Original file line number	Diff line number	Diff line change
`@@ -156,7 +156,7 @@ export function handleNewPeriod(event: NewPeriod): void {`
`156`	`156`	`updateTotalLeaderboardJurors(ONE, event.block.timestamp);`
`157`	`157`	`}`
`158`	`158`
`159`		`- // Since this is a ClassicVote entity, this will only work for the Classic DisputeKit (which has ID "1").`
	`159`	`+ // Since this is a ClassicVote entity, this will only work for the ClassicDisputeKit and ShutterDisputeKit.`
`160`	`160`	const vote = ClassicVote.load(`${round.disputeKit}-${draw.id}`);
`161`	`161`
`162`	`162`	`if (!vote) {`
Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,6 @@`
`1`	`1`	`{`
`2`	`2`	`"name": "@kleros/kleros-v2-subgraph",`
`3`		`- "version": "0.15.2",`
	`3`	`+ "version": "0.16.0",`
`4`	`4`	`"drtVersion": "0.12.0",`
`5`	`5`	`"license": "MIT",`
`6`	`6`	`"scripts": {`