DDRCompanion/lib/db/migrations.ts
Michael Zhang dfbf923d04
Some checks failed
ci/woodpecker/push/woodpecker Pipeline failed
download charts
2024-05-16 01:02:44 -05:00

73 lines
2.2 KiB
TypeScript

import type { Database } from "sql.js";
const migrations = [m01_initial];
export default async function executeMigrations(db: Database) {
// Check last migration status
const migrationsWithNames = migrations.map<
[string, (_: Database) => Promise<void>]
>((func) => [func.name, func]);
let startMigrationAt = 0;
try {
const rows = await db.exec("SELECT name FROM _migrations LIMIT 1");
if (rows.length < 1) throw new Error("wtf");
const migrationStatus = rows?.[0];
const lastMigrationName = migrationStatus.values[0][0];
console.log("status", migrationStatus);
if (lastMigrationName) {
const foundIndex = migrationsWithNames.findIndex(
([name]) => name === migrationStatus.values[0][0],
);
if (foundIndex >= 0) startMigrationAt = foundIndex + 1;
}
} catch (e) {
// Don't have the table
await db.run("BEGIN TRANSACTION");
await db.run(`
CREATE TABLE IF NOT EXISTS _migrations (name TEXT PRIMARY KEY);
INSERT INTO _migrations (name) VALUES (NULL);
CREATE TABLE IF NOT EXISTS _appDataVersion (version INTEGER PRIMARY KEY);
INSERT INTO _appDataVersion (version) VALUES (0);
`);
await db.exec("COMMIT TRANSACTION");
console.log("Created table.");
}
console.log(migrationsWithNames);
console.log(startMigrationAt);
const migrationsToRun = migrationsWithNames.slice(startMigrationAt);
console.log(`Running ${migrationsToRun.length} migrations...`);
for (const [name, migration] of migrationsToRun) {
console.log("Running migration", name);
await db.exec("BEGIN TRANSACTION");
await migration(db);
await db.exec("UPDATE _migrations SET name = $name", { $name: name });
await db.exec("COMMIT TRANSACTION");
}
}
async function m01_initial(db: Database) {
db.exec(`
CREATE TABLE charts (
chart_id INTEGER PRIMARY KEY AUTOINCREMENT,
artist TEXT,
title TEXT,
difficulty TEXT,
banner_image TEXT
);
CREATE TABLE scores (
score_id INTEGER PRIMARY KEY AUTOINCREMENT,
chart_id INTEGER,
FOREIGN KEY (chart_id) REFERENCES charts(chart_id)
);
CREATE TABLE banners (
name TEXT PRIMARY KEY
)
`);
}