Use externalMailInfo API call instead of mailboxes one to get ExternalMail info

Author: KevinBoulongne

app/src/main/java/com/infomaniak/mail/data/api/ApiRepository.kt

@@ -46,6 +46,7 @@ import com.infomaniak.mail.data.models.getMessages.ActivitiesResult
 import com.infomaniak.mail.data.models.getMessages.GetMessagesByUidsResult
 import com.infomaniak.mail.data.models.getMessages.NewMessagesResult
 import com.infomaniak.mail.data.models.mailbox.Mailbox
+import com.infomaniak.mail.data.models.mailbox.MailboxExternalMailInfo
 import com.infomaniak.mail.data.models.mailbox.MailboxLinkedResult
 import com.infomaniak.mail.data.models.mailbox.MailboxPermissions
 import com.infomaniak.mail.data.models.message.Message
@@ -145,6 +146,10 @@ object ApiRepository : ApiRepositoryCore() {
         return callApi(ApiRoutes.permissions(mailboxLinkId, mailboxHostingId), GET)
     }
 
+    fun getExternalMailInfo(mailboxHostingId: Int, mailboxName: String): ApiResponse<MailboxExternalMailInfo> {
+        return callApi(ApiRoutes.externalMailInfo(mailboxHostingId, mailboxName), GET)
+    }
+
     fun markMessagesAsSeen(mailboxUuid: String, messagesUids: List<String>): ApiResponse<Unit> {
         return callApi(ApiRoutes.messagesSeen(mailboxUuid), POST, mapOf("uids" to messagesUids))
     }

app/src/main/java/com/infomaniak/mail/data/api/ApiRoutes.kt

@@ -73,6 +73,10 @@ object ApiRoutes {
         return "$MAIL_API/api/securedProxy"
     }
 
+    fun externalMailInfo(mailboxHostingId: Int, mailboxName: String): String {
+        return "${securedProxy()}/1/mail_hostings/$mailboxHostingId/mailboxes/$mailboxName/external_mail_flag"
+    }
+
     fun updateMailboxPassword(mailboxId: Int): String {
         return "${securedProxy()}/cache/invalidation/profile/workspace/mailbox/$mailboxId/update_password"
     }

app/src/main/java/com/infomaniak/mail/data/cache/RealmDatabase.kt

@@ -158,7 +158,7 @@ object RealmDatabase {
 
         //region Configurations versions
         const val USER_INFO_SCHEMA_VERSION = 1L
-        const val MAILBOX_INFO_SCHEMA_VERSION = 4L
+        const val MAILBOX_INFO_SCHEMA_VERSION = 5L
         const val MAILBOX_CONTENT_SCHEMA_VERSION = 8L
         //endregion
 

app/src/main/java/com/infomaniak/mail/data/models/correspondent/Recipient.kt

@@ -64,10 +64,9 @@ open class Recipient : EmbeddedRealmObject, Correspondent {
 
     // Computes if the Recipient is external, according to the required conditions.
     // Does not tell anything about how to display the Recipient chip when composing a new Message.
-    fun isExternal(emailDictionary: MergedContactDictionary, aliases: List<String>): Boolean {
+    fun isExternal(emailDictionary: MergedContactDictionary, aliases: List<String>, trustedDomains: List<String>): Boolean {
         val isUnknownContact = email !in emailDictionary
         val isMailerDaemon = """mailer-daemon@(?:.+\.)?infomaniak\.ch""".toRegex(RegexOption.IGNORE_CASE).matches(email)
-        val trustedDomains = listOf("@infomaniak.com", "@infomaniak.event", "@swisstransfer.com")
         val isUntrustedDomain = email.isEmail() && trustedDomains.none { email.endsWith(it) }
         val isAlias = email in aliases
 

app/src/main/java/com/infomaniak/mail/data/models/mailbox/Mailbox.kt

@@ -62,8 +62,6 @@ class Mailbox : RealmObject {
     @SerialName("unseen_messages")
     var unreadCountRemote: Int = 0
     var aliases: RealmList<String> = realmListOf()
-    @SerialName("external_mail_flag_enabled")
-    var externalMailFlagEnabled: Boolean = false
     //endregion
 
     //region Local data (Transient)
@@ -80,6 +78,10 @@ class Mailbox : RealmObject {
     var permissions: MailboxPermissions? = null
     @Transient
     private var _featureFlags: RealmSet<String> = realmSetOf()
+    @Transient
+    var externalMailFlagEnabled: Boolean = false
+    @Transient
+    var trustedDomains: RealmList<String> = realmListOf()
     //endregion
 
     //region (Transient & Ignore)

app/src/main/java/com/infomaniak/mail/data/models/mailbox/MailboxExternalMailInfo.kt

@@ -0,0 +1,29 @@
+/*
+ * Infomaniak Mail - Android
+ * Copyright (C) 2024 Infomaniak Network SA
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package com.infomaniak.mail.data.models.mailbox
+
+import kotlinx.serialization.SerialName
+import kotlinx.serialization.Serializable
+
+@Serializable
+data class MailboxExternalMailInfo(
+    @SerialName("external_mail_flag_enabled")
+    val externalMailFlagEnabled: Boolean = false,
+    @SerialName("domains")
+    val trustedDomains: List<String> = emptyList(),
+)

app/src/main/java/com/infomaniak/mail/ui/MainViewModel.kt

@@ -63,6 +63,7 @@ import com.infomaniak.mail.utils.Utils.runCatchingRealm
 import com.infomaniak.mail.views.itemViews.AvatarMergedContactData
 import dagger.hilt.android.lifecycle.HiltViewModel
 import io.realm.kotlin.ext.copyFromRealm
+import io.realm.kotlin.ext.toRealmList
 import io.realm.kotlin.notifications.ResultsChange
 import kotlinx.coroutines.*
 import kotlinx.coroutines.flow.*
@@ -280,6 +281,7 @@ class MainViewModel @Inject constructor(
             updatePermissions(mailbox)
             updateSignatures(mailbox)
             updateFeatureFlag(mailbox)
+            updateExternalMailInfo(mailbox)
 
             // This update is blocking because we need it for the rest of the flow : `selectFolder()` needs the Folders.
             updateFolders(mailbox)
@@ -342,6 +344,16 @@ class MainViewModel @Inject constructor(
         sharedUtils.updateAiFeatureFlag(mailbox.objectId, mailbox.uuid)
     }
 
+    private fun updateExternalMailInfo(mailbox: Mailbox) = viewModelScope.launch(ioCoroutineContext) {
+        SentryLog.d(TAG, "Force refresh External Mail info")
+        with(ApiRepository.getExternalMailInfo(mailbox.hostingId, mailbox.mailboxName)) {
+            if (isSuccess() && data != null) mailboxController.updateMailbox(mailbox.objectId) {
+                it.externalMailFlagEnabled = data!!.externalMailFlagEnabled
+                it.trustedDomains = data!!.trustedDomains.toRealmList()
+            }
+        }
+    }
+
     private fun updateFolders(mailbox: Mailbox) {
         SentryLog.d(TAG, "Force refresh Folders")
         ApiRepository.getFolders(mailbox.uuid).data?.let { folders ->

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

@@ -375,6 +375,7 @@ class ThreadFragment : Fragment() {
                 emailDictionary = result.mergedContacts ?: emptyMap(),
                 aliases = result.mailbox?.aliases ?: emptyList(),
                 externalMailFlagEnabled = result.mailbox?.externalMailFlagEnabled ?: false,
+                trustedDomains = result.mailbox?.trustedDomains ?: emptyList(),
             )
 
             threadSubject.text = spannedSubject
@@ -551,11 +552,14 @@ class ThreadFragment : Fragment() {
         emailDictionary: MergedContactDictionary,
         aliases: List<String>,
         externalMailFlagEnabled: Boolean,
+        trustedDomains: List<String>,
     ): Pair<String, CharSequence> = with(binding) {
         val subject = context.formatSubject(thread.subject)
         if (!externalMailFlagEnabled) return subject to subject
 
-        val (externalRecipientEmail, externalRecipientQuantity) = thread.findExternalRecipients(emailDictionary, aliases)
+        val (externalRecipientEmail, externalRecipientQuantity) = thread.findExternalRecipients(
+            emailDictionary, aliases, trustedDomains,
+        )
         if (externalRecipientQuantity == 0) return subject to subject
 
         val spannedSubject = requireContext().postfixWithTag(

app/src/main/java/com/infomaniak/mail/ui/newMessage/NewMessageExternalsManager.kt

@@ -50,27 +50,40 @@ class NewMessageExternalsManager @Inject constructor() : NewMessageManager() {
     fun observeExternals(arrivedFromExistingDraft: Boolean) = with(newMessageViewModel) {
         Utils.waitInitMediator(initResult, mergedContacts).observe(viewLifecycleOwner) { (_, mergedContacts) ->
             val externalMailFlagEnabled = currentMailbox.externalMailFlagEnabled
+            val trustedDomains = currentMailbox.trustedDomains
             val shouldWarnForExternal = externalMailFlagEnabled && !arrivedFromExistingDraft
             val emailDictionary = mergedContacts.second
             val aliases = currentMailbox.aliases
 
-            updateFields(shouldWarnForExternal, emailDictionary, aliases)
-            updateBanner(shouldWarnForExternal, emailDictionary, aliases)
+            updateFields(shouldWarnForExternal, emailDictionary, aliases, trustedDomains)
+            updateBanner(shouldWarnForExternal, emailDictionary, aliases, trustedDomains)
         }
     }
 
-    private fun updateFields(shouldWarnForExternal: Boolean, emailDictionary: MergedContactDictionary, aliases: List<String>) {
+    private fun updateFields(
+        shouldWarnForExternal: Boolean,
+        emailDictionary: MergedContactDictionary,
+        aliases: List<String>,
+        trustedDomains: List<String>,
+    ) {
         with(binding) {
-            toField.updateExternals(shouldWarnForExternal, emailDictionary, aliases)
-            ccField.updateExternals(shouldWarnForExternal, emailDictionary, aliases)
-            bccField.updateExternals(shouldWarnForExternal, emailDictionary, aliases)
+            toField.updateExternals(shouldWarnForExternal, emailDictionary, aliases, trustedDomains)
+            ccField.updateExternals(shouldWarnForExternal, emailDictionary, aliases, trustedDomains)
+            bccField.updateExternals(shouldWarnForExternal, emailDictionary, aliases, trustedDomains)
         }
     }
 
-    private fun updateBanner(shouldWarnForExternal: Boolean, emailDictionary: MergedContactDictionary, aliases: List<String>) {
+    private fun updateBanner(
+        shouldWarnForExternal: Boolean,
+        emailDictionary: MergedContactDictionary,
+        aliases: List<String>,
+        trustedDomains: List<String>,
+    ) {
         with(newMessageViewModel) {
             if (shouldWarnForExternal && !isExternalBannerManuallyClosed) {
-                val (externalEmail, externalQuantity) = draft.findExternalRecipientForNewMessage(aliases, emailDictionary)
+                val (externalEmail, externalQuantity) = draft.findExternalRecipientForNewMessage(
+                    aliases, emailDictionary, trustedDomains,
+                )
                 externalRecipientCount.value = externalEmail to externalQuantity
             }
         }

app/src/main/java/com/infomaniak/mail/ui/newMessage/RecipientFieldView.kt

@@ -419,11 +419,16 @@ class RecipientFieldView @JvmOverloads constructor(
         return (if (recipientCount == 1) recipients.single().email else null) to recipientCount
     }
 
-    fun updateExternals(shouldWarnForExternal: Boolean, emailDictionary: MergedContactDictionary, aliases: List<String>) {
+    fun updateExternals(
+        shouldWarnForExternal: Boolean,
+        emailDictionary: MergedContactDictionary,
+        aliases: List<String>,
+        trustedDomains: List<String>,
+    ) {
         for (recipient in contactChipAdapter.getRecipients()) {
             if (recipient.isManuallyEntered) continue
 
-            val shouldDisplayAsExternal = shouldWarnForExternal && recipient.isExternal(emailDictionary, aliases)
+            val shouldDisplayAsExternal = shouldWarnForExternal && recipient.isExternal(emailDictionary, aliases, trustedDomains)
             recipient.initDisplayAsExternal(shouldDisplayAsExternal)
 
             updateCollapsedChipValues(isSelfCollapsed)

app/src/main/java/com/infomaniak/mail/utils/ExternalUtils.kt

@@ -22,18 +22,22 @@ import com.infomaniak.mail.data.models.draft.Draft
 import com.infomaniak.mail.data.models.thread.Thread
 
 object ExternalUtils {
+
     /**
      * Only returns a quantity of at most 2, used to differentiate between the singular or plural form of the dialog messages
      */
     fun Thread.findExternalRecipients(
         emailDictionary: MergedContactDictionary,
         aliases: List<String>,
+        trustedDomains: List<String>,
     ): Pair<String?, Int> {
         var externalRecipientEmail: String? = null
         var externalRecipientQuantity = 0
 
         messages.forEach { message ->
-            val (singleEmail, quantityForThisMessage) = findExternalRecipientInIterables(emailDictionary, aliases, message.from)
+            val (singleEmail, quantityForThisMessage) = findExternalRecipientInIterables(
+                emailDictionary, aliases, trustedDomains, message.from,
+            )
 
             externalRecipientQuantity += quantityForThisMessage
             if (externalRecipientQuantity > 1) return null to 2
@@ -47,11 +51,12 @@ object ExternalUtils {
     fun Draft.findExternalRecipientForNewMessage(
         aliases: List<String>,
         emailDictionary: MergedContactDictionary,
+        trustedDomains: List<String>,
     ): Pair<String?, Int> {
         val to = to.onlyAutomaticallyAddedOnes()
         val cc = cc.onlyAutomaticallyAddedOnes()
         val bcc = bcc.onlyAutomaticallyAddedOnes()
-        return findExternalRecipientInIterables(emailDictionary, aliases, to, cc, bcc)
+        return findExternalRecipientInIterables(emailDictionary, aliases, trustedDomains, to, cc, bcc)
     }
 
     private fun List<Recipient>.onlyAutomaticallyAddedOnes(): List<Recipient> = filter { !it.isManuallyEntered }
@@ -62,14 +67,15 @@ object ExternalUtils {
     private fun findExternalRecipientInIterables(
         emailDictionary: MergedContactDictionary,
         aliases: List<String>,
+        trustedDomains: List<String>,
         vararg recipientLists: Iterable<Recipient>,
     ): Pair<String?, Int> {
         var externalRecipientEmail: String? = null
         var externalRecipientQuantity = 0
 
         recipientLists.forEach { recipientList ->
             recipientList.forEach { recipient ->
-                if (recipient.isExternal(emailDictionary, aliases)) {
+                if (recipient.isExternal(emailDictionary, aliases, trustedDomains)) {
                     if (externalRecipientQuantity++ == 0) {
                         externalRecipientEmail = recipient.email
                     } else {