Build cron for data seed demo (#4142)

* Migrate command to cron

* Put back command using job as well

* Build service and module + move into folder

---------

Co-authored-by: Thomas Trompette <thomast@twenty.com>

packages/twenty-server/src/database/commands/data-seed-demo-workspace/crons/data-seed-demo-workspace-cron-pattern.ts

@@ -0,0 +1 @@
+export const dataSeedDemoWorkspaceCronPattern = '0 22 * * *'; // Every day at 10pm

packages/twenty-server/src/database/commands/data-seed-demo-workspace/crons/data-seed-demo-workspace.cron.command.ts

@@ -0,0 +1,29 @@
+import { Inject } from '@nestjs/common';
+
+import { Command, CommandRunner } from 'nest-commander';
+
+import { dataSeedDemoWorkspaceCronPattern } from 'src/database/commands/data-seed-demo-workspace/crons/data-seed-demo-workspace-cron-pattern';
+import { DataSeedDemoWorkspaceJob } from 'src/database/commands/data-seed-demo-workspace/jobs/data-seed-demo-workspace.job';
+import { MessageQueue } from 'src/integrations/message-queue/message-queue.constants';
+import { MessageQueueService } from 'src/integrations/message-queue/services/message-queue.service';
+
+@Command({
+  name: 'workspace:cron:seed:demo',
+  description: 'Seed workspace with demo data.',
+})
+export class DataSeedDemoWorkspaceCronCommand extends CommandRunner {
+  constructor(
+    @Inject(MessageQueue.cronQueue)
+    private readonly messageQueueService: MessageQueueService,
+  ) {
+    super();
+  }
+
+  async run(): Promise<void> {
+    await this.messageQueueService.addCron<undefined>(
+      DataSeedDemoWorkspaceJob.name,
+      undefined,
+      dataSeedDemoWorkspaceCronPattern,
+    );
+  }
+}

packages/twenty-server/src/database/commands/data-seed-demo-workspace/data-seed-demo-workspace-command.ts

@@ -0,0 +1,19 @@
+import { Command, CommandRunner } from 'nest-commander';
+
+import { DataSeedDemoWorkspaceService } from 'src/database/commands/data-seed-demo-workspace/services/data-seed-demo-workspace.service';
+
+@Command({
+  name: 'workspace:seed:demo',
+  description: 'Seed workspace with demo data. Use in development only.',
+})
+export class DataSeedDemoWorkspaceCommand extends CommandRunner {
+  constructor(
+    private readonly dataSeedDemoWorkspaceService: DataSeedDemoWorkspaceService,
+  ) {
+    super();
+  }
+
+  async run(): Promise<void> {
+    await this.dataSeedDemoWorkspaceService.seedDemo();
+  }
+}

packages/twenty-server/src/database/commands/data-seed-demo-workspace/data-seed-demo-workspace.module.ts

@@ -0,0 +1,12 @@
+import { Module } from '@nestjs/common';
+
+import { EnvironmentModule } from 'src/integrations/environment/environment.module';
+import { WorkspaceManagerModule } from 'src/workspace/workspace-manager/workspace-manager.module';
+import { DataSeedDemoWorkspaceService } from 'src/database/commands/data-seed-demo-workspace/services/data-seed-demo-workspace.service';
+
+@Module({
+  imports: [WorkspaceManagerModule, EnvironmentModule],
+  providers: [DataSeedDemoWorkspaceService],
+  exports: [DataSeedDemoWorkspaceService],
+})
+export class DataSeedDemoWorkspaceModule {}

packages/twenty-server/src/database/commands/data-seed-demo-workspace/jobs/data-seed-demo-workspace.job.ts

@@ -0,0 +1,16 @@
+import { Injectable } from '@nestjs/common';
+
+import { MessageQueueJob } from 'src/integrations/message-queue/interfaces/message-queue-job.interface';
+
+import { DataSeedDemoWorkspaceService } from 'src/database/commands/data-seed-demo-workspace/services/data-seed-demo-workspace.service';
+
+@Injectable()
+export class DataSeedDemoWorkspaceJob implements MessageQueueJob<undefined> {
+  constructor(
+    private readonly dataSeedDemoWorkspaceService: DataSeedDemoWorkspaceService,
+  ) {}
+
+  async handle(): Promise<void> {
+    await this.dataSeedDemoWorkspaceService.seedDemo();
+  }
+}

packages/twenty-server/src/database/commands/data-seed-demo-workspace/services/data-seed-demo-workspace.service.ts

@@ -0,0 +1,49 @@
+import { Injectable } from '@nestjs/common';
+
+import { DataSource } from 'typeorm';
+
+import {
+  deleteCoreSchema,
+  seedCoreSchema,
+} from 'src/database/typeorm-seeds/core';
+import { EnvironmentService } from 'src/integrations/environment/environment.service';
+import { WorkspaceManagerService } from 'src/workspace/workspace-manager/workspace-manager.service';
+
+@Injectable()
+export class DataSeedDemoWorkspaceService {
+  constructor(
+    private readonly environmentService: EnvironmentService,
+    private readonly workspaceManagerService: WorkspaceManagerService,
+  ) {}
+
+  async seedDemo(): Promise<void> {
+    try {
+      const dataSource = new DataSource({
+        url: this.environmentService.getPGDatabaseUrl(),
+        type: 'postgres',
+        logging: true,
+        schema: 'public',
+      });
+
+      await dataSource.initialize();
+      const demoWorkspaceIds = this.environmentService.getDemoWorkspaceIds();
+
+      if (demoWorkspaceIds.length === 0) {
+        throw new Error(
+          'Could not get DEMO_WORKSPACE_IDS. Please specify in .env',
+        );
+      }
+      for (const workspaceId of demoWorkspaceIds) {
+        await deleteCoreSchema(dataSource, workspaceId);
+        await this.workspaceManagerService.delete(workspaceId);
+
+        await seedCoreSchema(dataSource, workspaceId);
+        await this.workspaceManagerService.initDemo(workspaceId);
+      }
+    } catch (error) {
+      console.error(error);
+
+      return;
+    }
+  }
+}