feat: attachTagsToActivity in success send flow

Author: jvsena42

app/src/main/java/to/bitkit/viewmodels/AppViewModel.kt

@@ -7,6 +7,7 @@ import androidx.lifecycle.ViewModel
 import androidx.lifecycle.viewModelScope
 import dagger.hilt.android.lifecycle.HiltViewModel
 import kotlinx.coroutines.CompletableDeferred
+import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.delay
 import kotlinx.coroutines.flow.MutableSharedFlow
 import kotlinx.coroutines.flow.MutableStateFlow
@@ -39,8 +40,12 @@ import to.bitkit.ui.components.BottomSheetType
 import to.bitkit.ui.screens.wallets.send.SendRoute
 import to.bitkit.ui.shared.toast.ToastEventBus
 import to.bitkit.utils.Logger
+import uniffi.bitkitcore.Activity
+import uniffi.bitkitcore.ActivityFilter
+import uniffi.bitkitcore.ActivityType
 import uniffi.bitkitcore.LightningInvoice
 import uniffi.bitkitcore.OnChainInvoice
+import uniffi.bitkitcore.PaymentType
 import uniffi.bitkitcore.Scanner
 import javax.inject.Inject
 
@@ -455,6 +460,7 @@ class AppViewModel @Inject constructor(
                     val result = sendOnchain(validatedAddress.address, amount)
                     if (result.isSuccess) {
                         val txId = result.getOrNull()
+                        attachTagsToActivity(paymentHashOrTxId = txId, type = ActivityFilter.ONCHAIN)
                         Logger.info("Onchain send result txid: $txId")
                         setSendEffect(
                             SendEffect.PaymentSuccess(
@@ -482,6 +488,7 @@ class AppViewModel @Inject constructor(
                     if (result.isSuccess) {
                         val paymentHash = result.getOrNull()
                         Logger.info("Lightning send result payment hash: $paymentHash")
+                        attachTagsToActivity(paymentHashOrTxId = paymentHash, type = ActivityFilter.LIGHTNING)
                         setSendEffect(SendEffect.PaymentSuccess())
                         resetSendState()
                     } else {
@@ -493,6 +500,34 @@ class AppViewModel @Inject constructor(
         }
     }
 
+    private fun attachTagsToActivity(paymentHashOrTxId: String?, type: ActivityFilter) {
+        if (_sendUiState.value.selectedTags.isEmpty()) return
+        if (paymentHashOrTxId == null) return
+
+        viewModelScope.launch(Dispatchers.IO) {
+            val activity = coreService.activity.get(filter = type, txType = PaymentType.SENT, limit = 1u).firstOrNull()
+
+            if (activity == null) {
+                Logger.error(msg = "Activity not found")
+                return@launch
+            }
+
+            //TODO COMPARE PAYMENT HASH AND TX ID
+
+            when (activity) {
+                is Activity.Lightning -> coreService.activity.appendTags(
+                    toActivityId = activity.v1.id,
+                    tags = _sendUiState.value.selectedTags
+                )
+
+                is Activity.Onchain -> coreService.activity.appendTags(
+                    toActivityId = activity.v1.id,
+                    tags = _sendUiState.value.selectedTags
+                )
+            }
+        }
+    }
+
     private suspend fun sendOnchain(address: String, amount: ULong): Result<Txid> {
         return runCatching { lightningService.send(address = address, amount) }
             .onFailure {