Fix Transactions copy bug (#3581)

Co-authored-by: Priyatanu <[email protected]>
Ivy-Apps · Oct 4, 2024 · 9aafa36 · 9aafa36
1 parent 5c532c9
commit 9aafa36
Showing 1 changed file with 18 additions and 3 deletions.
diff --git a/screen/edit-transaction/src/main/java/com/ivy/transaction/EditTransactionViewModel.kt b/screen/edit-transaction/src/main/java/com/ivy/transaction/EditTransactionViewModel.kt
@@ -182,7 +182,6 @@ class EditTransactionViewModel @Inject constructor(
             transactionAssociatedTags =
                 tagRepository.findByAssociatedId(AssociationId(loadedTransaction().id)).map(Tag::id)
                     .toImmutableList()
-
             display(loadedTransaction!!)
         }
     }
@@ -622,17 +621,24 @@ class EditTransactionViewModel @Inject constructor(
     private fun duplicate() {
         viewModelScope.launch {
             ioThread {
-                loadedTransaction()
+                val id = UUID.randomUUID()
+                 loadedTransaction()
                     .copy(
-                        id = UUID.randomUUID(),
+                        id = id,
                         dateTime = timeProvider.utcNow(),
                     )
                     .toDomain(transactionMapper)
                     ?.let {
                         transactionRepo.save(it)
                     }
+
+                tagRepository.findByIds(transactionAssociatedTags).forEach {
+                    associateTagToTransaction(it, id)
+                }
+
                 refreshWidget(WalletBalanceWidgetReceiver::class.java)
             }
+
             closeScreen()
         }
     }
@@ -930,6 +936,15 @@ class EditTransactionViewModel @Inject constructor(
         }
     }
 
+    private fun associateTagToTransaction(selectedTag: Tag, id: UUID) {
+        viewModelScope.launch(Dispatchers.IO) {
+            val associatedId = AssociationId(id)
+            tagRepository.associateTagToEntity(associatedId, selectedTag.id)
+            transactionAssociatedTags =
+                tagRepository.findByAssociatedId(associatedId).map(Tag::id).toImmutableList()
+        }
+    }
+
     private fun removeTagAssociation(selectedTag: Tag) {
         viewModelScope.launch(Dispatchers.IO) {
             val associatedId = AssociationId(loadedTransaction().id)