fix migration number and add allowed_headers migration

server/setup/migrationsSqlite.ts

@@ -20,7 +20,7 @@ import m16 from "./scriptsSqlite/1.0.0";
 import m17 from "./scriptsSqlite/1.1.0";
 import m18 from "./scriptsSqlite/1.2.0";
 import m19 from "./scriptsSqlite/1.3.0";
-import m20 from "./scriptsSqlite/1.3.0";
+import m20 from "./scriptsSqlite/1.5.0";
 
 // THIS CANNOT IMPORT ANYTHING FROM THE SERVER
 // EXCEPT FOR THE DATABASE AND THE SCHEMA
@@ -41,7 +41,7 @@ const migrations = [
     { version: "1.1.0", run: m17 },
     { version: "1.2.0", run: m18 },
     { version: "1.3.0", run: m19 },
-    { version: "1.4.0", run: m20 },
+    { version: "1.5.0", run: m20 },
     // Add new migrations here as they are created
 ] as const;
 

server/setup/scriptsSqlite/1.4.0.ts

@@ -1,29 +0,0 @@
-import Database from "better-sqlite3";
-import path from "path";
-import { encodeBase32LowerCaseNoPadding } from "@oslojs/encoding";
-import { APP_PATH, configFilePath1, configFilePath2 } from "@server/lib/consts";
-
-const version = "1.4.0";
-const location = path.join(APP_PATH, "db", "db.sqlite");
-
-export default async function migration() {
-    console.log(`Running setup script ${version}...`);
-
-    const db = new Database(location);
-
-    try {
-        db.pragma("foreign_keys = OFF");
-        db.transaction(() => {
-            db.exec(`
-               ALTER TABLE 'sites' ADD 'dockerSocketEnabled' integer DEFAULT true NOT NULL;
-            `);
-        })(); // <-- executes the transaction immediately
-        db.pragma("foreign_keys = ON");
-        console.log(`Migrated database schema`);
-    } catch (e) {
-        console.log("Unable to migrate database schema");
-        throw e;
-    }
-
-    console.log(`${version} migration complete`);
-}

server/setup/scriptsSqlite/1.5.0.ts

@@ -0,0 +1,72 @@
+import Database from "better-sqlite3";
+import path from "path";
+import { APP_PATH, configFilePath1, configFilePath2 } from "@server/lib/consts";
+import fs from "fs";
+import yaml from "js-yaml";
+
+const version = "1.5.0";
+const location = path.join(APP_PATH, "db", "db.sqlite");
+
+export default async function migration() {
+    console.log(`Running setup script ${version}...`);
+
+    const db = new Database(location);
+
+    try {
+        db.pragma("foreign_keys = OFF");
+        db.transaction(() => {
+            db.exec(`
+               ALTER TABLE 'sites' ADD 'dockerSocketEnabled' integer DEFAULT true NOT NULL;
+            `);
+        })(); // <-- executes the transaction immediately
+        db.pragma("foreign_keys = ON");
+        console.log(`Migrated database schema`);
+    } catch (e) {
+        console.log("Unable to migrate database schema");
+        throw e;
+    }
+
+    try {
+        // Determine which config file exists
+        const filePaths = [configFilePath1, configFilePath2];
+        let filePath = "";
+        for (const path of filePaths) {
+            if (fs.existsSync(path)) {
+                filePath = path;
+                break;
+            }
+        }
+
+        if (!filePath) {
+            throw new Error(
+                `No config file found (expected config.yml or config.yaml).`
+            );
+        }
+
+        // Read and parse the YAML file
+        let rawConfig: any;
+        const fileContents = fs.readFileSync(filePath, "utf8");
+        rawConfig = yaml.load(fileContents);
+
+        if (rawConfig.cors?.headers) {
+            const headers = JSON.parse(
+                JSON.stringify(rawConfig.cors.headers)
+            );
+            rawConfig.cors.allowed_headers = headers;
+            delete rawConfig.cors.headers;
+        }
+
+        // Write the updated YAML back to the file
+        const updatedYaml = yaml.dump(rawConfig);
+        fs.writeFileSync(filePath, updatedYaml, "utf8");
+
+        console.log(`Migrated CORS headers to allowed_headers`);
+    } catch (e) {
+        console.log(
+            `Unable to migrate config file. Error: ${e}`
+        );
+        throw e;
+    }
+
+    console.log(`${version} migration complete`);
+}