Updated on 2026-08-14

core/datasource/src/main/java/com/tangem/datasource/di/StakingStoreModule.kt

@@ -1,6 +1,7 @@
 package com.tangem.datasource.di
 
 import android.content.Context
+import androidx.datastore.core.DataStore
 import androidx.datastore.core.DataStoreFactory
 import androidx.datastore.dataStoreFile
 import com.squareup.moshi.Moshi
@@ -49,21 +50,29 @@ internal object StakingStoreModule {
 
     @Provides
     @Singleton
-    fun provideStakingBalanceStore(
+    fun provideYieldsBalancesPersistenceStore(
         @NetworkMoshi moshi: Moshi,
         @ApplicationContext context: Context,
         dispatchers: CoroutineDispatcherProvider,
+    ): DataStore<Map<String, Set<YieldBalanceWrapperDTO>>> {
+        return DataStoreFactory.create(
+            serializer = MoshiDataStoreSerializer(
+                moshi = moshi,
+                types = mapWithStringKeyTypes(valueTypes = setTypes<YieldBalanceWrapperDTO>()),
+                defaultValue = emptyMap(),
+            ),
+            produceFile = { context.dataStoreFile(fileName = "yield_balances") },
+            scope = CoroutineScope(context = dispatchers.io + SupervisorJob()),
+        )
+    }
+
+    @Provides
+    @Singleton
+    fun provideStakingBalanceStore(
+        persistenceStore: DataStore<Map<String, Set<YieldBalanceWrapperDTO>>>,
     ): StakingBalanceStore {
         return DefaultStakingBalanceStore(
-            persistenceStore = DataStoreFactory.create(
+            persistenceStore = persistenceStore,
-                serializer = MoshiDataStoreSerializer(
-                    moshi = moshi,
-                    types = mapWithStringKeyTypes(valueTypes = setTypes<YieldBalanceWrapperDTO>()),
-                    defaultValue = emptyMap(),
-                ),
-                produceFile = { context.dataStoreFile(fileName = "yield_balances") },
-                scope = CoroutineScope(context = dispatchers.io + SupervisorJob()),
-            ),
             runtimeStore = RuntimeSharedStore(),
         )
     }

data/staking/src/main/java/com/tangem/data/staking/di/YieldBalanceFetcherModule.kt

@@ -0,0 +1,24 @@
+package com.tangem.data.staking.di
+
+import com.tangem.data.staking.multi.DefaultMultiYieldBalanceFetcher
+import com.tangem.data.staking.single.DefaultSingleYieldBalanceFetcher
+import com.tangem.domain.staking.multi.MultiYieldBalanceFetcher
+import com.tangem.domain.staking.single.SingleYieldBalanceFetcher
+import dagger.Binds
+import dagger.Module
+import dagger.hilt.InstallIn
+import dagger.hilt.components.SingletonComponent
+import javax.inject.Singleton
+
+@Module
+@InstallIn(SingletonComponent::class)
+internal interface YieldBalanceFetcherModule {
+
+    @Binds
+    @Singleton
+    fun bindSingleYieldBalanceFetcher(impl: DefaultSingleYieldBalanceFetcher): SingleYieldBalanceFetcher
+
+    @Binds
+    @Singleton
+    fun bindMultiYieldBalanceFetcher(impl: DefaultMultiYieldBalanceFetcher): MultiYieldBalanceFetcher
+}

data/staking/src/main/java/com/tangem/data/staking/di/YieldBalanceProducerFactoryModule.kt

@@ -0,0 +1,28 @@
+package com.tangem.data.staking.di
+
+import com.tangem.data.staking.multi.DefaultMultiYieldBalanceProducer
+import com.tangem.data.staking.single.DefaultSingleYieldBalanceProducer
+import com.tangem.domain.staking.multi.MultiYieldBalanceProducer
+import com.tangem.domain.staking.single.SingleYieldBalanceProducer
+import dagger.Binds
+import dagger.Module
+import dagger.hilt.InstallIn
+import dagger.hilt.components.SingletonComponent
+import javax.inject.Singleton
+
+@Module
+@InstallIn(SingletonComponent::class)
+internal interface YieldBalanceProducerFactoryModule {
+
+    @Binds
+    @Singleton
+    fun bindSingleYieldBalanceProducerFactory(
+        impl: DefaultSingleYieldBalanceProducer.Factory,
+    ): SingleYieldBalanceProducer.Factory
+
+    @Binds
+    @Singleton
+    fun bindMultiYieldBalanceProducerFactory(
+        impl: DefaultMultiYieldBalanceProducer.Factory,
+    ): MultiYieldBalanceProducer.Factory
+}

data/staking/src/main/java/com/tangem/data/staking/di/YieldBalanceSupplierModule.kt

@@ -0,0 +1,56 @@
+package com.tangem.data.staking.di
+
+import androidx.datastore.core.DataStore
+import com.tangem.data.staking.store.DefaultYieldsBalancesStore
+import com.tangem.data.staking.store.YieldsBalancesStore
+import com.tangem.datasource.api.stakekit.models.response.model.YieldBalanceWrapperDTO
+import com.tangem.datasource.local.datastore.RuntimeSharedStore
+import com.tangem.domain.staking.multi.MultiYieldBalanceProducer
+import com.tangem.domain.staking.multi.MultiYieldBalanceSupplier
+import com.tangem.domain.staking.single.SingleYieldBalanceProducer
+import com.tangem.domain.staking.single.SingleYieldBalanceSupplier
+import com.tangem.utils.coroutines.CoroutineDispatcherProvider
+import dagger.Module
+import dagger.Provides
+import dagger.hilt.InstallIn
+import dagger.hilt.components.SingletonComponent
+import javax.inject.Singleton
+
+@Module
+@InstallIn(SingletonComponent::class)
+internal object YieldBalanceSupplierModule {
+
+    @Provides
+    @Singleton
+    fun provideYieldsBalancesStore(
+        persistenceStore: DataStore<Map<String, Set<YieldBalanceWrapperDTO>>>,
+        dispatchers: CoroutineDispatcherProvider,
+    ): YieldsBalancesStore {
+        return DefaultYieldsBalancesStore(
+            runtimeStore = RuntimeSharedStore(),
+            persistenceStore = persistenceStore,
+            dispatchers = dispatchers,
+        )
+    }
+
+    @Provides
+    @Singleton
+    fun provideSingleYieldBalanceSupplier(factory: SingleYieldBalanceProducer.Factory): SingleYieldBalanceSupplier {
+        return object : SingleYieldBalanceSupplier(
+            factory = factory,
+            keyCreator = {
+                "single_yield_balance_${it.userWalletId.stringValue}_${it.network.id.value}_" +
+                    it.network.derivationPath.value
+            },
+        ) {}
+    }
+
+    @Provides
+    @Singleton
+    fun provideMultiYieldBalanceSupplier(factory: MultiYieldBalanceProducer.Factory): MultiYieldBalanceSupplier {
+        return object : MultiYieldBalanceSupplier(
+            factory = factory,
+            keyCreator = { "multi_yields_balances_${it.userWalletId.stringValue}" },
+        ) {}
+    }
+}

domain/staking/models/src/main/kotlin/com/tangem/domain/staking/model/stakekit/YieldBalance.kt

@@ -1,6 +1,7 @@
 package com.tangem.domain.staking.model.stakekit
 
 import com.tangem.domain.models.StatusSource
+import com.tangem.domain.staking.model.StakingID
 import com.tangem.domain.staking.model.stakekit.action.StakingActionType
 import org.joda.time.DateTime
 import java.math.BigDecimal
@@ -19,6 +20,15 @@ sealed class YieldBalance {
         }
     }
 
+    fun getStakingId(): StakingID? {
+        val integrationId = integrationId
+        val address = address
+
+        if (integrationId == null || address == null) return null
+
+        return StakingID(integrationId = integrationId, address = address)
+    }
+
     data class Data(
         override val integrationId: String?,
         override val address: String,