281 lines
11 KiB
Swift
281 lines
11 KiB
Swift
//
|
|
// ThreadViewModel+Diffable.swift
|
|
// Mastodon
|
|
//
|
|
// Created by MainasuK Cirno on 2021-4-12.
|
|
//
|
|
|
|
import UIKit
|
|
import Combine
|
|
import CoreData
|
|
import CoreDataStack
|
|
import MastodonCore
|
|
import MastodonUI
|
|
import MastodonSDK
|
|
|
|
extension ThreadViewModel {
|
|
|
|
@MainActor
|
|
func setupDiffableDataSource(
|
|
tableView: UITableView,
|
|
statusTableViewCellDelegate: StatusTableViewCellDelegate
|
|
) {
|
|
diffableDataSource = StatusSection.diffableDataSource(
|
|
tableView: tableView,
|
|
context: context,
|
|
configuration: StatusSection.Configuration(
|
|
context: context,
|
|
authContext: authContext,
|
|
statusTableViewCellDelegate: statusTableViewCellDelegate,
|
|
timelineMiddleLoaderTableViewCellDelegate: nil,
|
|
filterContext: .thread,
|
|
activeFilters: context.statusFilterService.$activeFilters
|
|
)
|
|
)
|
|
|
|
// make initial snapshot animation smooth
|
|
var snapshot = NSDiffableDataSourceSnapshot<StatusSection, StatusItem>()
|
|
snapshot.appendSections([.main])
|
|
if let root = self.root {
|
|
if case let .root(threadContext) = root,
|
|
let status = threadContext.status.object(in: context.managedObjectContext),
|
|
status.inReplyToID != nil
|
|
{
|
|
snapshot.appendItems([.topLoader], toSection: .main)
|
|
}
|
|
|
|
snapshot.appendItems([.thread(root)], toSection: .main)
|
|
} else {
|
|
|
|
}
|
|
diffableDataSource?.apply(snapshot, animatingDifferences: false)
|
|
|
|
$threadContext
|
|
.receive(on: DispatchQueue.main)
|
|
.sink { [weak self] threadContext in
|
|
guard let self = self else { return }
|
|
guard let _ = threadContext else {
|
|
return
|
|
}
|
|
|
|
self.loadThreadStateMachine.enter(LoadThreadState.Loading.self)
|
|
}
|
|
.store(in: &disposeBag)
|
|
|
|
Publishers.CombineLatest3(
|
|
$root,
|
|
mastodonStatusThreadViewModel.$ancestors,
|
|
mastodonStatusThreadViewModel.$descendants
|
|
)
|
|
.throttle(for: 1, scheduler: DispatchQueue.main, latest: true)
|
|
.sink { [weak self] root, ancestors, descendants in
|
|
guard let self = self else { return }
|
|
guard let diffableDataSource = self.diffableDataSource else { return }
|
|
|
|
Task { @MainActor in
|
|
let oldSnapshot = diffableDataSource.snapshot()
|
|
|
|
var newSnapshot = NSDiffableDataSourceSnapshot<StatusSection, StatusItem>()
|
|
newSnapshot.appendSections([.main])
|
|
|
|
// top loader
|
|
let _hasReplyTo: Bool? = try? await self.context.managedObjectContext.perform {
|
|
guard case let .root(threadContext) = root else { return nil }
|
|
guard let status = threadContext.status.object(in: self.context.managedObjectContext) else { return nil }
|
|
return status.inReplyToID != nil
|
|
}
|
|
if let hasReplyTo = _hasReplyTo, hasReplyTo {
|
|
let state = self.loadThreadStateMachine.currentState
|
|
if state is LoadThreadState.NoMore {
|
|
// do nothing
|
|
} else {
|
|
newSnapshot.appendItems([.topLoader], toSection: .main)
|
|
}
|
|
}
|
|
|
|
// replies
|
|
newSnapshot.appendItems(ancestors.reversed(), toSection: .main)
|
|
// root
|
|
if let root = root {
|
|
let item = StatusItem.thread(root)
|
|
newSnapshot.appendItems([item], toSection: .main)
|
|
}
|
|
// leafs
|
|
newSnapshot.appendItems(descendants, toSection: .main)
|
|
// bottom loader
|
|
if let currentState = self.loadThreadStateMachine.currentState {
|
|
switch currentState {
|
|
case is LoadThreadState.Initial,
|
|
is LoadThreadState.Loading,
|
|
is LoadThreadState.Fail:
|
|
newSnapshot.appendItems([.bottomLoader], toSection: .main)
|
|
default:
|
|
break
|
|
}
|
|
}
|
|
|
|
let hasChanges = newSnapshot.itemIdentifiers != oldSnapshot.itemIdentifiers
|
|
if !hasChanges && !self.hasPendingStatusEditReload {
|
|
return
|
|
}
|
|
|
|
guard let difference = self.calculateReloadSnapshotDifference(
|
|
tableView: tableView,
|
|
oldSnapshot: oldSnapshot,
|
|
newSnapshot: newSnapshot
|
|
) else {
|
|
await self.updateDataSource(snapshot: newSnapshot, animatingDifferences: false)
|
|
return
|
|
}
|
|
|
|
await self.updateSnapshotUsingReloadData(
|
|
tableView: tableView,
|
|
oldSnapshot: oldSnapshot,
|
|
newSnapshot: newSnapshot,
|
|
difference: difference
|
|
)
|
|
self.hasPendingStatusEditReload = false
|
|
} // end Task
|
|
}
|
|
.store(in: &disposeBag)
|
|
}
|
|
|
|
}
|
|
|
|
|
|
extension ThreadViewModel {
|
|
|
|
@MainActor func updateDataSource(
|
|
snapshot: NSDiffableDataSourceSnapshot<StatusSection, StatusItem>,
|
|
animatingDifferences: Bool
|
|
) async {
|
|
await diffableDataSource?.apply(snapshot, animatingDifferences: animatingDifferences)
|
|
}
|
|
|
|
@MainActor func updateSnapshotUsingReloadData(
|
|
snapshot: NSDiffableDataSourceSnapshot<StatusSection, StatusItem>
|
|
) async {
|
|
await self.diffableDataSource?.applySnapshotUsingReloadData(snapshot)
|
|
}
|
|
|
|
// Some UI tweaks to present replies and conversation smoothly
|
|
@MainActor private func updateSnapshotUsingReloadData(
|
|
tableView: UITableView,
|
|
oldSnapshot: NSDiffableDataSourceSnapshot<StatusSection, StatusItem>,
|
|
newSnapshot: NSDiffableDataSourceSnapshot<StatusSection, StatusItem>,
|
|
difference: ThreadViewModel.Difference // <StatusItem>
|
|
) async {
|
|
let replies: [StatusItem] = {
|
|
newSnapshot.itemIdentifiers.filter { item in
|
|
guard case let .thread(thread) = item else { return false }
|
|
guard case .reply = thread else { return false }
|
|
return true
|
|
}
|
|
}()
|
|
// additional margin for .topLoader
|
|
let oldTopMargin: CGFloat = {
|
|
let marginHeight = TimelineTopLoaderTableViewCell.cellHeight
|
|
if oldSnapshot.itemIdentifiers.contains(.topLoader) || !replies.isEmpty {
|
|
return marginHeight
|
|
}
|
|
return .zero
|
|
}()
|
|
|
|
await self.updateSnapshotUsingReloadData(snapshot: newSnapshot)
|
|
|
|
// note:
|
|
// tweak the content offset and bottom inset
|
|
// make the table view stable when data reload
|
|
// the keypoint is set the bottom inset to make the root padding with "TopLoaderHeight" to top edge
|
|
// and restore the "TopLoaderHeight" when bottom inset adjusted
|
|
|
|
// set bottom inset. Make root item pin to top.
|
|
if let item = root.flatMap({ StatusItem.thread($0) }),
|
|
let index = newSnapshot.indexOfItem(item),
|
|
let cell = tableView.cellForRow(at: IndexPath(row: index, section: 0))
|
|
{
|
|
// always set bottom inset due to lazy reply loading
|
|
// otherwise tableView will jump when insert replies
|
|
let bottomSpacing = tableView.safeAreaLayoutGuide.layoutFrame.height - cell.frame.height - oldTopMargin
|
|
let additionalInset = round(tableView.contentSize.height - cell.frame.maxY)
|
|
|
|
tableView.contentInset.bottom = max(0, bottomSpacing - additionalInset)
|
|
}
|
|
|
|
// set scroll position
|
|
tableView.scrollToRow(at: difference.targetIndexPath, at: .top, animated: false)
|
|
tableView.contentOffset.y = {
|
|
var offset: CGFloat = tableView.contentOffset.y - difference.sourceDistanceToTableViewTopEdge
|
|
if tableView.contentInset.bottom != 0.0 {
|
|
// needs restore top margin if bottom inset adjusted
|
|
offset += oldTopMargin
|
|
}
|
|
return offset
|
|
}()
|
|
}
|
|
}
|
|
|
|
extension ThreadViewModel {
|
|
struct Difference {
|
|
let item: StatusItem
|
|
let sourceIndexPath: IndexPath
|
|
let sourceDistanceToTableViewTopEdge: CGFloat
|
|
let targetIndexPath: IndexPath
|
|
}
|
|
|
|
@MainActor private func calculateReloadSnapshotDifference(
|
|
tableView: UITableView,
|
|
oldSnapshot: NSDiffableDataSourceSnapshot<StatusSection, StatusItem>,
|
|
newSnapshot: NSDiffableDataSourceSnapshot<StatusSection, StatusItem>
|
|
) -> Difference? {
|
|
guard oldSnapshot.numberOfItems != 0 else { return nil }
|
|
guard let indexPathsForVisibleRows = tableView.indexPathsForVisibleRows?.sorted() else { return nil }
|
|
|
|
// find index of the first visible item in both old and new snapshot
|
|
var _index: Int?
|
|
let items = oldSnapshot.itemIdentifiers(inSection: .main)
|
|
for (i, item) in items.enumerated() {
|
|
guard let indexPath = indexPathsForVisibleRows.first(where: { $0.row == i }) else { continue }
|
|
guard newSnapshot.indexOfItem(item) != nil else { continue }
|
|
let rectForCell = tableView.rectForRow(at: indexPath)
|
|
let distanceToTableViewTopEdge = tableView.convert(rectForCell, to: nil).origin.y - tableView.safeAreaInsets.top
|
|
guard distanceToTableViewTopEdge >= 0 else { continue }
|
|
_index = i
|
|
break
|
|
}
|
|
|
|
guard let index = _index else { return nil }
|
|
let sourceIndexPath = IndexPath(row: index, section: 0)
|
|
|
|
let rectForSourceItemCell = tableView.rectForRow(at: sourceIndexPath)
|
|
let sourceDistanceToTableViewTopEdge: CGFloat = {
|
|
if tableView.window != nil {
|
|
return tableView.convert(rectForSourceItemCell, to: nil).origin.y - tableView.safeAreaInsets.top
|
|
} else {
|
|
return rectForSourceItemCell.origin.y - tableView.contentOffset.y - tableView.safeAreaInsets.top
|
|
}
|
|
}()
|
|
|
|
guard sourceIndexPath.section < oldSnapshot.numberOfSections,
|
|
sourceIndexPath.row < oldSnapshot.numberOfItems(inSection: oldSnapshot.sectionIdentifiers[sourceIndexPath.section])
|
|
else { return nil }
|
|
|
|
let sectionIdentifier = oldSnapshot.sectionIdentifiers[sourceIndexPath.section]
|
|
let item = oldSnapshot.itemIdentifiers(inSection: sectionIdentifier)[sourceIndexPath.row]
|
|
|
|
guard let targetIndexPathRow = newSnapshot.indexOfItem(item),
|
|
let newSectionIdentifier = newSnapshot.sectionIdentifier(containingItem: item),
|
|
let targetIndexPathSection = newSnapshot.indexOfSection(newSectionIdentifier)
|
|
else { return nil }
|
|
|
|
let targetIndexPath = IndexPath(row: targetIndexPathRow, section: targetIndexPathSection)
|
|
|
|
return Difference(
|
|
item: item,
|
|
sourceIndexPath: sourceIndexPath,
|
|
sourceDistanceToTableViewTopEdge: sourceDistanceToTableViewTopEdge,
|
|
targetIndexPath: targetIndexPath
|
|
)
|
|
}
|
|
}
|