mirror of
https://github.com/metabolist/metatext.git
synced 2024-12-29 08:50:27 +00:00
136 lines
5.2 KiB
Swift
136 lines
5.2 KiB
Swift
// Copyright © 2021 Metabolist. All rights reserved.
|
|
|
|
import Combine
|
|
import Mastodon
|
|
import UIKit
|
|
import ViewModels
|
|
|
|
enum AutocompleteSection: Int, Hashable {
|
|
case search
|
|
case emoji
|
|
}
|
|
|
|
enum AutocompleteItem: Hashable {
|
|
case account(Account)
|
|
case tag(Tag)
|
|
case emoji(PickerEmoji)
|
|
}
|
|
|
|
final class AutocompleteDataSource: UICollectionViewDiffableDataSource<AutocompleteSection, AutocompleteItem> {
|
|
@Published private var searchViewModel: SearchViewModel
|
|
@Published private var emojiPickerViewModel: EmojiPickerViewModel
|
|
|
|
private let updateQueue =
|
|
DispatchQueue(label: "com.metabolist.metatext.autocomplete-data-source.update-queue")
|
|
private var cancellables = Set<AnyCancellable>()
|
|
|
|
init(collectionView: UICollectionView,
|
|
queryPublisher: AnyPublisher<String?, Never>,
|
|
parentViewModel: NewStatusViewModel) {
|
|
searchViewModel = SearchViewModel(identityContext: parentViewModel.identityContext)
|
|
emojiPickerViewModel = EmojiPickerViewModel(identityContext: parentViewModel.identityContext, queryOnly: true)
|
|
|
|
let registration = UICollectionView.CellRegistration<AutocompleteItemCollectionViewCell, AutocompleteItem> {
|
|
$0.item = $2
|
|
$0.identityContext = parentViewModel.identityContext
|
|
}
|
|
|
|
let emojiRegistration = UICollectionView.CellRegistration<EmojiCollectionViewCell, PickerEmoji> {
|
|
$0.viewModel = EmojiViewModel(emoji: $2, identityContext: parentViewModel.identityContext)
|
|
}
|
|
|
|
super.init(collectionView: collectionView) {
|
|
if case let .emoji(emoji) = $2 {
|
|
return $0.dequeueConfiguredReusableCell(using: emojiRegistration, for: $1, item: emoji)
|
|
} else {
|
|
return $0.dequeueConfiguredReusableCell(using: registration, for: $1, item: $2)
|
|
}
|
|
}
|
|
|
|
queryPublisher
|
|
.replaceNil(with: "")
|
|
.removeDuplicates()
|
|
.combineLatest($searchViewModel, $emojiPickerViewModel)
|
|
.sink(receiveValue: Self.combine(query:searchViewModel:emojiPickerViewModel:))
|
|
.store(in: &cancellables)
|
|
|
|
$searchViewModel.map(\.updates)
|
|
.switchToLatest()
|
|
.combineLatest($emojiPickerViewModel.map(\.$emoji).switchToLatest())
|
|
.receive(on: DispatchQueue.main)
|
|
.sink { [weak self] in self?.apply(searchViewModelUpdate: $0, emojiSections: $1) }
|
|
.store(in: &cancellables)
|
|
|
|
parentViewModel.$identityContext
|
|
.dropFirst()
|
|
.sink { [weak self] in
|
|
guard let self = self else { return }
|
|
|
|
self.searchViewModel = SearchViewModel(identityContext: $0)
|
|
self.emojiPickerViewModel = EmojiPickerViewModel(identityContext: $0, queryOnly: true)
|
|
}
|
|
.store(in: &cancellables)
|
|
}
|
|
|
|
override func apply(_ snapshot: NSDiffableDataSourceSnapshot<AutocompleteSection, AutocompleteItem>,
|
|
animatingDifferences: Bool = true,
|
|
completion: (() -> Void)? = nil) {
|
|
updateQueue.async {
|
|
super.apply(snapshot, animatingDifferences: animatingDifferences, completion: completion)
|
|
}
|
|
}
|
|
}
|
|
|
|
extension AutocompleteDataSource {
|
|
func updateUse(emoji: PickerEmoji) {
|
|
emojiPickerViewModel.updateUse(emoji: emoji)
|
|
}
|
|
}
|
|
|
|
private extension AutocompleteDataSource {
|
|
static func combine(query: String, searchViewModel: SearchViewModel, emojiPickerViewModel: EmojiPickerViewModel) {
|
|
if query.starts(with: ":") {
|
|
searchViewModel.query = ""
|
|
emojiPickerViewModel.query = String(query.dropFirst())
|
|
} else {
|
|
if query.starts(with: "@") {
|
|
searchViewModel.scope = .accounts
|
|
} else if query.starts(with: "#") {
|
|
searchViewModel.scope = .tags
|
|
}
|
|
|
|
searchViewModel.query = String(query.dropFirst())
|
|
emojiPickerViewModel.query = ""
|
|
}
|
|
}
|
|
|
|
func apply(searchViewModelUpdate: CollectionUpdate, emojiSections: [PickerEmoji.Category: [PickerEmoji]]) {
|
|
var newSnapshot = NSDiffableDataSourceSnapshot<AutocompleteSection, AutocompleteItem>()
|
|
let items: [AutocompleteItem] = searchViewModelUpdate.sections.map(\.items).reduce([], +).compactMap {
|
|
switch $0 {
|
|
case let .account(account, _, _):
|
|
return .account(account)
|
|
case let .tag(tag):
|
|
return .tag(tag)
|
|
default:
|
|
return nil
|
|
}
|
|
}
|
|
let emojis = emojiSections.sorted { $0.0 < $1.0 }.map(\.value).reduce([], +).map(AutocompleteItem.emoji)
|
|
|
|
newSnapshot.appendSections([.search])
|
|
|
|
if !items.isEmpty {
|
|
newSnapshot.appendItems(items, toSection: .search)
|
|
} else if !emojis.isEmpty {
|
|
newSnapshot.appendSections([.emoji])
|
|
newSnapshot.appendItems(emojis, toSection: .emoji)
|
|
}
|
|
|
|
apply(newSnapshot, animatingDifferences: !UIAccessibility.isReduceMotionEnabled) {
|
|
// animation causes issue with custom emoji images requiring reload
|
|
newSnapshot.reloadItems(newSnapshot.itemIdentifiers)
|
|
self.apply(newSnapshot, animatingDifferences: false)
|
|
}
|
|
}
|
|
}
|