fix(subgraph): correctly update counts and winningchoice

Author: alcercu

subgraph/src/DisputeKitClassic.ts

@@ -20,6 +20,7 @@ import { ensureClassicEvidenceGroup } from "./entities/ClassicEvidenceGroup";
 import {
   createClassicRound,
   updateChoiceFundingFromContributionEvent,
+  updateCounts,
 } from "./entities/ClassicRound";
 import { createClassicVote } from "./entities/ClassicVote";
 import { ONE, ZERO } from "./utils";
@@ -53,9 +54,9 @@ export function handleJustificationEvent(event: JustificationEvent): void {
   const classicDisputeID = `${DISPUTEKIT_ID}-${coreDisputeID}`;
   const classicDispute = ClassicDispute.load(classicDisputeID);
   if (!classicDispute) return;
-  const currentLocalRoundID = `${
-    classicDispute.id
-  }-${classicDispute.currentLocalRoundIndex.toString()}`;
+  const currentLocalRoundID =
+    classicDispute.id + "-" + classicDispute.currentLocalRoundIndex.toString();
+  updateCounts(currentLocalRoundID, event.params._choice);
   createClassicVote(currentLocalRoundID, event);
 }
 

subgraph/src/entities/ClassicRound.ts

@@ -25,6 +25,26 @@ export function createClassicRound(
   classicRound.save();
 }
 
+export function updateCounts(id: string, choice: BigInt): void {
+  const round = ClassicRound.load(id);
+  if (!round) return;
+  const choiceNum = choice.toI32();
+  const updatedCount = round.counts[choiceNum].plus(ONE);
+  round.counts[choiceNum] = updatedCount;
+  const currentWinningCount = round.counts[round.winningChoice.toI32()];
+  if (choice.equals(round.winningChoice)) {
+    if (round.tied) round.tied = false;
+  } else {
+    if (updatedCount.equals(currentWinningCount)) {
+      if (!round.tied) round.tied = true;
+    } else if (updatedCount.gt(currentWinningCount)) {
+      round.winningChoice = choice;
+      round.tied = false;
+    }
+  }
+  round.save();
+}
+
 export function updateChoiceFundingFromContributionEvent(
   event: Contribution
 ): void {