feat: attach tags on activity sync

Author: jvsena42

app/src/main/java/to/bitkit/data/dao/InvoiceTagDao.kt

@@ -13,6 +13,9 @@ interface InvoiceTagDao {
     @Insert(onConflict = OnConflictStrategy.REPLACE)
     suspend fun saveInvoice(invoiceTag: InvoiceTagEntity)
 
+    @Query("SELECT * FROM invoice_tag")
+    suspend fun getAll(): List<InvoiceTagEntity>
+
     @Query("SELECT * FROM invoice_tag WHERE paymentHash = :paymentHash LIMIT 1")
     suspend fun searchInvoice(paymentHash: String): InvoiceTagEntity?
 

app/src/main/java/to/bitkit/repositories/ActivityRepo.kt

@@ -105,7 +105,7 @@ class ActivityRepo @Inject constructor(
     suspend fun findActivityByPaymentId(
         paymentHashOrTxId: String,
         type: ActivityFilter,
-        txType: PaymentType,
+        txType: PaymentType?,
         retry: Boolean = true,
     ): Result<Activity> = withContext(bgDispatcher) {
         if (paymentHashOrTxId.isEmpty()) {

app/src/main/java/to/bitkit/repositories/WalletRepo.kt

@@ -453,6 +453,12 @@ class WalletRepo @Inject constructor(
         }
     }
 
+    suspend fun getAllInvoiceTags() : Result<List<InvoiceTagEntity>> = withContext(bgDispatcher) {
+        return@withContext runCatching {
+            db.invoiceTagDao().getAll()
+        }
+    }
+
     suspend fun searchInvoice(txId: Txid): Result<InvoiceTagEntity> = withContext(bgDispatcher) {
         return@withContext try {
             val invoiceTag = db.invoiceTagDao().searchInvoice(paymentHash = txId) ?: return@withContext Result.failure(

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

@@ -15,8 +15,9 @@ import kotlinx.coroutines.flow.debounce
 import kotlinx.coroutines.launch
 import kotlinx.coroutines.withContext
 import to.bitkit.di.BgDispatcher
+import to.bitkit.ext.rawId
 import to.bitkit.repositories.ActivityRepo
-import to.bitkit.repositories.LightningRepo
+import to.bitkit.repositories.WalletRepo
 import to.bitkit.services.CoreService
 import to.bitkit.services.LdkNodeEventBus
 import to.bitkit.ui.screens.wallets.activity.components.ActivityTab
@@ -27,9 +28,9 @@ import javax.inject.Inject
 class ActivityListViewModel @Inject constructor(
     @BgDispatcher private val bgDispatcher: CoroutineDispatcher,
     private val coreService: CoreService,
-    private val lightningRepo: LightningRepo,
+    private val walletRepo: WalletRepo,
     private val ldkNodeEventBus: LdkNodeEventBus,
-    private val activityRepo: ActivityRepo
+    private val activityRepo: ActivityRepo,
 ) : ViewModel() {
     private val _filteredActivities = MutableStateFlow<List<Activity>?>(null)
     val filteredActivities = _filteredActivities.asStateFlow()
@@ -224,6 +225,7 @@ class ActivityListViewModel @Inject constructor(
         viewModelScope.launch {
             activityRepo.syncActivities().onSuccess {
                 syncState()
+                syncActivityTags()
             }.onFailure { e ->
                 Logger.error("Failed to sync ldk-node payments", e)
             }
@@ -243,4 +245,25 @@ class ActivityListViewModel @Inject constructor(
             syncState()
         }
     }
+
+    private fun syncActivityTags() {
+        viewModelScope.launch {
+            walletRepo.getAllInvoiceTags().onSuccess { invoiceTags ->
+                invoiceTags.forEach { tagEntity ->
+                    activityRepo.findActivityByPaymentId(
+                        paymentHashOrTxId = tagEntity.paymentHash,
+                        type = ActivityFilter.ALL,
+                        txType = null
+                    ).onSuccess { activity ->
+                        activityRepo.addTagsToActivity(
+                            activityId = activity.rawId(),
+                            tags = tagEntity.tags
+                        ).onSuccess {
+                            walletRepo.deleteInvoice(tagEntity.paymentHash)
+                        }
+                    }
+                }
+            }
+        }
+    }
 }