add admin actions + set player rating at signup

This commit is contained in:
2026-03-09 22:00:25 +01:00
parent dfae3a4744
commit fd2ae80562
7 changed files with 373 additions and 19 deletions

44
scripts/cancel-matches.ts Normal file
View File

@@ -0,0 +1,44 @@
/**
* Cancel all active matches (CREATED, LOBBY, ROLLING, PICKING, PLAYING).
*
* Usage: bun scripts/cancel-matches.ts
*/
import { drizzle } from 'drizzle-orm/postgres-js';
import postgres from 'postgres';
import { sql } from 'drizzle-orm';
const DATABASE_URL = process.env.DATABASE_URL;
if (!DATABASE_URL) throw new Error('DATABASE_URL not set');
const client = postgres(DATABASE_URL);
const db = drizzle(client);
async function main() {
const result = await db.execute(sql`
UPDATE match
SET state = 'CANCELLED', finished_at = now()
WHERE state IN ('CREATED', 'LOBBY', 'ROLLING', 'PICKING', 'PLAYING')
RETURNING id, name, state
`);
const rows = result as any[];
console.log(`Cancelled ${rows.length} active match${rows.length !== 1 ? 'es' : ''}:`);
for (const r of rows) {
console.log(` ${r.id} — ${r.name ?? '(unnamed)'}`);
}
// Also clear the queue
const queueResult = await db.execute(sql`DELETE FROM match_queue RETURNING id`);
const queueRows = queueResult as any[];
if (queueRows.length > 0) {
console.log(`Cleared ${queueRows.length} queue entr${queueRows.length !== 1 ? 'ies' : 'y'}`);
}
process.exit(0);
}
main().catch((err) => {
console.error(err);
process.exit(1);
});

59
scripts/clear-db.ts Normal file
View File

@@ -0,0 +1,59 @@
/**
* Clear all match data from the database (matches, games, scores, queue, invites, notifications).
* Keeps users, teams, mappools, and ratings intact.
*
* Usage: bun scripts/clear-db.ts
* --all Also clear ratings, teams (except personal), and mappools
* --full Nuclear: clear everything except users and personal teams
*/
import { drizzle } from 'drizzle-orm/postgres-js';
import postgres from 'postgres';
import { sql } from 'drizzle-orm';
const DATABASE_URL = process.env.DATABASE_URL;
if (!DATABASE_URL) throw new Error('DATABASE_URL not set');
const client = postgres(DATABASE_URL);
const db = drizzle(client);
const flag = process.argv[2];
async function main() {
console.log('Clearing match data...');
// Always clear match-related data
await db.execute(sql`DELETE FROM match_game_score`);
await db.execute(sql`DELETE FROM match_game`);
await db.execute(sql`DELETE FROM match_participant_player`);
await db.execute(sql`DELETE FROM match_participant`);
await db.execute(sql`UPDATE match SET winner_id = NULL`);
await db.execute(sql`DELETE FROM match`);
await db.execute(sql`DELETE FROM match_queue`);
await db.execute(sql`DELETE FROM match_invite`);
await db.execute(sql`DELETE FROM notification`);
console.log(' Cleared: matches, games, scores, queue, invites, notifications');
if (flag === '--all' || flag === '--full') {
await db.execute(sql`DELETE FROM player_rating`);
console.log(' Cleared: player ratings');
}
if (flag === '--full') {
await db.execute(sql`DELETE FROM team_member WHERE team_id IN (SELECT id FROM team WHERE is_personal = false)`);
await db.execute(sql`DELETE FROM team WHERE is_personal = false`);
console.log(' Cleared: non-personal teams');
await db.execute(sql`DELETE FROM mappool_slot`);
await db.execute(sql`DELETE FROM mappool`);
console.log(' Cleared: mappools');
}
console.log('Done.');
process.exit(0);
}
main().catch((err) => {
console.error(err);
process.exit(1);
});

99
scripts/seed-db.ts Normal file
View File

@@ -0,0 +1,99 @@
/**
* Seed the database with test data for development.
* Creates fake teams and mappools if none exist.
*
* Usage: bun scripts/seed-db.ts
*/
import { drizzle } from 'drizzle-orm/postgres-js';
import postgres from 'postgres';
import { sql } from 'drizzle-orm';
import { pgTable, text, uuid, boolean, timestamp, integer, real, jsonb } from 'drizzle-orm/pg-core';
const DATABASE_URL = process.env.DATABASE_URL;
if (!DATABASE_URL) throw new Error('DATABASE_URL not set');
const client = postgres(DATABASE_URL);
const db = drizzle(client);
const mappool = pgTable('mappool', {
id: uuid('id').primaryKey().defaultRandom(),
name: text('name').notNull(),
createdBy: text('created_by'),
createdAt: timestamp('created_at').defaultNow().notNull(),
verifiedAt: timestamp('verified_at')
});
const mappoolSlot = pgTable('mappool_slot', {
id: uuid('id').primaryKey().defaultRandom(),
mappoolId: uuid('mappool_id').notNull(),
category: text('category').notNull(),
orderInCategory: integer('order_in_category').notNull(),
beatmapId: text('beatmap_id').notNull(),
starRating: real('star_rating'),
bpm: real('bpm'),
totalLength: integer('total_length'),
mods: text('mods').array().default([]).notNull(),
title: text('title'),
artist: text('artist'),
version: text('version'),
coverUrl: text('cover_url'),
listCoverUrl: text('list_cover_url')
});
// Sample beatmap IDs (real osu! maps)
const SAMPLE_MAPS = [
{ id: '75', category: 'NM', title: 'Disco Prince', artist: 'Kenji Ninuma', sr: 2.5, bpm: 120 },
{ id: '163', category: 'NM', title: 'Crazy Noisy Bizarre Town', artist: 'The DU', sr: 3.2, bpm: 140 },
{ id: '250', category: 'HD', title: 'Sakura Kagetsu', artist: 'Mutsuhiko Izumi', sr: 4.1, bpm: 165 },
{ id: '306', category: 'HR', title: 'Airman ga Taosenai', artist: 'Team Nekokan', sr: 5.0, bpm: 200 },
{ id: '390', category: 'DT', title: 'Kira Kira Days', artist: 'MOSAIC.WAV', sr: 4.5, bpm: 175 },
{ id: '455', category: 'FM', title: 'FREEDOM DiVE', artist: 'xi', sr: 6.0, bpm: 222 },
{ id: '500', category: 'TB', title: 'Blue Zenith', artist: 'xi', sr: 6.5, bpm: 200 }
];
async function main() {
// Check if mappools exist
const [{ count }] = await db.select({ count: sql<number>`count(*)` }).from(mappool);
if (Number(count) > 0) {
console.log(`${count} mappool(s) already exist, skipping seed.`);
console.log('Run "bun scripts/clear-db.ts --full" first to clear everything.');
process.exit(0);
}
// Create a sample mappool
const [pool] = await db.insert(mappool).values({
name: 'Dev Test Pool',
verifiedAt: new Date()
}).returning();
console.log(`Created mappool: ${pool.name} (${pool.id})`);
for (let i = 0; i < SAMPLE_MAPS.length; i++) {
const m = SAMPLE_MAPS[i];
const sameCategory = SAMPLE_MAPS.filter((x, j) => j <= i && x.category === m.category);
await db.insert(mappoolSlot).values({
mappoolId: pool.id,
category: m.category,
orderInCategory: sameCategory.length,
beatmapId: m.id,
starRating: m.sr,
bpm: m.bpm,
totalLength: 120,
mods: m.category === 'NM' || m.category === 'TB' ? [] : [m.category],
title: m.title,
artist: m.artist,
version: 'Normal'
});
}
console.log(` Added ${SAMPLE_MAPS.length} maps (${[...new Set(SAMPLE_MAPS.map((m) => m.category))].join(', ')})`);
console.log('Done. Sign in via osu! OAuth to create a user and personal team.');
process.exit(0);
}
main().catch((err) => {
console.error(err);
process.exit(1);
});