diff --git a/server/setup/migrationsPg.ts b/server/setup/migrationsPg.ts index c4eda0a0f..d65e09cfa 100644 --- a/server/setup/migrationsPg.ts +++ b/server/setup/migrationsPg.ts @@ -30,6 +30,7 @@ import m21 from "./scriptsPg/1.20.0"; import m22 from "./scriptsPg/1.21.0"; import m23 from "./scriptsPg/1.22.0"; import m24 from "./scriptsPg/1.23.0"; +import m25 from "./scriptsPg/1.24.0"; // THIS CANNOT IMPORT ANYTHING FROM THE SERVER // EXCEPT FOR THE DATABASE AND THE SCHEMA @@ -59,7 +60,8 @@ const migrations = [ { version: "1.20.0", run: m21 }, { version: "1.21.0", run: m22 }, { version: "1.22.0", run: m23 }, - { version: "1.23.0", run: m24 } + { version: "1.23.0", run: m24 }, + { version: "1.24.0", run: m25 } // Add new migrations here as they are created ] as { version: string; diff --git a/server/setup/migrationsSqlite.ts b/server/setup/migrationsSqlite.ts index eb2846610..b47e7989c 100644 --- a/server/setup/migrationsSqlite.ts +++ b/server/setup/migrationsSqlite.ts @@ -50,6 +50,7 @@ import m43 from "./scriptsSqlite/1.20.0"; import m44 from "./scriptsSqlite/1.21.0"; import m45 from "./scriptsSqlite/1.22.0"; import m46 from "./scriptsSqlite/1.23.0"; +import m47 from "./scriptsSqlite/1.24.0"; // THIS CANNOT IMPORT ANYTHING FROM THE SERVER // EXCEPT FOR THE DATABASE AND THE SCHEMA @@ -96,7 +97,8 @@ const migrations = [ { version: "1.20.0", run: m43 }, { version: "1.21.0", run: m44 }, { version: "1.22.0", run: m45 }, - { version: "1.23.0", run: m46 } + { version: "1.23.0", run: m46 }, + { version: "1.24.0", run: m47 } // Add new migrations here as they are created ] as const; diff --git a/server/setup/scriptsPg/1.24.0.ts b/server/setup/scriptsPg/1.24.0.ts new file mode 100644 index 000000000..887c2651a --- /dev/null +++ b/server/setup/scriptsPg/1.24.0.ts @@ -0,0 +1,34 @@ +import { db } from "@server/db/pg/driver"; +import { sql } from "drizzle-orm"; + +const version = "1.24.0"; + +export default async function migration() { + console.log(`Running setup script ${version}...`); + + try { + await db.execute(sql`BEGIN`); + + await db.execute(sql` + ALTER TABLE "resources" RENAME COLUMN "headers" TO "requestHeaders" + `); + + await db.execute(sql` + ALTER TABLE "orgs" ADD COLUMN "settingsJitModeLimit" integer DEFAULT 250 NOT NULL; + `); + + await db.execute(sql` + ALTER TABLE "resources" ADD COLUMN "responseHeaders" text; + `); + + await db.execute(sql`COMMIT`); + console.log("Migrated database"); + } catch (e) { + await db.execute(sql`ROLLBACK`); + console.log("Unable to migrate database"); + console.log(e); + throw e; + } + + console.log(`${version} migration complete`); +} diff --git a/server/setup/scriptsSqlite/1.24.0.ts b/server/setup/scriptsSqlite/1.24.0.ts new file mode 100644 index 000000000..d2ce5d35e --- /dev/null +++ b/server/setup/scriptsSqlite/1.24.0.ts @@ -0,0 +1,45 @@ +import { APP_PATH } from "@server/lib/consts"; +import Database from "better-sqlite3"; +import path from "path"; + +const version = "1.24.0"; + +export default async function migration() { + console.log(`Running setup script ${version}...`); + + const location = path.join(APP_PATH, "db", "db.sqlite"); + const db = new Database(location); + + try { + db.pragma("foreign_keys = OFF"); + + db.transaction(() => { + db.prepare( + ` + ALTER TABLE 'resources' RENAME COLUMN "headers" TO "requestHeaders"; + ` + ).run(); + + db.prepare( + ` + ALTER TABLE 'resources' ADD 'responseHeaders' text; + ` + ).run(); + + db.prepare( + ` + ALTER TABLE 'orgs' ADD 'settingsJitModeLimit' integer DEFAULT 250 NOT NULL; + ` + ).run(); + })(); + + db.pragma("foreign_keys = ON"); + + console.log("Migrated database"); + } catch (e) { + console.log("Failed to migrate db:", e); + throw e; + } + + console.log(`${version} migration complete`); +}