feat: Save summary in realm

Author: Elouan1411

app/src/main/java/com/infomaniak/mail/data/models/message/Body.kt

@@ -33,7 +33,10 @@ class Body : EmbeddedRealmObject {
     @Serializable(FlatteningSubBodiesSerializer::class)
     @SerialName("subBody")
     var subBodies = realmListOf<SubBody>()
+    var summary: String? = null
     //endregion
 
+    val hasSummary: Boolean get() = summary != null
+
     companion object
 }

app/src/main/java/com/infomaniak/mail/ui/main/thread/ThreadAdapter.kt

@@ -541,6 +541,7 @@ class ThreadAdapter(
             closeButton.setOnClickListener {
                 threadAdapterState.aiSummaryStateMap.remove(messageUid)
                 root.isVisible = false
+                threadAdapterCallbacks?.onAiSummaryClose?.invoke(messageUid)
             }
 
             informationButton.setOnClickListener {
@@ -1200,6 +1201,7 @@ class ThreadAdapter(
         var onAddEmoji: ((emoji: String, messageUid: String) -> Unit)? = null,
         var showEmojiDetails: ((messageUid: String, emoji: String) -> Unit)? = null,
         var onAiSummaryRetry: ((messageUid: String) -> Unit)? = null,
+        var onAiSummaryClose: ((messageUid: String) -> Unit)? = null,
         var showSnackbarRetry: (() -> Unit)? = null,
     )
 

app/src/main/java/com/infomaniak/mail/ui/main/thread/ThreadFragment.kt

@@ -446,6 +446,7 @@ class ThreadFragment : Fragment(), PickerEmojiObserver {
                     }
                 },
                 onAiSummaryRetry = { messageUid -> doAiAction(messageUid, AiAction.SUMMARY) },
+                onAiSummaryClose = { messageUid -> threadViewModel.removeSummary(messageUid)},
                 showSnackbarRetry = ::createSnackBarRetry,
             ),
         )
@@ -1026,6 +1027,12 @@ class ThreadFragment : Fragment(), PickerEmojiObserver {
         val wasLoaderShown = currentState is AiProcessState.Retrying && currentState.isLoaderVisible
 
         val finalState = mapApiResultToState(result, isRetry, wasLoaderShown)
+
+        val summaryContent = result.data
+        if (summaryContent != null) {
+            threadViewModel.saveSummary(messageUid,summaryContent)
+        }
+
         updateAiProcessState(messageUid, aiAction, finalState)
     }
 

app/src/main/java/com/infomaniak/mail/ui/main/thread/ThreadViewModel.kt

@@ -409,12 +409,35 @@ class ThreadViewModel @Inject constructor(
                 val isNotAlreadySplit = !threadState.cachedSplitBodies.contains(message.uid)
                 if (isNotAlreadySplit) threadState.cachedSplitBodies[message.uid] = MessageBodyUtils.splitContentAndQuote(it)
                 splitBody = threadState.cachedSplitBodies[message.uid]
+
+                it.summary?.let { summaryText ->
+                    if (threadState.aiSummaryStateMap[message.uid] == null) {
+                        threadState.aiSummaryStateMap[message.uid] =
+                            AiProcessState.Success(content = summaryText)
+                    }
+                }
             }
         }
 
         return@withContext message
     }
 
+    fun saveSummary(messageUid: String, summaryText: String) = viewModelScope.launch(ioCoroutineContext) {
+        mailboxContentRealm().write {
+            MessageController.updateMessageBlocking(messageUid, realm = this) { localMessage ->
+                localMessage?.body?.summary = summaryText
+            }
+        }
+    }
+
+    fun removeSummary(messageUid: String) = viewModelScope.launch(ioCoroutineContext) {
+        mailboxContentRealm().write {
+            MessageController.updateMessageBlocking(messageUid, realm = this) { localMessage ->
+                localMessage?.body?.summary = null
+            }
+        }
+    }
+
     private fun markThreadAsSeen(thread: Thread) = viewModelScope.launch(ioCoroutineContext) {
         sharedUtils.markAsSeen(mailbox(), listOf(thread))
     }