kleros/kleros-v2

import { BigInt } from "@graphprotocol/graph-ts";
import {
  DisputeKitClassic,
  DisputeCreation,
  VoteCast,
  Contribution as ContributionEvent,
  ChoiceFunded,
  Withdrawal,
  CommitCast,
} from "../generated/DisputeKitClassic/DisputeKitClassic";
import { KlerosCore } from "../generated/KlerosCore/KlerosCore";
import { ClassicDispute, ClassicJustification, ClassicRound, ClassicVote, Dispute } from "../generated/schema";
import { ensureClassicContributionFromEvent } from "./entities/ClassicContribution";
import { createClassicDisputeFromEvent } from "./entities/ClassicDispute";
import {
  createClassicRound,
  updateChoiceFundingFromContributionEvent,
  updateCountsAndGetCurrentRuling,
} from "./entities/ClassicRound";
import { ensureClassicVote } from "./entities/ClassicVote";
import { ONE, ZERO } from "./utils";

export const DISPUTEKIT_ID = "1";

export function handleDisputeCreation(event: DisputeCreation): void {
  const disputeID = event.params._coreDisputeID.toString();
  createClassicDisputeFromEvent(event);
  const numberOfChoices = event.params._numberOfChoices;
  createClassicRound(disputeID, numberOfChoices, ZERO);
}

export function handleCommitCast(event: CommitCast): void {
  const coreDisputeID = event.params._coreDisputeID;
  const coreDispute = Dispute.load(coreDisputeID.toString());
  const classicDisputeID = `${DISPUTEKIT_ID}-${coreDisputeID}`;
  const classicDispute = ClassicDispute.load(classicDisputeID);
  if (!classicDispute || !coreDispute) return;
  const currentLocalRoundID = classicDispute.id + "-" + classicDispute.currentLocalRoundIndex.toString();
  const voteIDs = event.params._voteIDs;
  for (let i = 0; i < voteIDs.length; i++) {
    const classicVote = ensureClassicVote(
      currentLocalRoundID,
      event.params._juror.toHexString(),
      voteIDs[i],
      coreDispute
    );
    classicVote.commited = true;
    classicVote.commit = event.params._commit;
    classicVote.save();
  }
}

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}`;
  const classicDispute = ClassicDispute.load(classicDisputeID);
  if (!classicDispute || !coreDispute) return;
  const choice = event.params._choice;
  const currentLocalRoundID = classicDispute.id + "-" + classicDispute.currentLocalRoundIndex.toString();
  const voteIDs = event.params._voteIDs;
  const justification = new ClassicJustification(`${currentLocalRoundID}-${voteIDs.toString()}`);
  justification.juror = juror;
  justification.coreDispute = coreDisputeID;
  justification.localRound = currentLocalRoundID;
  justification.choice = choice;
  justification.reference = event.params._justification;
  justification.save();
  const currentRulingInfo = updateCountsAndGetCurrentRuling(
    currentLocalRoundID,
    choice,
    BigInt.fromI32(voteIDs.length)
  );
  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);
    classicVote.voted = true;
    classicVote.choice = choice;
    classicVote.justification = justification.id;
    classicVote.save();
  }
}

export function handleContributionEvent(event: ContributionEvent): void {
  ensureClassicContributionFromEvent(event);
  updateChoiceFundingFromContributionEvent(event);
}

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 localRound = ClassicRound.load(roundID);
  if (!localRound) return;

  const currentFeeRewards = localRound.feeRewards;
  const deltaFeeRewards = localRound.paidFees[choice.toI32()];
  localRound.feeRewards = currentFeeRewards.plus(deltaFeeRewards);
  localRound.fundedChoices = localRound.fundedChoices.concat([choice]);

  if (localRound.fundedChoices.length > 1) {
    const disputeKitClassic = DisputeKitClassic.bind(event.address);
    const klerosCore = KlerosCore.bind(disputeKitClassic.core());

    // cannot use core.appealCost as that will give the cost for the newly created round
    const numberOfRounds = klerosCore.getNumberOfRounds(BigInt.fromString(coreDisputeID));
    const roundInfo = klerosCore.getRoundInfo(BigInt.fromString(coreDisputeID), numberOfRounds.minus(ONE));
    const appealCost = roundInfo.totalFeesForJurors;

    localRound.feeRewards = localRound.feeRewards.minus(appealCost);

    const localDispute = ClassicDispute.load(`${DISPUTEKIT_ID}-${coreDisputeID}`);
    if (!localDispute) return;
    const newRoundIndex = localDispute.currentLocalRoundIndex.plus(ONE);
    const numberOfChoices = localDispute.numberOfChoices;
    localDispute.currentLocalRoundIndex = newRoundIndex;
    localDispute.save();
    createClassicRound(coreDisputeID, numberOfChoices, newRoundIndex);
  }

  localRound.save();
}

export function handleWithdrawal(event: Withdrawal): void {
  const contribution = ensureClassicContributionFromEvent(event);
  if (!contribution) return;
  contribution.rewardWithdrawn = true;
  contribution.rewardAmount = event.params._amount;

  // 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 localRound = ClassicRound.load(roundID);
  if (!localRound) return;

  localRound.totalFeeDispersed = localRound.totalFeeDispersed.plus(event.params._amount);

  if (localRound.totalFeeDispersed.equals(localRound.feeRewards)) {
    localRound.appealFeesDispersed = true;
  }

  contribution.save();
  localRound.save();
}