[trello.com/c/Q5t2dsq4] fix: update the wallet visibility status in t…

…he search mode
Adamant-im · Jun 12, 2023 · 0ca7474 · 0ca7474
1 parent 6ec4234
commit 0ca7474
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/Adamant/Stories/Settings/VisibleWallets/VisibleWalletsTableViewCell.swift b/Adamant/Stories/Settings/VisibleWallets/VisibleWalletsTableViewCell.swift
@@ -59,6 +59,8 @@ class VisibleWalletsTableViewCell: UITableViewCell {
         set { checkmarkRowView.balance = newValue }
     }
 
+    var unicId: String?
+
     required override init(style: UITableViewCell.CellStyle, reuseIdentifier: String?) {
         super.init(style: style, reuseIdentifier: reuseIdentifier)
         setupView()

diff --git a/Adamant/Stories/Settings/VisibleWallets/VisibleWalletsViewController.swift b/Adamant/Stories/Settings/VisibleWallets/VisibleWalletsViewController.swift
@@ -234,6 +234,7 @@ extension VisibleWalletsViewController: UITableViewDataSource, UITableViewDelega
         cell.balance = wallet.wallet?.balance
         cell.delegate = self
         cell.isChecked = !isInvisible(wallet)
+        cell.unicId = wallet.tokenUnicID
 
         return cell
     }
@@ -292,11 +293,16 @@ extension VisibleWalletsViewController: UITableViewDataSource, UITableViewDelega
 
 // MARK: - AdamantVisibleWalletsCellDelegate
 extension VisibleWalletsViewController: AdamantVisibleWalletsCellDelegate {
-    func delegateCell(_ cell: VisibleWalletsTableViewCell, didChangeCheckedStateTo state: Bool) {
-        guard let indexPath = tableView.indexPath(for: cell) else {
-            return
-        }
-        let wallet = wallets[indexPath.row]
+    func delegateCell(
+        _ cell: VisibleWalletsTableViewCell,
+        didChangeCheckedStateTo state: Bool
+    ) {
+        let wallet = wallets.first(where: {
+            $0.tokenUnicID == cell.unicId
+        })
+
+        guard let wallet = wallet else { return }
+
         if !isInvisible(wallet) {
             visibleWalletsService.addToInvisibleWallets(wallet)
         } else {