Add migrations

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;

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;
 

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`);
+}

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`);
+}