mastodon-ios/Mastodon/Scene/Profile/RemoteProfileViewModel.swift

//
//  RemoteProfileViewModel.swift
//  Mastodon
//
//  Created by MainasuK Cirno on 2021-4-2.
//

import os.log
import Foundation
import Combine
import CoreDataStack
import MastodonSDK
import MastodonCore

final class RemoteProfileViewModel: ProfileViewModel {
    
    init(context: AppContext, userID: Mastodon.Entity.Account.ID) {
        super.init(context: context, optionalMastodonUser: nil)
        
        guard let activeMastodonAuthenticationBox = context.authenticationService.activeMastodonAuthenticationBox.value else {
            return
        }
        let domain = activeMastodonAuthenticationBox.domain
        let authorization = activeMastodonAuthenticationBox.userAuthorization
        Just(userID)
            .asyncMap { userID in
                try await context.apiService.accountInfo(
                    domain: domain,
                    userID: userID,
                    authorization: authorization
                )
            }
            .retry(3)
            .sink { completion in
                switch completion {
                case .failure(let error):
                    // TODO: handle error
                    os_log(.info, log: .debug, "%{public}s[%{public}ld], %{public}s: remote user %s fetch failed: %s", ((#file as NSString).lastPathComponent), #line, #function, userID, error.localizedDescription)
                case .finished:
                    os_log(.info, log: .debug, "%{public}s[%{public}ld], %{public}s: remote user %s fetched", ((#file as NSString).lastPathComponent), #line, #function, userID)
                }
            } 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, notificationID: Mastodon.Entity.Notification.ID) {
        super.init(context: context, optionalMastodonUser: nil)

        guard let authenticationBox = context.authenticationService.activeMastodonAuthenticationBox.value else {
            return
        }

        Task { @MainActor in
            let response = try await context.apiService.notification(
                notificationID: notificationID,
                authenticationBox: authenticationBox
            )
            let userID = response.value.account.id
            
            let _user: MastodonUser? = try await context.managedObjectContext.perform {
                let request = MastodonUser.sortedFetchRequest
                request.predicate = MastodonUser.predicate(domain: authenticationBox.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: authenticationBox.domain,
                    userID: userID,
                    authorization: authenticationBox.userAuthorization
                )
                
                let _user: MastodonUser? = try await context.managedObjectContext.perform {
                    let request = MastodonUser.sortedFetchRequest
                    request.predicate = MastodonUser.predicate(domain: authenticationBox.domain, id: userID)
                    request.fetchLimit = 1
                    return context.managedObjectContext.safeFetch(request).first
                }
                
                self.user = _user
            }
        }   // end Task
    }
    
}