Workspace creation prefill fix

packages/twenty-server/src/engine/workspace-manager/standard-objects-prefill-data/utils/prefill-created-workspace-records.util.ts

@@ -0,0 +1,71 @@
+import { Logger } from '@nestjs/common';
+
+import { type FlatEntityMaps } from 'src/engine/metadata-modules/flat-entity/types/flat-entity-maps.type';
+import { type FlatFieldMetadata } from 'src/engine/metadata-modules/flat-field-metadata/types/flat-field-metadata.type';
+import { type FlatObjectMetadata } from 'src/engine/metadata-modules/flat-object-metadata/types/flat-object-metadata.type';
+import { type GlobalWorkspaceOrmManager } from 'src/engine/twenty-orm/global-workspace-datasource/global-workspace-orm.manager';
+import { buildSystemAuthContext } from 'src/engine/twenty-orm/utils/build-system-auth-context.util';
+import { prefillCompanies } from 'src/engine/workspace-manager/standard-objects-prefill-data/prefill-companies';
+import { prefillPeople } from 'src/engine/workspace-manager/standard-objects-prefill-data/prefill-people';
+import { prefillWorkflows } from 'src/engine/workspace-manager/standard-objects-prefill-data/prefill-workflows';
+
+const logger = new Logger('PrefillWorkspaceDataWithTransaction');
+
+export const prefillCreatedWorkspaceRecords = async ({
+  globalWorkspaceOrmManager,
+  workspaceId,
+  schemaName,
+  flatObjectMetadataMaps,
+  flatFieldMetadataMaps,
+}: {
+  globalWorkspaceOrmManager: GlobalWorkspaceOrmManager;
+  workspaceId: string;
+  schemaName: string;
+  flatObjectMetadataMaps: FlatEntityMaps<FlatObjectMetadata>;
+  flatFieldMetadataMaps: FlatEntityMaps<FlatFieldMetadata>;
+}): Promise<void> => {
+  const authContext = buildSystemAuthContext(workspaceId);
+
+  await globalWorkspaceOrmManager.executeInWorkspaceContext(
+    authContext,
+    async () => {
+      const workspaceDataSource =
+        await globalWorkspaceOrmManager.getGlobalWorkspaceDataSource();
+
+      const queryRunner = workspaceDataSource.createQueryRunner();
+
+      await queryRunner.connect();
+
+      try {
+        await queryRunner.startTransaction();
+
+        await prefillCompanies(queryRunner.manager, schemaName);
+
+        await prefillPeople(queryRunner.manager, schemaName);
+
+        await prefillWorkflows(
+          queryRunner.manager,
+          schemaName,
+          flatObjectMetadataMaps,
+          flatFieldMetadataMaps,
+        );
+
+        await queryRunner.commitTransaction();
+      } catch (error) {
+        if (queryRunner.isTransactionActive) {
+          try {
+            await queryRunner.rollbackTransaction();
+          } catch (rollbackError) {
+            logger.error(
+              `Failed to rollback prefill transaction: ${rollbackError.message}`,
+            );
+          }
+        }
+        throw error;
+      } finally {
+        await queryRunner.release();
+      }
+    },
+  );
+};
+

packages/twenty-server/src/engine/workspace-manager/workspace-manager.service.ts

@@ -24,6 +24,7 @@ import { GlobalWorkspaceOrmManager } from 'src/engine/twenty-orm/global-workspac
 import { WorkspaceDataSourceService } from 'src/engine/workspace-datasource/workspace-datasource.service';
 import { prefillCoreViews } from 'src/engine/workspace-manager/standard-objects-prefill-data/prefill-core-views';
 import { standardObjectsPrefillData } from 'src/engine/workspace-manager/standard-objects-prefill-data/standard-objects-prefill-data';
+import { prefillCreatedWorkspaceRecords } from 'src/engine/workspace-manager/standard-objects-prefill-data/utils/prefill-created-workspace-records.util';
 import { TwentyStandardApplicationService } from 'src/engine/workspace-manager/twenty-standard-application/services/twenty-standard-application.service';
 import { ADMIN_ROLE } from 'src/engine/workspace-manager/workspace-sync-metadata/standard-roles/roles/admin-role';
 import { WorkspaceSyncMetadataService } from 'src/engine/workspace-manager/workspace-sync-metadata/workspace-sync-metadata.service';
@@ -103,6 +104,22 @@ export class WorkspaceManagerService {
           workspaceId,
         },
       );
+
+      const { flatObjectMetadataMaps, flatFieldMetadataMaps } =
+        await this.flatEntityMapsCacheService.getOrRecomputeManyOrAllFlatEntityMaps(
+          {
+            workspaceId,
+            flatMapsKeys: ['flatObjectMetadataMaps', 'flatFieldMetadataMaps'],
+          },
+        );
+
+      await prefillCreatedWorkspaceRecords({
+        globalWorkspaceOrmManager: this.globalWorkspaceOrmManager,
+        workspaceId,
+        schemaName,
+        flatObjectMetadataMaps,
+        flatFieldMetadataMaps,
+      });
     } else {
       await this.workspaceSyncMetadataService.synchronize({
         workspaceId,