sendou.ink/app/features/sendouq-match/ReportedWeaponRepository.server.ts

import type { NotNull, Transaction } from "kysely";
import { db } from "~/db/sql";
import type { DB, TablesInsertable } from "~/db/tables";

export function createMany(
	weapons: TablesInsertable["ReportedWeapon"][],
	trx?: Transaction<DB>,
) {
	if (weapons.length === 0) return;

	return (trx ?? db).insertInto("ReportedWeapon").values(weapons).execute();
}

export async function upsertOne({
	groupMatchMapId,
	userId,
	weaponSplId,
}: TablesInsertable["ReportedWeapon"] & { groupMatchMapId: number }) {
	await db
		.deleteFrom("ReportedWeapon")
		.where("groupMatchMapId", "=", groupMatchMapId)
		.where("userId", "=", userId)
		.execute();

	await db
		.insertInto("ReportedWeapon")
		.values({ groupMatchMapId, userId, weaponSplId })
		.execute();
}

export async function replaceByMatchId(
	matchId: number,
	weapons: TablesInsertable["ReportedWeapon"][],
	trx?: Transaction<DB>,
) {
	const executor = trx ?? db;

	const groupMatchMaps = await executor
		.selectFrom("GroupMatchMap")
		.select("id")
		.where("matchId", "=", matchId)
		.execute();

	if (groupMatchMaps.length > 0) {
		await executor
			.deleteFrom("ReportedWeapon")
			.where(
				"groupMatchMapId",
				"in",
				groupMatchMaps.map((m) => m.id),
			)
			.execute();
	}

	if (weapons.length > 0) {
		await executor.insertInto("ReportedWeapon").values(weapons).execute();
	}
}

export async function deleteByUserMapIndex({
	matchId,
	userId,
	mapIndex,
}: {
	matchId: number;
	userId: number;
	mapIndex: number;
}) {
	const groupMatchMap = await db
		.selectFrom("GroupMatchMap")
		.select("id")
		.where("matchId", "=", matchId)
		.where("index", "=", mapIndex)
		.executeTakeFirst();

	if (!groupMatchMap) return;

	await db
		.deleteFrom("ReportedWeapon")
		.where("groupMatchMapId", "=", groupMatchMap.id)
		.where("userId", "=", userId)
		.execute();
}

export async function findByMatchId(matchId: number) {
	const rows = await db
		.selectFrom("ReportedWeapon")
		.innerJoin(
			"GroupMatchMap",
			"GroupMatchMap.id",
			"ReportedWeapon.groupMatchMapId",
		)
		.select([
			"ReportedWeapon.groupMatchMapId",
			"ReportedWeapon.weaponSplId",
			"ReportedWeapon.userId",
			"GroupMatchMap.index as mapIndex",
		])
		.where("GroupMatchMap.matchId", "=", matchId)
		.orderBy("GroupMatchMap.index", "asc")
		.orderBy("ReportedWeapon.userId", "asc")
		.$narrowType<{ groupMatchMapId: NotNull }>()
		.execute();

	if (rows.length === 0) return null;

	return rows;
}