Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
76 changes: 76 additions & 0 deletions internal-packages/dashboard-agent-db/migrate-status.mjs
Original file line number Diff line number Diff line change
@@ -0,0 +1,76 @@
// Pending-status check for the `trigger_dashboard_agent` schema (sibling of
// migrate.mjs). Exit 0 = up to date, 1 = pending, 2 = error.
import { readFileSync } from "node:fs";
import { dirname, join } from "node:path";
import { fileURLToPath } from "node:url";
import postgres from "postgres";

const MIGRATIONS_SCHEMA = "drizzle";
const MIGRATIONS_TABLE = "__dashboard_agent_migrations";

const connectionString = process.env.DASHBOARD_AGENT_DATABASE_URL ?? process.env.DATABASE_URL;

if (!connectionString) {
console.error(
"[dashboard-agent-db] DASHBOARD_AGENT_DATABASE_URL / DATABASE_URL not set; cannot check status."
);
process.exit(2);
}

// Match migrate.mjs: drop the Prisma-style `?schema=` param postgres.js forwards.
function normalizeConnectionString(value) {
try {
const url = new URL(value);
url.searchParams.delete("schema");
return url.toString();
} catch {
return value;
}
}

const journalPath = join(
dirname(fileURLToPath(import.meta.url)),
"drizzle/meta/_journal.json"
);
const sql = postgres(normalizeConnectionString(connectionString), {
max: 1,
prepare: false,
onnotice: () => {},
});

async function main() {
const journal = JSON.parse(readFileSync(journalPath, "utf-8"));
const entries = [...journal.entries].sort((a, b) => a.when - b.when);

let lastAppliedAt = -1;
try {
const rows = await sql`SELECT MAX(created_at)::bigint AS last FROM ${sql(
MIGRATIONS_SCHEMA
)}.${sql(MIGRATIONS_TABLE)}`;
lastAppliedAt = rows[0].last === null ? -1 : Number(rows[0].last);
} catch (err) {
// 42P01: journal table absent (fresh database), so nothing is applied.
if (err.code !== "42P01") throw err;
}

const pending = entries.filter((e) => e.when > lastAppliedAt);
console.log(
`${entries.length} migration(s) found, ${entries.length - pending.length} applied`
);

if (pending.length > 0) {
console.log(`${pending.length} pending migration(s):`);
for (const e of pending) console.log(` - ${e.tag}`);
return 1;
}

console.log("Dashboard agent schema is up to date");
return 0;
}

main()
.then((code) => sql.end({ timeout: 5 }).then(() => process.exit(code)))
.catch((err) => {
console.error(err);
sql.end({ timeout: 5 }).finally(() => process.exit(2));
});
1 change: 1 addition & 0 deletions internal-packages/dashboard-agent-db/package.json
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@
"db:generate": "drizzle-kit generate",
"db:migrate": "drizzle-kit migrate",
"db:migrate:deploy": "node migrate.mjs",
"db:migrate:status": "node migrate-status.mjs",
"db:studio": "drizzle-kit studio"
}
}