chore: map appeal funding in subgraph

Author: epiqueras

README.md

@@ -4,4 +4,5 @@
 - chore: generate first changelog ([2e41de7](https://github.com/kleros/governor-web/commit/2e41de7))
 - chore: map governance functions in subgraph ([360d007](https://github.com/kleros/governor-web/commit/360d007))
 - chore: map list submissions in subgraph ([8cbbe41](https://github.com/kleros/governor-web/commit/8cbbe41))
+- chore: map the withdrawal and execution of submissions in subgraph ([f74acb7](https://github.com/kleros/governor-web/commit/f74acb7))
 - chore: set up project and architecture ([d28fb74](https://github.com/kleros/governor-web/commit/d28fb74))

subgraph/src/mapping.ts

@@ -11,49 +11,34 @@ import {
   ChangeWinnerMultiplierCall,
   ChangeWithdrawTimeoutCall,
   ExecuteSubmissionsCall,
+  FundAppealCall,
   Governor,
   SetMetaEvidenceCall,
   SubmitListCall,
   WithdrawTransactionListCall,
 } from "../generated/Governor/Governor";
 import {
   Contract,
+  Contribution,
   MetaEvidence,
   Round,
   Session,
   Submission,
   Transaction,
 } from "../generated/schema";
 
-function getStatus(status: number): string {
-  if (status == 0) return "NoDispute";
-  if (status == 1) return "DisputeCreated";
-  if (status == 2) return "Resolved";
-  return "Error";
-}
-
 function concatByteArrays(a: ByteArray, b: ByteArray): ByteArray {
   let out = new Uint8Array(a.length + b.length);
   for (let i = 0; i < a.length; i++) out[i] = a[i];
   for (let j = 0; j < b.length; j++) out[a.length + j] = b[j];
   return out as ByteArray;
 }
 
-function newSession(contract: Contract, creationTime: BigInt): Session {
-  let session = new Session(contract.sessionsLength.toHexString());
-  session.creationTime = creationTime;
-  session.sumDeposit = BigInt.fromI32(0);
-  session.status = getStatus(0);
-  session.durationOffset = BigInt.fromI32(0);
-  session.submissionsLength = BigInt.fromI32(0);
-  session.roundsLength = BigInt.fromI32(0);
-  session.save();
-
-  contract.sessionsLength.plus(BigInt.fromI32(1));
-  contract.save();
-
-  // Something is broken with The Graph's null type guards so we need these explicit casts in some places.
-  return session as Session;
+function getStatus(status: number): string {
+  if (status == 0) return "NoDispute";
+  if (status == 1) return "DisputeCreated";
+  if (status == 2) return "Resolved";
+  return "Error";
 }
 
 function initializeContract(
@@ -89,12 +74,29 @@ function initializeContract(
   return contract as Contract;
 }
 
+function newSession(contract: Contract, creationTime: BigInt): Session {
+  let session = new Session(contract.sessionsLength.toHexString());
+  session.creationTime = creationTime;
+  session.sumDeposit = BigInt.fromI32(0);
+  session.status = getStatus(0);
+  session.durationOffset = BigInt.fromI32(0);
+  session.submissionsLength = BigInt.fromI32(0);
+  session.roundsLength = BigInt.fromI32(0);
+  session.save();
+
+  contract.sessionsLength.plus(BigInt.fromI32(1));
+  contract.save();
+
+  // Something is broken with The Graph's null type guards so we need these explicit casts in some places.
+  return session as Session;
+}
+
 function newRound(session: Session, creationTime: BigInt): Round {
   let round = new Round(
     crypto
       .keccak256(
         concatByteArrays(
-          ByteArray.fromUTF8(session.id),
+          ByteArray.fromHexString(session.id),
           ByteArray.fromUTF8(session.roundsLength.toString())
         )
       )
@@ -115,6 +117,49 @@ function newRound(session: Session, creationTime: BigInt): Round {
   return round as Round;
 }
 
+function updateContribution(
+  governor: Governor,
+  sessionID: BigInt,
+  roundIndex: BigInt,
+  contributor: Address,
+  time: BigInt
+): Contribution {
+  let roundInfo = governor.getRoundInfo(sessionID, roundIndex);
+  let contributions = governor.getContributions(
+    sessionID,
+    roundIndex,
+    contributor
+  );
+
+  let roundID = crypto.keccak256(
+    concatByteArrays(
+      ByteArray.fromUTF8(sessionID.toString()),
+      ByteArray.fromUTF8(roundIndex.toString())
+    )
+  );
+  let round = Round.load(roundID.toHexString());
+  round.paidFees = roundInfo.value0;
+  round.hasPaid = roundInfo.value1;
+  round.feeRewards = roundInfo.value2;
+  round.successfullyPaid = roundInfo.value3;
+  round.save();
+
+  let contributionID = crypto
+    .keccak256(concatByteArrays(roundID, contributor))
+    .toHexString();
+  let contribution = Contribution.load(contributionID);
+  if (contribution == null) {
+    contribution = new Contribution(contributionID);
+    contribution.creationTime = time;
+    contribution.round = round.id;
+    contribution.contributor = contributor;
+  }
+  contribution.values = contributions;
+  contribution.save();
+
+  return contribution as Contribution;
+}
+
 export function setMetaEvidence(call: SetMetaEvidenceCall): void {
   let contract = initializeContract(call.to, call.from, call.block.timestamp);
 
@@ -215,7 +260,7 @@ export function submitList(call: SubmitListCall): void {
   for (let i = 0; i < call.inputs._target.length; i++) {
     let transaction = new Transaction(
       concatByteArrays(
-        crypto.keccak256(ByteArray.fromUTF8(submission.id)),
+        crypto.keccak256(ByteArray.fromHexString(submission.id)),
         ByteArray.fromUTF8(i.toString())
       ).toHexString()
     );
@@ -309,3 +354,26 @@ export function executeSubmissions(call: ExecuteSubmissionsCall): void {
     newRound(session as Session, call.block.timestamp);
   }
 }
+
+export function fundAppeal(call: FundAppealCall): void {
+  let governor = Governor.bind(call.to);
+  let contract = initializeContract(call.to, call.from, call.block.timestamp);
+
+  let sessionID = contract.sessionsLength.minus(BigInt.fromI32(1));
+  let session = Session.load(sessionID.toHexString());
+  updateContribution(
+    governor,
+    sessionID,
+    session.roundsLength.minus(BigInt.fromI32(1)),
+    call.from,
+    call.block.timestamp
+  );
+
+  contract.reservedETH = governor.reservedETH();
+  contract.shadowWinner = governor.shadowWinner().toHexString();
+  contract.save();
+
+  let sessionRoundsNumber = governor.getSessionRoundsNumber(sessionID);
+  if (sessionRoundsNumber.gt(session.roundsLength))
+    newRound(session as Session, call.block.timestamp);
+}

subgraph/subgraph.template.yaml

@@ -45,4 +45,6 @@ dataSources:
           handler: withdrawTransactionList
         - function: executeSubmissions()
           handler: executeSubmissions
+        - function: fundAppeal(uint256)
+          handler: fundAppeal
       file: ./src/mapping.ts