From 9f6d837a9b055f981795638a872b08540a9ca11b Mon Sep 17 00:00:00 2001
From: Meet <meetcshah19@gmail.com>
Date: Mon, 7 Oct 2024 17:28:32 +0530
Subject: [PATCH] feat: add migration script to migrate org

---
 backend/package.json                    |  1 +
 backend/scripts/migrate-organization.ts | 55 +++++++++++++++++++++++++
 2 files changed, 56 insertions(+)
 create mode 100644 backend/scripts/migrate-organization.ts

diff --git a/backend/package.json b/backend/package.json
index 314a77306..a51a5b3e4 100644
--- a/backend/package.json
+++ b/backend/package.json
@@ -52,6 +52,7 @@
     "migration:latest": "knex --knexfile ./src/db/knexfile.ts --client pg migrate:latest",
     "migration:status": "knex --knexfile ./src/db/knexfile.ts --client pg migrate:status",
     "migration:rollback": "knex --knexfile ./src/db/knexfile.ts migrate:rollback",
+    "migrate:org": "tsx ./scripts/migrate-organization.ts",
     "seed:new": "tsx ./scripts/create-seed-file.ts",
     "seed": "knex --knexfile ./src/db/knexfile.ts --client pg seed:run",
     "db:reset": "npm run migration:rollback -- --all && npm run migration:latest"
diff --git a/backend/scripts/migrate-organization.ts b/backend/scripts/migrate-organization.ts
new file mode 100644
index 000000000..88ffbdd25
--- /dev/null
+++ b/backend/scripts/migrate-organization.ts
@@ -0,0 +1,55 @@
+/* eslint-disable */
+import promptSync from "prompt-sync";
+import { execSync } from "child_process";
+import path from "path";
+import { existsSync } from "fs";
+
+const prompt = promptSync({
+    sigint: true
+});
+
+const exportDb = () => {
+    const exportHost = prompt("Enter your Postgres Host to migrate from: ");
+    const exportPort = prompt("Enter your Postgres Port to migrate from [Default = 5432]: ") ?? "5432";
+    const exportUser = prompt("Enter your Postgres User to migrate from: [Default = infisical]: ") ?? "infisical";
+    const exportPassword = prompt("Enter your Postgres Password to migrate from: ");
+    const exportDatabase = prompt("Enter your Postgres Database to migrate from [Default = infisical]: ") ?? "infisical";
+
+    execSync(
+        `PGDATABASE="${exportDatabase}" PGPASSWORD="${exportPassword}" PGHOST="${exportHost}" PGPORT=${exportPort} PGUSER=${exportUser} pg_dump infisical --exclude-table "audit_log*" > ${path.join(__dirname, "../src/db/dump.sql")}`,
+        { stdio: "inherit" }
+    );
+}
+
+const importDbForOrg = () => {
+    const importHost = prompt("Enter your Postgres Host to migrate to: ");
+    const importPort = prompt("Enter your Postgres Port to migrate to [Default = 5432]: ") ?? "5432";
+    const importUser = prompt("Enter your Postgres User to migrate to: [Default = infisical]: ") ?? "infisical";
+    const importPassword = prompt("Enter your Postgres Password to migrate to: ");
+    const importDatabase = prompt("Enter your Postgres Database to migrate to [Default = infisical]: ") ?? "infisical";
+    const orgId = prompt("Enter the organization ID to migrate: ");
+
+    if (!existsSync(path.join(__dirname, "../src/db/dump.sql"))) {
+        console.log("File not found, please export the database first.");
+        return;
+    }
+
+    execSync(`PGDATABASE="${importDatabase}" PGPASSWORD="${importPassword}" PGHOST="${importHost}" PGPORT=${importPort} PGUSER=${importUser} psql -f ${path.join(__dirname, "../src/db/dump.sql")}`);
+
+    execSync(`PGDATABASE="${importDatabase}" PGPASSWORD="${importPassword}" PGHOST="${importHost}" PGPORT=${importPort} PGUSER=${importUser} psql -c "DELETE FROM public.organizations WHERE id != '${orgId}'"`);
+
+    console.log("Organization migrated successfully.");
+}
+
+const main = () => {
+    const action = prompt("Enter the action to perform\n 1. Export from existing instance.\n 2. Import org to instance.\n \n Action: ");
+    if (action === "1") {
+        exportDb();
+    } else if (action === "2") {
+        importDbForOrg();
+    } else {
+        console.log("Invalid action");
+    }
+}
+
+main();
\ No newline at end of file