feat: api endpoint to create comment

Author: hentrymartin

src/api/ai-workflow/ai-workflow.controller.ts

@@ -24,6 +24,7 @@ import {
   UpdateAiWorkflowDto,
   CreateAiWorkflowRunItemsDto,
   UpdateAiWorkflowRunDto,
+  CreateRunItemCommentDto,
 } from '../../dto/aiWorkflow.dto';
 import { Scopes } from 'src/shared/decorators/scopes.decorator';
 import { UserRole } from 'src/shared/enums/userRole.enum';
@@ -38,6 +39,43 @@ import { User } from 'src/shared/decorators/user.decorator';
 export class AiWorkflowController {
   constructor(private readonly aiWorkflowService: AiWorkflowService) {}
 
+  @Post('/:workflowId/runs/:runId/items/:itemId/comments')
+  @Roles(
+    UserRole.Submitter,
+    UserRole.Copilot,
+    UserRole.ProjectManager,
+    UserRole.Admin,
+  )
+  @Scopes(Scope.CreateWorkflowRun)
+  @ApiOperation({ summary: 'Create a comment for a specific run item' })
+  @ApiResponse({
+    status: 201,
+    description: 'Comment created successfully.',
+  })
+  @ApiResponse({ status: 400, description: 'Bad Request.' })
+  @ApiResponse({ status: 401, description: 'Unauthorized.' })
+  @ApiResponse({ status: 403, description: 'Forbidden.' })
+  @ApiResponse({
+    status: 404,
+    description: 'Workflow, Run, or Item not found.',
+  })
+  async createRunItemComment(
+    @Param('workflowId') workflowId: string,
+    @Param('runId') runId: string,
+    @Param('itemId') itemId: string,
+    @Body(new ValidationPipe({ whitelist: true, transform: true }))
+    body: CreateRunItemCommentDto,
+    @User() user: JwtUser,
+  ) {
+    return this.aiWorkflowService.createRunItemComment(
+      workflowId,
+      runId,
+      itemId,
+      body,
+      user,
+    );
+  }
+
   @Post()
   @Roles(UserRole.Admin)
   @Scopes(Scope.CreateWorkflow)

src/api/ai-workflow/ai-workflow.service.ts

@@ -10,6 +10,7 @@ import { PrismaService } from '../../shared/modules/global/prisma.service';
 import {
   CreateAiWorkflowDto,
   CreateAiWorkflowRunDto,
+  CreateRunItemCommentDto,
   UpdateAiWorkflowDto,
   UpdateAiWorkflowRunDto,
 } from '../../dto/aiWorkflow.dto';
@@ -36,6 +37,51 @@ export class AiWorkflowService {
     this.logger = LoggerService.forRoot('AiWorkflowService');
   }
 
+  async createRunItemComment(
+    workflowId: string,
+    runId: string,
+    itemId: string,
+    body: CreateRunItemCommentDto,
+    user: JwtUser,
+  ) {
+    const workflow = await this.prisma.aiWorkflow.findUnique({
+      where: { id: workflowId },
+    });
+    if (!workflow) {
+      throw new NotFoundException(`Workflow with id ${workflowId} not found.`);
+    }
+
+    const run = await this.prisma.aiWorkflowRun.findUnique({
+      where: { id: runId },
+    });
+    if (!run || run.workflowId !== workflowId) {
+      throw new NotFoundException(
+        `Run with id ${runId} not found or does not belong to workflow ${workflowId}.`,
+      );
+    }
+
+    const item = await this.prisma.aiWorkflowRunItem.findUnique({
+      where: { id: itemId },
+    });
+    if (!item || item.workflowRunId !== runId) {
+      throw new NotFoundException(
+        `Item with id ${itemId} not found or does not belong to run ${runId}.`,
+      );
+    }
+
+    const createdComment = await this.prisma.aiWorkflowRunItemComment.create({
+      data: {
+        workflowRunItemId: itemId,
+        content: body.content,
+        parentId: body.parentId ?? null,
+        userId: user.userId!,
+        createdAt: new Date(),
+      },
+    });
+
+    return createdComment;
+  }
+
   async scorecardExists(scorecardId: string): Promise<boolean> {
     const count = await this.prisma.scorecard.count({
       where: { id: scorecardId, status: ScorecardStatus.ACTIVE },

src/dto/aiWorkflow.dto.ts

@@ -138,3 +138,15 @@ export class CreateAiWorkflowRunItemsDto {
   @Type(() => CreateAiWorkflowRunItemDto)
   items: CreateAiWorkflowRunItemDto[];
 }
+
+export class CreateRunItemCommentDto {
+  @ApiProperty()
+  @IsString()
+  @IsNotEmpty()
+  content: string;
+
+  @ApiProperty({ required: false })
+  @IsString()
+  @IsOptional()
+  parentId?: string;
+}