sendou.ink/app/modules/brackets-manager/update.ts

import type {
  Match,
  MatchGame,
  Round,
  Seeding,
  SeedOrdering,
} from "brackets-model";
import { Status } from "brackets-model";
import { ordering } from "./ordering";
import { BaseUpdater } from "./base/updater";
import type { ChildCountLevel, DeepPartial } from "./types";
import * as helpers from "./helpers";

export class Update extends BaseUpdater {
  /**
   * Updates partial information of a match. Its id must be given.
   *
   * This will update related matches accordingly.
   *
   * @param match Values to change in a match.
   */
  public match<M extends Match = Match>(match: DeepPartial<M>): void {
    if (match.id === undefined) throw Error("No match id given.");

    const stored = this.storage.select("match", match.id);
    if (!stored) throw Error("Match not found.");

    this.updateMatch(stored, match);
  }

  /**
   * Updates partial information of a match game. Its id must be given.
   *
   * This will update the parent match accordingly.
   *
   * @param game Values to change in a match game.
   */
  public matchGame<G extends MatchGame = MatchGame>(
    game: DeepPartial<G>
  ): void {
    const stored = this.findMatchGame(game);

    this.updateMatchGame(stored, game);
  }

  /**
   * Updates the seed ordering of every ordered round in a stage.
   *
   * @param stageId ID of the stage.
   * @param seedOrdering A list of ordering methods.
   */
  public ordering(stageId: number, seedOrdering: SeedOrdering[]): void {
    const stage = this.storage.select("stage", stageId);
    if (!stage) throw Error("Stage not found.");

    helpers.ensureNotRoundRobin(stage);

    const roundsToOrder = this.getOrderedRounds(stage);
    if (seedOrdering.length !== roundsToOrder.length)
      throw Error("The count of seed orderings is incorrect.");

    for (let i = 0; i < roundsToOrder.length; i++)
      this.updateRoundOrdering(roundsToOrder[i], seedOrdering[i]);
  }

  /**
   * Updates the seed ordering of a round.
   *
   * @param roundId ID of the round.
   * @param method Seed ordering method.
   */
  public roundOrdering(roundId: number, method: SeedOrdering): void {
    const round = this.storage.select("round", roundId);
    if (!round) throw Error("This round does not exist.");

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

    helpers.ensureNotRoundRobin(stage);

    this.updateRoundOrdering(round, method);
  }

  /**
   * Updates child count of all matches of a given level.
   *
   * @param level The level at which to act.
   * @param id ID of the chosen level.
   * @param childCount The target child count.
   */
  public matchChildCount(
    level: ChildCountLevel,
    id: number,
    childCount: number
  ): void {
    switch (level) {
      case "stage":
        this.updateStageMatchChildCount(id, childCount);
        break;
      case "group":
        this.updateGroupMatchChildCount(id, childCount);
        break;
      case "round":
        this.updateRoundMatchChildCount(id, childCount);
        break;
      case "match":
        // eslint-disable-next-line no-case-declarations
        const match = this.storage.select("match", id);
        if (!match) throw Error("Match not found.");
        this.adjustMatchChildGames(match, childCount);
        break;
      default:
        throw Error("Unknown child count level.");
    }
  }

  /**
   * Updates the seeding of a stage.
   *
   * @param stageId ID of the stage.
   * @param seeding The new seeding.
   */
  public seeding(stageId: number, seeding: Seeding): void {
    this.updateSeeding(stageId, seeding);
  }

  /**
   * Confirms the seeding of a stage.
   *
   * This will convert TBDs to BYEs and propagate them.
   *
   * @param stageId ID of the stage.
   */
  public confirmSeeding(stageId: number): void {
    this.confirmCurrentSeeding(stageId);
  }

  /**
   * Update the seed ordering of a round.
   *
   * @param round The round of which to update the ordering.
   * @param method The new ordering method.
   */
  private updateRoundOrdering(round: Round, method: SeedOrdering): void {
    const matches = this.storage.select("match", { round_id: round.id });
    if (!matches) throw Error("This round has no match.");

    if (matches.some((match) => match.status > Status.Ready))
      throw Error("At least one match has started or is completed.");

    const stage = this.storage.select("stage", round.stage_id);
    if (!stage) throw Error("Stage not found.");
    if (stage.settings.size === undefined) throw Error("Undefined stage size.");

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

    const inLoserBracket = helpers.isLoserBracket(stage.type, group.number);
    const roundCountLB = helpers.getLowerBracketRoundCount(stage.settings.size);
    const seeds = helpers.getSeeds(
      inLoserBracket,
      round.number,
      roundCountLB,
      matches.length
    );
    const positions = ordering[method](seeds);

    this.applyRoundOrdering(round.number, matches, positions);
  }

  /**
   * Updates child count of all matches of a stage.
   *
   * @param stageId ID of the stage.
   * @param childCount The target child count.
   */
  private updateStageMatchChildCount(
    stageId: number,
    childCount: number
  ): void {
    if (
      !this.storage.update(
        "match",
        { stage_id: stageId },
        { child_count: childCount }
      )
    )
      throw Error("Could not update the match.");

    const matches = this.storage.select("match", { stage_id: stageId });
    if (!matches) throw Error("This stage has no match.");

    for (const match of matches) this.adjustMatchChildGames(match, childCount);
  }

  /**
   * Updates child count of all matches of a group.
   *
   * @param groupId ID of the group.
   * @param childCount The target child count.
   */
  private updateGroupMatchChildCount(
    groupId: number,
    childCount: number
  ): void {
    if (
      !this.storage.update(
        "match",
        { group_id: groupId },
        { child_count: childCount }
      )
    )
      throw Error("Could not update the match.");

    const matches = this.storage.select("match", { group_id: groupId });
    if (!matches) throw Error("This group has no match.");

    for (const match of matches) this.adjustMatchChildGames(match, childCount);
  }

  /**
   * Updates child count of all matches of a round.
   *
   * @param roundId ID of the round.
   * @param childCount The target child count.
   */
  private updateRoundMatchChildCount(
    roundId: number,
    childCount: number
  ): void {
    if (
      !this.storage.update(
        "match",
        { round_id: roundId },
        { child_count: childCount }
      )
    )
      throw Error("Could not update the match.");

    const matches = this.storage.select("match", { round_id: roundId });
    if (!matches) throw Error("This round has no match.");

    for (const match of matches) this.adjustMatchChildGames(match, childCount);
  }

  /**
   * Updates the ordering of participants in a round's matches.
   *
   * @param roundNumber The number of the round.
   * @param matches The matches of the round.
   * @param positions The new positions.
   */
  private applyRoundOrdering(
    roundNumber: number,
    matches: Match[],
    positions: number[]
  ): void {
    for (const match of matches) {
      const updated = { ...match };
      updated.opponent1 = helpers.findPosition(matches, positions.shift()!);

      // The only rounds where we have a second ordered participant are first rounds of brackets (upper and lower).
      if (roundNumber === 1)
        updated.opponent2 = helpers.findPosition(matches, positions.shift()!);

      if (!this.storage.update("match", updated.id, updated))
        throw Error("Could not update the match.");
    }
  }

  /**
   * Adds or deletes match games of a match based on a target child count.
   *
   * @param match The match of which child games need to be adjusted.
   * @param targetChildCount The target child count.
   */
  private adjustMatchChildGames(match: Match, targetChildCount: number): void {
    const games = this.storage.select("match_game", {
      parent_id: match.id,
    });
    let childCount = games ? games.length : 0;

    while (childCount < targetChildCount) {
      const id = this.storage.insert("match_game", {
        number: childCount + 1,
        stage_id: match.stage_id,
        parent_id: match.id,
        status: match.status,
        opponent1: { id: null },
        opponent2: { id: null },
      });

      if (id === -1)
        throw Error("Could not adjust the match games when inserting.");

      childCount++;
    }

    while (childCount > targetChildCount) {
      const deleted = this.storage.delete("match_game", {
        parent_id: match.id,
        number: childCount,
      });

      if (!deleted)
        throw Error("Could not adjust the match games when deleting.");

      childCount--;
    }

    if (
      !this.storage.update("match", match.id, {
        ...match,
        child_count: targetChildCount,
      })
    )
      throw Error("Could not update the match.");
  }
}