[Chat] 채팅방에 초대할 때 사용할 코드 생성 API 구현

chat/src/main/kotlin/kpring/chat/chatroom/repository/UserChatRoomInvitationRepository.kt → chat/src/main/kotlin/kpring/chat/chatroom/repository/InvitationRepository.kt

@@ -1,45 +1,57 @@
 package kpring.chat.chatroom.repository
 
-import kpring.chat.global.config.PropertyConfig
-import kpring.chat.global.exception.ErrorCode
-import kpring.chat.global.exception.GlobalException
-import kpring.core.chat.chat.dto.response.InvitationResponse
+import kpring.chat.global.config.ChatRoomProperty
 import org.springframework.data.redis.core.RedisTemplate
 import org.springframework.data.redis.core.ValueOperations
 import org.springframework.stereotype.Component
-import java.time.LocalDateTime
+import java.nio.charset.StandardCharsets
+import java.security.MessageDigest
 import java.util.*
 
 @Component
-class UserChatRoomInvitationRepository(
+class InvitationRepository(
   private val redisTemplate: RedisTemplate<String, String>,
-  private val propertyConfig: PropertyConfig,
-  private val invitationChatRoomRepository: InvitationChatRoomRepository,
+  private val chatRoomProperty: ChatRoomProperty,
 ) {
-  fun getInvitation(
+  fun getInvitationCode(
     userId: String,
     chatRoomId: String,
-  ): InvitationResponse {
+  ): String {
     val key = generateKey(userId, chatRoomId)
     var value = redisTemplate.opsForValue().get(key)
     if (value == null) {
       value = setInvitation(key, chatRoomId)
     }
-
-    val expiration = redisTemplate.getExpire(key)
-
-    return InvitationResponse(value, LocalDateTime.now().plusSeconds(expiration).toString())
+    return generateCode(key, value)
   }
 
   fun setInvitation(
     key: String,
     chatRoomId: String,
   ): String {
-    val invitationCode = generateCode()
+    val value = generateValue()
     val ops: ValueOperations<String, String> = redisTemplate.opsForValue()
-    ops.set(key, invitationCode, propertyConfig.getExpiration())
-    invitationChatRoomRepository.setInvitationCode(invitationCode, chatRoomId)
-    return invitationCode
+    ops.set(key, value, chatRoomProperty.getExpiration())
+    return value
+  }
+
+  fun getExpiration(
+    userId: String,
+    chatRoomId: String,
+  ): Long {
+    val key = generateKey(userId, chatRoomId)
+    val expiration = redisTemplate.getExpire(key)
+    return expiration
+  }
+
+  fun generateCode(
+    key: String,
+    value: String,
+  ): String {
+    val combine = "$key,$value"
+    val digest = MessageDigest.getInstance("SHA-256")
+    val hash = digest.digest(combine.toByteArray(StandardCharsets.UTF_8))
+    return Base64.getEncoder().encodeToString(hash)
   }
 
   private fun generateKey(
@@ -49,7 +61,7 @@ class UserChatRoomInvitationRepository(
     return "$userId:$chatRoomId"
   }
 
-  private fun generateCode(): String {
+  private fun generateValue(): String {
     return UUID.randomUUID().toString()
   }
 }

chat/src/main/kotlin/kpring/chat/chatroom/service/ChatRoomService.kt

@@ -2,7 +2,7 @@ package kpring.chat.chatroom.service
 
 import kpring.chat.chatroom.model.ChatRoom
 import kpring.chat.chatroom.repository.ChatRoomRepository
-import kpring.chat.chatroom.repository.UserChatRoomInvitationRepository
+import kpring.chat.chatroom.repository.InvitationRepository
 import kpring.chat.global.exception.ErrorCode
 import kpring.chat.global.exception.GlobalException
 import kpring.core.chat.chat.dto.response.InvitationResponse
@@ -12,7 +12,7 @@ import org.springframework.stereotype.Service
 @Service
 class ChatRoomService(
   private val chatRoomRepository: ChatRoomRepository,
-  private val userChatRoomInvitationRepository: UserChatRoomInvitationRepository,
+  private val invitationRepository: InvitationRepository,
 ) {
   fun createChatRoom(
     request: CreateChatRoomRequest,
@@ -38,7 +38,8 @@ class ChatRoomService(
     userId: String,
   ): InvitationResponse {
     verifyChatRoomAccess(chatRoomId, userId)
-    return userChatRoomInvitationRepository.getInvitation(chatRoomId, userId)
+    val code = invitationRepository.getInvitationCode(userId, chatRoomId)
+    return InvitationResponse(code)
   }
 
   fun verifyChatRoomAccess(

chat/src/main/kotlin/kpring/chat/global/config/PropertyConfig.kt → chat/src/main/kotlin/kpring/chat/global/config/ChatRoomProperty.kt

@@ -6,7 +6,7 @@ import java.time.Duration
 
 @Configuration
 @ConfigurationProperties(prefix = "chatroom")
-class PropertyConfig {
+class ChatRoomProperty {
   private lateinit var expiration: Duration
 
   fun getExpiration(): Duration {

chat/src/test/kotlin/kpring/chat/chatroom/ChatRoomServiceTest.kt

@@ -7,7 +7,7 @@ import io.mockk.mockk
 import io.mockk.verify
 import kpring.chat.chatroom.model.ChatRoom
 import kpring.chat.chatroom.repository.ChatRoomRepository
-import kpring.chat.chatroom.repository.UserChatRoomInvitationRepository
+import kpring.chat.chatroom.repository.InvitationRepository
 import kpring.chat.chatroom.service.ChatRoomService
 import kpring.chat.global.ChatRoomTest
 import kpring.chat.global.CommonTest
@@ -17,8 +17,8 @@ import java.util.*
 class ChatRoomServiceTest : FunSpec({
 
   val chatRoomRepository = mockk<ChatRoomRepository>()
-  val userChatRoomInvitationRepository = mockk<UserChatRoomInvitationRepository>()
-  val chatRoomService = ChatRoomService(chatRoomRepository, userChatRoomInvitationRepository)
+  val invitationRepository = mockk<InvitationRepository>()
+  val chatRoomService = ChatRoomService(chatRoomRepository, invitationRepository)
 
   test("createChatRoom 는 새 ChatRoom을 저장해야 한다") {
     // Given

chat/src/test/kotlin/kpring/chat/chatroom/api/v1/ChatRoomControllerTest.kt

@@ -26,7 +26,6 @@ import org.springframework.test.context.junit.jupiter.SpringExtension
 import org.springframework.test.web.reactive.server.WebTestClient
 import org.springframework.test.web.servlet.client.MockMvcWebTestClient
 import org.springframework.web.context.WebApplicationContext
-import java.time.LocalDateTime
 
 @WebMvcTest(controllers = [ChatRoomController::class])
 @ExtendWith(RestDocumentationExtension::class)
@@ -59,8 +58,7 @@ class ChatRoomControllerTest(
         val chatRoomId = ChatRoomTest.TEST_ROOM_ID
         val userId = CommonTest.TEST_USER_ID
         val key = "62e9df6b-13cb-4673-a6fe-8566451b7f15"
-        val expiration = LocalDateTime.now().plusSeconds(3600).toString()
-        val data = InvitationResponse(key, expiration)
+        val data = InvitationResponse(key)
 
         every { authClient.getTokenInfo(any()) } returns
           ApiResponse(
@@ -96,7 +94,6 @@ class ChatRoomControllerTest(
           response {
             body {
               "data.code" type JsonDataType.Strings mean "참여 코드"
-              "data.expiration" type JsonDataType.Strings mean "참여코드 유효시간"
             }
           }
         }

core/src/main/kotlin/kpring/core/chat/chat/dto/request/ChatType.kt

@@ -2,9 +2,10 @@ package kpring.core.chat.chat.dto.request
 
 enum class ChatType(val type: String) {
   Room("Room"),
-  Server("Server");
+  Server("Server"),
+  ;
 
-  override fun toString(): String{
-    return type;
+  override fun toString(): String {
+    return type
   }
 }

core/src/main/kotlin/kpring/core/chat/chat/dto/response/InvitationResponse.kt

@@ -2,5 +2,4 @@ package kpring.core.chat.chat.dto.response
 
 data class InvitationResponse(
   val code: String,
-  val expiration: String,
 )