Select exit node for local sites

server/setup/scriptsPg/1.11.1.ts

@@ -0,0 +1,45 @@
+import { db } from "@server/db/pg/driver";
+import { sql } from "drizzle-orm";
+
+const version = "1.11.1";
+
+export default async function migration() {
+    console.log(`Running setup script ${version}...`);
+
+    try {
+        // Get the first exit node with type 'gerbil'
+        const exitNodesQuery = await db.execute(
+            sql`SELECT "exitNodeId" FROM "exitNodes" WHERE "type" = 'gerbil' LIMIT 1`
+        );
+        const exitNodes = exitNodesQuery.rows as {
+            exitNodeId: number;
+        }[];
+
+        const exitNodeId = exitNodes.length > 0 ? exitNodes[0].exitNodeId : null;
+
+        // Get all sites with type 'local'
+        const sitesQuery = await db.execute(
+            sql`SELECT "siteId" FROM "sites" WHERE "type" = 'local'`
+        );
+        const sites = sitesQuery.rows as {
+            siteId: number;
+        }[];
+
+        // Update sites to use the exit node
+        for (const site of sites) {
+            await db.execute(sql`
+                UPDATE "sites" SET "exitNodeId" = ${exitNodeId} WHERE "siteId" = ${site.siteId}
+            `);
+        }
+
+        await db.execute(sql`COMMIT`);
+        console.log(`Updated sites with exit node`);
+    } catch (e) {
+        await db.execute(sql`ROLLBACK`);
+        console.log("Unable to update sites with exit node");
+        console.log(e);
+        throw e;
+    }
+
+    console.log(`${version} migration complete`);
+}