Updated on 2026-08-14

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}" },
+        ) {}
+    }
+}