From 2f8ecf17ed0a23a04f5a825a4252046010c4cdcb Mon Sep 17 00:00:00 2001 From: miloschwartz Date: Sun, 21 Dec 2025 15:43:38 -0500 Subject: [PATCH] add postgres migration --- :w | 96 ++++++++++++++++++++++++++++ server/setup/migrationsPg.ts | 4 +- server/setup/scriptsPg/1.14.0.ts | 96 ++++++++++++++++++++++++++++ server/setup/scriptsSqlite/1.14.0.ts | 2 - 4 files changed, 195 insertions(+), 3 deletions(-) create mode 100644 :w create mode 100644 server/setup/scriptsPg/1.14.0.ts diff --git a/:w b/:w new file mode 100644 index 00000000..7ccded5a --- /dev/null +++ b/:w @@ -0,0 +1,96 @@ +import { db } from "@server/db/pg/driver"; +import { sql } from "drizzle-orm"; +import { __DIRNAME } from "@server/lib/consts"; + +const version = "1.14.0"; + +export default async function migration() { + console.log(`Running setup script ${version}...`); + + try { + await db.execute(sql`BEGIN`); + + await db.execute(sql` + CREATE TABLE "loginPageBranding" ( + "loginPageBrandingId" serial PRIMARY KEY NOT NULL, + "logoUrl" text NOT NULL, + "logoWidth" integer NOT NULL, + "logoHeight" integer NOT NULL, + "primaryColor" text, + "resourceTitle" text NOT NULL, + "resourceSubtitle" text, + "orgTitle" text, + "orgSubtitle" text + ); + `); + + await db.execute(sql` + CREATE TABLE "loginPageBrandingOrg" ( + "loginPageBrandingId" integer NOT NULL, + "orgId" varchar NOT NULL + ); + `); + + await db.execute(sql` + CREATE TABLE "resourceHeaderAuthExtendedCompatibility" ( + "headerAuthExtendedCompatibilityId" serial PRIMARY KEY NOT NULL, + "resourceId" integer NOT NULL, + "extendedCompatibilityIsActivated" boolean DEFAULT false NOT NULL + ); + `); + + await db.execute( + sql`ALTER TABLE "resources" ADD COLUMN "maintenanceModeEnabled" boolean DEFAULT false NOT NULL;` + ); + + await db.execute( + sql`ALTER TABLE "resources" ADD COLUMN "maintenanceModeType" text DEFAULT 'forced';` + ); + + await db.execute( + sql`ALTER TABLE "resources" ADD COLUMN "maintenanceTitle" text;` + ); + + await db.execute( + sql`ALTER TABLE "resources" ADD COLUMN "maintenanceMessage" text;` + ); + + await db.execute( + sql`ALTER TABLE "resources" ADD COLUMN "maintenanceEstimatedTime" text;` + ); + + await db.execute( + sql`ALTER TABLE "siteResources" ADD COLUMN "tcpPortRangeString" varchar;` + ); + + await db.execute( + sql`ALTER TABLE "siteResources" ADD COLUMN "udpPortRangeString" varchar;` + ); + + await db.execute( + sql`ALTER TABLE "siteResources" ADD COLUMN "disableIcmp" boolean DEFAULT false NOT NULL;` + ); + + await db.execute( + sql`ALTER TABLE "loginPageBrandingOrg" ADD CONSTRAINT "loginPageBrandingOrg_loginPageBrandingId_loginPageBranding_loginPageBrandingId_fk" FOREIGN KEY ("loginPageBrandingId") REFERENCES "public"."loginPageBranding"("loginPageBrandingId") ON DELETE cascade ON UPDATE no action;` + ); + + await db.execute( + sql`ALTER TABLE "loginPageBrandingOrg" ADD CONSTRAINT "loginPageBrandingOrg_orgId_orgs_orgId_fk" FOREIGN KEY ("orgId") REFERENCES "public"."orgs"("orgId") ON DELETE cascade ON UPDATE no action;` + ); + + await db.execute( + sql`ALTER TABLE "resourceHeaderAuthExtendedCompatibility" ADD CONSTRAINT "resourceHeaderAuthExtendedCompatibility_resourceId_resources_resourceId_fk" FOREIGN KEY ("resourceId") REFERENCES "public"."resources"("resourceId") ON DELETE cascade ON UPDATE no action;` + ); + + 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/migrationsPg.ts b/server/setup/migrationsPg.ts index 0fc42f9d..0fc8c574 100644 --- a/server/setup/migrationsPg.ts +++ b/server/setup/migrationsPg.ts @@ -15,6 +15,7 @@ import m7 from "./scriptsPg/1.11.0"; import m8 from "./scriptsPg/1.11.1"; import m9 from "./scriptsPg/1.12.0"; import m10 from "./scriptsPg/1.13.0"; +import m11 from "./scriptsPg/1.14.0"; // THIS CANNOT IMPORT ANYTHING FROM THE SERVER // EXCEPT FOR THE DATABASE AND THE SCHEMA @@ -30,7 +31,8 @@ const migrations = [ { version: "1.11.0", run: m7 }, { version: "1.11.1", run: m8 }, { version: "1.12.0", run: m9 }, - { version: "1.13.0", run: m10 } + { version: "1.13.0", run: m10 }, + { version: "1.14.0", run: m11 } // Add new migrations here as they are created ] as { version: string; diff --git a/server/setup/scriptsPg/1.14.0.ts b/server/setup/scriptsPg/1.14.0.ts new file mode 100644 index 00000000..7ccded5a --- /dev/null +++ b/server/setup/scriptsPg/1.14.0.ts @@ -0,0 +1,96 @@ +import { db } from "@server/db/pg/driver"; +import { sql } from "drizzle-orm"; +import { __DIRNAME } from "@server/lib/consts"; + +const version = "1.14.0"; + +export default async function migration() { + console.log(`Running setup script ${version}...`); + + try { + await db.execute(sql`BEGIN`); + + await db.execute(sql` + CREATE TABLE "loginPageBranding" ( + "loginPageBrandingId" serial PRIMARY KEY NOT NULL, + "logoUrl" text NOT NULL, + "logoWidth" integer NOT NULL, + "logoHeight" integer NOT NULL, + "primaryColor" text, + "resourceTitle" text NOT NULL, + "resourceSubtitle" text, + "orgTitle" text, + "orgSubtitle" text + ); + `); + + await db.execute(sql` + CREATE TABLE "loginPageBrandingOrg" ( + "loginPageBrandingId" integer NOT NULL, + "orgId" varchar NOT NULL + ); + `); + + await db.execute(sql` + CREATE TABLE "resourceHeaderAuthExtendedCompatibility" ( + "headerAuthExtendedCompatibilityId" serial PRIMARY KEY NOT NULL, + "resourceId" integer NOT NULL, + "extendedCompatibilityIsActivated" boolean DEFAULT false NOT NULL + ); + `); + + await db.execute( + sql`ALTER TABLE "resources" ADD COLUMN "maintenanceModeEnabled" boolean DEFAULT false NOT NULL;` + ); + + await db.execute( + sql`ALTER TABLE "resources" ADD COLUMN "maintenanceModeType" text DEFAULT 'forced';` + ); + + await db.execute( + sql`ALTER TABLE "resources" ADD COLUMN "maintenanceTitle" text;` + ); + + await db.execute( + sql`ALTER TABLE "resources" ADD COLUMN "maintenanceMessage" text;` + ); + + await db.execute( + sql`ALTER TABLE "resources" ADD COLUMN "maintenanceEstimatedTime" text;` + ); + + await db.execute( + sql`ALTER TABLE "siteResources" ADD COLUMN "tcpPortRangeString" varchar;` + ); + + await db.execute( + sql`ALTER TABLE "siteResources" ADD COLUMN "udpPortRangeString" varchar;` + ); + + await db.execute( + sql`ALTER TABLE "siteResources" ADD COLUMN "disableIcmp" boolean DEFAULT false NOT NULL;` + ); + + await db.execute( + sql`ALTER TABLE "loginPageBrandingOrg" ADD CONSTRAINT "loginPageBrandingOrg_loginPageBrandingId_loginPageBranding_loginPageBrandingId_fk" FOREIGN KEY ("loginPageBrandingId") REFERENCES "public"."loginPageBranding"("loginPageBrandingId") ON DELETE cascade ON UPDATE no action;` + ); + + await db.execute( + sql`ALTER TABLE "loginPageBrandingOrg" ADD CONSTRAINT "loginPageBrandingOrg_orgId_orgs_orgId_fk" FOREIGN KEY ("orgId") REFERENCES "public"."orgs"("orgId") ON DELETE cascade ON UPDATE no action;` + ); + + await db.execute( + sql`ALTER TABLE "resourceHeaderAuthExtendedCompatibility" ADD CONSTRAINT "resourceHeaderAuthExtendedCompatibility_resourceId_resources_resourceId_fk" FOREIGN KEY ("resourceId") REFERENCES "public"."resources"("resourceId") ON DELETE cascade ON UPDATE no action;` + ); + + 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.14.0.ts b/server/setup/scriptsSqlite/1.14.0.ts index ce7e0ae5..a4883b8f 100644 --- a/server/setup/scriptsSqlite/1.14.0.ts +++ b/server/setup/scriptsSqlite/1.14.0.ts @@ -4,8 +4,6 @@ import path from "path"; const version = "1.14.0"; -await migration(); - export default async function migration() { console.log(`Running setup script ${version}...`);