Fix #8463: Only display NFTs after user updates any NFT's visibility …

…status (#8470) after NFT status chanegd, the re-computation needs to apply on only NFTs.
brave · Nov 23, 2023 · 2c1110b · 2c1110b
1 parent fff9d0e
commit 2c1110b
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/Sources/BraveWallet/Crypto/Stores/NFTStore.swift b/Sources/BraveWallet/Crypto/Stores/NFTStore.swift
@@ -582,7 +582,21 @@ public class NFTStore: ObservableObject, WalletObserverStore {
  let selectedAccounts = self.filters.accounts.filter(\.isSelected).map(\.model)
  let selectedNetworks = self.filters.networks.filter(\.isSelected).map(\.model)
  let userVisibleAssets = self.assetManager.getAllUserAssetsInNetworkAssetsByVisibility(networks: selectedNetworks, visible: true)
+ .map { networkAssets in
+ NetworkAssets(
+ network: networkAssets.network,
+ tokens: networkAssets.tokens.filter { $0.isNft || $0.isErc721 },
+ sortOrder: networkAssets.sortOrder
+ )
+ }
  let userHiddenAssets = self.assetManager.getAllUserAssetsInNetworkAssetsByVisibility(networks: selectedNetworks, visible: false)
+ .map { networkAssets in
+ NetworkAssets(
+ network: networkAssets.network,
+ tokens: networkAssets.tokens.filter { $0.isNft || $0.isErc721 },
+ sortOrder: networkAssets.sortOrder
+ )
+ }
  let unionedSpamNFTs = computeSpamNFTs(
  selectedNetworks: selectedNetworks,
  selectedAccounts: selectedAccounts,