sendou.ink/app/modules/brackets-manager/base/updater.ts

import type { GroupType, Match, Stage } from "~/modules/brackets-model";
import { Status } from "~/modules/brackets-model";
import type { SetNextOpponent } from "../helpers";
import * as helpers from "../helpers";
import type { DeepPartial, Side } from "../types";
import { BaseGetter } from "./getter";

export class BaseUpdater extends BaseGetter {
  /**
   * Updates the matches related (previous and next) to a match.
   *
   * @param match A match.
   * @param updatePrevious Whether to update the previous matches.
   * @param updateNext Whether to update the next matches.
   */
  protected updateRelatedMatches(
    match: Match,
    updatePrevious: boolean,
    updateNext: boolean,
  ): void {
    const { roundNumber, roundCount } = this.getRoundPositionalInfo(
      match.round_id,
    );

    const stage = this.storage.select("stage", match.stage_id);
    if (!stage) throw Error("Stage not found.");

    const group = this.storage.select("group", match.group_id);
    if (!group) throw Error("Group not found.");

    const matchLocation = helpers.getMatchLocation(stage.type, group.number);

    updatePrevious &&
      this.updatePrevious(match, matchLocation, stage, roundNumber);
    updateNext &&
      this.updateNext(match, matchLocation, stage, roundNumber, roundCount);
  }

  /**
   * Updates a match based on a partial match.
   *
   * @param stored A reference to what will be updated in the storage.
   * @param match Input of the update.
   * @param force Whether to force update locked matches.
   */
  protected updateMatch(
    stored: Match,
    match: DeepPartial<Match>,
    force?: boolean,
  ): void {
    if (!force && helpers.isMatchUpdateLocked(stored))
      throw Error("The match is locked.");

    const stage = this.storage.select("stage", stored.stage_id);
    if (!stage) throw Error("Stage not found.");

    const inRoundRobin = helpers.isRoundRobin(stage);

    const { statusChanged, resultChanged } = helpers.setMatchResults(
      stored,
      match,
      inRoundRobin,
    );
    this.applyMatchUpdate(stored);

    // Don't update related matches if it's a simple score update.
    if (!statusChanged && !resultChanged) return;

    if (!helpers.isRoundRobin(stage) && !helpers.isSwiss(stage))
      this.updateRelatedMatches(stored, statusChanged, resultChanged);
  }

  /**
   * Updates the opponents and status of a match and its child games.
   *
   * @param match A match.
   */
  protected applyMatchUpdate(match: Match): void {
    if (!this.storage.update("match", match.id, match))
      throw Error("Could not update the match.");
  }

  /**
   * Updates the match(es) leading to the current match based on this match results.
   *
   * @param match Input of the update.
   * @param matchLocation Location of the current match.
   * @param stage The parent stage.
   * @param roundNumber Number of the round.
   */
  protected updatePrevious(
    match: Match,
    matchLocation: GroupType,
    stage: Stage,
    roundNumber: number,
  ): void {
    const previousMatches = this.getPreviousMatches(
      match,
      matchLocation,
      stage,
      roundNumber,
    );
    if (previousMatches.length === 0) return;

    if (match.status < Status.Running) {
      this.resetMatchesStatus(previousMatches);
    }
  }

  /**
   * Resets the status of a list of matches to what it should currently be.
   *
   * @param matches The matches to update.
   */
  protected resetMatchesStatus(matches: Match[]): void {
    for (const match of matches) {
      match.status = helpers.getMatchStatus(match);
      this.applyMatchUpdate(match);
    }
  }

  /**
   * Updates the match(es) following the current match based on this match results.
   *
   * @param match Input of the update.
   * @param matchLocation Location of the current match.
   * @param stage The parent stage.
   * @param roundNumber Number of the round.
   * @param roundCount Count of rounds.
   */
  protected updateNext(
    match: Match,
    matchLocation: GroupType,
    stage: Stage,
    roundNumber: number,
    roundCount: number,
  ): void {
    const nextMatches = this.getNextMatches(
      match,
      matchLocation,
      stage,
      roundNumber,
      roundCount,
    );
    if (nextMatches.length === 0) {
      return;
    }

    const winnerSide = helpers.getMatchResult(match);
    const actualRoundNumber =
      stage.settings.skipFirstRound && matchLocation === "winner_bracket"
        ? roundNumber + 1
        : roundNumber;

    if (winnerSide)
      this.applyToNextMatches(
        helpers.setNextOpponent,
        match,
        matchLocation,
        actualRoundNumber,
        roundCount,
        nextMatches,
        winnerSide,
      );
    else
      this.applyToNextMatches(
        helpers.resetNextOpponent,
        match,
        matchLocation,
        actualRoundNumber,
        roundCount,
        nextMatches,
      );
  }

  /**
   * Applies a SetNextOpponent function to matches following the current match.
   *
   * @param setNextOpponent The SetNextOpponent function.
   * @param match The current match.
   * @param matchLocation Location of the current match.
   * @param roundNumber Number of the current round.
   * @param roundCount Count of rounds.
   * @param nextMatches The matches following the current match.
   * @param winnerSide Side of the winner in the current match.
   */
  protected applyToNextMatches(
    setNextOpponent: SetNextOpponent,
    match: Match,
    matchLocation: GroupType,
    roundNumber: number,
    roundCount: number,
    nextMatches: (Match | null)[],
    winnerSide?: Side,
  ): void {
    if (matchLocation === "final_group") {
      if (!nextMatches[0]) throw Error("First next match is null.");
      setNextOpponent(nextMatches[0], "opponent1", match, "opponent1");
      setNextOpponent(nextMatches[0], "opponent2", match, "opponent2");
      this.applyMatchUpdate(nextMatches[0]);
      return;
    }

    const nextSide = helpers.getNextSide(
      match.number,
      roundNumber,
      roundCount,
      matchLocation,
    );

    if (nextMatches[0]) {
      setNextOpponent(nextMatches[0], nextSide, match, winnerSide);
      this.propagateByeWinners(nextMatches[0]);
    }

    if (nextMatches.length !== 2) return;
    if (!nextMatches[1]) throw Error("Second next match is null.");

    // The second match is either the consolation final (single elimination) or a loser bracket match (double elimination).

    if (matchLocation === "single_bracket") {
      setNextOpponent(
        nextMatches[1],
        nextSide,
        match,
        winnerSide && helpers.getOtherSide(winnerSide),
      );
      this.applyMatchUpdate(nextMatches[1]);
    } else {
      const nextSideLB = helpers.getNextSideLoserBracket(
        match.number,
        nextMatches[1],
        roundNumber,
      );
      setNextOpponent(
        nextMatches[1],
        nextSideLB,
        match,
        winnerSide && helpers.getOtherSide(winnerSide),
      );
      this.propagateByeWinners(nextMatches[1]);
    }
  }

  /**
   * Propagates winner against BYEs in related matches.
   *
   * @param match The current match.
   */
  protected propagateByeWinners(match: Match): void {
    helpers.setMatchResults(match, match, false); // BYE propagation is only in non round-robin stages.
    this.applyMatchUpdate(match);

    if (helpers.hasBye(match)) this.updateRelatedMatches(match, true, true);
  }
}