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,98 @@
+import { rawDataSource } from 'src/database/typeorm/raw/raw.datasource';
+
+import { camelToSnakeCase, performQuery } from './setup-db-utils';
+
+rawDataSource
+  .initialize()
+  .then(async () => {
+    await performQuery(
+      'CREATE SCHEMA IF NOT EXISTS "public"',
+      'create schema "public"',
+    );
+    await performQuery(
+      'CREATE SCHEMA IF NOT EXISTS "core"',
+      'create schema "core"',
+    );
+
+    await performQuery(
+      'CREATE EXTENSION IF NOT EXISTS "uuid-ossp"',
+      'create extension "uuid-ossp"',
+    );
+
+    await performQuery(
+      'CREATE EXTENSION IF NOT EXISTS "unaccent"',
+      'create extension "unaccent"',
+    );
+
+    await performQuery(
+      `CREATE OR REPLACE FUNCTION public.unaccent_immutable(input text)
+    RETURNS text
+    LANGUAGE sql
+    IMMUTABLE
+AS $$
+SELECT public.unaccent('public.unaccent'::regdictionary, input)
+$$;`,
+      'create immutable unaccent wrapper function',
+    );
+
+    // We paused the work on FDW
+    if (process.env.IS_FDW_ENABLED !== 'true') {
+      return;
+    }
+
+    await performQuery(
+      'CREATE EXTENSION IF NOT EXISTS "postgres_fdw"',
+      'create extension "postgres_fdw"',
+    );
+
+    await performQuery(
+      'CREATE EXTENSION IF NOT EXISTS "wrappers"',
+      'create extension "wrappers"',
+    );
+
+    await performQuery(
+      'CREATE EXTENSION IF NOT EXISTS "mysql_fdw"',
+      'create extension "mysql_fdw"',
+    );
+
+    const supabaseWrappers = [
+      'airtable',
+      'bigQuery',
+      'clickHouse',
+      'firebase',
+      'logflare',
+      's3',
+      'stripe',
+    ]; // See https://supabase.github.io/wrappers/
+
+    for (const wrapper of supabaseWrappers) {
+      if (await checkForeignDataWrapperExists(`${wrapper.toLowerCase()}_fdw`)) {
+        continue;
+      }
+      await performQuery(
+        `
+          CREATE FOREIGN DATA WRAPPER "${wrapper.toLowerCase()}_fdw"
+          HANDLER "${camelToSnakeCase(wrapper)}_fdw_handler"
+          VALIDATOR "${camelToSnakeCase(wrapper)}_fdw_validator";
+          `,
+        `create ${wrapper} "wrappers"`,
+        true,
+        true,
+      );
+    }
+  })
+  .catch((err) => {
+    // oxlint-disable-next-line no-console
+    console.error('Error during Data Source initialization:', err);
+  });
+
+async function checkForeignDataWrapperExists(
+  wrapperName: string,
+): Promise<boolean> {
+  const result = await rawDataSource.query(
+    `SELECT 1 FROM pg_foreign_data_wrapper WHERE fdwname = $1`,
+    [wrapperName],
+  );
+
+  return result.length > 0;
+}