105 lines
4.0 KiB
Swift
105 lines
4.0 KiB
Swift
//
|
|
// MastodonAttachmentService+UploadState.swift
|
|
// Mastodon
|
|
//
|
|
// Created by MainasuK Cirno on 2021-3-18.
|
|
//
|
|
|
|
import os.log
|
|
import Foundation
|
|
import GameplayKit
|
|
import Kingfisher
|
|
import MastodonSDK
|
|
|
|
extension MastodonAttachmentService {
|
|
class UploadState: GKState {
|
|
weak var service: MastodonAttachmentService?
|
|
|
|
init(service: MastodonAttachmentService) {
|
|
self.service = service
|
|
}
|
|
|
|
override func didEnter(from previousState: GKState?) {
|
|
os_log("%{public}s[%{public}ld], %{public}s: enter %s, previous: %s", ((#file as NSString).lastPathComponent), #line, #function, self.debugDescription, previousState.debugDescription)
|
|
service?.uploadStateMachineSubject.send(self)
|
|
}
|
|
}
|
|
}
|
|
|
|
extension MastodonAttachmentService.UploadState {
|
|
|
|
class Initial: MastodonAttachmentService.UploadState {
|
|
override func isValidNextState(_ stateClass: AnyClass) -> Bool {
|
|
guard service?.authenticationBox != nil else { return false }
|
|
guard service?.imageData.value != nil else { return false }
|
|
return stateClass == Uploading.self
|
|
}
|
|
}
|
|
|
|
class Uploading: MastodonAttachmentService.UploadState {
|
|
override func isValidNextState(_ stateClass: AnyClass) -> Bool {
|
|
return stateClass == Fail.self || stateClass == Finish.self
|
|
}
|
|
|
|
override func didEnter(from previousState: GKState?) {
|
|
super.didEnter(from: previousState)
|
|
|
|
guard let service = service, let stateMachine = stateMachine else { return }
|
|
guard let authenticationBox = service.authenticationBox else { return }
|
|
guard let imageData = service.imageData.value else { return }
|
|
|
|
let file: Mastodon.Query.MediaAttachment = {
|
|
if imageData.kf.imageFormat == .PNG {
|
|
return .png(imageData)
|
|
} else {
|
|
return .jpeg(imageData)
|
|
}
|
|
}()
|
|
let description = service.description.value
|
|
let query = Mastodon.API.Media.UploadMeidaQuery(
|
|
file: file,
|
|
thumbnail: nil,
|
|
description: description,
|
|
focus: nil
|
|
)
|
|
|
|
service.context.apiService.uploadMedia(
|
|
domain: authenticationBox.domain,
|
|
query: query,
|
|
mastodonAuthenticationBox: authenticationBox
|
|
)
|
|
.receive(on: DispatchQueue.main)
|
|
.sink { completion in
|
|
switch completion {
|
|
case .failure(let error):
|
|
os_log(.info, log: .debug, "%{public}s[%{public}ld], %{public}s: upload attachment fail: %s", ((#file as NSString).lastPathComponent), #line, #function, error.localizedDescription)
|
|
service.error.send(error)
|
|
case .finished:
|
|
os_log(.info, log: .debug, "%{public}s[%{public}ld], %{public}s: upload attachment success", ((#file as NSString).lastPathComponent), #line, #function)
|
|
break
|
|
}
|
|
} receiveValue: { response in
|
|
os_log(.info, log: .debug, "%{public}s[%{public}ld], %{public}s: upload attachment %s success: %s", ((#file as NSString).lastPathComponent), #line, #function, response.value.id, response.value.url)
|
|
service.attachment.value = response.value
|
|
stateMachine.enter(Finish.self)
|
|
}
|
|
.store(in: &service.disposeBag)
|
|
}
|
|
}
|
|
|
|
class Fail: MastodonAttachmentService.UploadState {
|
|
override func isValidNextState(_ stateClass: AnyClass) -> Bool {
|
|
// allow discard publishing
|
|
return stateClass == Uploading.self || stateClass == Finish.self
|
|
}
|
|
}
|
|
|
|
class Finish: MastodonAttachmentService.UploadState {
|
|
override func isValidNextState(_ stateClass: AnyClass) -> Bool {
|
|
return false
|
|
}
|
|
}
|
|
|
|
}
|
|
|