Updated on 2026-08-14

features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/domain/WalletImageResolver.kt

@@ -19,11 +19,15 @@ internal object WalletImageResolver {
     private const val WALLET_WITH_TWO_BACKUPS_COUNT = 3
 
     /** Get a specified wallet [userWallet] image */
+    @Suppress("CyclomaticComplexMethod")
     @DrawableRes
     fun resolve(userWallet: UserWallet): Int? {
         val cardTypesResolver = userWallet.scanResponse.cardTypesResolver
         return when {
             cardTypesResolver.isDevKit() -> R.drawable.ill_dev_120_106
+            cardTypesResolver.isWhiteWallet2() -> userWallet.resolveWhiteWallet2()
+            cardTypesResolver.isAvroraWallet() -> userWallet.resolveAvroraWallet()
+            cardTypesResolver.isTraillantWallet() -> userWallet.resolveTraillantWallet()
             cardTypesResolver.isTronWallet() -> userWallet.resolveTronWallet()
             cardTypesResolver.isKaspaWallet() -> userWallet.resolveKaspaWallet()
             cardTypesResolver.isBadWallet() -> userWallet.resolveBadWallet()
@@ -81,6 +85,27 @@ internal object WalletImageResolver {
         )
     }
 
+    private fun UserWallet.resolveWhiteWallet2(): Int? {
+        return resolveWallet2(
+            oneBackupResId = R.drawable.ill_white_card2_120_106,
+            twoBackupResId = R.drawable.ill_white_card3_120_106,
+        )
+    }
+
+    private fun UserWallet.resolveAvroraWallet(): Int? {
+        return resolveWallet2(
+            oneBackupResId = R.drawable.ill_avrora_card2_120_106,
+            twoBackupResId = R.drawable.ill_avrora_card3_120_106,
+        )
+    }
+
+    private fun UserWallet.resolveTraillantWallet(): Int? {
+        return resolveWallet2(
+            oneBackupResId = R.drawable.ill_traillant_card2_120_106,
+            twoBackupResId = R.drawable.ill_traillant_card3_120_106,
+        )
+    }
+
     private fun UserWallet.resolveWallet1(): Int? {
         return resolveWalletWithBackups { count ->
             when (count) {