[Chat] 채팅 수정 API 구현

chat/src/main/kotlin/kpring/chat/ChatApplication.kt

@@ -2,7 +2,9 @@ package kpring.chat
 
 import org.springframework.boot.autoconfigure.SpringBootApplication
 import org.springframework.boot.runApplication
+import org.springframework.data.mongodb.config.EnableMongoAuditing
 
+@EnableMongoAuditing
 @SpringBootApplication
 class ChatApplication
 

chat/src/main/kotlin/kpring/chat/chat/api/v1/ChatController.kt

@@ -6,6 +6,7 @@ import kpring.chat.global.exception.GlobalException
 import kpring.core.auth.client.AuthClient
 import kpring.core.chat.chat.dto.request.ChatType
 import kpring.core.chat.chat.dto.request.CreateChatRequest
+import kpring.core.chat.chat.dto.request.UpdateChatRequest
 import kpring.core.global.dto.response.ApiResponse
 import kpring.core.server.client.ServerClient
 import kpring.core.server.dto.request.GetServerCondition
@@ -18,8 +19,8 @@ import org.springframework.web.bind.annotation.*
 @RequestMapping("/api/v1")
 class ChatController(
   private val chatService: ChatService,
-  val authClient: AuthClient,
-  val serverClient: ServerClient,
+  private val authClient: AuthClient,
+  private val serverClient: ServerClient,
 ) {
   @PostMapping("/chat")
   fun createChat(
@@ -59,4 +60,18 @@ class ChatController(
       }
     return ResponseEntity.ok().body(ApiResponse(data = result, status = 200))
   }
+
+  @PatchMapping("/chat")
+  fun updateChat(
+    @Validated @RequestBody request: UpdateChatRequest,
+    @RequestHeader("Authorization") token: String,
+  ): ResponseEntity<*> {
+    val userId = authClient.getTokenInfo(token).data!!.userId
+    val result =
+      when (request.type) {
+        ChatType.Room -> chatService.updateRoomChat(request, userId)
+        ChatType.Server -> chatService.updateServerChat(request, userId)
+      }
+    return ResponseEntity.ok().body(ApiResponse<Nothing>(status = 200))
+  }
 }

chat/src/main/kotlin/kpring/chat/chat/model/RoomChat.kt → chat/src/main/kotlin/kpring/chat/chat/model/Chat.kt

@@ -7,15 +7,20 @@ import org.springframework.data.mongodb.core.mapping.Document
 
 @NoArg
 @Document(collection = "chats")
-class RoomChat(
+class Chat(
   val userId: String,
-  val roomId: String,
-  val content: String,
+  // roomId or serverId
+  val contextId: String,
+  var content: String,
 ) : BaseTime() {
   @Id
   var id: String? = null
 
   fun isEdited(): Boolean {
     return !createdAt.equals(updatedAt)
   }
+
+  fun updateContent(content: String) {
+    this.content = content
+  }
 }

chat/src/main/kotlin/kpring/chat/chat/repository/RoomChatRepository.kt

@@ -1,15 +1,15 @@
 package kpring.chat.chat.repository
 
-import kpring.chat.chat.model.RoomChat
+import kpring.chat.chat.model.Chat
 import org.springframework.data.domain.Pageable
 import org.springframework.data.mongodb.repository.MongoRepository
 import org.springframework.data.querydsl.QuerydslPredicateExecutor
 import org.springframework.stereotype.Repository
 
 @Repository
-interface RoomChatRepository : MongoRepository<RoomChat, String>, QuerydslPredicateExecutor<RoomChat> {
-  fun findAllByRoomId(
-    roomId: String,
+interface RoomChatRepository : MongoRepository<Chat, String>, QuerydslPredicateExecutor<Chat> {
+  fun findAllByContextId(
+    contextId: String,
     pageable: Pageable,
-  ): List<RoomChat>
+  ): List<Chat>
 }

chat/src/main/kotlin/kpring/chat/chat/repository/ServerChatRepository.kt

@@ -1,15 +1,15 @@
 package kpring.chat.chat.repository
 
-import kpring.chat.chat.model.ServerChat
+import kpring.chat.chat.model.Chat
 import org.springframework.data.domain.Pageable
 import org.springframework.data.mongodb.repository.MongoRepository
 import org.springframework.data.querydsl.QuerydslPredicateExecutor
 import org.springframework.stereotype.Repository
 
 @Repository
-interface ServerChatRepository : MongoRepository<ServerChat, String>, QuerydslPredicateExecutor<ServerChat> {
-  fun findAllByServerId(
+interface ServerChatRepository : MongoRepository<Chat, String>, QuerydslPredicateExecutor<Chat> {
+  fun findAllByContextId(
     serverId: String,
     pageable: Pageable,
-  ): List<ServerChat>
+  ): List<Chat>
 }

chat/src/main/kotlin/kpring/chat/chat/service/ChatService.kt

@@ -1,13 +1,13 @@
 package kpring.chat.chat.service
 
-import kpring.chat.chat.model.RoomChat
-import kpring.chat.chat.model.ServerChat
+import kpring.chat.chat.model.Chat
 import kpring.chat.chat.repository.RoomChatRepository
 import kpring.chat.chat.repository.ServerChatRepository
 import kpring.chat.chatroom.repository.ChatRoomRepository
 import kpring.chat.global.exception.ErrorCode
 import kpring.chat.global.exception.GlobalException
 import kpring.core.chat.chat.dto.request.CreateChatRequest
+import kpring.core.chat.chat.dto.request.UpdateChatRequest
 import kpring.core.chat.chat.dto.response.ChatResponse
 import kpring.core.server.dto.ServerSimpleInfo
 import org.springframework.beans.factory.annotation.Value
@@ -28,9 +28,9 @@ class ChatService(
   ): Boolean {
     val chat =
       roomChatRepository.save(
-        RoomChat(
+        Chat(
           userId = userId,
-          roomId = request.id,
+          contextId = request.id,
           content = request.content,
         ),
       )
@@ -43,9 +43,9 @@ class ChatService(
   ): Boolean {
     val chat =
       serverChatRepository.save(
-        ServerChat(
+        Chat(
           userId = userId,
-          serverId = request.id,
+          contextId = request.id,
           content = request.content,
         ),
       )
@@ -60,9 +60,9 @@ class ChatService(
     verifyChatRoomAccess(chatRoomId, userId)
 
     val pageable: Pageable = PageRequest.of(page, pageSize)
-    val roomChats: List<RoomChat> = roomChatRepository.findAllByRoomId(chatRoomId, pageable)
+    val roomChats: List<Chat> = roomChatRepository.findAllByContextId(chatRoomId, pageable)
 
-    return convertRoomChatsToResponses(roomChats)
+    return convertChatsToResponses(roomChats)
   }
 
   fun getServerChats(
@@ -74,9 +74,40 @@ class ChatService(
     verifyServerAccess(servers, serverId)
 
     val pageable: Pageable = PageRequest.of(page, pageSize)
-    val chats: List<ServerChat> = serverChatRepository.findAllByServerId(serverId, pageable)
+    val chats: List<Chat> = serverChatRepository.findAllByContextId(serverId, pageable)
 
-    return convertServerChatsToResponses(chats)
+    return convertChatsToResponses(chats)
+  }
+
+  fun updateRoomChat(
+    request: UpdateChatRequest,
+    userId: String,
+  ): Boolean {
+    val chat = roomChatRepository.findById(request.id).orElseThrow { GlobalException(ErrorCode.CHAT_NOT_FOUND) }
+    verifyIfAuthor(userId, chat)
+    chat.updateContent(request.content)
+    roomChatRepository.save(chat)
+    return true
+  }
+
+  fun updateServerChat(
+    request: UpdateChatRequest,
+    userId: String,
+  ): Boolean {
+    val chat = serverChatRepository.findById(request.id).orElseThrow { GlobalException(ErrorCode.CHAT_NOT_FOUND) }
+    verifyIfAuthor(userId, chat)
+    chat.updateContent(request.content)
+    serverChatRepository.save(chat)
+    return true
+  }
+
+  fun verifyIfAuthor(
+    userId: String,
+    chat: Chat,
+  ) {
+    if (userId != chat.userId) {
+      throw GlobalException(ErrorCode.FORBIDDEN_CHAT)
+    }
   }
 
   private fun verifyServerAccess(
@@ -100,15 +131,7 @@ class ChatService(
     }
   }
 
-  private fun convertRoomChatsToResponses(roomChats: List<RoomChat>): List<ChatResponse> {
-    val chatResponse =
-      roomChats.map { chat ->
-        ChatResponse(chat.id!!, chat.isEdited(), chat.createdAt.toString(), chat.content)
-      }
-    return chatResponse
-  }
-
-  private fun convertServerChatsToResponses(chats: List<ServerChat>): List<ChatResponse> {
+  private fun convertChatsToResponses(chats: List<Chat>): List<ChatResponse> {
     val chatResponse =
       chats.map { chat ->
         ChatResponse(chat.id!!, chat.isEdited(), chat.createdAt.toString(), chat.content)

chat/src/main/kotlin/kpring/chat/global/exception/ErrorCode.kt

@@ -9,9 +9,11 @@ enum class ErrorCode(val httpStatus: Int, val message: String) {
   // 403
   FORBIDDEN_CHATROOM(HttpStatus.FORBIDDEN.value(), "접근이 제한된 채팅방 입니다"),
   FORBIDDEN_SERVER(HttpStatus.FORBIDDEN.value(), "접근이 제한된 서버 입니다"),
+  FORBIDDEN_CHAT(HttpStatus.FORBIDDEN.value(), "접근이 제한된 채팅 입니다"),
 
   // 404
   CHATROOM_NOT_FOUND(HttpStatus.NOT_FOUND.value(), "해당 id로 chatroom을 찾을 수 없습니다"),
+  CHAT_NOT_FOUND(HttpStatus.NOT_FOUND.value(), "해당 id로 채팅을 찾을 수 없습니다"),
 
   // 500
   INVITATION_LINK_SAVE_FAILURE(HttpStatus.INTERNAL_SERVER_ERROR.value(), "Invitation Code가 저장되지 않았습니다"),

chat/src/main/resources/application.yml

@@ -1,6 +1,8 @@
 spring:
   application:
     name: chat
+  main:
+    allow-bean-definition-overriding: true
   data:
     mongodb:
       host: localhost
@@ -10,6 +12,8 @@ spring:
       database: mongodb
       authentication-database: admin
       authSource: admin
+      repositories:
+        enabled: true
 
     redis:
       host: localhost