lockdown-iOS-mirror/LockdowniOS/Scenes/Questionnaire/ViewModel/StepsViewModel.swift

//
//  StepsViewModel.swift
//  Lockdown
//
//  Created by Pavel Vilbik on 23.06.23.
//  Copyright © 2023 Confirmed Inc. All rights reserved.
//

import UIKit

protocol StepsViewProtocol: AnyObject {
    func changeContent()
    func close(completion: (() -> Void)?)
    func showSelectCountry(with viewModel: SelectCountryViewModelProtocol)
}

protocol StepViewModelProtocol {
    func contentView() -> UITableView
    var isSkiped: Bool { get set }
    var step: Steps { get }
    var message: String? { get }
}

class StepsViewModel {
    private lazy var steps: [StepViewModelProtocol] = [
        WhatProblemStepViewModel(),
        questionsStep
    ]
    private lazy var questionsStep: QuestionsStepViewModel = {
        let viewModel = QuestionsStepViewModel()
        viewModel.selectCountry = { [weak self] in self?.selectCountry(viewModel: $0) }
        return viewModel
    }()
    
    private weak var view: StepsViewProtocol?
    
    var showSkipButton: Bool {
        stepViewModel.step.showSkipButton
    }
    
    var stepsCount: Int {
        steps.count
    }
    
    var actionTitle: String {
        stepViewModel.step.actionTitle
    }
    
    var currentStepIndex = 0
    
    var stepViewModel: StepViewModelProtocol {
        steps[currentStepIndex]
    }
    
    var sendMessage: ((String) -> Void)?
    
    init(sendMessage: ((String) -> Void)?) {
        self.sendMessage = sendMessage
    }
    
    func bind(_ view: StepsViewProtocol) {
        self.view = view
        view.changeContent()
    }
    
    func performStepAction() {
        guard currentStepIndex != steps.count - 1 else {
            finishFlow()
            return
        }
        
        currentStepIndex += 1
        view?.changeContent()
    }
    
    func skipStep() {
        guard stepViewModel.step.showSkipButton else { return }
        
        steps[currentStepIndex].isSkiped = true
        performStepAction()
    }
    
    func backPressed() {
        guard currentStepIndex > 0 else {
            view?.close(completion: nil)
            return
        }
        
        currentStepIndex -= 1
        view?.changeContent()
        steps[currentStepIndex].isSkiped = false
    }
    
    func selectCountry(viewModel: SelectCountryViewModelProtocol) {
        view?.showSelectCountry(with: viewModel)
    }
    
    func finishFlow() {
        let sendMessage = sendMessage
        let message = message()
        view?.close {
            if let message {
                sendMessage?(message)
            }
        }
    }
    
    private func message() -> String? {
        steps
            .compactMap { $0.message }
            .reduce("") { partialResult, message in
                partialResult + "\n" + message
            }
        
    }
}