aevgarik/paperclip
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

server: auto-apply embedded postgres migrations on first run

Browse Source
This commit is contained in:
Dotta
2026-03-03 10:49:18 -06:00
parent 20737e2a11
commit c77e2b3c19
1 changed files with 22 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
server/src/index.ts
View File

@@ -73,8 +73,8 @@ function formatPendingMigrationSummary(migrations: string[]): string {
async function promptApplyMigrations(migrations: string[]): Promise<boolean> { async function promptApplyMigrations(migrations: string[]): Promise<boolean> {
if (process.env.PAPERCLIP_MIGRATION_PROMPT === "never") return false; if (process.env.PAPERCLIP_MIGRATION_PROMPT === "never") return false;
if (!stdin.isTTY || !stdout.isTTY) return true;
if (process.env.PAPERCLIP_MIGRATION_AUTO_APPLY === "true") return true; if (process.env.PAPERCLIP_MIGRATION_AUTO_APPLY === "true") return true;
if (!stdin.isTTY || !stdout.isTTY) return true;
const prompt = createInterface({ input: stdin, output: stdout }); const prompt = createInterface({ input: stdin, output: stdout });
try { try {
@@ -87,7 +87,16 @@ async function promptApplyMigrations(migrations: string[]): Promise<boolean> {
} }
} }
async function ensureMigrations(connectionString: string, label: string): Promise<MigrationSummary> { type EnsureMigrationsOptions = {
autoApply?: boolean;
};
async function ensureMigrations(
connectionString: string,
label: string,
opts?: EnsureMigrationsOptions,
): Promise<MigrationSummary> {
const autoApply = opts?.autoApply === true;
let state = await inspectMigrations(connectionString); let state = await inspectMigrations(connectionString);
if (state.status === "needsMigrations" && state.reason === "pending-migrations") { if (state.status === "needsMigrations" && state.reason === "pending-migrations") {
const repair = await reconcilePendingMigrationHistory(connectionString); const repair = await reconcilePendingMigrationHistory(connectionString);
@@ -106,7 +115,7 @@ async function ensureMigrations(connectionString: string, label: string): Promis
{ tableCount: state.tableCount }, { tableCount: state.tableCount },
`${label} has existing tables but no migration journal. Run migrations manually to sync schema.`, `${label} has existing tables but no migration journal. Run migrations manually to sync schema.`,
); );
const apply = await promptApplyMigrations(state.pendingMigrations); const apply = autoApply ? true : await promptApplyMigrations(state.pendingMigrations);
if (!apply) { if (!apply) {
logger.warn( logger.warn(
{ pendingMigrations: state.pendingMigrations }, { pendingMigrations: state.pendingMigrations },
@@ -120,7 +129,7 @@ async function ensureMigrations(connectionString: string, label: string): Promis
return "applied (pending migrations)"; return "applied (pending migrations)";
} }
const apply = await promptApplyMigrations(state.pendingMigrations); const apply = autoApply ? true : await promptApplyMigrations(state.pendingMigrations);
if (!apply) { if (!apply) {
logger.warn( logger.warn(
{ pendingMigrations: state.pendingMigrations }, { pendingMigrations: state.pendingMigrations },
@@ -233,6 +242,7 @@ if (config.databaseUrl) {
} }
const clusterVersionFile = resolve(dataDir, "PG_VERSION"); const clusterVersionFile = resolve(dataDir, "PG_VERSION");
const clusterAlreadyInitialized = existsSync(clusterVersionFile);
const postmasterPidFile = resolve(dataDir, "postmaster.pid"); const postmasterPidFile = resolve(dataDir, "postmaster.pid");
const isPidRunning = (pid: number): boolean => { const isPidRunning = (pid: number): boolean => {
try { try {
@@ -277,7 +287,7 @@ if (config.databaseUrl) {
persistent: true, persistent: true,
}); });
if (!existsSync(clusterVersionFile)) { if (!clusterAlreadyInitialized) {
await embeddedPostgres.initialise(); await embeddedPostgres.initialise();
} else { } else {
logger.info(`Embedded PostgreSQL cluster already exists (${clusterVersionFile}); skipping init`); logger.info(`Embedded PostgreSQL cluster already exists (${clusterVersionFile}); skipping init`);
@@ -298,7 +308,13 @@ if (config.databaseUrl) {
} }
const embeddedConnectionString = `postgres://paperclip:paperclip@127.0.0.1:${port}/paperclip`; const embeddedConnectionString = `postgres://paperclip:paperclip@127.0.0.1:${port}/paperclip`;
migrationSummary = await ensureMigrations(embeddedConnectionString, "Embedded PostgreSQL"); const shouldAutoApplyFirstRunMigrations = !clusterAlreadyInitialized || dbStatus === "created";
if (shouldAutoApplyFirstRunMigrations) {
logger.info("Detected first-run embedded PostgreSQL setup; applying pending migrations automatically");
}
migrationSummary = await ensureMigrations(embeddedConnectionString, "Embedded PostgreSQL", {
autoApply: shouldAutoApplyFirstRunMigrations,
});
db = createDb(embeddedConnectionString); db = createDb(embeddedConnectionString);
logger.info("Embedded PostgreSQL ready"); logger.info("Embedded PostgreSQL ready");