CipherLabs
/
alpha-wallet-ios
mirror of https://github.com/AlphaWallet/alpha-wallet-ios
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #6952 from AlphaWallet/improve-wallet-tab-state

Browse Source 
Fix/improve Wallet tab state management so more view state is stored in a single object and kept in sync. This reduces crashes and bugs
pull/6953/head
Hwee-Boon Yar 1 year ago committed by GitHub
parent e540003eea fd88c4c21d
commit 2691104da8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 108 additions and 73 deletions
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Show Stats Download Patch File Download Diff File
  1. 31
      AlphaWallet/Tokens/ViewControllers/TokensViewController.swift
  2. 120
      AlphaWallet/Tokens/ViewModels/TokensViewModel.swift
  3. 1
      modules/AlphaWalletFoundation/AlphaWalletFoundation/Tokens/TokensFilter.swift
  4. 27
      modules/AlphaWalletFoundation/AlphaWalletFoundation/Tokens/Types/WalletFilter.swift

31
AlphaWallet/Tokens/ViewControllers/TokensViewController.swift
Unescape View File

@ -19,6 +19,9 @@ final class TokensViewController: UIViewController {
private let appear = PassthroughSubject<Void, Never>() private let appear = PassthroughSubject<Void, Never>()
private let selection = PassthroughSubject<TokensViewModel.SelectionSource, Never>() private let selection = PassthroughSubject<TokensViewModel.SelectionSource, Never>()
private let viewModel: TokensViewModel private let viewModel: TokensViewModel
private var viewState: TokensViewModel.ViewState {
return viewModel.viewState
}
lazy private var filterView: ScrollableSegmentedControl = { lazy private var filterView: ScrollableSegmentedControl = {
let control = ScrollableSegmentedControl(cells: viewModel.filterViewModel.cells, configuration: viewModel.filterViewModel.configuration) let control = ScrollableSegmentedControl(cells: viewModel.filterViewModel.cells, configuration: viewModel.filterViewModel.configuration)
@ -252,9 +255,10 @@ final class TokensViewController: UIViewController {
let output = viewModel.transform(input: input) let output = viewModel.transform(input: input)
dataSource.numberOfRowsInSection dataSource.numberOfRowsInSection
.sink { [weak self, viewModel] section in .sink { [weak self] section in
guard viewModel.filteredTokensAndSections.sections[section] == .tokens || viewModel.filteredTokensAndSections.sections[section] == .collectiblePairs else { return } guard let strongSelf = self else { return }
self?.handleTokensCountChange(rows: viewModel.numberOfItems(for: section)) guard strongSelf.viewState.sections[section] == .tokens || strongSelf.viewState.sections[section] == .collectiblePairs else { return }
self?.handleTokensCountChange(rows: strongSelf.viewState.numberOfItems(for: section))
}.store(in: &cancellable) }.store(in: &cancellable)
output.viewState output.viewState
@ -262,6 +266,7 @@ final class TokensViewController: UIViewController {
//Throttle with a small interval so that we only handle the last value in a bursts of updates. This is because each update might have changed `TokensViewModel.filteredTokensAndSections` and we don't want to use an outdated value of it and crash //Throttle with a small interval so that we only handle the last value in a bursts of updates. This is because each update might have changed `TokensViewModel.filteredTokensAndSections` and we don't want to use an outdated value of it and crash
.throttle(for: 0.01, scheduler: RunLoop.main, latest: true) .throttle(for: 0.01, scheduler: RunLoop.main, latest: true)
.sink { [weak self, weak walletSummaryView, blockieImageView, navigationItem] state in .sink { [weak self, weak walletSummaryView, blockieImageView, navigationItem] state in
self?.viewModel.viewState = state
self?.showOrHideBackupWalletViewHolder() self?.showOrHideBackupWalletViewHolder()
walletSummaryView?.configure(viewModel: .init(walletSummary: state.summary, alignment: .center)) walletSummaryView?.configure(viewModel: .init(walletSummary: state.summary, alignment: .center))
@ -271,16 +276,14 @@ final class TokensViewController: UIViewController {
navigationItem.title = state.title navigationItem.title = state.title
self?.isConsoleButtonHidden = state.isConsoleButtonHidden self?.isConsoleButtonHidden = state.isConsoleButtonHidden
self?.footerBar.isHidden = state.isFooterHidden self?.footerBar.isHidden = state.isFooterHidden
self?.applySnapshot(with: state.sections, animatingDifferences: false) self?.applySnapshot(with: state.viewModels, animatingDifferences: false)
}.store(in: &cancellable) }.store(in: &cancellable)
output.deletion output.deletion
.sink { [dataSource] indexPaths in .sink { [dataSource] indexPaths in
var snapshot = dataSource.snapshot() var snapshot = dataSource.snapshot()
let ids = indexPaths.compactMap { dataSource.itemIdentifier(for: $0) } let ids = indexPaths.compactMap { dataSource.itemIdentifier(for: $0) }
snapshot.deleteItems(ids) snapshot.deleteItems(ids)
dataSource.apply(snapshot, animatingDifferences: true) dataSource.apply(snapshot, animatingDifferences: true)
}.store(in: &cancellable) }.store(in: &cancellable)
@ -360,11 +363,11 @@ extension TokensViewController: UITableViewDelegate {
} }
func tableView(_ tableView: UITableView, heightForHeaderInSection section: Int) -> CGFloat { func tableView(_ tableView: UITableView, heightForHeaderInSection section: Int) -> CGFloat {
return viewModel.heightForHeaderInSection(for: section) return viewState.heightForHeaderInSection(for: section)
} }
func tableView(_ tableView: UITableView, viewForHeaderInSection section: Int) -> UIView? { func tableView(_ tableView: UITableView, viewForHeaderInSection section: Int) -> UIView? {
switch viewModel.filteredTokensAndSections.sections[section] { switch viewState.sections[section] {
case .walletSummary: case .walletSummary:
let header: TokensViewController.GeneralTableViewSectionHeader<WalletSummaryView> = tableView.dequeueReusableHeaderFooterView() let header: TokensViewController.GeneralTableViewSectionHeader<WalletSummaryView> = tableView.dequeueReusableHeaderFooterView()
header.subview = walletSummaryView header.subview = walletSummaryView
@ -378,7 +381,7 @@ extension TokensViewController: UITableViewDelegate {
return header return header
case .activeWalletSession: case .activeWalletSession:
let header: ActiveWalletSessionView = tableView.dequeueReusableHeaderFooterView() let header: ActiveWalletSessionView = tableView.dequeueReusableHeaderFooterView()
header.configure(viewModel: .init(count: viewModel.walletConnectSessions)) header.configure(viewModel: .init(count: viewState.walletConnectSessions))
header.delegate = self header.delegate = self
return header return header
@ -441,10 +444,10 @@ extension TokensViewController {
private func tableHeight() -> CGFloat? { private func tableHeight() -> CGFloat? {
guard let delegate = tableView.delegate else { return nil } guard let delegate = tableView.delegate else { return nil }
let sectionCount = viewModel.filteredTokensAndSections.sections.count let sectionCount = viewState.sections.count
var height: CGFloat = 0 var height: CGFloat = 0
for sectionIndex in 0..<sectionCount { for sectionIndex in 0..<sectionCount {
let rows = viewModel.numberOfItems(for: sectionIndex) let rows = viewState.numberOfItems(for: sectionIndex)
for rowIndex in 0..<rows { for rowIndex in 0..<rows {
height += (delegate.tableView?(tableView, heightForRowAt: IndexPath(row: rowIndex, section: sectionIndex))) ?? 0 height += (delegate.tableView?(tableView, heightForRowAt: IndexPath(row: rowIndex, section: sectionIndex))) ?? 0
} }
@ -516,8 +519,8 @@ extension TokensViewController {
} }
private func apply(filter: WalletFilter, withSegmentAtSelection selection: ControlSelection?) { private func apply(filter: WalletFilter, withSegmentAtSelection selection: ControlSelection?) {
let previousFilter = viewModel.filter let previousFilter = viewModel.filterInUserInterface
viewModel.set(filter: filter) viewModel.set(filterInUserInterface: filter)
DispatchQueue.main.asyncAfter(deadline: .now() + 0.1) { [filterView] in DispatchQueue.main.asyncAfter(deadline: .now() + 0.1) { [filterView] in
//Important to update the segmented control (and hence add the segmented control back to the table) after they have been re-added to the table header through the table reload. Otherwise adding to the table header will break the animation for segmented control //Important to update the segmented control (and hence add the segmented control back to the table) after they have been re-added to the table header through the table reload. Otherwise adding to the table header will break the animation for segmented control
@ -562,7 +565,7 @@ extension TokensViewController: UISearchResultsUpdating {
private func processSearchWithKeywords() { private func processSearchWithKeywords() {
shouldHidePromptBackupWalletViewHolderBecauseSearchIsActive = searchController.isActive shouldHidePromptBackupWalletViewHolderBecauseSearchIsActive = searchController.isActive
guard searchController.isActive else { guard searchController.isActive else {
switch viewModel.filter { switch viewModel.filterInUserInterface {
case .all, .defi, .governance, .assets, .collectiblesOnly, .filter, .attestations: case .all, .defi, .governance, .assets, .collectiblesOnly, .filter, .attestations:
break break
case .keyword: case .keyword:

120
AlphaWallet/Tokens/ViewModels/TokensViewModel.swift
Unescape View File

@ -34,13 +34,16 @@ final class TokensViewModel {
private let tokensFilter: TokensFilter private let tokensFilter: TokensFilter
private (set) var isSearchActive: Bool = false private (set) var isSearchActive: Bool = false
//This can be updated anytime; only used to compute the UI state when the UI is updated //These can be updated anytime; only used to compute the next UI state and must not be read by others
private (set) var tokens: [TokenViewModel] = [] private var tokens: [TokenViewModel] = []
private var walletConnectSessions: Int = 0
//This is controlled by the UI and other view state and the other parts of the UI might or might not have handled this yet
private (set) var filterInUserInterface: WalletFilter = .all
//This is the committed state of the UI
var viewState: ViewState!
//These state must be the current displayed values, i.e they should only be updated only when the UI is going to be refreshed or just prior to be updated
private (set) var filter: WalletFilter = .all
private (set) var walletConnectSessions: Int = 0
private (set) var filteredTokensAndSections: (tokens: [TokenOrRpcServer], sections: [Section]) = ([], [])
private let attestationsStore: AttestationsStore private let attestationsStore: AttestationsStore
lazy private var _attestations: [Attestation] = attestationsStore.attestations lazy private var _attestations: [Attestation] = attestationsStore.attestations
private var attestations: [Attestation] { private var attestations: [Attestation] {
@ -70,7 +73,7 @@ final class TokensViewModel {
} }
var emptyTokensTitle: String { var emptyTokensTitle: String {
switch filter { switch filterInUserInterface {
case .assets: case .assets:
return R.string.localizable.emptyTableViewWalletTitle(R.string.localizable.aWalletContentsFilterAssetsOnlyTitle()) return R.string.localizable.emptyTableViewWalletTitle(R.string.localizable.aWalletContentsFilterAssetsOnlyTitle())
case .collectiblesOnly: case .collectiblesOnly:
@ -108,7 +111,7 @@ final class TokensViewModel {
var shouldShowBackupPromptViewHolder: Bool { var shouldShowBackupPromptViewHolder: Bool {
//TODO show the prompt in both ASSETS and COLLECTIBLES tab too //TODO show the prompt in both ASSETS and COLLECTIBLES tab too
switch filter { switch filterInUserInterface {
case .all, .keyword: case .all, .keyword:
return true return true
case .assets, .collectiblesOnly, .filter, .defi, .governance, .attestations: case .assets, .collectiblesOnly, .filter, .defi, .governance, .attestations:
@ -116,25 +119,6 @@ final class TokensViewModel {
} }
} }
func heightForHeaderInSection(for section: Int) -> CGFloat {
switch filteredTokensAndSections.sections[section] {
case .walletSummary:
return 80
case .filters:
return DataEntry.Metric.Tokens.Filter.height
case .activeWalletSession:
return 80
case .search:
return DataEntry.Metric.AddHideToken.Header.height
case .tokens, .collectiblePairs:
return 0.01
}
}
func numberOfItems(for section: Int) -> Int {
return functional.numberOfItems(for: section, sections: filteredTokensAndSections.sections, filteredTokens: filteredTokensAndSections.tokens, collectiblePairs: functional.collectiblePairs(filteredTokens: filteredTokensAndSections.tokens), filter: filter)
}
init(wallet: Wallet, init(wallet: Wallet,
tokensPipeline: TokensProcessingPipeline, tokensPipeline: TokensProcessingPipeline,
tokensFilter: TokensFilter, tokensFilter: TokensFilter,
@ -193,12 +177,10 @@ final class TokensViewModel {
let selection = selection(trigger: input.selection) let selection = selection(trigger: input.selection)
let titleWithListOfBadTokenScriptFiles = Publishers.CombineLatest(title, assetDefinitionStore.listOfBadTokenScriptFiles) let titleWithListOfBadTokenScriptFiles = Publishers.CombineLatest(title, assetDefinitionStore.listOfBadTokenScriptFiles)
let firstNonZeroTokens = sectionViewModelsSubject.filter { [tokenImageFetcher] sections in let firstNonZeroTokens = sectionViewModelsSubject.filter {
let filteredTokens = functional.filteredAndSortedTokens(tokens: self.tokens, attestations: self.attestations, tokensFilter: self.tokensFilter, filter: self.filter) let (sectionsViewModels, _, _) = functional.generateDisplayState(tokens: self.tokens, attestations: self.attestations, tokensFilter: self.tokensFilter, filterInUserInterface: self.filterInUserInterface, walletConnectSessions: self.walletConnectSessions, isSearchActive: self.isSearchActive, tokenImageFetcher: self.tokenImageFetcher)
let sections = functional.refreshSections(walletConnectSessions: self.walletConnectSessions, isSearchActive: self.isSearchActive, filter: self.filter) //We could simplify the above code to just generate and the list of tokens for `.tokens` and perform less work, but it's not called often, so lets keep it simple
let sectionViewModels = functional.buildSectionViewModels(sections: sections, filteredTokens: filteredTokens, collectiblePairs: functional.collectiblePairs(filteredTokens: filteredTokens), filter: self.filter, tokenImageFetcher: tokenImageFetcher) if let s = sectionsViewModels.first(where: { $0.section == .tokens }) {
if let s = sectionViewModels.first(where: { $0.section == .tokens }) {
return !s.views.isEmpty return !s.views.isEmpty
} else { } else {
return false return false
@ -213,9 +195,8 @@ final class TokensViewModel {
let viewState = Publishers.CombineLatest4(compositeSectionViewModelsSubject, walletSummary, blockieImage, titleWithListOfBadTokenScriptFiles) let viewState = Publishers.CombineLatest4(compositeSectionViewModelsSubject, walletSummary, blockieImage, titleWithListOfBadTokenScriptFiles)
//Prevent crash, keeping updates serialized so receiving end can update with atomic state //Prevent crash, keeping updates serialized so receiving end can update with atomic state
.receive(on: RunLoop.main) .receive(on: RunLoop.main)
.map { [tokenImageFetcher] _, summary, blockiesImage, data -> TokensViewModel.ViewState in .map { _, summary, blockiesImage, data -> TokensViewModel.ViewState in
self.filteredTokensAndSections = (tokens: functional.filteredAndSortedTokens(tokens: self.tokens, attestations: self.attestations, tokensFilter: self.tokensFilter, filter: self.filter), sections: functional.refreshSections(walletConnectSessions: self.walletConnectSessions, isSearchActive: self.isSearchActive, filter: self.filter)) let (sectionsViewModels, filteredTokens, sections) = functional.generateDisplayState(tokens: self.tokens, attestations: self.attestations, tokensFilter: self.tokensFilter, filterInUserInterface: self.filterInUserInterface, walletConnectSessions: self.walletConnectSessions, isSearchActive: self.isSearchActive, tokenImageFetcher: self.tokenImageFetcher)
let sections = functional.buildSectionViewModels(sections: self.filteredTokensAndSections.sections, filteredTokens: self.filteredTokensAndSections.tokens, collectiblePairs: functional.collectiblePairs(filteredTokens: self.filteredTokensAndSections.tokens), filter: self.filter, tokenImageFetcher: tokenImageFetcher)
let isConsoleButtonHidden = data.1.isEmpty let isConsoleButtonHidden = data.1.isEmpty
return TokensViewModel.ViewState( return TokensViewModel.ViewState(
title: data.0, title: data.0,
@ -223,7 +204,11 @@ final class TokensViewModel {
blockiesImage: blockiesImage, blockiesImage: blockiesImage,
isConsoleButtonHidden: isConsoleButtonHidden, isConsoleButtonHidden: isConsoleButtonHidden,
isFooterHidden: self.isFooterHidden, isFooterHidden: self.isFooterHidden,
sections: sections) viewModels: sectionsViewModels,
filteredRows: filteredTokens,
sections: sections,
filter: self.filterInUserInterface,
walletConnectSessions: self.walletConnectSessions)
}.removeDuplicates() }.removeDuplicates()
.eraseToAnyPublisher() .eraseToAnyPublisher()
@ -302,9 +287,9 @@ final class TokensViewModel {
asFuture { () -> TokenOrAttestation? in asFuture { () -> TokenOrAttestation? in
switch source { switch source {
case .gridItem(let indexPath, let isLeftCardSelected): case .gridItem(let indexPath, let isLeftCardSelected):
switch self.filteredTokensAndSections.sections[indexPath.section] { switch self.viewState.sections[indexPath.section] {
case .collectiblePairs: case .collectiblePairs:
let pair = functional.collectiblePairs(filteredTokens: self.filteredTokensAndSections.tokens)[indexPath.row] let pair = functional.collectiblePairs(filteredTokens: self.viewState.filteredRows)[indexPath.row]
guard let viewModel: TokenViewModel = isLeftCardSelected ? pair.left : pair.right else { return nil } guard let viewModel: TokenViewModel = isLeftCardSelected ? pair.left : pair.right else { return nil }
return await tokensService.token(for: viewModel.contractAddress, server: viewModel.server).flatMap { TokenOrAttestation.token($0) } return await tokensService.token(for: viewModel.contractAddress, server: viewModel.server).flatMap { TokenOrAttestation.token($0) }
@ -313,7 +298,7 @@ final class TokensViewModel {
} }
case .cell(let indexPath): case .cell(let indexPath):
let tokenOrServer = self.tokenOrServer(at: indexPath) let tokenOrServer = self.tokenOrServer(at: indexPath)
switch (self.filteredTokensAndSections.sections[indexPath.section], tokenOrServer) { switch (self.viewState.sections[indexPath.section], tokenOrServer) {
case (.tokens, .token(let viewModel)): case (.tokens, .token(let viewModel)):
return .token(await tokensService.token(for: viewModel.contractAddress, server: viewModel.server)!) return .token(await tokensService.token(for: viewModel.contractAddress, server: viewModel.server)!)
case (.tokens, .attestation(let attestation)): case (.tokens, .attestation(let attestation)):
@ -339,14 +324,13 @@ final class TokensViewModel {
reloadData(immediately: true) reloadData(immediately: true)
} }
func set(filter: WalletFilter) { func set(filterInUserInterface: WalletFilter) {
//TODO possible that layout will be broken due to `filter` and refresh not in sync? self.filterInUserInterface = filterInUserInterface
self.filter = filter
reloadData(immediately: true) reloadData(immediately: true)
} }
private func tokenOrServer(at indexPath: IndexPath) -> TokenOrRpcServer { private func tokenOrServer(at indexPath: IndexPath) -> TokenOrRpcServer {
return filteredTokensAndSections.tokens[indexPath.row] return viewState.filteredRows[indexPath.row]
} }
func trailingSwipeActionsConfiguration(for indexPath: IndexPath) -> UISwipeActionsConfiguration? { func trailingSwipeActionsConfiguration(for indexPath: IndexPath) -> UISwipeActionsConfiguration? {
@ -399,7 +383,7 @@ final class TokensViewModel {
} }
} }
switch filteredTokensAndSections.sections[indexPath.section] { switch viewState.sections[indexPath.section] {
case .collectiblePairs, .search, .walletSummary, .filters, .activeWalletSession: case .collectiblePairs, .search, .walletSummary, .filters, .activeWalletSession:
return nil return nil
case .tokens: case .tokens:
@ -408,7 +392,7 @@ final class TokensViewModel {
} }
func cellHeight(for indexPath: IndexPath) -> CGFloat { func cellHeight(for indexPath: IndexPath) -> CGFloat {
switch filteredTokensAndSections.sections[indexPath.section] { switch viewState.sections[indexPath.section] {
case .tokens: case .tokens:
switch tokenOrServer(at: indexPath) { switch tokenOrServer(at: indexPath) {
case .rpcServer: case .rpcServer:
@ -425,21 +409,15 @@ final class TokensViewModel {
} }
} }
@discardableResult private func markTokenHidden(token: TokenViewModel) -> Bool { private func markTokenHidden(token: TokenViewModel) {
tokensService.mark(token: token, isHidden: true) tokensService.mark(token: token, isHidden: true)
if let index = tokens.firstIndex(where: { $0 == token }) { if let index = tokens.firstIndex(where: { $0 == token }) {
tokens.remove(at: index) tokens.remove(at: index)
filteredTokensAndSections = (tokens: functional.filteredAndSortedTokens(tokens: tokens, attestations: attestations, tokensFilter: tokensFilter, filter: filter), sections: functional.refreshSections(walletConnectSessions: walletConnectSessions, isSearchActive: isSearchActive, filter: filter))
return true
} }
return false
} }
private func removeAttestation(_ attestation: Attestation) { private func removeAttestation(_ attestation: Attestation) {
attestationsStore.removeAttestation(attestation, forWallet: wallet.address) attestationsStore.removeAttestation(attestation, forWallet: wallet.address)
filteredTokensAndSections = (tokens: functional.filteredAndSortedTokens(tokens: tokens, attestations: attestations, tokensFilter: tokensFilter, filter: filter), sections: functional.refreshSections(walletConnectSessions: walletConnectSessions, isSearchActive: isSearchActive, filter: filter))
} }
func convertSegmentedControlSelectionToFilter(_ selection: ControlSelection) -> WalletFilter? { func convertSegmentedControlSelectionToFilter(_ selection: ControlSelection) -> WalletFilter? {
@ -455,7 +433,7 @@ final class TokensViewModel {
let canRemoveCurrentItem: Bool = tokenOrServer(at: current).isRemovable let canRemoveCurrentItem: Bool = tokenOrServer(at: current).isRemovable
let canRemovePreviousItem: Bool = current.row > 0 ? tokenOrServer(at: current.previous).isRemovable : false let canRemovePreviousItem: Bool = current.row > 0 ? tokenOrServer(at: current.previous).isRemovable : false
let canRemoveNextItem: Bool = { let canRemoveNextItem: Bool = {
guard (current.row + 1) < filteredTokensAndSections.tokens.count else { return false } guard (current.row + 1) < viewState.filteredRows.count else { return false }
return tokenOrServer(at: current.next).isRemovable return tokenOrServer(at: current.next).isRemovable
}() }()
switch (canRemovePreviousItem, canRemoveCurrentItem, canRemoveNextItem) { switch (canRemovePreviousItem, canRemoveCurrentItem, canRemoveNextItem) {
@ -502,7 +480,7 @@ extension TokensViewModel {
case failure case failure
} }
enum TokenOrRpcServer { enum TokenOrRpcServer: Hashable {
case token(TokenViewModel) case token(TokenViewModel)
case rpcServer(RPCServer) case rpcServer(RPCServer)
case attestation(Attestation) case attestation(Attestation)
@ -635,7 +613,30 @@ extension TokensViewModel {
let blockiesImage: BlockiesImage let blockiesImage: BlockiesImage
let isConsoleButtonHidden: Bool let isConsoleButtonHidden: Bool
let isFooterHidden: Bool let isFooterHidden: Bool
let sections: [TokensViewModel.SectionViewModel] let viewModels: [TokensViewModel.SectionViewModel]
let filteredRows: [TokenOrRpcServer]
let sections: [Section]
let filter: WalletFilter
let walletConnectSessions: Int
func heightForHeaderInSection(for section: Int) -> CGFloat {
switch sections[section] {
case .walletSummary:
return 80
case .filters:
return DataEntry.Metric.Tokens.Filter.height
case .activeWalletSession:
return 80
case .search:
return DataEntry.Metric.AddHideToken.Header.height
case .tokens, .collectiblePairs:
return 0.01
}
}
func numberOfItems(for section: Int) -> Int {
return TokensViewModel.functional.numberOfItems(for: section, sections: sections, filteredTokens: filteredRows, collectiblePairs: functional.collectiblePairs(filteredTokens: filteredRows), filter: filter)
}
} }
} }
@ -839,6 +840,13 @@ fileprivate extension TokensViewModel.functional {
return .init(left: left, right: right) return .init(left: left, right: right)
} }
} }
static func generateDisplayState(tokens: [TokenViewModel], attestations: [Attestation], tokensFilter: TokensFilter, filterInUserInterface: WalletFilter, walletConnectSessions: Int, isSearchActive: Bool, tokenImageFetcher: TokenImageFetcher) -> ([TokensViewModel.SectionViewModel], [TokensViewModel.TokenOrRpcServer], [TokensViewModel.Section]) {
let filteredTokens = filteredAndSortedTokens(tokens: tokens, attestations: attestations, tokensFilter: tokensFilter, filter: filterInUserInterface)
let sections = refreshSections(walletConnectSessions: walletConnectSessions, isSearchActive: isSearchActive, filter: filterInUserInterface)
let sectionsViewModels = buildSectionViewModels(sections: sections, filteredTokens: filteredTokens, collectiblePairs: collectiblePairs(filteredTokens: filteredTokens), filter: filterInUserInterface, tokenImageFetcher: tokenImageFetcher)
return (sectionsViewModels, filteredTokens, sections)
}
} }
fileprivate extension IndexPath { fileprivate extension IndexPath {

1
modules/AlphaWalletFoundation/AlphaWalletFoundation/Tokens/TokensFilter.swift
Unescape View File

@ -33,6 +33,7 @@ public extension TokenSortable {
} }
} }
//TODO rename. This name is misleading. It contains filtering logic, but it's not the filter like `WalletFilter` (which is for both tabs and search keyword). Or we rename `WalletFilter` to tabs? But it represent the search keyword too
public class TokensFilter { public class TokensFilter {
private enum FilterKeys { private enum FilterKeys {
case all case all

27
modules/AlphaWalletFoundation/AlphaWalletFoundation/Tokens/Types/WalletFilter.swift
Unescape View File

@ -3,15 +3,38 @@
import Foundation import Foundation
import Combine import Combine
public enum WalletFilter: Equatable { public enum WalletFilter: Equatable, Hashable {
case all case all
case attestations case attestations
case filter(TokenFilterProtocol) case filter(any TokenFilterProtocol)
case defi case defi
case governance case governance
case assets case assets
case collectiblesOnly case collectiblesOnly
case keyword(String) case keyword(String)
public func hash(into hasher: inout Hasher) {
switch self {
case .all:
hasher.combine(0)
case .attestations:
hasher.combine(1)
case .filter:
//The associated value doesn't need to be checked
hasher.combine(2)
case .defi:
hasher.combine(3)
case .governance:
hasher.combine(4)
case .assets:
hasher.combine(5)
case .collectiblesOnly:
hasher.combine(6)
case .keyword(let keyword):
hasher.combine(7)
hasher.combine(keyword)
}
}
} }
public protocol TokenFilterProtocol { public protocol TokenFilterProtocol {

Write Preview
Loading…
Cancel
Save