element-ios/RiotSwiftUI/Modules/Spaces/SpaceCreation/SpaceCreationPostProcess/ViewModel/SpaceCreationPostProcessViewModel.swift

// File created from SimpleUserProfileExample
// $ createScreen.sh Spaces/SpaceCreation/SpaceCreationPostProcess SpaceCreationPostProcess
//
// Copyright 2021 New Vector Ltd
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//

import SwiftUI
import Combine



@available(iOS 14, *)
typealias SpaceCreationPostProcessViewModelType = StateStoreViewModel<SpaceCreationPostProcessViewState,
                                                                 SpaceCreationPostProcessStateAction,
                                                                 SpaceCreationPostProcessViewAction>
@available(iOS 14, *)
class SpaceCreationPostProcessViewModel: SpaceCreationPostProcessViewModelType, SpaceCreationPostProcessViewModelProtocol {

    // MARK: - Properties

    // MARK: Private

    private let spaceCreationPostProcessService: SpaceCreationPostProcessServiceProtocol
    private var updateNotificationObserver: Any?

    // MARK: Public

    var completion: ((SpaceCreationPostProcessViewModelResult) -> Void)?

    // MARK: - Setup

    static func makeSpaceCreationPostProcessViewModel(spaceCreationPostProcessService: SpaceCreationPostProcessServiceProtocol) -> SpaceCreationPostProcessViewModelProtocol {
        return SpaceCreationPostProcessViewModel(spaceCreationPostProcessService: spaceCreationPostProcessService)
    }

    private init(spaceCreationPostProcessService: SpaceCreationPostProcessServiceProtocol) {
        self.spaceCreationPostProcessService = spaceCreationPostProcessService
        super.init(initialViewState: Self.defaultState(spaceCreationPostProcessService: spaceCreationPostProcessService))
        setupTasksObserving()
    }

    private static func defaultState(spaceCreationPostProcessService: SpaceCreationPostProcessServiceProtocol) -> SpaceCreationPostProcessViewState {
        let tasks = spaceCreationPostProcessService.tasksSubject.value
        return SpaceCreationPostProcessViewState(
            avatar: spaceCreationPostProcessService.avatar,
            avatarImage: spaceCreationPostProcessService.avatarImage,
            tasks: tasks,
            isFinished: tasks.first?.state == .failure || tasks.reduce(true, { result, task in result && task.isFinished }),
            errorCount: tasks.reduce(0, { result, task in result + (task.state == .failure ? 1 : 0) })
        )
    }
    
    private func setupTasksObserving() {
        let tasksUpdatePublisher = spaceCreationPostProcessService.tasksSubject
            .map(SpaceCreationPostProcessStateAction.updateTasks)
            .eraseToAnyPublisher()
        dispatch(actionPublisher: tasksUpdatePublisher)
        updateNotificationObserver = NotificationCenter.default.addObserver(forName: SpaceCreationPostProcessViewModel.didUpdate, object: nil, queue: OperationQueue.main) { [weak self] notification in
            guard let self = self else {
                return
            }
            
            guard let state = notification.userInfo?[SpaceCreationPostProcessViewModel.newStateKey] as? SpaceCreationPostProcessViewState else {
                return
            }
            
            if state.isFinished && state.errorCount == 0 {
                guard let spaceId = self.spaceCreationPostProcessService.createdSpaceId else {
                    self.cancel()
                    return
                }
                
                self.done(spaceId: spaceId)
            }
        }
    }
    
    deinit {
        if let updateNotificationObserver = self.updateNotificationObserver {
            NotificationCenter.default.removeObserver(updateNotificationObserver)
        }
    }

    // MARK: - Public

    override func process(viewAction: SpaceCreationPostProcessViewAction) {
        switch viewAction {
        case .cancel:
            cancel()
        case .runTasks:
            runTasks()
        case .retry:
            runTasks()
        }
    }

    override class func reducer(state: inout SpaceCreationPostProcessViewState, action: SpaceCreationPostProcessStateAction) {
        switch action {
        case .updateTasks(let tasks):
            state.tasks = tasks
            state.isFinished = tasks.first?.state == .failure || tasks.reduce(true, { result, task in result && task.isFinished })
            state.errorCount = tasks.reduce(0, { result, task in result + (task.state == .failure ? 1 : 0) })
        }
        
        NotificationCenter.default.post(name: SpaceCreationPostProcessViewModel.didUpdate, object: nil, userInfo: [SpaceCreationPostProcessViewModel.newStateKey : state])
        
        UILog.debug("[SpaceCreationPostProcessViewModel] reducer with action \(action) produced state: \(state)")
    }

    private func done(spaceId: String) {
        completion?(.done(spaceId))
    }

    private func cancel() {
        completion?(.cancel)
    }
    
    private func runTasks() {
        spaceCreationPostProcessService.run()
    }
}

// MARK: - MXSpaceService notification constants
@available(iOS 14, *)
extension SpaceCreationPostProcessViewModel {
    /// Posted once the process is finished
    public static let didUpdate = Notification.Name("SpaceCreationPostProcessViewModelDidUpdate")
    
    public static let newStateKey = "newState"
}