Fix comment

apps/server/src/core/comment/comment.controller.ts

@@ -9,7 +9,7 @@ import {
 import { CommentService } from './comment.service';
 import { CreateCommentDto } from './dto/create-comment.dto';
 import { UpdateCommentDto } from './dto/update-comment.dto';
-import { CommentsInput, SingleCommentInput } from './dto/comments.input';
+import { PageIdDto, CommentIdDto } from './dto/comments.input';
 import { AuthUser } from '../../decorators/auth-user.decorator';
 import { AuthWorkspace } from '../../decorators/auth-workspace.decorator';
 import { JwtAuthGuard } from '../../guards/jwt-auth.guard';
@@ -34,7 +34,7 @@ export class CommentController {
   @HttpCode(HttpStatus.OK)
   @Post()
   findPageComments(
-    @Body() input: CommentsInput,
+    @Body() input: PageIdDto,
     @Body()
     pagination: PaginationOptions,
     //@AuthUser() user: User,
@@ -45,19 +45,22 @@ export class CommentController {
 
   @HttpCode(HttpStatus.OK)
   @Post('info')
-  findOne(@Body() input: SingleCommentInput) {
-    return this.commentService.findWithCreator(input.id);
+  findOne(@Body() input: CommentIdDto) {
+    return this.commentService.findById(input.commentId);
   }
 
   @HttpCode(HttpStatus.OK)
   @Post('update')
   update(@Body() updateCommentDto: UpdateCommentDto) {
-    return this.commentService.update(updateCommentDto.id, updateCommentDto);
+    return this.commentService.update(
+      updateCommentDto.commentId,
+      updateCommentDto,
+    );
   }
 
   @HttpCode(HttpStatus.OK)
   @Post('delete')
-  remove(@Body() input: SingleCommentInput) {
-    return this.commentService.remove(input.id);
+  remove(@Body() input: CommentIdDto) {
+    return this.commentService.remove(input.commentId);
   }
 }

apps/server/src/core/comment/comment.service.ts

@@ -1,4 +1,8 @@
-import { BadRequestException, Injectable } from '@nestjs/common';
+import {
+  BadRequestException,
+  Injectable,
+  NotFoundException,
+} from '@nestjs/common';
 import { CreateCommentDto } from './dto/create-comment.dto';
 import { UpdateCommentDto } from './dto/update-comment.dto';
 import { PageService } from '../page/services/page.service';
@@ -14,8 +18,14 @@ export class CommentService {
     private pageService: PageService,
   ) {}
 
-  async findWithCreator(commentId: string) {
-    // todo: find comment with creator object
+  async findById(commentId: string) {
+    const comment = this.commentRepo.findById(commentId, {
+      includeCreator: true,
+    });
+    if (!comment) {
+      throw new NotFoundException('Comment not found');
+    }
+    return comment;
   }
 
   async create(
@@ -55,15 +65,21 @@ export class CommentService {
       creatorId: userId,
       workspaceId: workspaceId,
     });
-    // todo return created comment and creator relation
 
-    return createdComment;
+    // return created comment and creator relation
+    return this.findById(createdComment.id);
   }
 
   async findByPageId(
     pageId: string,
     pagination: PaginationOptions,
   ): Promise<PaginationResult<Comment>> {
+    const page = await this.pageService.findById(pageId);
+
+    if (!page) {
+      throw new BadRequestException('Page not found');
+    }
+
     const pageComments = await this.commentRepo.findPageComments(
       pageId,
       pagination,
@@ -78,15 +94,24 @@ export class CommentService {
   ): Promise<Comment> {
     const commentContent = JSON.parse(updateCommentDto.content);
 
+    const comment = await this.commentRepo.findById(commentId);
+    if (!comment) {
+      throw new NotFoundException('Comment not found');
+    }
+
+    const editedAt = new Date();
+
     await this.commentRepo.updateComment(
       {
         content: commentContent,
-        editedAt: new Date(),
+        editedAt: editedAt,
       },
       commentId,
     );
+    comment.content = commentContent;
+    comment.editedAt = editedAt;
 
-    return this.commentRepo.findById(commentId);
+    return comment;
   }
 
   async remove(id: string): Promise<void> {

apps/server/src/core/comment/dto/comments.input.ts

@@ -1,11 +1,11 @@
 import { IsUUID } from 'class-validator';
 
-export class CommentsInput {
+export class PageIdDto {
   @IsUUID()
   pageId: string;
 }
 
-export class SingleCommentInput {
+export class CommentIdDto {
   @IsUUID()
-  id: string;
+  commentId: string;
 }

apps/server/src/core/comment/dto/create-comment.dto.ts

@@ -1,10 +1,6 @@
 import { IsJSON, IsOptional, IsString, IsUUID } from 'class-validator';
 
 export class CreateCommentDto {
-  @IsOptional()
-  @IsUUID()
-  id?: string;
-
   @IsUUID()
   pageId: string;
 

apps/server/src/core/comment/dto/update-comment.dto.ts

@@ -2,7 +2,7 @@ import { IsJSON, IsUUID } from 'class-validator';
 
 export class UpdateCommentDto {
   @IsUUID()
-  id: string;
+  commentId: string;
 
   @IsJSON()
   content: any;