Updated on 2026-08-14

features/wallet/impl/src/main/java/com/tangem/feature/wallet/child/wallet/model/WalletsUpdateActionResolver.kt

@@ -12,6 +12,7 @@ import com.tangem.feature.wallet.presentation.wallet.state.model.WalletCardState
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletNotification
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletScreenState
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletType
 import timber.log.Timber
 import javax.inject.Inject
 
@@ -110,7 +111,7 @@ internal class WalletsUpdateActionResolver @Inject constructor(
             when (walletState) {
                 is WalletState.MultiCurrency -> {
                     val wallet = wallets.firstOrNull { it.walletId == walletState.walletCardState.id }
-                    walletState.type == WalletState.MultiCurrency.WalletType.Hot && wallet is UserWallet.Cold
+                    walletState.type == WalletType.Hot && wallet is UserWallet.Cold
                 }
                 else -> false
             }
@@ -212,7 +213,7 @@ internal class WalletsUpdateActionResolver @Inject constructor(
             val previousState = state.wallets.firstOrNull { it.walletCardState.id == wallet.walletId }
                 ?: return@filter false
             wallet is UserWallet.Cold && previousState is WalletState.MultiCurrency &&
-                previousState.type == WalletState.MultiCurrency.WalletType.Hot
+                previousState.type == WalletType.Hot
         }
         return Action.ReinitializeWallets(selectedWallet, walletsToUpdate)
     }

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/common/preview/WalletScreenPreviewData.kt

@@ -186,7 +186,7 @@ internal object WalletScreenPreviewData {
                 onItemClick = { },
             ),
             tangemPayState = TangemPayState.Empty,
-            type = WalletState.MultiCurrency.WalletType.Cold,
+            type = WalletType.Cold,
         )
     }
 
@@ -218,6 +218,7 @@ internal object WalletScreenPreviewData {
             singleWalletLockedState,
             multiWalletState,
         ),
+        wallets2 = persistentListOf(),
         onWalletChange = { _, _ -> },
         event = consumedEvent(),
         isHidingMode = false,

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/WalletStateController.kt

@@ -1,12 +1,10 @@
 package com.tangem.feature.wallet.presentation.wallet.state
 
+import com.tangem.core.ui.DesignFeatureToggles
 import com.tangem.core.ui.components.bottomsheets.TangemBottomSheetConfigContent
 import com.tangem.core.ui.event.consumedEvent
 import com.tangem.domain.models.wallet.UserWalletId
-import com.tangem.feature.wallet.presentation.wallet.state.model.NOT_INITIALIZED_WALLET_INDEX
-import com.tangem.feature.wallet.presentation.wallet.state.model.WalletScreenState
-import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
-import com.tangem.feature.wallet.presentation.wallet.state.model.WalletTopBarConfig
+import com.tangem.feature.wallet.presentation.wallet.state.model.*
 import com.tangem.feature.wallet.presentation.wallet.state.transformers.CloseBottomSheetTransformer
 import com.tangem.feature.wallet.presentation.wallet.state.transformers.OpenBottomSheetTransformer
 import com.tangem.feature.wallet.presentation.wallet.state.transformers.WalletScreenStateTransformer
@@ -25,7 +23,9 @@ import javax.inject.Singleton
 [REDACTED_AUTHOR]
  */
 @Singleton
-internal class WalletStateController @Inject constructor() {
+internal class WalletStateController @Inject constructor(
+    private val designFeatureToggles: DesignFeatureToggles,
+) {
 
     val uiState: StateFlow<WalletScreenState> get() = mutableUiState
 
@@ -53,6 +53,10 @@ internal class WalletStateController @Inject constructor() {
         return value.wallets.firstOrNull { it.walletCardState.id == userWalletId }
     }
 
+    fun getWalletUM(userWalletId: UserWalletId): WalletUM? {
+        return value.wallets2.firstOrNull { it.walletsBalanceUM.id == userWalletId }
+    }
+
     fun getWalletStateIfSelected(walletId: UserWalletId): WalletState? {
         val selectedWalletId = getSelectedWalletId()
 
@@ -61,16 +65,40 @@ internal class WalletStateController @Inject constructor() {
         }
     }
 
+    fun getWalletUMIfSelected(walletId: UserWalletId): WalletUM? {
+        val selectedWalletId = getSelectedWalletId()
+
+        return value.wallets2.firstOrNull {
+            it.walletsBalanceUM.id == walletId && it.walletsBalanceUM.id == selectedWalletId
+        }
+    }
+
     fun getSelectedWallet(): WalletState {
         return with(value) { wallets[selectedWalletIndex] }
     }
 
+    fun getSelectedWalletUM(): WalletUM {
+        return with(value) { wallets2[selectedWalletIndex] }
+    }
+
     fun getSelectedWalletId(): UserWalletId {
-        return with(value) { wallets[selectedWalletIndex].walletCardState.id }
+        return with(value) {
+            if (designFeatureToggles.isRedesignEnabled) {
+                wallets2[selectedWalletIndex].walletsBalanceUM.id
+            } else {
+                wallets[selectedWalletIndex].walletCardState.id
+            }
+        }
     }
 
     fun getWalletIndexByWalletId(userWalletId: UserWalletId): Int? {
-        return with(value) { wallets.indexOfFirstOrNull { it.walletCardState.id == userWalletId } }
+        return with(value) {
+            if (designFeatureToggles.isRedesignEnabled) {
+                wallets2.indexOfFirstOrNull { it.walletsBalanceUM.id == userWalletId }
+            } else {
+                wallets.indexOfFirstOrNull { it.walletCardState.id == userWalletId }
+            }
+        }
     }
 
     fun showBottomSheet(
@@ -105,6 +133,7 @@ internal class WalletStateController @Inject constructor() {
             topBarConfig = WalletTopBarConfig(onDetailsClick = {}),
             selectedWalletIndex = NOT_INITIALIZED_WALLET_INDEX,
             wallets = persistentListOf(),
+            wallets2 = persistentListOf(),
             onWalletChange = { _, _ -> },
             event = consumedEvent(),
             isHidingMode = false,

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/model/WalletNotificationUM.kt

@@ -227,7 +227,7 @@ internal sealed class WalletNotificationUM(val messageUM: TangemMessageUM, val t
         messageUM = TangemMessageUM(
             id = "DemoCardNotification",
             title = resourceReference(id = R.string.warning_demo_mode_title),
-            subtitle = resourceReference(id = R.string.warning_demo_mode_title),
+            subtitle = resourceReference(id = R.string.warning_demo_mode_message),
             messageEffect = TangemMessageEffect.None,
         ),
         type = WalletNotificationType.Warning,
@@ -401,11 +401,6 @@ internal sealed class WalletNotificationUM(val messageUM: TangemMessageUM, val t
                     type = TangemButtonType.PrimaryInverse,
                     onClick = onCloseClick,
                 ),
-                TangemMessageButtonUM(
-                    text = resourceReference(R.string.notification_one_plus_one_button),
-                    type = TangemButtonType.Primary,
-                    onClick = onClick,
-                ),
             ),
         ),
         type = WalletNotificationType.Promo,

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/model/WalletScreenState.kt

@@ -9,6 +9,7 @@ internal data class WalletScreenState(
     val topBarConfig: WalletTopBarConfig,
     val selectedWalletIndex: Int,
     val wallets: ImmutableList<WalletState>,
+    val wallets2: ImmutableList<WalletUM>,
     val onWalletChange: (index: Int, onlyState: Boolean) -> Unit,
     val event: StateEvent<WalletEvent>,
     val isHidingMode: Boolean,

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/CloseBottomSheetTransformer.kt

@@ -2,6 +2,7 @@ package com.tangem.feature.wallet.presentation.wallet.state.transformers
 
 import com.tangem.domain.models.wallet.UserWalletId
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 
 internal class CloseBottomSheetTransformer(userWalletId: UserWalletId) : WalletStateTransformer(userWalletId) {
 
@@ -22,6 +23,10 @@ internal class CloseBottomSheetTransformer(userWalletId: UserWalletId) : WalletS
         }
     }
 
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
+
     private fun updateConfig(prevState: WalletState) = prevState.bottomSheetConfig?.copy(
         isShown = false,
     )

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/InitializeWalletsTransformer.kt

@@ -7,7 +7,6 @@ import com.tangem.feature.wallet.child.wallet.model.intents.WalletClickIntents
 import com.tangem.feature.wallet.presentation.wallet.domain.WalletAdditionalInfoFactory
 import com.tangem.feature.wallet.presentation.wallet.domain.WalletImageResolver
 import com.tangem.feature.wallet.presentation.wallet.state.model.*
-import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState.MultiCurrency.WalletType
 import com.tangem.feature.wallet.presentation.wallet.state.utils.WalletLoadingStateFactory
 import com.tangem.feature.wallet.presentation.wallet.state.utils.createStateByWalletType
 import kotlinx.collections.immutable.PersistentList

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/OpenBottomSheetTransformer.kt

@@ -4,6 +4,7 @@ import com.tangem.core.ui.components.bottomsheets.TangemBottomSheetConfig
 import com.tangem.core.ui.components.bottomsheets.TangemBottomSheetConfigContent
 import com.tangem.domain.models.wallet.UserWalletId
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 
 internal class OpenBottomSheetTransformer(
     userWalletId: UserWalletId,
@@ -28,6 +29,10 @@ internal class OpenBottomSheetTransformer(
         }
     }
 
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
+
     private fun updateConfig() = TangemBottomSheetConfig(
         isShown = true,
         onDismissRequest = onDismissBottomSheet,

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/ReinitializeWalletTransformer.kt

@@ -4,6 +4,7 @@ import com.tangem.domain.models.wallet.UserWallet
 import com.tangem.feature.wallet.child.wallet.model.intents.WalletClickIntents
 import com.tangem.feature.wallet.presentation.wallet.domain.WalletImageResolver
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 import com.tangem.feature.wallet.presentation.wallet.state.utils.WalletLoadingStateFactory
 
 /**
@@ -26,6 +27,10 @@ internal class ReinitializeWalletTransformer(
         )
     }
 
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
+
     override fun transform(prevState: WalletState): WalletState {
         return walletLoadingStateFactory.create(
             userWallet = userWallet,

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/RemoveNFTCollectionsTransformer.kt

@@ -3,6 +3,7 @@ package com.tangem.feature.wallet.presentation.wallet.state.transformers
 import com.tangem.domain.models.wallet.UserWalletId
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletNFTItemUM
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 
 internal class RemoveNFTCollectionsTransformer(
     userWalletId: UserWalletId,
@@ -17,4 +18,8 @@ internal class RemoveNFTCollectionsTransformer(
         is WalletState.SingleCurrency.Locked,
         -> prevState
     }
+
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
 }

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/SetCryptoCurrencyActionsTransformer.kt

@@ -8,6 +8,7 @@ import com.tangem.domain.tokens.model.TokenActionsState
 import com.tangem.feature.wallet.child.wallet.model.intents.WalletClickIntents
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletManageButton
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 import kotlinx.collections.immutable.PersistentList
 import kotlinx.collections.immutable.toPersistentList
 import timber.log.Timber
@@ -35,6 +36,10 @@ internal class SetCryptoCurrencyActionsTransformer(
         }
     }
 
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
+
     private fun TokenActionsState.toManageButtons(): PersistentList<WalletManageButton> {
         return states
             .filterIfS2C()

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/SetExpressStatusesTransformer.kt

@@ -10,6 +10,7 @@ import com.tangem.domain.models.wallet.UserWalletId
 import com.tangem.domain.onramp.model.cache.OnrampTransaction
 import com.tangem.feature.wallet.child.wallet.model.intents.WalletClickIntents
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 import com.tangem.feature.wallet.presentation.wallet.state.transformers.converter.SingleWalletOnrampTransactionConverter
 import kotlinx.collections.immutable.toPersistentList
 import timber.log.Timber
@@ -56,6 +57,10 @@ internal class SetExpressStatusesTransformer(
         }
     }
 
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
+
     private fun TangemBottomSheetConfig.updateStateWithExpressStatusBottomSheet(
         expressState: ExpressTransactionStateUM?,
     ): TangemBottomSheetConfig {

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/SetNFTCollectionsTransformer.kt

@@ -6,6 +6,7 @@ import com.tangem.domain.nft.models.NFTCollections
 import com.tangem.domain.nft.models.allLoadedCollectionsEmpty
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletNFTItemUM
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 import kotlinx.collections.immutable.toPersistentList
 
 internal class SetNFTCollectionsTransformer(
@@ -28,6 +29,10 @@ internal class SetNFTCollectionsTransformer(
         -> prevState
     }
 
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
+
     private fun createContentNFTItemUM(onItemClick: () -> Unit): WalletNFTItemUM.Content {
         val collectionsContent = nftCollections
             .map { it.content }

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/SetPrimaryCurrencyTransformer.kt

@@ -6,6 +6,7 @@ import com.tangem.domain.models.currency.CryptoCurrencyStatus
 import com.tangem.domain.models.wallet.UserWallet
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletCardState
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 import com.tangem.feature.wallet.presentation.wallet.state.transformers.converter.SingleWalletCardStateConverter
 import com.tangem.feature.wallet.presentation.wallet.state.transformers.converter.SingleWalletMarketPriceConverter
 import timber.log.Timber
@@ -35,6 +36,10 @@ internal class SetPrimaryCurrencyTransformer(
         }
     }
 
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
+
     private fun WalletCardState.toLoadedSingleCurrencyState(): WalletCardState {
         return SingleWalletCardStateConverter(status.value, userWallet, appCurrency).convert(value = this)
     }

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/SetRefreshStateTransformer.kt

@@ -5,6 +5,7 @@ import com.tangem.domain.models.wallet.UserWalletId
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletManageButton
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletTokensListState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 import kotlinx.collections.immutable.PersistentList
 import kotlinx.collections.immutable.mutate
 
@@ -33,6 +34,10 @@ internal class SetRefreshStateTransformer(
         }
     }
 
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
+
     private fun PullToRefreshConfig.toUpdatedState(isRefreshing: Boolean): PullToRefreshConfig {
         return copy(isRefreshing = isRefreshing)
     }

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/SetTokenListErrorTransformer.kt

@@ -10,6 +10,7 @@ import com.tangem.feature.wallet.presentation.wallet.domain.WalletAdditionalInfo
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletCardState
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletTokensListState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 import com.tangem.feature.wallet.presentation.wallet.state.utils.disableButtons
 import timber.log.Timber
 import java.math.BigDecimal
@@ -51,6 +52,10 @@ internal class SetTokenListErrorTransformer(
         }
     }
 
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
+
     private fun WalletCardState.toLoadedState(): WalletCardState {
         return WalletCardState.Content(
             id = id,

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/SetTokenListTransformer.kt

@@ -8,6 +8,7 @@ import com.tangem.feature.wallet.child.wallet.model.intents.WalletClickIntents
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletCardState
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletTokensListState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 import com.tangem.feature.wallet.presentation.wallet.state.transformers.converter.MultiWalletCardStateConverter
 import com.tangem.feature.wallet.presentation.wallet.state.transformers.converter.TokenListStateConverter
 import com.tangem.feature.wallet.presentation.wallet.state.utils.enableButtons
@@ -45,6 +46,10 @@ internal class SetTokenListTransformer(
         }
     }
 
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
+
     private fun WalletCardState.toLoadedState(): WalletCardState {
         val fiatBalance = when (params) {
             is TokenConverterParams.Account -> params.accountList.totalFiatBalance

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/SetTxHistoryCountErrorTransformer.kt

@@ -9,6 +9,7 @@ import com.tangem.domain.models.wallet.UserWallet
 import com.tangem.domain.txhistory.models.TxHistoryStateError
 import com.tangem.feature.wallet.child.wallet.model.intents.WalletClickIntents
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 import com.tangem.feature.wallet.presentation.wallet.state.transformers.converter.TxHistoryItemStateConverter
 import kotlinx.collections.immutable.toImmutableList
 import timber.log.Timber
@@ -35,6 +36,10 @@ internal class SetTxHistoryCountErrorTransformer(
         )
     }
 
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
+
     override fun transform(prevState: WalletState): WalletState {
         return when (prevState) {
             is WalletState.SingleCurrency.Content -> prevState.copy(txHistoryState = createErrorState())

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/SetTxHistoryCountTransformer.kt

@@ -6,6 +6,7 @@ import com.tangem.core.ui.components.transactions.state.TxHistoryState
 import com.tangem.domain.models.wallet.UserWalletId
 import com.tangem.feature.wallet.child.wallet.model.intents.WalletClickIntents
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 import kotlinx.coroutines.flow.MutableStateFlow
 import kotlinx.coroutines.flow.update
 import timber.log.Timber
@@ -33,6 +34,10 @@ internal class SetTxHistoryCountTransformer(
         }
     }
 
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
+
     private fun TxHistoryState.toLoadingState(): TxHistoryState {
         return if (this is TxHistoryState.Content) {
             Timber.d("Load transactions history: $transactionsCount")

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/SetTxHistoryItemsErrorTransformer.kt

@@ -5,6 +5,7 @@ import com.tangem.domain.models.wallet.UserWalletId
 import com.tangem.domain.txhistory.models.TxHistoryListError
 import com.tangem.feature.wallet.child.wallet.model.intents.WalletClickIntents
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 import timber.log.Timber
 
 internal class SetTxHistoryItemsErrorTransformer(
@@ -27,6 +28,10 @@ internal class SetTxHistoryItemsErrorTransformer(
         }
     }
 
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
+
     private fun createErrorState(): TxHistoryState.Error = when (error) {
         is TxHistoryListError.DataError -> {
             TxHistoryState.Error(

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/SetTxHistoryItemsTransformer.kt

@@ -6,6 +6,7 @@ import com.tangem.core.ui.components.transactions.state.TxHistoryState
 import com.tangem.domain.models.wallet.UserWallet
 import com.tangem.feature.wallet.child.wallet.model.intents.WalletClickIntents
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 import com.tangem.feature.wallet.presentation.wallet.state.transformers.converter.TxHistoryItemFlowConverter
 import kotlinx.coroutines.flow.Flow
 import timber.log.Timber
@@ -32,6 +33,10 @@ internal class SetTxHistoryItemsTransformer(
         }
     }
 
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
+
     private fun TxHistoryState.toContentState(): TxHistoryState {
         val converter = TxHistoryItemFlowConverter(
             currentState = this,

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/SetWarningsTransformer.kt

@@ -3,6 +3,7 @@ package com.tangem.feature.wallet.presentation.wallet.state.transformers
 import com.tangem.domain.models.wallet.UserWalletId
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletNotification
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 import kotlinx.collections.immutable.ImmutableList
 import timber.log.Timber
 
@@ -23,4 +24,8 @@ internal class SetWarningsTransformer(
             }
         }
     }
+
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
 }

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/TangemPayExposedDeviceTransformer.kt

@@ -3,6 +3,7 @@ package com.tangem.feature.wallet.presentation.wallet.state.transformers
 import com.tangem.domain.models.wallet.UserWalletId
 import com.tangem.feature.wallet.presentation.wallet.state.model.TangemPayState
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 
 internal class TangemPayExposedDeviceTransformer(
     userWalletId: UserWalletId,
@@ -14,4 +15,8 @@ internal class TangemPayExposedDeviceTransformer(
             prevState
         }
     }
+
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
 }

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/TangemPayHiddenStateTransformer.kt

@@ -3,6 +3,7 @@ package com.tangem.feature.wallet.presentation.wallet.state.transformers
 import com.tangem.domain.models.wallet.UserWalletId
 import com.tangem.feature.wallet.presentation.wallet.state.model.TangemPayState
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 
 internal class TangemPayHiddenStateTransformer(
     userWalletId: UserWalletId,
@@ -15,4 +16,8 @@ internal class TangemPayHiddenStateTransformer(
             prevState
         }
     }
+
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
 }

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/TangemPayHideOnboardingStateTransformer.kt

@@ -3,6 +3,7 @@ package com.tangem.feature.wallet.presentation.wallet.state.transformers
 import com.tangem.domain.models.wallet.UserWalletId
 import com.tangem.feature.wallet.presentation.wallet.state.model.TangemPayState
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 
 internal class TangemPayHideOnboardingStateTransformer(
     userWalletId: UserWalletId,
@@ -15,4 +16,8 @@ internal class TangemPayHideOnboardingStateTransformer(
             prevState
         }
     }
+
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
 }

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/TangemPayLoadingStateTransformer.kt

@@ -3,6 +3,7 @@ package com.tangem.feature.wallet.presentation.wallet.state.transformers
 import com.tangem.domain.models.wallet.UserWalletId
 import com.tangem.feature.wallet.presentation.wallet.state.model.TangemPayState
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 
 internal class TangemPayLoadingStateTransformer(userWalletId: UserWalletId) : WalletStateTransformer(userWalletId) {
     override fun transform(prevState: WalletState): WalletState {
@@ -12,4 +13,8 @@ internal class TangemPayLoadingStateTransformer(userWalletId: UserWalletId) : Wa
             prevState
         }
     }
+
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
 }

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/TangemPayOnboardingBannerStateTransformer.kt

@@ -3,6 +3,7 @@ package com.tangem.feature.wallet.presentation.wallet.state.transformers
 import com.tangem.domain.models.wallet.UserWalletId
 import com.tangem.feature.wallet.presentation.wallet.state.model.TangemPayState
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 
 internal class TangemPayOnboardingBannerStateTransformer(
     userWalletId: UserWalletId,
@@ -22,4 +23,8 @@ internal class TangemPayOnboardingBannerStateTransformer(
             prevState
         }
     }
+
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
 }

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/TangemPayRefreshNeededStateTransformer.kt

@@ -7,6 +7,7 @@ import com.tangem.feature.wallet.impl.R
 import com.tangem.feature.wallet.presentation.wallet.state.model.TangemPayState
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletNotification.Warning.TangemPayRefreshNeeded
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 
 internal class TangemPayRefreshNeededStateTransformer(
     userWalletId: UserWalletId,
@@ -32,4 +33,8 @@ internal class TangemPayRefreshNeededStateTransformer(
             prevState
         }
     }
+
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
 }

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/TangemPayRefreshShowProgressTransformer.kt

@@ -4,6 +4,7 @@ import com.tangem.domain.models.wallet.UserWalletId
 import com.tangem.feature.wallet.presentation.wallet.state.model.TangemPayState
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletNotification
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 
 internal class TangemPayRefreshShowProgressTransformer(
     userWalletId: UserWalletId,
@@ -21,4 +22,8 @@ internal class TangemPayRefreshShowProgressTransformer(
             ),
         )
     }
+
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
 }

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/TangemPayUnavailableStateTransformer.kt

@@ -4,6 +4,7 @@ import com.tangem.domain.models.wallet.UserWalletId
 import com.tangem.feature.wallet.presentation.wallet.state.model.TangemPayState
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletNotification
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 
 internal class TangemPayUnavailableStateTransformer(
     userWalletId: UserWalletId,
@@ -20,4 +21,8 @@ internal class TangemPayUnavailableStateTransformer(
             prevState
         }
     }
+
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
 }

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/TangemPayUpdateInfoStateTransformer.kt

@@ -18,6 +18,7 @@ import com.tangem.feature.wallet.presentation.wallet.state.model.TangemPayState.
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
 import com.tangem.domain.pay.model.CustomerInfo.KycStatus.APPROVED
 import com.tangem.domain.pay.model.CustomerInfo
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 import java.util.Currency
 
 /**
@@ -42,6 +43,10 @@ internal class TangemPayUpdateInfoStateTransformer(
         }
     }
 
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
+
     private fun createInitialState(): TangemPayState {
         val cardInfo = value.info.cardInfo
         val productInstance = value.info.productInstance

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/UpdateMultiWalletActionButtonBadgeTransformer.kt

@@ -2,6 +2,7 @@ package com.tangem.feature.wallet.presentation.wallet.state.transformers
 
 import com.tangem.domain.models.wallet.UserWalletId
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 import com.tangem.feature.wallet.presentation.wallet.state.utils.showSwapBadge
 
 internal class UpdateMultiWalletActionButtonBadgeTransformer(
@@ -16,4 +17,8 @@ internal class UpdateMultiWalletActionButtonBadgeTransformer(
             else -> prevState
         }
     }
+
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
 }

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/UpdateWalletCardsCountTransformer.kt

@@ -6,6 +6,7 @@ import com.tangem.feature.wallet.presentation.wallet.domain.WalletAdditionalInfo
 import com.tangem.feature.wallet.presentation.wallet.domain.WalletImageResolver
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletCardState
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 import timber.log.Timber
 
 internal class UpdateWalletCardsCountTransformer(
@@ -30,6 +31,10 @@ internal class UpdateWalletCardsCountTransformer(
         }
     }
 
+    override fun transform(walletUM: WalletUM): WalletUM {
+        return walletUM // todo redesign main
+    }
+
     private fun WalletCardState.toUpdatedState(): WalletCardState {
         return when (this) {
             is WalletCardState.Content -> copy(

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/transformers/WalletStateTransformer.kt

@@ -3,6 +3,7 @@ package com.tangem.feature.wallet.presentation.wallet.state.transformers
 import com.tangem.domain.models.wallet.UserWalletId
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletScreenState
 import com.tangem.feature.wallet.presentation.wallet.state.model.WalletState
+import com.tangem.feature.wallet.presentation.wallet.state.model.WalletUM
 import kotlinx.collections.immutable.toImmutableList
 
 internal abstract class WalletStateTransformer(
@@ -11,6 +12,8 @@ internal abstract class WalletStateTransformer(
 
     abstract fun transform(prevState: WalletState): WalletState
 
+    abstract fun transform(walletUM: WalletUM): WalletUM
+
     final override fun transform(prevState: WalletScreenState): WalletScreenState {
         return prevState.copy(
             wallets = prevState.wallets
@@ -18,6 +21,11 @@ internal abstract class WalletStateTransformer(
                     if (state.walletCardState.id == userWalletId) transform(state) else state
                 }
                 .toImmutableList(),
+            wallets2 = prevState.wallets2
+                .map { walletUM ->
+                    if (walletUM.walletsBalanceUM.id == userWalletId) transform(walletUM) else walletUM
+                }
+                .toImmutableList(),
         )
     }
 }

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/state/utils/WalletLoadingStateFactory.kt

@@ -52,7 +52,7 @@ internal class WalletLoadingStateFactory(
             bottomSheetConfig = null,
             tokensListState = WalletTokensListState.ContentState.Loading,
             nftState = WalletNFTItemUM.Hidden,
-            type = WalletState.MultiCurrency.WalletType.Hot,
+            type = WalletType.Hot,
             tangemPayState = TangemPayState.Empty,
         )
     }
@@ -66,7 +66,7 @@ internal class WalletLoadingStateFactory(
             bottomSheetConfig = null,
             tokensListState = WalletTokensListState.ContentState.Loading,
             nftState = WalletNFTItemUM.Hidden,
-            type = WalletState.MultiCurrency.WalletType.Cold,
+            type = WalletType.Cold,
             tangemPayState = TangemPayState.Empty,
         )
     }