sendou.ink/apps/web-react/e2e/helpers/counterpick-map-pool.ts

import type { Page } from "@playwright/test";
import { rankedModesShort } from "@sendou/in-game-lists/modes";
import type { ModeShort, StageId } from "@sendou/in-game-lists/types";
import { BANNED_MAPS } from "~/features/match-profile/banned-maps";
import { TOURNAMENT } from "~/features/tournament/tournament-constants";

/** Stage the counterpick picking starts from, leaving the lowest ids to tiebreakers. */
const FIRST_COUNTERPICK_STAGE_ID = 5;

/** A stage of a `ModeMapPoolPicker`, whether picked or not. */
export function counterpickMap(page: Page, mode: ModeShort, stageId: StageId) {
	return page.getByTestId(`map-pool-${mode}-${stageId}`);
}

/** The check overlaying a picked stage. It covers the stage itself, so unpicking goes through it. */
export function pickedCounterpickMap(
	page: Page,
	mode: ModeShort,
	stageId: StageId,
) {
	return page.getByTestId(`map-pool-${mode}-${stageId}-picked`);
}

/**
 * Picks the required amount of counterpick maps for every ranked mode, skipping
 * banned ones, and returns what was picked.
 */
export async function pickCounterpickMaps(page: Page) {
	const picked: Array<{ mode: ModeShort; stageId: StageId }> = [];
	let stageId = FIRST_COUNTERPICK_STAGE_ID;

	for (const mode of rankedModesShort) {
		for (let i = 0; i < TOURNAMENT.COUNTERPICK_MAPS_PER_MODE; i++) {
			while (BANNED_MAPS[mode].includes(stageId as StageId)) {
				stageId++;
			}

			await counterpickMap(page, mode, stageId as StageId).click();
			picked.push({ mode, stageId: stageId as StageId });
			stageId++;
		}
	}

	return picked;
}