fix: add migration for new user columns (tier, auth_provider, etc.)
This commit is contained in:
@@ -81,6 +81,38 @@ async function ensureToolRelationsTable(): Promise<void> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async function ensureUserColumns(): Promise<void> {
|
||||||
|
try {
|
||||||
|
const cols = await db.execute(
|
||||||
|
sql`SELECT column_name FROM information_schema.columns WHERE table_name = 'users'`,
|
||||||
|
);
|
||||||
|
const existing = new Set((cols.rows as [{ column_name: string }]).map((r) => r.column_name));
|
||||||
|
const toAdd: { name: string; def: string }[] = [];
|
||||||
|
if (!existing.has("tier")) toAdd.push({ name: "tier", def: "text NOT NULL DEFAULT 'free'" });
|
||||||
|
if (!existing.has("auth_provider")) toAdd.push({ name: "auth_provider", def: "text NOT NULL DEFAULT 'local'" });
|
||||||
|
if (!existing.has("auth_provider_id")) toAdd.push({ name: "auth_provider_id", def: "text" });
|
||||||
|
if (!existing.has("display_name")) toAdd.push({ name: "display_name", def: "text" });
|
||||||
|
if (!existing.has("password_hash")) {
|
||||||
|
toAdd.push({ name: "password_hash", def: "text" });
|
||||||
|
} else {
|
||||||
|
const [ph] = await db.execute(
|
||||||
|
sql`SELECT is_nullable FROM information_schema.columns WHERE table_name = 'users' AND column_name = 'password_hash'`,
|
||||||
|
);
|
||||||
|
const nullable = (ph.rows as [{ is_nullable: string }])[0]?.is_nullable;
|
||||||
|
if (nullable === "NO") {
|
||||||
|
await db.execute(sql`ALTER TABLE "users" ALTER COLUMN "password_hash" DROP NOT NULL`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for (const col of toAdd) {
|
||||||
|
const stmt = `ALTER TABLE "users" ADD COLUMN "${col.name}" ${col.def}`;
|
||||||
|
await db.execute(sql.raw(stmt));
|
||||||
|
logger.info({ column: col.name }, "Added column to users table");
|
||||||
|
}
|
||||||
|
} catch (err) {
|
||||||
|
logger.error({ err }, "Failed to ensure user columns");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
async function ensureToolCostsTable(): Promise<void> {
|
async function ensureToolCostsTable(): Promise<void> {
|
||||||
try {
|
try {
|
||||||
const exists = await db.execute(
|
const exists = await db.execute(
|
||||||
|
|||||||
Reference in New Issue
Block a user