Add workflow query hooks

packages/twenty-server/src/modules/workflow/common/query-hooks/workflow-query-hook.exception.ts

@@ -0,0 +1,11 @@
+import { CustomException } from 'src/utils/custom-exception';
+
+export class WorkflowQueryHookException extends CustomException {
+  constructor(message: string, code: WorkflowQueryHookExceptionCode) {
+    super(message, code);
+  }
+}
+
+export enum WorkflowQueryHookExceptionCode {
+  FORBIDDEN = 'FORBIDDEN',
+}

packages/twenty-server/src/modules/workflow/common/query-hooks/workflow-query-hook.module.ts

@@ -1,12 +1,34 @@
 import { Module } from '@nestjs/common';
 
-import { WorkflowVersionUpdateOnePreQueryHook } from 'src/modules/workflow/common/query-hooks/workflow-version/workflow-version-update-one.pre-query-hook';
-import { WorkflowVersionValidationWorkspaceService } from 'src/modules/workflow/common/query-hooks/workflow-version/workflow-version-validation.workspace-service';
-import { WorkflowCommonWorkspaceService } from 'src/modules/workflow/common/workflow-common.workspace-service';
+import { WorkflowRunCreateManyPreQueryHook } from 'src/modules/workflow/common/query-hooks/workflow-run/workflow-run-create-many.pre-query.hook';
+import { WorkflowRunCreateOnePreQueryHook } from 'src/modules/workflow/common/query-hooks/workflow-run/workflow-run-create-one.pre-query.hook';
+import { WorkflowVersionValidationWorkspaceService } from 'src/modules/workflow/common/query-hooks/workflow-version/services/workflow-version-validation.workspace-service';
+import { WorkflowVersionCreateManyPreQueryHook } from 'src/modules/workflow/common/query-hooks/workflow-version/workflow-version-create-many.pre-query.hook';
+import { WorkflowVersionCreateOnePreQueryHook } from 'src/modules/workflow/common/query-hooks/workflow-version/workflow-version-create-one.pre-query.hook';
+import { WorkflowVersionDeleteManyPreQueryHook } from 'src/modules/workflow/common/query-hooks/workflow-version/workflow-version-delete-many.pre-query.hook';
+import { WorkflowVersionDeleteOnePreQueryHook } from 'src/modules/workflow/common/query-hooks/workflow-version/workflow-version-delete-one.pre-query.hook';
+import { WorkflowVersionUpdateManyPreQueryHook } from 'src/modules/workflow/common/query-hooks/workflow-version/workflow-version-update-many.pre-query.hook';
+import { WorkflowVersionUpdateOnePreQueryHook } from 'src/modules/workflow/common/query-hooks/workflow-version/workflow-version-update-one.pre-query.hook';
+import { WorkflowCreateManyPreQueryHook } from 'src/modules/workflow/common/query-hooks/workflow/workflow-create-many.pre-query.hook';
+import { WorkflowCreateOnePreQueryHook } from 'src/modules/workflow/common/query-hooks/workflow/workflow-create-one.pre-query.hook';
+import { WorkflowUpdateManyPreQueryHook } from 'src/modules/workflow/common/query-hooks/workflow/workflow-update-many.pre-query.hook';
+import { WorkflowUpdateOnePreQueryHook } from 'src/modules/workflow/common/query-hooks/workflow/workflow-update-one.pre-query.hook';
+import { WorkflowCommonWorkspaceService } from 'src/modules/workflow/common/services/workflow-common.workspace-service';
 
 @Module({
   providers: [
+    WorkflowCreateOnePreQueryHook,
+    WorkflowCreateManyPreQueryHook,
+    WorkflowUpdateOnePreQueryHook,
+    WorkflowUpdateManyPreQueryHook,
+    WorkflowRunCreateOnePreQueryHook,
+    WorkflowRunCreateManyPreQueryHook,
+    WorkflowVersionCreateOnePreQueryHook,
+    WorkflowVersionCreateManyPreQueryHook,
     WorkflowVersionUpdateOnePreQueryHook,
+    WorkflowVersionUpdateManyPreQueryHook,
+    WorkflowVersionDeleteOnePreQueryHook,
+    WorkflowVersionDeleteManyPreQueryHook,
     WorkflowVersionValidationWorkspaceService,
     WorkflowCommonWorkspaceService,
   ],

packages/twenty-server/src/modules/workflow/common/query-hooks/workflow-run/workflow-run-create-many.pre-query.hook.ts

@@ -0,0 +1,21 @@
+import { WorkspaceQueryHookInstance } from 'src/engine/api/graphql/workspace-query-runner/workspace-query-hook/interfaces/workspace-query-hook.interface';
+import { CreateManyResolverArgs } from 'src/engine/api/graphql/workspace-resolver-builder/interfaces/workspace-resolvers-builder.interface';
+
+import { WorkspaceQueryHook } from 'src/engine/api/graphql/workspace-query-runner/workspace-query-hook/decorators/workspace-query-hook.decorator';
+import {
+  WorkflowQueryHookException,
+  WorkflowQueryHookExceptionCode,
+} from 'src/modules/workflow/common/query-hooks/workflow-query-hook.exception';
+import { WorkflowRunWorkspaceEntity } from 'src/modules/workflow/common/standard-objects/workflow-run.workspace-entity';
+
+@WorkspaceQueryHook(`workflowRun.createMany`)
+export class WorkflowRunCreateManyPreQueryHook
+  implements WorkspaceQueryHookInstance
+{
+  async execute(): Promise<CreateManyResolverArgs<WorkflowRunWorkspaceEntity>> {
+    throw new WorkflowQueryHookException(
+      'Method not allowed.',
+      WorkflowQueryHookExceptionCode.FORBIDDEN,
+    );
+  }
+}

packages/twenty-server/src/modules/workflow/common/query-hooks/workflow-run/workflow-run-create-one.pre-query.hook.ts

@@ -0,0 +1,21 @@
+import { WorkspaceQueryHookInstance } from 'src/engine/api/graphql/workspace-query-runner/workspace-query-hook/interfaces/workspace-query-hook.interface';
+import { CreateOneResolverArgs } from 'src/engine/api/graphql/workspace-resolver-builder/interfaces/workspace-resolvers-builder.interface';
+
+import { WorkspaceQueryHook } from 'src/engine/api/graphql/workspace-query-runner/workspace-query-hook/decorators/workspace-query-hook.decorator';
+import {
+  WorkflowQueryHookException,
+  WorkflowQueryHookExceptionCode,
+} from 'src/modules/workflow/common/query-hooks/workflow-query-hook.exception';
+import { WorkflowRunWorkspaceEntity } from 'src/modules/workflow/common/standard-objects/workflow-run.workspace-entity';
+
+@WorkspaceQueryHook(`workflowRun.createOne`)
+export class WorkflowRunCreateOnePreQueryHook
+  implements WorkspaceQueryHookInstance
+{
+  async execute(): Promise<CreateOneResolverArgs<WorkflowRunWorkspaceEntity>> {
+    throw new WorkflowQueryHookException(
+      'Method not allowed.',
+      WorkflowQueryHookExceptionCode.FORBIDDEN,
+    );
+  }
+}

packages/twenty-server/src/modules/workflow/common/query-hooks/workflow-run/workflow-run-delete-many.pre-query.hook.ts

@@ -0,0 +1,21 @@
+import { WorkspaceQueryHookInstance } from 'src/engine/api/graphql/workspace-query-runner/workspace-query-hook/interfaces/workspace-query-hook.interface';
+import { DeleteManyResolverArgs } from 'src/engine/api/graphql/workspace-resolver-builder/interfaces/workspace-resolvers-builder.interface';
+
+import { WorkspaceQueryHook } from 'src/engine/api/graphql/workspace-query-runner/workspace-query-hook/decorators/workspace-query-hook.decorator';
+import {
+  WorkflowQueryHookException,
+  WorkflowQueryHookExceptionCode,
+} from 'src/modules/workflow/common/query-hooks/workflow-query-hook.exception';
+import { WorkflowRunWorkspaceEntity } from 'src/modules/workflow/common/standard-objects/workflow-run.workspace-entity';
+
+@WorkspaceQueryHook(`workflowRun.deleteMany`)
+export class WorkflowRunDeleteManyPreQueryHook
+  implements WorkspaceQueryHookInstance
+{
+  async execute(): Promise<DeleteManyResolverArgs<WorkflowRunWorkspaceEntity>> {
+    throw new WorkflowQueryHookException(
+      'Method not allowed.',
+      WorkflowQueryHookExceptionCode.FORBIDDEN,
+    );
+  }
+}

packages/twenty-server/src/modules/workflow/common/query-hooks/workflow-run/workflow-run-delete-one.pre-query.hook.ts

@@ -0,0 +1,20 @@
+import { WorkspaceQueryHookInstance } from 'src/engine/api/graphql/workspace-query-runner/workspace-query-hook/interfaces/workspace-query-hook.interface';
+import { DeleteOneResolverArgs } from 'src/engine/api/graphql/workspace-resolver-builder/interfaces/workspace-resolvers-builder.interface';
+
+import { WorkspaceQueryHook } from 'src/engine/api/graphql/workspace-query-runner/workspace-query-hook/decorators/workspace-query-hook.decorator';
+import {
+  WorkflowQueryHookException,
+  WorkflowQueryHookExceptionCode,
+} from 'src/modules/workflow/common/query-hooks/workflow-query-hook.exception';
+
+@WorkspaceQueryHook(`workflowRun.deleteOne`)
+export class WorkflowRunDeleteOnePreQueryHook
+  implements WorkspaceQueryHookInstance
+{
+  async execute(): Promise<DeleteOneResolverArgs> {
+    throw new WorkflowQueryHookException(
+      'Method not allowed.',
+      WorkflowQueryHookExceptionCode.FORBIDDEN,
+    );
+  }
+}

packages/twenty-server/src/modules/workflow/common/query-hooks/workflow-run/workflow-run-update-many.pre-query.hook.ts

@@ -0,0 +1,21 @@
+import { WorkspaceQueryHookInstance } from 'src/engine/api/graphql/workspace-query-runner/workspace-query-hook/interfaces/workspace-query-hook.interface';
+import { UpdateManyResolverArgs } from 'src/engine/api/graphql/workspace-resolver-builder/interfaces/workspace-resolvers-builder.interface';
+
+import { WorkspaceQueryHook } from 'src/engine/api/graphql/workspace-query-runner/workspace-query-hook/decorators/workspace-query-hook.decorator';
+import {
+  WorkflowQueryHookException,
+  WorkflowQueryHookExceptionCode,
+} from 'src/modules/workflow/common/query-hooks/workflow-query-hook.exception';
+import { WorkflowRunWorkspaceEntity } from 'src/modules/workflow/common/standard-objects/workflow-run.workspace-entity';
+
+@WorkspaceQueryHook(`workflowRun.updateMany`)
+export class WorkflowRunUpdateManyPreQueryHook
+  implements WorkspaceQueryHookInstance
+{
+  async execute(): Promise<UpdateManyResolverArgs<WorkflowRunWorkspaceEntity>> {
+    throw new WorkflowQueryHookException(
+      'Method not allowed.',
+      WorkflowQueryHookExceptionCode.FORBIDDEN,
+    );
+  }
+}

packages/twenty-server/src/modules/workflow/common/query-hooks/workflow-run/workflow-run-update-one.pre-query.hook.ts

@@ -0,0 +1,21 @@
+import { WorkspaceQueryHookInstance } from 'src/engine/api/graphql/workspace-query-runner/workspace-query-hook/interfaces/workspace-query-hook.interface';
+import { UpdateOneResolverArgs } from 'src/engine/api/graphql/workspace-resolver-builder/interfaces/workspace-resolvers-builder.interface';
+
+import { WorkspaceQueryHook } from 'src/engine/api/graphql/workspace-query-runner/workspace-query-hook/decorators/workspace-query-hook.decorator';
+import {
+  WorkflowQueryHookException,
+  WorkflowQueryHookExceptionCode,
+} from 'src/modules/workflow/common/query-hooks/workflow-query-hook.exception';
+import { WorkflowRunWorkspaceEntity } from 'src/modules/workflow/common/standard-objects/workflow-run.workspace-entity';
+
+@WorkspaceQueryHook(`workflowRun.updateOne`)
+export class WorkflowRunUpdateOnePreQueryHook
+  implements WorkspaceQueryHookInstance
+{
+  async execute(): Promise<UpdateOneResolverArgs<WorkflowRunWorkspaceEntity>> {
+    throw new WorkflowQueryHookException(
+      'Method not allowed.',
+      WorkflowQueryHookExceptionCode.FORBIDDEN,
+    );
+  }
+}

packages/twenty-server/src/modules/workflow/common/query-hooks/workflow-version/services/workflow-version-validation.workspace-service.ts

@@ -0,0 +1,101 @@
+import { Injectable } from '@nestjs/common';
+
+import {
+  CreateOneResolverArgs,
+  DeleteOneResolverArgs,
+  UpdateOneResolverArgs,
+} from 'src/engine/api/graphql/workspace-resolver-builder/interfaces/workspace-resolvers-builder.interface';
+
+import { TwentyORMManager } from 'src/engine/twenty-orm/twenty-orm.manager';
+import {
+  WorkflowQueryHookException,
+  WorkflowQueryHookExceptionCode,
+} from 'src/modules/workflow/common/query-hooks/workflow-query-hook.exception';
+import { WorkflowCommonWorkspaceService } from 'src/modules/workflow/common/services/workflow-common.workspace-service';
+import {
+  WorkflowVersionStatus,
+  WorkflowVersionWorkspaceEntity,
+} from 'src/modules/workflow/common/standard-objects/workflow-version.workspace-entity';
+import { WorkflowTrigger } from 'src/modules/workflow/common/types/workflow-trigger.type';
+
+@Injectable()
+export class WorkflowVersionValidationWorkspaceService {
+  constructor(
+    private readonly workflowCommonWorkspaceService: WorkflowCommonWorkspaceService,
+    private readonly twentyORMManager: TwentyORMManager,
+  ) {}
+
+  async validateWorkflowVersionForCreateOne(
+    payload: CreateOneResolverArgs<WorkflowVersionWorkspaceEntity>,
+  ) {
+    if (
+      payload.data.status &&
+      payload.data.status !== WorkflowVersionStatus.DRAFT
+    ) {
+      throw new WorkflowQueryHookException(
+        'Cannot create workflow version with status other than draft',
+        WorkflowQueryHookExceptionCode.FORBIDDEN,
+      );
+    }
+
+    const workflowVersionRepository =
+      await this.twentyORMManager.getRepository<WorkflowVersionWorkspaceEntity>(
+        'workflowVersion',
+      );
+
+    const workflowAlreadyHasDraftVersion =
+      await workflowVersionRepository.exists({
+        where: {
+          workflowId: payload.data.workflowId,
+          status: WorkflowVersionStatus.DRAFT,
+        },
+      });
+
+    if (workflowAlreadyHasDraftVersion) {
+      throw new WorkflowQueryHookException(
+        'Cannot create multiple draft versions for the same workflow',
+        WorkflowQueryHookExceptionCode.FORBIDDEN,
+      );
+    }
+  }
+
+  async validateWorkflowVersionForUpdateOne(
+    payload: UpdateOneResolverArgs<WorkflowVersionWorkspaceEntity>,
+  ) {
+    const workflowVersion =
+      await this.workflowCommonWorkspaceService.getWorkflowVersionOrFail(
+        payload.id,
+      );
+
+    this.validateWorkflowVersionIsDraft(workflowVersion);
+
+    if (payload.data.status !== workflowVersion.status) {
+      throw new WorkflowQueryHookException(
+        'Cannot update workflow version status manually',
+        WorkflowQueryHookExceptionCode.FORBIDDEN,
+      );
+    }
+  }
+
+  async validateWorkflowVersionForDeleteOne(payload: DeleteOneResolverArgs) {
+    const workflowVersion =
+      await this.workflowCommonWorkspaceService.getWorkflowVersionOrFail(
+        payload.id,
+      );
+
+    this.validateWorkflowVersionIsDraft(workflowVersion);
+  }
+
+  private validateWorkflowVersionIsDraft(
+    workflowVersion: Omit<WorkflowVersionWorkspaceEntity, 'trigger'> & {
+      trigger: WorkflowTrigger;
+    },
+  ) {
+    if (workflowVersion.status !== WorkflowVersionStatus.DRAFT) {
+      throw new WorkflowQueryHookException(
+        'Workflow version is not in draft status',
+        WorkflowQueryHookExceptionCode.FORBIDDEN,
+      );
+    }
+  }
+}

packages/twenty-server/src/modules/workflow/common/query-hooks/workflow-version/workflow-version-create-many.pre-query.hook.ts

@@ -0,0 +1,37 @@
+import { WorkspaceQueryHookInstance } from 'src/engine/api/graphql/workspace-query-runner/workspace-query-hook/interfaces/workspace-query-hook.interface';
+import {
+  CreateManyResolverArgs,
+  CreateOneResolverArgs,
+} from 'src/engine/api/graphql/workspace-resolver-builder/interfaces/workspace-resolvers-builder.interface';
+
+import { WorkspaceQueryHook } from 'src/engine/api/graphql/workspace-query-runner/workspace-query-hook/decorators/workspace-query-hook.decorator';
+import { AuthContext } from 'src/engine/core-modules/auth/types/auth-context.type';
+import { WorkflowVersionValidationWorkspaceService } from 'src/modules/workflow/common/query-hooks/workflow-version/services/workflow-version-validation.workspace-service';
+import { WorkflowVersionWorkspaceEntity } from 'src/modules/workflow/common/standard-objects/workflow-version.workspace-entity';
+
+@WorkspaceQueryHook(`workflowVersion.createMany`)
+export class WorkflowVersionCreateManyPreQueryHook
+  implements WorkspaceQueryHookInstance
+{
+  constructor(
+    private readonly workflowVersionValidationWorkspaceService: WorkflowVersionValidationWorkspaceService,
+  ) {}
+
+  async execute(
+    _authContext: AuthContext,
+    _objectName: string,
+    payload: CreateManyResolverArgs<WorkflowVersionWorkspaceEntity>,
+  ): Promise<CreateManyResolverArgs<WorkflowVersionWorkspaceEntity>> {
+    await Promise.all(
+      payload.data.map(async (workflowVersion) => {
+        await this.workflowVersionValidationWorkspaceService.validateWorkflowVersionForCreateOne(
+          {
+            data: workflowVersion,
+          } satisfies CreateOneResolverArgs<WorkflowVersionWorkspaceEntity>,
+        );
+      }),
+    );
+
+    return payload;
+  }
+}

packages/twenty-server/src/modules/workflow/common/query-hooks/workflow-version/workflow-version-create-one.pre-query.hook.ts

@@ -0,0 +1,28 @@
+import { WorkspaceQueryHookInstance } from 'src/engine/api/graphql/workspace-query-runner/workspace-query-hook/interfaces/workspace-query-hook.interface';
+import { CreateOneResolverArgs } from 'src/engine/api/graphql/workspace-resolver-builder/interfaces/workspace-resolvers-builder.interface';
+
+import { WorkspaceQueryHook } from 'src/engine/api/graphql/workspace-query-runner/workspace-query-hook/decorators/workspace-query-hook.decorator';
+import { AuthContext } from 'src/engine/core-modules/auth/types/auth-context.type';
+import { WorkflowVersionValidationWorkspaceService } from 'src/modules/workflow/common/query-hooks/workflow-version/services/workflow-version-validation.workspace-service';
+import { WorkflowVersionWorkspaceEntity } from 'src/modules/workflow/common/standard-objects/workflow-version.workspace-entity';
+
+@WorkspaceQueryHook(`workflowVersion.createOne`)
+export class WorkflowVersionCreateOnePreQueryHook
+  implements WorkspaceQueryHookInstance
+{
+  constructor(
+    private readonly workflowVersionValidationWorkspaceService: WorkflowVersionValidationWorkspaceService,
+  ) {}
+
+  async execute(
+    _authContext: AuthContext,
+    _objectName: string,
+    payload: CreateOneResolverArgs<WorkflowVersionWorkspaceEntity>,
+  ): Promise<CreateOneResolverArgs<WorkflowVersionWorkspaceEntity>> {
+    await this.workflowVersionValidationWorkspaceService.validateWorkflowVersionForCreateOne(
+      payload,
+    );
+
+    return payload;
+  }
+}

packages/twenty-server/src/modules/workflow/common/query-hooks/workflow-version/workflow-version-delete-many.pre-query.hook.ts

@@ -0,0 +1,23 @@
+import { WorkspaceQueryHookInstance } from 'src/engine/api/graphql/workspace-query-runner/workspace-query-hook/interfaces/workspace-query-hook.interface';
+import { DeleteManyResolverArgs } from 'src/engine/api/graphql/workspace-resolver-builder/interfaces/workspace-resolvers-builder.interface';
+
+import { WorkspaceQueryHook } from 'src/engine/api/graphql/workspace-query-runner/workspace-query-hook/decorators/workspace-query-hook.decorator';
+import {
+  WorkflowQueryHookException,
+  WorkflowQueryHookExceptionCode,
+} from 'src/modules/workflow/common/query-hooks/workflow-query-hook.exception';
+import { WorkflowVersionWorkspaceEntity } from 'src/modules/workflow/common/standard-objects/workflow-version.workspace-entity';
+
+@WorkspaceQueryHook(`workflowVersion.deleteMany`)
+export class WorkflowVersionDeleteManyPreQueryHook
+  implements WorkspaceQueryHookInstance
+{
+  async execute(): Promise<
+    DeleteManyResolverArgs<WorkflowVersionWorkspaceEntity>
+  > {
+    throw new WorkflowQueryHookException(
+      'Method not allowed.',
+      WorkflowQueryHookExceptionCode.FORBIDDEN,
+    );
+  }
+}