forked from zelo72/mastodon-ios
67 lines
2.8 KiB
Swift
67 lines
2.8 KiB
Swift
|
//
|
||
|
|
// SearchHistoryViewModel+Diffable.swift
|
||
|
|
// Mastodon
|
||
|
|
//
|
||
|
|
// Created by MainasuK on 2022-1-20.
|
||
|
|
//
|
||
|
|
|
||
|
|
import UIKit
|
||
|
|
|
||
|
|
extension SearchHistoryViewModel {
|
||
|
|
|
||
|
|
func setupDiffableDataSource(
|
||
|
|
collectionView: UICollectionView,
|
||
|
|
searchHistorySectionHeaderCollectionReusableViewDelegate: SearchHistorySectionHeaderCollectionReusableViewDelegate
|
||
|
|
) {
|
||
|
|
diffableDataSource = SearchHistorySection.diffableDataSource(
|
||
|
|
collectionView: collectionView,
|
||
|
|
context: context,
|
||
|
|
configuration: SearchHistorySection.Configuration(
|
||
|
|
searchHistorySectionHeaderCollectionReusableViewDelegate: searchHistorySectionHeaderCollectionReusableViewDelegate
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
var snapshot = NSDiffableDataSourceSnapshot<SearchHistorySection, SearchHistoryItem>()
|
||
|
|
snapshot.appendSections([.main])
|
||
|
|
diffableDataSource?.apply(snapshot, animatingDifferences: false)
|
||
|
|
|
||
|
|
searchHistoryFetchedResultController.$records
|
||
|
|
.receive(on: DispatchQueue.main)
|
||
|
|
.sink { [weak self] records in
|
||
|
|
guard let self = self else { return }
|
||
|
|
guard let diffableDataSource = self.diffableDataSource else { return }
|
||
|
|
|
||
|
|
Task {
|
||
|
|
do {
|
||
|
|
let managedObjectContext = self.context.managedObjectContext
|
||
|
|
let items: [SearchHistoryItem] = try await managedObjectContext.perform {
|
||
|
|
var users: [SearchHistoryItem] = []
|
||
|
|
var hashtags: [SearchHistoryItem] = []
|
||
|
|
|
||
|
|
for record in records {
|
||
|
|
guard let searchHistory = record.object(in: managedObjectContext) else { continue }
|
||
|
|
if let user = searchHistory.account {
|
||
|
|
users.append(.user(.init(objectID: user.objectID)))
|
||
|
|
} else if let hashtag = searchHistory.hashtag {
|
||
|
|
hashtags.append(.hashtag(.init(objectID: hashtag.objectID)))
|
||
|
|
} else {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return users + hashtags
|
||
|
|
}
|
||
|
|
var snapshot = NSDiffableDataSourceSnapshot<SearchHistorySection, SearchHistoryItem>()
|
||
|
|
snapshot.appendSections([.main])
|
||
|
|
snapshot.appendItems(items, toSection: .main)
|
||
|
|
diffableDataSource.apply(snapshot, animatingDifferences: false)
|
||
|
|
} catch {
|
||
|
|
// do nothing
|
||
|
|
}
|
||
|
|
} // end Task
|
||
|
|
}
|
||
|
|
.store(in: &disposeBag)
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|