sendou.ink/scripts/create-league-divisions.ts

// for testing use the command `npx tsx ./scripts/create-league-divisions.ts 6 'https://gist.githubusercontent.com/sendou-ink/38aa4d5d8426035ce178c09598ae627f/raw/17be9bb53a9f017c2097d0624f365d1c5a029f01/league.csv'`

import "dotenv/config";
import { z } from "zod";
import { db } from "~/db/sql";
import { ADMIN_ID } from "~/features/admin/admin-constants";
import * as CalendarRepository from "~/features/calendar/CalendarRepository.server";
import * as TournamentTeamRepository from "~/features/tournament/TournamentTeamRepository.server";
import type { Tournament } from "~/features/tournament-bracket/core/Tournament";
import { tournamentFromDB } from "~/features/tournament-bracket/core/Tournament.server";
import { dateToDatabaseTimestamp } from "~/utils/dates";
import invariant from "~/utils/invariant";
import { logger } from "~/utils/logger";

const tournamentId = Number(process.argv[2]?.trim());

invariant(
	tournamentId && !Number.isNaN(tournamentId),
	"tournament id is required (argument 1)",
);

const csvUrl = process.argv[3]?.trim();

invariant(z.string().url().parse(csvUrl), "csv url is required (argument 2)");

async function main() {
	const tournament = await tournamentFromDB({
		tournamentId,
		user: { id: ADMIN_ID },
	});
	invariant(tournament.isLeagueSignup, "Tournament is not a league signup");

	const csv = await loadCsv();

	const teams = parseCsv(csv);
	for (const team of teams) {
		validateTeam(team, tournament);
	}
	validateDivs(teams);

	const grouped = Object.entries(Object.groupBy(teams, (t) => t.division)).sort(
		(a, b) => {
			const divAIndex = teams.findIndex((t) => t.division === a[0]);
			const divBIndex = teams.findIndex((t) => t.division === b[0]);

			return divAIndex - divBIndex;
		},
	);

	for (const [, divsTeams] of grouped) {
		divsTeams!.sort((a, b) => {
			const teamAIndex = teams.findIndex((t) => t.id === a.id);
			const teamBIndex = teams.findIndex((t) => t.id === b.id);

			return teamAIndex - teamBIndex;
		});
	}

	const calendarEvent = await db
		.selectFrom("CalendarEvent")
		.selectAll()
		.where("CalendarEvent.id", "=", tournament.ctx.eventId)
		.executeTakeFirstOrThrow();

	for (const [div, divsTeams] of grouped) {
		logger.info(`Creating division ${div}...`);

		const createdEvent = await CalendarRepository.create({
			parentTournamentId: tournament.ctx.id,
			authorId: tournament.ctx.author.id,
			bracketProgression: tournament.ctx.settings.bracketProgression,
			description: tournament.ctx.description,
			discordInviteCode:
				tournament.ctx.discordUrl?.replace("https://discord.gg/", "") ?? null,
			mapPickingStyle: tournament.ctx.mapPickingStyle,
			name: `${tournament.ctx.name} - ${div.startsWith("Division") ? div : `Division ${div}`}`,
			organizationId: tournament.ctx.organization?.id ?? null,
			rules: tournament.ctx.rules,
			startTimes: [dateToDatabaseTimestamp(tournament.ctx.startTime)],
			tags: null,
			tournamentToCopyId: tournament.ctx.id,
			avatarImgId: calendarEvent.avatarImgId ?? undefined,
			avatarFileName: undefined,
			mapPoolMaps:
				tournament.ctx.mapPickingStyle !== "TO"
					? tournament.ctx.tieBreakerMapPool
					: tournament.ctx.toSetMapPool,
			badges: [],
			enableNoScreenToggle: tournament.ctx.settings.enableNoScreenToggle,
			enableSubs: false,
			isInvitational: true,
			autonomousSubs: false,
			isRanked: tournament.ctx.settings.isRanked,
			minMembersPerTeam: tournament.ctx.settings.minMembersPerTeam,
			maxMembersPerTeam: tournament.ctx.settings.maxMembersPerTeam,
			regClosesAt: tournament.ctx.settings.regClosesAt,
			requireInGameNames: tournament.ctx.settings.requireInGameNames,
			bracketUrl: "https://sendou.ink",
			isFullTournament: true,
			autoValidateAvatar: true,
			// these come from progression
			swissGroupCount: undefined,
			swissRoundCount: undefined,
			teamsPerGroup: undefined,
			thirdPlaceMatch: undefined,
		});

		for (const [idx, team] of divsTeams!.entries()) {
			await TournamentTeamRepository.copyFromAnotherTournament({
				destinationTournamentId: createdEvent.tournamentId!,
				tournamentTeamId: team.id,
				seed: idx + 1,
				defaultCheckedIn: true,
			});
		}

		logger.info(`Created division ${div} (id: ${createdEvent.tournamentId})`);
	}
}

async function loadCsv() {
	const response = await fetch(csvUrl);
	return response.text();
}

const csvSchema = z.array(
	z.object({
		"Team id": z.coerce.number(),
		Div: z.string(),
	}),
);

type ParsedTeam = ReturnType<typeof parseCsv>[number];

function parseCsv(csv: string) {
	const lines = csv.trim().split("\n");
	const headers = splitCsvRow(lines[0]).map((h) => h.trim());
	const rows = lines.slice(1).map((line) => {
		const row = splitCsvRow(line);
		return headers.reduce(
			(acc, header, i) => {
				acc[header] = row[i];
				return acc;
			},
			{} as Record<string, string>,
		);
	});

	const validated = csvSchema.parse(rows);

	return validated.map((row) => ({
		id: row["Team id"],
		division: row.Div,
	}));
}

function validateTeam(team: ParsedTeam, tournament: Tournament) {
	invariant(
		tournament.ctx.teams.some((t) => t.id === team.id),
		`Team with id ${team.id} not found in tournament`,
	);
}

const MIN_TEAMS_COUNT_PER_DIV = 6;
function validateDivs(teams: ParsedTeam[]) {
	const counts = teams.reduce(
		(acc, team) => {
			acc[team.division] = (acc[team.division] ?? 0) + 1;
			return acc;
		},
		{} as Record<string, number>,
	);

	for (const [div, count] of Object.entries(counts)) {
		invariant(
			count >= MIN_TEAMS_COUNT_PER_DIV,
			`Division ${div} has ${count} teams, expected at least ${MIN_TEAMS_COUNT_PER_DIV}`,
		);
	}
}

function splitCsvRow(line: string) {
	const fields: string[] = [];
	let current = "";
	let inQuotes = false;

	for (let i = 0; i < line.length; i++) {
		const char = line[i];

		if (inQuotes) {
			if (char === '"') {
				if (line[i + 1] === '"') {
					current += '"';
					i++;
				} else {
					inQuotes = false;
				}
			} else {
				current += char;
			}
		} else if (char === '"') {
			inQuotes = true;
		} else if (char === ",") {
			fields.push(current);
			current = "";
		} else {
			current += char;
		}
	}

	fields.push(current);
	return fields;
}

main();