Updated on 2026-08-14

domain/staking/src/main/java/com/tangem/domain/staking/StakingIdFactory.kt

@@ -2,23 +2,27 @@ package com.tangem.domain.staking
 
 import arrow.core.Either
 import arrow.core.raise.either
+import arrow.core.raise.ensure
 import arrow.core.raise.ensureNotNull
 import com.tangem.domain.models.currency.CryptoCurrency
 import com.tangem.domain.models.network.Network
 import com.tangem.domain.models.wallet.UserWalletId
 import com.tangem.domain.models.staking.StakingID
 import com.tangem.domain.staking.model.StakingIntegrationID
+import com.tangem.domain.staking.toggles.StakingFeatureToggles
 import com.tangem.domain.walletmanager.WalletManagersFacade
 
 /**
  * Factory class for creating instances of [StakingID]
  *
- * @property walletManagersFacade wallet manager facade
+ * @property walletManagersFacade  wallet manager facade
+ * @property stakingFeatureToggles staking feature toggles
  *
 [REDACTED_AUTHOR]
  */
 class StakingIdFactory(
     private val walletManagersFacade: WalletManagersFacade,
+    private val stakingFeatureToggles: StakingFeatureToggles,
 ) {
 
     /**
@@ -72,6 +76,8 @@ class StakingIdFactory(
 
         ensureNotNull(integrationId) { Error.UnsupportedCurrency }
 
+        ensure(stakingFeatureToggles.isIntegrationEnabled(integrationId)) { Error.UnsupportedCurrency }
+
         val address = defaultAddressProvider().takeUnless { it.isNullOrEmpty() }
 
         ensureNotNull(address) { Error.UnableToGetAddress(integrationId = integrationId) }

domain/staking/src/main/java/com/tangem/domain/staking/toggles/StakingFeatureToggles.kt

@@ -1,5 +1,8 @@
 package com.tangem.domain.staking.toggles
 
+import com.tangem.domain.staking.model.StakingIntegrationID
+
 interface StakingFeatureToggles {
-    val isEthStakingEnabled: Boolean
+
+    fun isIntegrationEnabled(integrationId: StakingIntegrationID): Boolean
 }

domain/staking/src/test/kotlin/com/tangem/domain/staking/StakingIdFactoryTest.kt

@@ -10,11 +10,13 @@ import com.tangem.domain.models.currency.CryptoCurrency
 import com.tangem.domain.models.staking.StakingID
 import com.tangem.domain.models.wallet.UserWalletId
 import com.tangem.domain.staking.model.StakingIntegrationID
+import com.tangem.domain.staking.toggles.StakingFeatureToggles
 import com.tangem.domain.walletmanager.WalletManagersFacade
 import com.tangem.test.core.ProvideTestModels
 import io.mockk.clearMocks
 import io.mockk.coEvery
 import io.mockk.coVerify
+import io.mockk.every
 import io.mockk.mockk
 import kotlinx.coroutines.test.runTest
 import org.junit.jupiter.api.BeforeEach
@@ -30,11 +32,16 @@ import org.junit.jupiter.params.ParameterizedTest
 internal class StakingIdFactoryTest {
 
     private val walletManagersFacade: WalletManagersFacade = mockk()
-    private val factory = StakingIdFactory(walletManagersFacade = walletManagersFacade)
+    private val stakingFeatureToggles: StakingFeatureToggles = mockk()
+    private val factory = StakingIdFactory(
+        walletManagersFacade = walletManagersFacade,
+        stakingFeatureToggles = stakingFeatureToggles,
+    )
 
     @BeforeEach
     fun resetMocks() {
-        clearMocks(walletManagersFacade)
+        clearMocks(walletManagersFacade, stakingFeatureToggles)
+        every { stakingFeatureToggles.isIntegrationEnabled(any()) } returns true
     }
 
     @Nested
@@ -66,6 +73,33 @@ internal class StakingIdFactoryTest {
             }
         }
 
+        @Test
+        fun `create returns UnsupportedCurrency if integration is disabled by toggle`() = runTest {
+            // Arrange
+            val userWalletId = UserWalletId(stringValue = "011")
+            val currency = MockCryptoCurrencyFactory().createCoin(Blockchain.TON)
+
+            every {
+                stakingFeatureToggles.isIntegrationEnabled(StakingIntegrationID.StakeKit.Coin.Ton)
+            } returns false
+
+            // Act
+            val actual = factory.create(
+                userWalletId = userWalletId,
+                currencyId = currency.id,
+                network = currency.network,
+            )
+
+            // Assert
+            val expected = StakingIdFactory.Error.UnsupportedCurrency
+
+            Truth.assertThat(actual.leftOrNull()).isEqualTo(expected)
+
+            coVerify(inverse = true) {
+                walletManagersFacade.getDefaultAddress(userWalletId = any(), network = any())
+            }
+        }
+
         @Test
         fun `create returns UnableToGetAddress if address is null`() = runTest {
             // Arrange