feat: add schema validation and event table checks in scenario seeder

This commit is contained in:
2026-01-18 18:13:55 +00:00
parent 4647b67ba3
commit 9ac59893c4
3 changed files with 56 additions and 4 deletions
+2 -2
View File
@@ -13,7 +13,7 @@ describe('formatSseFrame', () => {
data: 'ok',
});
expect(output).toBe(['event: ping', 'id: 1', 'retry: 1500', 'data: ok', ''].join('\n'));
expect(output).toBe(['event: ping', 'id: 1', 'retry: 1500', 'data: ok', ''].join('\n') + '\n');
});
it('splits multiline data', () => {
@@ -22,7 +22,7 @@ describe('formatSseFrame', () => {
data: 'first\nsecond',
});
expect(output).toBe(['event: notice', 'data: first', 'data: second', ''].join('\n'));
expect(output).toBe(['event: notice', 'data: first', 'data: second', ''].join('\n') + '\n');
});
});
+29 -2
View File
@@ -58,6 +58,29 @@ export interface ScenarioSeedResult {
const asJson = (value: unknown): InputJsonValue => value as InputJsonValue;
type RawQueryClient = {
$queryRawUnsafe<T>(query: string): Promise<T>;
};
const resolveSchemaName = (databaseUrl: string): string => {
try {
return new URL(databaseUrl).searchParams.get('schema') ?? 'public';
} catch {
return 'public';
}
};
const hasEventTable = async (prisma: RawQueryClient, schema: string): Promise<boolean> => {
try {
const result = await prisma.$queryRawUnsafe<Array<{ regclass: string | null }>>(
`SELECT to_regclass('${schema}.event') as regclass`
);
return Array.isArray(result) && result.length > 0 && result[0]?.regclass !== null;
} catch {
return false;
}
};
const formatDateTime = (date: Date): string => {
const pad = (value: number): string => String(value).padStart(2, '0');
return [
@@ -236,9 +259,13 @@ export const seedScenarioToDatabase = async (options: ScenarioSeedOptions): Prom
await connector.connect();
try {
const prisma = connector.prisma;
const schema = resolveSchemaName(options.databaseUrl);
const eventTableReady = await hasEventTable(prisma, schema);
if (options.resetTables ?? true) {
await prisma.event.deleteMany();
if (eventTableReady) {
await prisma.event.deleteMany();
}
await prisma.diplomacy.deleteMany();
await prisma.general.deleteMany();
await prisma.troop.deleteMany();
@@ -408,7 +435,7 @@ export const seedScenarioToDatabase = async (options: ScenarioSeedOptions): Prom
}
const eventRows = [...buildEventRows(seed.events), ...buildEventRows(seed.initialEvents, 'initial')];
if (eventRows.length > 0) {
if (eventRows.length > 0 && eventTableReady) {
await prisma.event.createMany({
data: eventRows,
});
@@ -37,12 +37,37 @@ type ScenarioSeederPrismaClient = {
};
};
const requiredTables = [
'world_state',
'nation',
'city',
'general',
'diplomacy',
'troop',
'event',
];
const hasRequiredTables = async (prisma: ScenarioSeederPrismaClient, schemaName: string): Promise<boolean> => {
for (const table of requiredTables) {
const result = (await prisma.$queryRawUnsafe(
`SELECT to_regclass('${schemaName}.${table}') as regclass`
)) as Array<{ regclass: string | null }>;
if (!Array.isArray(result) || result.length === 0 || result[0]?.regclass === null) {
return false;
}
}
return true;
};
const canConnectToDatabase = async (url: string): Promise<boolean> => {
const connector = createGamePostgresConnector({ url });
try {
await connector.connect();
const prisma = connector.prisma as unknown as ScenarioSeederPrismaClient;
await prisma.$queryRawUnsafe('SELECT 1');
if (!(await hasRequiredTables(prisma, schema))) {
return false;
}
return true;
} catch {
return false;