mirror of https://github.com/xaoxuu/ProHUD
116 lines
3.2 KiB
Swift
116 lines
3.2 KiB
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)
|
||
}
|
||
}
|
||
|
||
|
||
}
|