ProHUD/Source/Alert/AlertModel.swift

//
//  AlertModel.swift
//  ProHUD
//
//  Created by xaoxuu on 2019/7/29.
//  Copyright © 2019 Titan Studio. All rights reserved.
//

import UIKit

public extension Alert {
    class ViewModel {
        
        /// 使用场景
        public var scene = ProHUD.Scene.default
        
        /// 标题
        public var title: String? {
            didSet {
                vc?.titleLabel?.text = title
            }
        }
        
        /// 正文
        public var message: String? {
            didSet {
                vc?.bodyLabel?.text = message
            }
        }
        
        /// 图标
        public var icon: UIImage? {
            didSet {
                vc?.imageView.image = icon
            }
        }
        
        /// 持续时间（为空代表根据场景不同的默认配置，为0代表无穷大）
        public var duration: TimeInterval? {
            didSet {
                updateDuration()
            }
        }
        
        public weak var vc: Alert?
        
        // MARK: 私有
        
        /// 持续时间
        var durationBlock: DispatchWorkItem?
        
        /// 强制退出按钮
        var hideTimerBlock: DispatchWorkItem?
        
        /// 强制退出代码
        var forceQuitCallback: (() -> Void)?
        
        func updateDuration() {
            durationBlock?.cancel()
            if let t = duration ?? scene.alertDuration, t > 0 {
                durationBlock = DispatchWorkItem(block: { [weak self] in
                    if let vc = self?.vc {
                        if vc.buttonEvents.count == 0 {
                            vc.pop()
                        }
                    }
                })
                DispatchQueue.main.asyncAfter(deadline: .now()+t, execute: durationBlock!)
            } else {
                durationBlock = nil
            }
        }
        
    }
    
}

public extension Alert.ViewModel {
    
    /// 添加按钮
    /// - Parameter style: 样式
    /// - Parameter text: 标题
    /// - Parameter handler: 事件处理
    @discardableResult func add(action style: UIAlertAction.Style, title: String?, handler: (() -> Void)?) -> UIButton {
        return vc!.insert(action: nil, style: style, title: title, handler: handler)
    }
    
    /// 插入按钮
    /// - Parameter index: 索引
    /// - Parameter style: 样式
    /// - Parameter title: 标题
    /// - Parameter handler: 事件处理
    @discardableResult func insert(action index: Int, style: UIAlertAction.Style, title: String?, handler: (() -> Void)?) -> UIButton {
        return vc!.insert(action: index, style: style, title: title, handler: handler)
    }
    
    /// 更新按钮
    /// - Parameter index: 索引
    /// - Parameter style: 样式
    /// - Parameter title: 标题
    /// - Parameter handler: 事件处理
    func update(action index: Int, style: UIAlertAction.Style, title: String?, handler: (() -> Void)?) {
        vc?.update(action: index, style: style, title: title, handler: handler)
    }
    
    /// 移除按钮
    /// - Parameter index: 索引
    func remove(action index: Int...) {
        for (i, idx) in index.enumerated() {
            vc?.remove(action: idx-i)
        }
    }
    
    
}