element-ios/RiotSwiftUI/Modules/Spaces/SpaceCreation/SpaceCreationEmailInvites/ViewModel/SpaceCreationEmailInvitesViewModel.swift

// File created from SimpleUserProfileExample
// $ createScreen.sh Spaces/SpaceCreation/SpaceCreationEmailInvites SpaceCreationEmailInvites
//
// 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 Combine
import SwiftUI

typealias SpaceCreationEmailInvitesViewModelType = StateStoreViewModel<SpaceCreationEmailInvitesViewState, SpaceCreationEmailInvitesStateAction, SpaceCreationEmailInvitesViewAction>

class SpaceCreationEmailInvitesViewModel: SpaceCreationEmailInvitesViewModelType, SpaceCreationEmailInvitesViewModelProtocol {
    // MARK: - Properties

    // MARK: Private

    private let creationParameters: SpaceCreationParameters
    private let service: SpaceCreationEmailInvitesServiceProtocol

    // MARK: Public

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

    // MARK: - Setup

    init(creationParameters: SpaceCreationParameters, service: SpaceCreationEmailInvitesServiceProtocol) {
        self.creationParameters = creationParameters
        self.service = service
        super.init(initialViewState: SpaceCreationEmailInvitesViewModel.defaultState(creationParameters: creationParameters, service: service))
    }

    private func setupServiceObserving() {
        let publisher = service.isLoadingSubject
            .map(SpaceCreationEmailInvitesStateAction.updateLoading)
            .eraseToAnyPublisher()
        dispatch(actionPublisher: publisher)
    }

    private static func defaultState(creationParameters: SpaceCreationParameters, service: SpaceCreationEmailInvitesServiceProtocol) -> SpaceCreationEmailInvitesViewState {
        let emailValidation = service.validate(creationParameters.emailInvites)
        let bindings = SpaceCreationEmailInvitesViewModelBindings(emailInvites: creationParameters.emailInvites)
        return SpaceCreationEmailInvitesViewState(
            title: creationParameters.isPublic ? VectorL10n.spacesCreationPublicSpaceTitle : VectorL10n.spacesCreationPrivateSpaceTitle,
            emailAddressesValid: emailValidation,
            loading: service.isLoadingSubject.value,
            bindings: bindings
        )
    }
    
    // MARK: - Public

    override func process(viewAction: SpaceCreationEmailInvitesViewAction) {
        switch viewAction {
        case .cancel:
            cancel()
        case .back:
            back()
        case .done:
            done()
        case .inviteByUsername:
            inviteByUsername()
        }
    }

    override class func reducer(state: inout SpaceCreationEmailInvitesViewState, action: SpaceCreationEmailInvitesStateAction) {
        switch action {
        case .updateEmailValidity(let emailValidity):
            state.emailAddressesValid = emailValidity
        case .updateLoading(let isLoading):
            state.loading = isLoading
        }
    }

    private func done() {
        creationParameters.emailInvites = context.emailInvites
        creationParameters.inviteType = .email
        let emailAddressesValidity = service.validate(context.emailInvites)
        
        dispatch(action: .updateEmailValidity(emailAddressesValidity))
        if context.emailInvites.reduce(true, { $0 && $1.isEmpty }) {
            completion?(.done)
        } else if emailAddressesValidity.reduce(true, { $0 && $1 }) {
            if service.isIdentityServiceReady {
                completion?(.done)
            } else {
                service.prepareIdentityService { [weak self] baseURL, accessToken in
                    self?.completion?(.needIdentityServiceTerms(baseURL, accessToken))
                } failure: { [weak self] error in
                    self?.completion?(.identityServiceFailure(error))
                }
            }
        }
    }

    private func cancel() {
        completion?(.cancel)
    }
    
    private func back() {
        completion?(.back)
    }
    
    private func inviteByUsername() {
        completion?(.inviteByUsername)
    }
}