Guard yarn database:migrate:prod (#19008)

## Motivations A lot of self hosters hands up using the `yarn database:migrated:prod` either manually or through AI assisted debug while they try to upgrade an instance while their workspace is still blocked in a previous one Leading to their whole database permanent corruption ## What happened Replaced the direct call the the typeorm cli to a command calling it programmatically, adding a layer of security in case a workspace seems to be blocked in a previous version than the one just before the one being installed ( e.g 1.0 when you try to upgrade from 1.1 to 1.2 ) For our cloud we still need a way to bypass this security explaining the -f flag ## Remark Centralized this logic and refactored creating new services `WorkspaceVersionService` and `CoreEngineVersionService` that will become useful for the upcoming upgrade refactor Related to https://github.com/twentyhq/twenty-infra/pull/529
2026-03-27 15:39:18 +01:00
parent c96c034908
commit 281bb6d783
25 changed files with 401 additions and 216 deletions
@@ -0,0 +1,44 @@
+import { rawDataSource } from 'src/database/typeorm/raw/raw.datasource';
+
+import { performQuery } from './setup-db-utils';
+
+async function dropSchemasSequentially() {
+  try {
+    await rawDataSource.initialize();
+
+    // Fetch all schemas excluding the ones we want to keep
+    const schemas =
+      (await performQuery<{ schema_name: string }[]>(
+        `
+      SELECT n.nspname AS "schema_name"
+      FROM pg_catalog.pg_namespace n
+      WHERE n.nspname !~ '^pg_'
+        AND n.nspname <> 'information_schema'
+        AND n.nspname NOT IN ('metric_helpers', 'user_management', 'public')
+    `,
+        'Fetching schemas...',
+      )) ?? [];
+
+    const batchSize = 10;
+
+    for (let i = 0; i < schemas.length; i += batchSize) {
+      const batch = schemas.slice(i, i + batchSize);
+
+      await Promise.all(
+        batch.map((schema) =>
+          performQuery(
+            `DROP SCHEMA IF EXISTS "${schema.schema_name}" CASCADE;`,
+            `Dropping schema ${schema.schema_name}...`,
+          ),
+        ),
+      );
+    }
+    // oxlint-disable-next-line no-console
+    console.log('All schemas dropped successfully.');
+  } catch (err) {
+    // oxlint-disable-next-line no-console
+    console.error('Error during schema dropping:', err);
+  }
+}
+
+dropSchemasSequentially();