feat: Add the transfer to the database if upload finish with success

Infomaniak · Oct 22, 2024 · 2ed95ac · 2ed95ac
1 parent 5cf20ba
commit 2ed95ac
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/...rc/commonMain/kotlin/com/infomaniak/multiplatform_swisstransfer/SwissTransferInjection.kt b/...rc/commonMain/kotlin/com/infomaniak/multiplatform_swisstransfer/SwissTransferInjection.kt
@@ -65,7 +65,7 @@ class SwissTransferInjection {
     val accountManager by lazy { AccountManager(appSettingsController, uploadController, transferController, realmProvider) }
 
     /** A manager used to orchestrate Uploads operations. */
-    val uploadManager by lazy { UploadManager(uploadController, uploadRepository) }
+    val uploadManager by lazy { UploadManager(uploadController, uploadRepository, transferManager) }
 
     /** An utils to help use shared routes  */
     val sharedApiUrlCreator by lazy { SharedApiUrlCreator(transferController, uploadController) }

diff --git a/...rc/commonMain/kotlin/com/infomaniak/multiplatform_swisstransfer/managers/UploadManager.kt b/...rc/commonMain/kotlin/com/infomaniak/multiplatform_swisstransfer/managers/UploadManager.kt
@@ -42,10 +42,12 @@ import kotlin.coroutines.cancellation.CancellationException
  *
  * @property uploadController The controller for managing upload data in the database.
  * @property uploadRepository The repository for interacting with the SwissTransfer API for uploads.
+ * @property transferManager Transfer operations
  */
 class UploadManager(
     private val uploadController: UploadController,
     private val uploadRepository: UploadRepository,
+    private val transferManager: TransferManager,
 ) {
 
     /**
@@ -184,7 +186,7 @@ class UploadManager(
     }
 
     /**
-     * Finishes an upload session.
+     * Finishes an upload session and add the transfer to the database .
      *
      * This method retrieves an upload session from the database using the provided `uuid`.
      * If the session is found and has a remote container UUID, it creates a `FinishUploadBody` object
@@ -217,7 +219,12 @@ class UploadManager(
             language = uploadSession.language.code,
             recipientsEmails = uploadSession.recipientsEmails,
         )
-        uploadRepository.finishUpload(finishUploadBody)
-        uploadController.removeUploadSession(containerUUID)
+        val finishUploadResponse = runCatching {
+            uploadRepository.finishUpload(finishUploadBody).first()
+        }.getOrElse { throw UnknownException(it) }
+        uploadController.removeUploadSession(uploadSession.uuid)
+
+        transferManager.addTransferByLinkUUID(finishUploadResponse.linkUUID)
+        // TODO: If we can't retrieve the transfer cause of the Internet, we should put it in Realm and try again later.
     }
 }