forked from zelo72/mastodon-ios
209 lines
8.1 KiB
Swift
209 lines
8.1 KiB
Swift
//
|
|
// SearchResultViewModel.swift
|
|
// Mastodon
|
|
//
|
|
// Created by MainasuK Cirno on 2021-7-14.
|
|
//
|
|
|
|
import Foundation
|
|
import Combine
|
|
import CoreData
|
|
import CoreDataStack
|
|
import GameplayKit
|
|
import CommonOSLog
|
|
import MastodonSDK
|
|
|
|
final class SearchResultViewModel {
|
|
|
|
var disposeBag = Set<AnyCancellable>()
|
|
|
|
// input
|
|
let context: AppContext
|
|
let searchScope: SearchDetailViewModel.SearchScope
|
|
let searchText = CurrentValueSubject<String, Never>("")
|
|
@Published var hashtags: [Mastodon.Entity.Tag] = []
|
|
let userFetchedResultsController: UserFetchedResultsController
|
|
let statusFetchedResultsController: StatusFetchedResultsController
|
|
let listBatchFetchViewModel = ListBatchFetchViewModel()
|
|
|
|
let viewDidAppear = CurrentValueSubject<Bool, Never>(false)
|
|
var cellFrameCache = NSCache<NSNumber, NSValue>()
|
|
var navigationBarFrame = CurrentValueSubject<CGRect, Never>(.zero)
|
|
|
|
// output
|
|
var diffableDataSource: UITableViewDiffableDataSource<SearchResultSection, SearchResultItem>!
|
|
@Published var items: [SearchResultItem] = []
|
|
|
|
private(set) lazy var stateMachine: GKStateMachine = {
|
|
let stateMachine = GKStateMachine(states: [
|
|
State.Initial(viewModel: self),
|
|
State.Loading(viewModel: self),
|
|
State.Fail(viewModel: self),
|
|
State.Idle(viewModel: self),
|
|
State.NoMore(viewModel: self),
|
|
])
|
|
stateMachine.enter(State.Initial.self)
|
|
return stateMachine
|
|
}()
|
|
let didDataSourceUpdate = PassthroughSubject<Void, Never>()
|
|
|
|
init(context: AppContext, searchScope: SearchDetailViewModel.SearchScope) {
|
|
self.context = context
|
|
self.searchScope = searchScope
|
|
self.userFetchedResultsController = UserFetchedResultsController(
|
|
managedObjectContext: context.managedObjectContext,
|
|
domain: nil,
|
|
additionalTweetPredicate: nil
|
|
)
|
|
self.statusFetchedResultsController = StatusFetchedResultsController(
|
|
managedObjectContext: context.managedObjectContext,
|
|
domain: nil,
|
|
additionalTweetPredicate: nil
|
|
)
|
|
|
|
context.authenticationService.activeMastodonAuthenticationBox
|
|
.map { $0?.domain }
|
|
.assign(to: \.domain, on: userFetchedResultsController)
|
|
.store(in: &disposeBag)
|
|
|
|
context.authenticationService.activeMastodonAuthenticationBox
|
|
.map { $0?.domain }
|
|
.assign(to: \.value, on: statusFetchedResultsController.domain)
|
|
.store(in: &disposeBag)
|
|
|
|
// Publishers.CombineLatest(
|
|
// items,
|
|
// statusFetchedResultsController.objectIDs.removeDuplicates()
|
|
// )
|
|
// .receive(on: DispatchQueue.main)
|
|
// .sink { [weak self] items, statusObjectIDs in
|
|
// guard let self = self else { return }
|
|
// guard let diffableDataSource = self.diffableDataSource else { return }
|
|
//
|
|
// var snapshot = NSDiffableDataSourceSnapshot<SearchResultSection, SearchResultItem>()
|
|
// snapshot.appendSections([.main])
|
|
//
|
|
// // append account & hashtag items
|
|
//
|
|
// var items = items
|
|
// if self.searchScope == .all {
|
|
// // all search scope not paging. it's safe sort on whole dataset
|
|
// items.sort(by: { ($0.sortKey ?? "") < ($1.sortKey ?? "")})
|
|
// }
|
|
// snapshot.appendItems(items, toSection: .main)
|
|
//
|
|
// var oldSnapshotAttributeDict: [NSManagedObjectID : Item.StatusAttribute] = [:]
|
|
// let oldSnapshot = diffableDataSource.snapshot()
|
|
// for item in oldSnapshot.itemIdentifiers {
|
|
// guard case let .status(objectID, attribute) = item else { continue }
|
|
// oldSnapshotAttributeDict[objectID] = attribute
|
|
// }
|
|
//
|
|
// // append statuses
|
|
// var statusItems: [SearchResultItem] = []
|
|
// for objectID in statusObjectIDs {
|
|
// let attribute = oldSnapshotAttributeDict[objectID] ?? Item.StatusAttribute()
|
|
// statusItems.append(.status(statusObjectID: objectID, attribute: attribute))
|
|
// }
|
|
// snapshot.appendItems(statusItems, toSection: .main)
|
|
//
|
|
// if let currentState = self.stateMachine.currentState {
|
|
// switch currentState {
|
|
// case is State.Loading, is State.Fail, is State.Idle:
|
|
// let attribute = SearchResultItem.BottomLoaderAttribute(isEmptyResult: false)
|
|
// snapshot.appendItems([.bottomLoader(attribute: attribute)], toSection: .main)
|
|
// case is State.Fail:
|
|
// break
|
|
// case is State.NoMore:
|
|
// if snapshot.itemIdentifiers.isEmpty {
|
|
// let attribute = SearchResultItem.BottomLoaderAttribute(isEmptyResult: true)
|
|
// snapshot.appendItems([.bottomLoader(attribute: attribute)], toSection: .main)
|
|
// }
|
|
// default:
|
|
// break
|
|
// }
|
|
// }
|
|
//
|
|
// diffableDataSource.defaultRowAnimation = .fade
|
|
// diffableDataSource.apply(snapshot, animatingDifferences: true) { [weak self] in
|
|
// guard let self = self else { return }
|
|
// self.didDataSourceUpdate.send()
|
|
// }
|
|
//
|
|
// }
|
|
// .store(in: &disposeBag)
|
|
}
|
|
|
|
}
|
|
|
|
extension SearchResultViewModel {
|
|
func persistSearchHistory(for item: SearchResultItem) {
|
|
fatalError()
|
|
// guard let box = context.authenticationService.activeMastodonAuthenticationBox.value else { return }
|
|
// let property = SearchHistory.Property(domain: box.domain, userID: box.userID)
|
|
// let domain = box.domain
|
|
//
|
|
// switch item {
|
|
// case .account(let entity):
|
|
// let managedObjectContext = context.backgroundManagedObjectContext
|
|
// managedObjectContext.performChanges {
|
|
// let (user, _) = APIService.CoreData.createOrMergeMastodonUser(
|
|
// into: managedObjectContext,
|
|
// for: nil,
|
|
// in: domain,
|
|
// entity: entity,
|
|
// userCache: nil,
|
|
// networkDate: Date(),
|
|
// log: OSLog.api
|
|
// )
|
|
// if let searchHistory = user.findSearchHistory(domain: box.domain, userID: box.userID) {
|
|
// searchHistory.update(updatedAt: Date())
|
|
// } else {
|
|
// SearchHistory.insert(into: managedObjectContext, property: property, account: user)
|
|
// }
|
|
// }
|
|
// .sink { result in
|
|
// switch result {
|
|
// case .failure(let error):
|
|
// assertionFailure(error.localizedDescription)
|
|
// case .success:
|
|
// break
|
|
// }
|
|
// }
|
|
// .store(in: &context.disposeBag)
|
|
//
|
|
// case .hashtag(let entity):
|
|
// let managedObjectContext = context.backgroundManagedObjectContext
|
|
// var tag: Tag?
|
|
// managedObjectContext.performChanges {
|
|
// let (hashtag, _) = APIService.CoreData.createOrMergeTag(
|
|
// into: managedObjectContext,
|
|
// entity: entity
|
|
// )
|
|
// tag = hashtag
|
|
// if let searchHistory = hashtag.findSearchHistory(domain: box.domain, userID: box.userID) {
|
|
// searchHistory.update(updatedAt: Date())
|
|
// } else {
|
|
// _ = SearchHistory.insert(into: managedObjectContext, property: property, hashtag: hashtag)
|
|
// }
|
|
// }
|
|
// .sink { result in
|
|
// switch result {
|
|
// case .failure(let error):
|
|
// assertionFailure(error.localizedDescription)
|
|
// case .success:
|
|
// print(tag?.searchHistories)
|
|
// break
|
|
// }
|
|
// }
|
|
// .store(in: &context.disposeBag)
|
|
//
|
|
// case .status:
|
|
// // FIXME:
|
|
// break
|
|
// case .bottomLoader:
|
|
// break
|
|
// }
|
|
}
|
|
}
|