forked from zelo72/mastodon-ios
44 lines
1.2 KiB
Swift
44 lines
1.2 KiB
Swift
//
|
|
// DiscoveryForYouViewModel+Diffable.swift
|
|
// Mastodon
|
|
//
|
|
// Created by MainasuK on 2022-4-14.
|
|
//
|
|
|
|
import UIKit
|
|
import Combine
|
|
|
|
extension DiscoveryForYouViewModel {
|
|
|
|
func setupDiffableDataSource(
|
|
tableView: UITableView
|
|
) {
|
|
diffableDataSource = DiscoverySection.diffableDataSource(
|
|
tableView: tableView,
|
|
context: context,
|
|
configuration: DiscoverySection.Configuration()
|
|
)
|
|
|
|
Task {
|
|
try await fetch()
|
|
}
|
|
|
|
userFetchedResultsController.$records
|
|
.receive(on: DispatchQueue.main)
|
|
.sink { [weak self] records in
|
|
guard let self = self else { return }
|
|
guard let diffableDataSource = self.diffableDataSource else { return }
|
|
|
|
var snapshot = NSDiffableDataSourceSnapshot<DiscoverySection, DiscoveryItem>()
|
|
snapshot.appendSections([.forYou])
|
|
|
|
let items = records.map { DiscoveryItem.user($0) }
|
|
snapshot.appendItems(items, toSection: .forYou)
|
|
|
|
diffableDataSource.applySnapshot(snapshot, animated: false)
|
|
}
|
|
.store(in: &disposeBag)
|
|
}
|
|
|
|
}
|