Refactor to simplify code in GenerativeChatRepository

Author: jja08111

feature/chat/data/src/main/java/io/jja08111/gemini/feature/chat/data/repository/GenerativeChatRepository.kt

@@ -121,18 +121,16 @@ class GenerativeChatRepository @Inject constructor(
     return runCatching {
       val model = generativeModel ?: throw NotJoinedException()
       val roomId = joinedRoomId ?: throw NotJoinedException()
-      val messageGroupStream = chatLocalDataSource.getMessageGroupStream(roomId)
-      val messageGroups = messageGroupStream.first()
+      val messageGroups = chatLocalDataSource.getMessageGroupsBy(roomId)
       val lastMessageGroup = messageGroups.lastOrNull() ?: throw EmptyMessageGroupsException()
       val lastPrompt = lastMessageGroup.prompt
-      val lastPromptId = lastPrompt.id
       val responseTextBuilders = List(CANDIDATE_COUNT) { ResponseTextBuilder() }
 
       chatLocalDataSource.insertResponsesAndRemoveError(
         newResponseIds = responseTextBuilders.map { it.id },
         errorResponseId = lastMessageGroup.selectedResponse.id,
         roomId = roomId,
-        promptId = lastPromptId,
+        promptId = lastPrompt.id,
       )
 
       return model.generateTextMessageStream(
@@ -141,7 +139,7 @@ class GenerativeChatRepository @Inject constructor(
         history = messageGroups
           .dropLast(1)
           .flatMap(MessageGroup::toContents),
-        promptId = lastPromptId,
+        promptId = lastPrompt.id,
         responseTextBuilders = responseTextBuilders,
       )
     }
@@ -151,26 +149,24 @@ class GenerativeChatRepository @Inject constructor(
     return runCatching {
       val model = generativeModel ?: throw NotJoinedException()
       val roomId = joinedRoomId ?: throw NotJoinedException()
-      val messageGroupStream = chatLocalDataSource.getMessageGroupStream(roomId)
-      val messageGroups = messageGroupStream.first()
+      val messageGroups = chatLocalDataSource.getMessageGroupsBy(roomId)
       val messageGroup = messageGroups.firstOrNull {
         it.selectedResponse.id == responseId
       } ?: throw EmptyMessageGroupsException()
       val prompt = messageGroup.prompt
-      val promptId = prompt.id
       val responseTextBuilders = List(CANDIDATE_COUNT) { ResponseTextBuilder() }
 
       chatLocalDataSource.insertAndUnselectOldResponses(
         newResponseIds = responseTextBuilders.map { it.id },
         roomId = roomId,
-        promptId = promptId,
+        promptId = prompt.id,
       )
 
       return model.generateTextMessageStream(
         message = prompt.text,
         images = prompt.images.map { promptImageLocalDataSource.loadImage(it.path) },
         history = messageGroups.flatMap(MessageGroup::toContents),
-        promptId = promptId,
+        promptId = prompt.id,
         responseTextBuilders = responseTextBuilders,
       )
     }

feature/chat/data/src/main/java/io/jja08111/gemini/feature/chat/data/source/ChatLocalDataSource.kt

@@ -22,6 +22,7 @@ import kotlinx.coroutines.async
 import kotlinx.coroutines.awaitAll
 import kotlinx.coroutines.coroutineScope
 import kotlinx.coroutines.flow.Flow
+import kotlinx.coroutines.flow.first
 import kotlinx.coroutines.flow.map
 import kotlinx.coroutines.flow.mapLatest
 import java.time.LocalDateTime
@@ -45,6 +46,11 @@ class ChatLocalDataSource @Inject constructor(
     return messageDao.getPromptWithResponsesAndImages(roomId).mapLatest(::convertToMessageGroups)
   }
 
+  suspend fun getMessageGroupsBy(roomId: String): List<MessageGroup> {
+    val messageGroupStream = getMessageGroupStream(roomId)
+    return messageGroupStream.first()
+  }
+
   suspend fun updateResponseContentPartials(partials: List<ModelResponseContentPartial>) {
     messageDao.updateAll(partials)
   }