478 lines
21 KiB
Swift
478 lines
21 KiB
Swift
//
|
|
// ComposeViewModel.swift
|
|
// Mastodon
|
|
//
|
|
// Created by MainasuK Cirno on 2021-3-11.
|
|
//
|
|
|
|
import os.log
|
|
import UIKit
|
|
import Combine
|
|
import CoreData
|
|
import CoreDataStack
|
|
import GameplayKit
|
|
import MastodonSDK
|
|
|
|
final class ComposeViewModel {
|
|
|
|
static let composeContentLimit: Int = 500
|
|
|
|
var disposeBag = Set<AnyCancellable>()
|
|
|
|
// input
|
|
let context: AppContext
|
|
let composeKind: ComposeStatusSection.ComposeKind
|
|
let composeStatusAttribute = ComposeStatusItem.ComposeStatusAttribute()
|
|
let isPollComposing = CurrentValueSubject<Bool, Never>(false)
|
|
let isCustomEmojiComposing = CurrentValueSubject<Bool, Never>(false)
|
|
let isContentWarningComposing = CurrentValueSubject<Bool, Never>(false)
|
|
let selectedStatusVisibility: CurrentValueSubject<ComposeToolbarView.VisibilitySelectionType, Never>
|
|
let activeAuthentication: CurrentValueSubject<MastodonAuthentication?, Never>
|
|
let activeAuthenticationBox: CurrentValueSubject<AuthenticationService.MastodonAuthenticationBox?, Never>
|
|
let traitCollectionDidChangePublisher = CurrentValueSubject<Void, Never>(Void()) // use CurrentValueSubject to make intial event emit
|
|
let repliedToCellFrame = CurrentValueSubject<CGRect, Never>(.zero)
|
|
let autoCompleteRetryLayoutTimes = CurrentValueSubject<Int, Never>(0)
|
|
let autoCompleteInfo = CurrentValueSubject<ComposeViewController.AutoCompleteInfo?, Never>(nil)
|
|
|
|
// output
|
|
var diffableDataSource: UICollectionViewDiffableDataSource<ComposeStatusSection, ComposeStatusItem>!
|
|
var customEmojiPickerDiffableDataSource: UICollectionViewDiffableDataSource<CustomEmojiPickerSection, CustomEmojiPickerItem>!
|
|
private(set) lazy var publishStateMachine: GKStateMachine = {
|
|
// exclude timeline middle fetcher state
|
|
let stateMachine = GKStateMachine(states: [
|
|
PublishState.Initial(viewModel: self),
|
|
PublishState.Publishing(viewModel: self),
|
|
PublishState.Fail(viewModel: self),
|
|
PublishState.Discard(viewModel: self),
|
|
PublishState.Finish(viewModel: self),
|
|
])
|
|
stateMachine.enter(PublishState.Initial.self)
|
|
return stateMachine
|
|
}()
|
|
private(set) lazy var publishStateMachinePublisher = CurrentValueSubject<PublishState?, Never>(nil)
|
|
private(set) var publishDate = Date() // update it when enter Publishing state
|
|
|
|
// UI & UX
|
|
let title: CurrentValueSubject<String, Never>
|
|
let shouldDismiss = CurrentValueSubject<Bool, Never>(true)
|
|
let isPublishBarButtonItemEnabled = CurrentValueSubject<Bool, Never>(false)
|
|
let isMediaToolbarButtonEnabled = CurrentValueSubject<Bool, Never>(true)
|
|
let isPollToolbarButtonEnabled = CurrentValueSubject<Bool, Never>(true)
|
|
let characterCount = CurrentValueSubject<Int, Never>(0)
|
|
let collectionViewState = CurrentValueSubject<CollectionViewState, Never>(.fold)
|
|
|
|
// for hashtag: "#<hashag> "
|
|
// for mention: "@<mention> "
|
|
private(set) var preInsertedContent: String?
|
|
|
|
// custom emojis
|
|
var customEmojiViewModelSubscription: AnyCancellable?
|
|
let customEmojiViewModel = CurrentValueSubject<EmojiService.CustomEmojiViewModel?, Never>(nil)
|
|
let customEmojiPickerInputViewModel = CustomEmojiPickerInputViewModel()
|
|
let isLoadingCustomEmoji = CurrentValueSubject<Bool, Never>(false)
|
|
|
|
// attachment
|
|
let attachmentServices = CurrentValueSubject<[MastodonAttachmentService], Never>([])
|
|
|
|
// polls
|
|
let pollOptionAttributes = CurrentValueSubject<[ComposeStatusItem.ComposePollOptionAttribute], Never>([])
|
|
let pollExpiresOptionAttribute = ComposeStatusItem.ComposePollExpiresOptionAttribute()
|
|
|
|
init(
|
|
context: AppContext,
|
|
composeKind: ComposeStatusSection.ComposeKind
|
|
) {
|
|
self.context = context
|
|
self.composeKind = composeKind
|
|
switch composeKind {
|
|
case .post, .hashtag, .mention: self.title = CurrentValueSubject(L10n.Scene.Compose.Title.newPost)
|
|
case .reply: self.title = CurrentValueSubject(L10n.Scene.Compose.Title.newReply)
|
|
}
|
|
self.selectedStatusVisibility = CurrentValueSubject(context.authenticationService.activeMastodonAuthentication.value?.user.locked == true ? .private : .public)
|
|
self.activeAuthentication = CurrentValueSubject(context.authenticationService.activeMastodonAuthentication.value)
|
|
self.activeAuthenticationBox = CurrentValueSubject(context.authenticationService.activeMastodonAuthenticationBox.value)
|
|
// end init
|
|
if case let .reply(repliedToStatusObjectID) = composeKind {
|
|
context.managedObjectContext.performAndWait {
|
|
guard let status = context.managedObjectContext.object(with: repliedToStatusObjectID) as? Status else { return }
|
|
let composeAuthor: MastodonUser? = {
|
|
guard let objectID = self.activeAuthentication.value?.user.objectID else { return nil }
|
|
guard let author = context.managedObjectContext.object(with: objectID) as? MastodonUser else { return nil }
|
|
return author
|
|
}()
|
|
|
|
var mentionAccts: [String] = []
|
|
if composeAuthor?.id != status.author.id {
|
|
mentionAccts.append("@" + status.author.acct)
|
|
}
|
|
let mentions = (status.mentions ?? Set())
|
|
.sorted(by: { $0.index.intValue < $1.index.intValue })
|
|
.filter { $0.id != composeAuthor?.id }
|
|
for mention in mentions {
|
|
mentionAccts.append("@" + mention.acct)
|
|
}
|
|
for acct in mentionAccts {
|
|
UITextChecker.learnWord(acct)
|
|
}
|
|
if let spoilerText = status.spoilerText, !spoilerText.isEmpty {
|
|
self.isContentWarningComposing.value = true
|
|
self.composeStatusAttribute.contentWarningContent.value = spoilerText
|
|
}
|
|
|
|
let initialComposeContent = mentionAccts.joined(separator: " ")
|
|
let preInsertedContent: String? = initialComposeContent.isEmpty ? nil : initialComposeContent + " "
|
|
self.preInsertedContent = preInsertedContent
|
|
self.composeStatusAttribute.composeContent.value = preInsertedContent
|
|
}
|
|
|
|
} else if case let .hashtag(text) = composeKind {
|
|
let initialComposeContent = "#" + text
|
|
UITextChecker.learnWord(initialComposeContent)
|
|
let preInsertedContent = initialComposeContent + " "
|
|
self.preInsertedContent = preInsertedContent
|
|
self.composeStatusAttribute.composeContent.value = preInsertedContent
|
|
} else if case let .mention(mastodonUserObjectID) = composeKind {
|
|
context.managedObjectContext.performAndWait {
|
|
let mastodonUser = context.managedObjectContext.object(with: mastodonUserObjectID) as! MastodonUser
|
|
let initialComposeContent = "@" + mastodonUser.acct
|
|
UITextChecker.learnWord(initialComposeContent)
|
|
let preInsertedContent = initialComposeContent + " "
|
|
self.preInsertedContent = preInsertedContent
|
|
self.composeStatusAttribute.composeContent.value = preInsertedContent
|
|
}
|
|
} else {
|
|
self.preInsertedContent = nil
|
|
}
|
|
|
|
isCustomEmojiComposing
|
|
.assign(to: \.value, on: customEmojiPickerInputViewModel.isCustomEmojiComposing)
|
|
.store(in: &disposeBag)
|
|
|
|
isContentWarningComposing
|
|
.assign(to: \.value, on: composeStatusAttribute.isContentWarningComposing)
|
|
.store(in: &disposeBag)
|
|
|
|
// bind active authentication
|
|
context.authenticationService.activeMastodonAuthentication
|
|
.assign(to: \.value, on: activeAuthentication)
|
|
.store(in: &disposeBag)
|
|
context.authenticationService.activeMastodonAuthenticationBox
|
|
.assign(to: \.value, on: activeAuthenticationBox)
|
|
.store(in: &disposeBag)
|
|
|
|
// bind avatar and names
|
|
activeAuthentication
|
|
.sink { [weak self] mastodonAuthentication in
|
|
guard let self = self else { return }
|
|
let mastodonUser = mastodonAuthentication?.user
|
|
let username = mastodonUser?.username ?? " "
|
|
|
|
self.composeStatusAttribute.avatarURL.value = mastodonUser?.avatarImageURL()
|
|
self.composeStatusAttribute.displayName.value = {
|
|
guard let displayName = mastodonUser?.displayName, !displayName.isEmpty else {
|
|
return username
|
|
}
|
|
return displayName
|
|
}()
|
|
self.composeStatusAttribute.username.value = username
|
|
}
|
|
.store(in: &disposeBag)
|
|
|
|
// bind character count
|
|
Publishers.CombineLatest3(
|
|
composeStatusAttribute.composeContent.eraseToAnyPublisher(),
|
|
composeStatusAttribute.isContentWarningComposing.eraseToAnyPublisher(),
|
|
composeStatusAttribute.contentWarningContent.eraseToAnyPublisher()
|
|
)
|
|
.map { composeContent, isContentWarningComposing, contentWarningContent -> Int in
|
|
let composeContent = composeContent ?? ""
|
|
var count = composeContent.count
|
|
if isContentWarningComposing {
|
|
count += contentWarningContent.count
|
|
}
|
|
return count
|
|
}
|
|
.assign(to: \.value, on: characterCount)
|
|
.store(in: &disposeBag)
|
|
// bind compose bar button item UI state
|
|
let isComposeContentEmpty = composeStatusAttribute.composeContent
|
|
.map { ($0 ?? "").isEmpty }
|
|
let isComposeContentValid = composeStatusAttribute.composeContent
|
|
.map { composeContent -> Bool in
|
|
let composeContent = composeContent ?? ""
|
|
return composeContent.count <= ComposeViewModel.composeContentLimit
|
|
}
|
|
let isMediaEmpty = attachmentServices
|
|
.map { $0.isEmpty }
|
|
let isMediaUploadAllSuccess = attachmentServices
|
|
.map { services in
|
|
services.allSatisfy { $0.uploadStateMachineSubject.value is MastodonAttachmentService.UploadState.Finish }
|
|
}
|
|
let isPollAttributeAllValid = pollOptionAttributes
|
|
.map { pollAttributes in
|
|
pollAttributes.allSatisfy { attribute -> Bool in
|
|
!attribute.option.value.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty
|
|
}
|
|
}
|
|
|
|
let isPublishBarButtonItemEnabledPrecondition1 = Publishers.CombineLatest4(
|
|
isComposeContentEmpty.eraseToAnyPublisher(),
|
|
isComposeContentValid.eraseToAnyPublisher(),
|
|
isMediaEmpty.eraseToAnyPublisher(),
|
|
isMediaUploadAllSuccess.eraseToAnyPublisher()
|
|
)
|
|
.map { isComposeContentEmpty, isComposeContentValid, isMediaEmpty, isMediaUploadAllSuccess -> Bool in
|
|
if isMediaEmpty {
|
|
return isComposeContentValid && !isComposeContentEmpty
|
|
} else {
|
|
return isComposeContentValid && isMediaUploadAllSuccess
|
|
}
|
|
}
|
|
.eraseToAnyPublisher()
|
|
|
|
let isPublishBarButtonItemEnabledPrecondition2 = Publishers.CombineLatest4(
|
|
isComposeContentEmpty.eraseToAnyPublisher(),
|
|
isComposeContentValid.eraseToAnyPublisher(),
|
|
isPollComposing.eraseToAnyPublisher(),
|
|
isPollAttributeAllValid.eraseToAnyPublisher()
|
|
)
|
|
.map { isComposeContentEmpty, isComposeContentValid, isPollComposing, isPollAttributeAllValid -> Bool in
|
|
if isPollComposing {
|
|
return isComposeContentValid && !isComposeContentEmpty && isPollAttributeAllValid
|
|
} else {
|
|
return isComposeContentValid && !isComposeContentEmpty
|
|
}
|
|
}
|
|
.eraseToAnyPublisher()
|
|
|
|
Publishers.CombineLatest(
|
|
isPublishBarButtonItemEnabledPrecondition1,
|
|
isPublishBarButtonItemEnabledPrecondition2
|
|
)
|
|
.map { $0 && $1 }
|
|
.assign(to: \.value, on: isPublishBarButtonItemEnabled)
|
|
.store(in: &disposeBag)
|
|
|
|
// bind modal dismiss state
|
|
composeStatusAttribute.composeContent
|
|
.receive(on: DispatchQueue.main)
|
|
.map { [weak self] content in
|
|
let content = content ?? ""
|
|
if content.isEmpty {
|
|
return true
|
|
}
|
|
// if preInsertedContent plus a space is equal to the content, simply dismiss the modal
|
|
if let preInsertedContent = self?.preInsertedContent {
|
|
return content == (preInsertedContent + " ")
|
|
}
|
|
return false
|
|
}
|
|
.assign(to: \.value, on: shouldDismiss)
|
|
.store(in: &disposeBag)
|
|
|
|
// bind custom emojis
|
|
context.authenticationService.activeMastodonAuthenticationBox
|
|
.receive(on: DispatchQueue.main)
|
|
.sink { [weak self] activeMastodonAuthenticationBox in
|
|
guard let self = self else { return }
|
|
guard let activeMastodonAuthenticationBox = activeMastodonAuthenticationBox else { return }
|
|
let domain = activeMastodonAuthenticationBox.domain
|
|
|
|
// trigger dequeue to preload emojis
|
|
self.customEmojiViewModel.value = self.context.emojiService.dequeueCustomEmojiViewModel(for: domain)
|
|
}
|
|
.store(in: &disposeBag)
|
|
|
|
// bind snapshot
|
|
Publishers.CombineLatest3(
|
|
attachmentServices.eraseToAnyPublisher(),
|
|
isPollComposing.eraseToAnyPublisher(),
|
|
pollOptionAttributes.eraseToAnyPublisher()
|
|
)
|
|
.receive(on: DispatchQueue.main)
|
|
.sink { [weak self] attachmentServices, isPollComposing, pollAttributes in
|
|
os_log(.info, log: .debug, "%{public}s[%{public}ld], %{public}s: trigger attachments upload…", ((#file as NSString).lastPathComponent), #line, #function)
|
|
|
|
guard let self = self else { return }
|
|
guard let diffableDataSource = self.diffableDataSource else { return }
|
|
var snapshot = diffableDataSource.snapshot()
|
|
|
|
snapshot.deleteItems(snapshot.itemIdentifiers(inSection: .attachment))
|
|
var attachmentItems: [ComposeStatusItem] = []
|
|
for attachmentService in attachmentServices {
|
|
let item = ComposeStatusItem.attachment(attachmentService: attachmentService)
|
|
attachmentItems.append(item)
|
|
}
|
|
snapshot.appendItems(attachmentItems, toSection: .attachment)
|
|
|
|
snapshot.deleteItems(snapshot.itemIdentifiers(inSection: .poll))
|
|
if isPollComposing {
|
|
var pollItems: [ComposeStatusItem] = []
|
|
for pollAttribute in pollAttributes {
|
|
let item = ComposeStatusItem.pollOption(attribute: pollAttribute)
|
|
pollItems.append(item)
|
|
}
|
|
snapshot.appendItems(pollItems, toSection: .poll)
|
|
if pollAttributes.count < 4 {
|
|
snapshot.appendItems([ComposeStatusItem.pollOptionAppendEntry], toSection: .poll)
|
|
}
|
|
snapshot.appendItems([ComposeStatusItem.pollExpiresOption(attribute: self.pollExpiresOptionAttribute)], toSection: .poll)
|
|
}
|
|
|
|
diffableDataSource.apply(snapshot)
|
|
|
|
// drive service upload state
|
|
// make image upload in the queue
|
|
for attachmentService in attachmentServices {
|
|
// skip when prefix N task when task finish OR fail OR uploading
|
|
guard let currentState = attachmentService.uploadStateMachine.currentState else { break }
|
|
if currentState is MastodonAttachmentService.UploadState.Fail {
|
|
continue
|
|
}
|
|
if currentState is MastodonAttachmentService.UploadState.Finish {
|
|
continue
|
|
}
|
|
if currentState is MastodonAttachmentService.UploadState.Uploading {
|
|
break
|
|
}
|
|
// trigger uploading one by one
|
|
if currentState is MastodonAttachmentService.UploadState.Initial {
|
|
attachmentService.uploadStateMachine.enter(MastodonAttachmentService.UploadState.Uploading.self)
|
|
break
|
|
}
|
|
}
|
|
}
|
|
.store(in: &disposeBag)
|
|
|
|
// bind delegate
|
|
attachmentServices
|
|
.sink { [weak self] attachmentServices in
|
|
guard let self = self else { return }
|
|
attachmentServices.forEach { $0.delegate = self }
|
|
}
|
|
.store(in: &disposeBag)
|
|
|
|
pollOptionAttributes
|
|
.sink { [weak self] pollAttributes in
|
|
guard let self = self else { return }
|
|
pollAttributes.forEach { $0.delegate = self }
|
|
}
|
|
.store(in: &disposeBag)
|
|
|
|
// bind compose toolbar UI state
|
|
Publishers.CombineLatest(
|
|
isPollComposing.eraseToAnyPublisher(),
|
|
attachmentServices.eraseToAnyPublisher()
|
|
)
|
|
.receive(on: DispatchQueue.main)
|
|
.sink(receiveValue: { [weak self] isPollComposing, attachmentServices in
|
|
guard let self = self else { return }
|
|
let shouldMediaDisable = isPollComposing || attachmentServices.count >= 4
|
|
let shouldPollDisable = attachmentServices.count > 0
|
|
|
|
self.isMediaToolbarButtonEnabled.value = !shouldMediaDisable
|
|
self.isPollToolbarButtonEnabled.value = !shouldPollDisable
|
|
})
|
|
.store(in: &disposeBag)
|
|
|
|
if let preInsertedContent = preInsertedContent {
|
|
// add a space after the injected text
|
|
composeStatusAttribute.composeContent.send(preInsertedContent + " ")
|
|
}
|
|
}
|
|
|
|
deinit {
|
|
os_log(.info, log: .debug, "%{public}s[%{public}ld], %{public}s", ((#file as NSString).lastPathComponent), #line, #function)
|
|
}
|
|
|
|
}
|
|
|
|
extension ComposeViewModel {
|
|
enum CollectionViewState {
|
|
case fold // snap to input
|
|
case expand // snap to reply
|
|
}
|
|
}
|
|
|
|
extension ComposeViewModel {
|
|
func createNewPollOptionIfPossible() {
|
|
guard pollOptionAttributes.value.count < 4 else { return }
|
|
|
|
let attribute = ComposeStatusItem.ComposePollOptionAttribute()
|
|
pollOptionAttributes.value = pollOptionAttributes.value + [attribute]
|
|
}
|
|
|
|
func updatePublishDate() {
|
|
publishDate = Date()
|
|
}
|
|
}
|
|
|
|
extension ComposeViewModel {
|
|
|
|
enum AttachmentPrecondition: Error, LocalizedError {
|
|
case videoAttachWithPhoto
|
|
case moreThanOneVideo
|
|
|
|
var errorDescription: String? {
|
|
return L10n.Common.Alerts.PublishPostFailure.title
|
|
}
|
|
|
|
var failureReason: String? {
|
|
switch self {
|
|
case .videoAttachWithPhoto:
|
|
return L10n.Common.Alerts.PublishPostFailure.AttchmentsMessage.videoAttachWithPhoto
|
|
case .moreThanOneVideo:
|
|
return L10n.Common.Alerts.PublishPostFailure.AttchmentsMessage.moreThanOneVideo
|
|
}
|
|
}
|
|
}
|
|
|
|
// check exclusive limit:
|
|
// - up to 1 video
|
|
// - up to 4 photos
|
|
func checkAttachmentPrecondition() throws {
|
|
let attachmentServices = self.attachmentServices.value
|
|
guard !attachmentServices.isEmpty else { return }
|
|
var photoAttachmentServices: [MastodonAttachmentService] = []
|
|
var videoAttachmentServices: [MastodonAttachmentService] = []
|
|
attachmentServices.forEach { service in
|
|
guard let file = service.file.value else {
|
|
assertionFailure()
|
|
return
|
|
}
|
|
switch file {
|
|
case .jpeg, .png, .gif:
|
|
photoAttachmentServices.append(service)
|
|
case .other:
|
|
videoAttachmentServices.append(service)
|
|
}
|
|
}
|
|
|
|
if !videoAttachmentServices.isEmpty {
|
|
guard videoAttachmentServices.count == 1 else {
|
|
throw AttachmentPrecondition.moreThanOneVideo
|
|
}
|
|
guard photoAttachmentServices.isEmpty else {
|
|
throw AttachmentPrecondition.videoAttachWithPhoto
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
// MARK: - MastodonAttachmentServiceDelegate
|
|
extension ComposeViewModel: MastodonAttachmentServiceDelegate {
|
|
func mastodonAttachmentService(_ service: MastodonAttachmentService, uploadStateDidChange state: MastodonAttachmentService.UploadState?) {
|
|
// trigger new output event
|
|
attachmentServices.value = attachmentServices.value
|
|
}
|
|
}
|
|
|
|
// MARK: - ComposePollAttributeDelegate
|
|
extension ComposeViewModel: ComposePollAttributeDelegate {
|
|
func composePollAttribute(_ attribute: ComposeStatusItem.ComposePollOptionAttribute, pollOptionDidChange: String?) {
|
|
// trigger update
|
|
pollOptionAttributes.value = pollOptionAttributes.value
|
|
}
|
|
}
|