mastodon-ios/Mastodon/Scene/Profile/Following/FollowingListViewModel+Diff...

67 lines
2.7 KiB
Swift
Raw Normal View History

2021-11-02 07:56:42 +01:00
//
// FollowingListViewModel+Diffable.swift
// Mastodon
//
// Created by Cirno MainasuK on 2021-11-2.
//
import UIKit
extension FollowingListViewModel {
func setupDiffableDataSource(
for tableView: UITableView,
dependency: NeedsDependency
) {
diffableDataSource = UserSection.tableViewDiffableDataSource(
for: tableView,
dependency: dependency,
managedObjectContext: userFetchedResultsController.fetchedResultsController.managedObjectContext
)
// workaround to append loader wrong animation issue
// set empty section to make update animation top-to-bottom style
var snapshot = NSDiffableDataSourceSnapshot<UserSection, UserItem>()
snapshot.appendSections([.main])
snapshot.appendItems([.bottomLoader], toSection: .main)
if #available(iOS 15.0, *) {
diffableDataSource?.applySnapshotUsingReloadData(snapshot, completion: nil)
} else {
// Fallback on earlier versions
diffableDataSource?.apply(snapshot, animatingDifferences: false)
}
userFetchedResultsController.objectIDs
.receive(on: DispatchQueue.main)
.sink { [weak self] objectIDs in
guard let self = self else { return }
guard let diffableDataSource = self.diffableDataSource else { return }
var snapshot = NSDiffableDataSourceSnapshot<UserSection, UserItem>()
snapshot.appendSections([.main])
let items: [UserItem] = objectIDs.map {
UserItem.following(objectID: $0)
}
snapshot.appendItems(items, toSection: .main)
if let currentState = self.stateMachine.currentState {
switch currentState {
case is State.Idle, is State.Loading, is State.Fail:
snapshot.appendItems([.bottomLoader], toSection: .main)
case is State.NoMore:
guard let activeMastodonAuthenticationBox = self.context.authenticationService.activeMastodonAuthenticationBox.value,
let userID = self.userID.value,
userID != activeMastodonAuthenticationBox.userID
else { break }
let text = L10n.Scene.Following.footer
snapshot.appendItems([.bottomHeader(text: text)], toSection: .main)
2021-11-02 07:56:42 +01:00
default:
break
}
}
diffableDataSource.apply(snapshot)
}
.store(in: &disposeBag)
}
}