130 lines
5.2 KiB
Swift
130 lines
5.2 KiB
Swift
//
|
|
// RemoteProfileViewModel.swift
|
|
// Mastodon
|
|
//
|
|
// Created by MainasuK Cirno on 2021-4-2.
|
|
//
|
|
|
|
import Foundation
|
|
import Combine
|
|
import CoreDataStack
|
|
import MastodonSDK
|
|
import MastodonCore
|
|
|
|
final class RemoteProfileViewModel: ProfileViewModel {
|
|
|
|
init(context: AppContext, authContext: AuthContext, userID: Mastodon.Entity.Account.ID) {
|
|
super.init(context: context, authContext: authContext, optionalMastodonUser: nil)
|
|
|
|
let domain = authContext.mastodonAuthenticationBox.domain
|
|
let authorization = authContext.mastodonAuthenticationBox.userAuthorization
|
|
Just(userID)
|
|
.asyncMap { userID in
|
|
try await context.apiService.accountInfo(
|
|
domain: domain,
|
|
userID: userID,
|
|
authorization: authorization
|
|
)
|
|
}
|
|
.retry(3)
|
|
.receive(on: DispatchQueue.main)
|
|
.sink { completion in
|
|
switch completion {
|
|
case .failure(let error):
|
|
// TODO: handle error
|
|
break
|
|
case .finished:
|
|
break
|
|
}
|
|
} receiveValue: { [weak self] response in
|
|
guard let self = self else { return }
|
|
let managedObjectContext = context.managedObjectContext
|
|
let request = MastodonUser.sortedFetchRequest
|
|
request.fetchLimit = 1
|
|
request.predicate = MastodonUser.predicate(domain: domain, id: response.value.id)
|
|
guard let mastodonUser = managedObjectContext.safeFetch(request).first else {
|
|
assertionFailure()
|
|
return
|
|
}
|
|
self.user = mastodonUser
|
|
}
|
|
.store(in: &disposeBag)
|
|
}
|
|
|
|
init(context: AppContext, authContext: AuthContext, notificationID: Mastodon.Entity.Notification.ID) {
|
|
super.init(context: context, authContext: authContext, optionalMastodonUser: nil)
|
|
|
|
Task { @MainActor in
|
|
let response = try await context.apiService.notification(
|
|
notificationID: notificationID,
|
|
authenticationBox: authContext.mastodonAuthenticationBox
|
|
)
|
|
let userID = response.value.account.id
|
|
|
|
let _user: MastodonUser? = try await context.managedObjectContext.perform {
|
|
let request = MastodonUser.sortedFetchRequest
|
|
request.predicate = MastodonUser.predicate(domain: authContext.mastodonAuthenticationBox.domain, id: userID)
|
|
request.fetchLimit = 1
|
|
return context.managedObjectContext.safeFetch(request).first
|
|
}
|
|
|
|
if let user = _user {
|
|
self.user = user
|
|
} else {
|
|
_ = try await context.apiService.accountInfo(
|
|
domain: authContext.mastodonAuthenticationBox.domain,
|
|
userID: userID,
|
|
authorization: authContext.mastodonAuthenticationBox.userAuthorization
|
|
)
|
|
|
|
let _user: MastodonUser? = try await context.managedObjectContext.perform {
|
|
let request = MastodonUser.sortedFetchRequest
|
|
request.predicate = MastodonUser.predicate(domain: authContext.mastodonAuthenticationBox.domain, id: userID)
|
|
request.fetchLimit = 1
|
|
return context.managedObjectContext.safeFetch(request).first
|
|
}
|
|
|
|
self.user = _user
|
|
}
|
|
} // end Task
|
|
}
|
|
|
|
init(context: AppContext, authContext: AuthContext, acct: String){
|
|
super.init(context: context, authContext: authContext, optionalMastodonUser: nil)
|
|
|
|
let domain = authContext.mastodonAuthenticationBox.domain
|
|
let authenticationBox = authContext.mastodonAuthenticationBox
|
|
|
|
Just(acct)
|
|
.asyncMap { acct -> Mastodon.Response.Content<Mastodon.Entity.Account?> in
|
|
try await context.apiService.search(
|
|
query: .init(q: acct, type: .accounts, resolve: true),
|
|
authenticationBox: authenticationBox
|
|
).map { $0.accounts.first }
|
|
}
|
|
.retry(3)
|
|
.receive(on: DispatchQueue.main)
|
|
.sink { completion in
|
|
switch completion {
|
|
case .failure(let error):
|
|
// TODO: handle error
|
|
break
|
|
case .finished:
|
|
break
|
|
}
|
|
} receiveValue: { [weak self] response in
|
|
guard let self = self, let value = response.value else { return }
|
|
let managedObjectContext = context.managedObjectContext
|
|
let request = MastodonUser.sortedFetchRequest
|
|
request.fetchLimit = 1
|
|
request.predicate = MastodonUser.predicate(domain: domain, id: value.id)
|
|
guard let mastodonUser = managedObjectContext.safeFetch(request).first else {
|
|
assertionFailure()
|
|
return
|
|
}
|
|
self.user = mastodonUser
|
|
}
|
|
.store(in: &disposeBag)
|
|
}
|
|
}
|