wjtx-client-ios/ios-template/common/ZExtensions.swift

//
//  ZExtensions.swift
//  武极天下
//
//  Created by zhl on 2021/7/22.
//  Copyright © 2021 egret. All rights reserved.
//

import Foundation
import UIKit

fileprivate var LoadingViewKey = "LoadingViewKey"

extension NSObject {
    
    /// Returns The view controller associated with the currently visible view.
    ///
    /// - Returns: The view controller associated with the currently visible view.
    public func currentViewController() -> UIViewController? {
        let sharedApp = UIApplication.shared
        
        let window = sharedApp.keyWindow ?? sharedApp.windows[0]
        let viewController = window.rootViewController
        
        return findCurrentViewController(from: viewController)
    }
    
    private func findCurrentViewController(from viewController: UIViewController?) -> UIViewController? {
        
        guard var vc = viewController else {
            return nil
        }
        
        while true {
            if let tvc = vc.presentedViewController {
                vc = tvc
            } else if vc.isKind(of: UITabBarController.self) {
                let tbc = vc as! UITabBarController
                if let tvc = tbc.selectedViewController {
                    vc = tvc
                }
            } else if vc.isKind(of: UINavigationController.self) {
                let nc = vc as! UINavigationController
                if let tvc = nc.visibleViewController {
                    vc = tvc
                }
            } else {
                if vc.children.count > 0 {
                    if let tvc = vc.children.last {
                        vc = tvc
                    }
                }
                break
            }
        }
        
        return vc
    }
    
    /// Shows the tips for user.
    public func showTipsMessage(_ message: String) -> Void {
        
        guard let vc = self.currentViewController(), !vc.isKind(of: UIAlertController.self) else {
            return
        }
        
        let alertController = UIAlertController(title: message, message: nil, preferredStyle: UIAlertController.Style.alert)
        
        vc.present(alertController, animated: true, completion: nil)
        
        DispatchQueue.main.asyncAfter(deadline: .now() + 2.0) {
            alertController.dismiss(animated: true, completion: nil)
        }
    }
    
    /// Shows an alert view controller.
    public func showAlert(withTitle title: String?,
                          message: String?,
                          cancelButtonTitle: String? = nil,
                          cancel cancelHandler: ((UIAlertAction) -> Void)? = nil,
                          confirmButtonTitle: String?,
                          execute executableHandler: ((UIAlertAction) -> Void)? = nil) {
        
        guard let vc = self.currentViewController() else {
            return
        }
        
        let alertController = UIAlertController(title: title, message: message, preferredStyle: UIAlertController.Style.alert)
        
        if let t = cancelButtonTitle, t.count > 0 {
            let action = UIAlertAction(title: t, style: UIAlertAction.Style.cancel, handler: cancelHandler)
            alertController.addAction(action)
        }
        
        if let t = confirmButtonTitle, t.count > 0 {
            let action = UIAlertAction(title: t, style: UIAlertAction.Style.default, handler: executableHandler)
            alertController.addAction(action)
        }
        
        vc.present(alertController, animated: true, completion: nil)
    }
    
    /// Shows a loading panel.
    public func showLoading(_ text: String) {
        
        let value = objc_getAssociatedObject(self, &LoadingViewKey)
        if value != nil {
            return
        }
        
        let loadingView = ZLoadingView()
        loadingView.show(text)
        loadingView.color = COLOR_RGBA(10, 10, 10, 0.75)
        loadingView.indicatorColor = COLOR_RGB(54, 205, 64)
        loadingView.textColor = COLOR_RGB(248, 248, 248)
        
        objc_setAssociatedObject(self, &LoadingViewKey, loadingView, .OBJC_ASSOCIATION_RETAIN_NONATOMIC)
    }
    
    /// Hides a loading panel.
    public func hideLoading() {
        
        let value = objc_getAssociatedObject(self, &LoadingViewKey)
        guard let loadingView = value as? ZLoadingView else {
            return
        }
        
        loadingView.hide()
        
        objc_setAssociatedObject(self, &LoadingViewKey, nil, .OBJC_ASSOCIATION_RETAIN_NONATOMIC)
    }
    
}

extension UIView {
    
    /// This method is used to set the corner.
    ///
    /// - Parameters:
    ///   - rectCorner: The corners of a rectangle.
    ///   - radius: The radius of each corner.
    public func setCorner(rectCorner: UIRectCorner = UIRectCorner.allCorners, radius: CGFloat) {
        
        let maskLayer = CAShapeLayer()
        let w = self.bounds.size.width
        let h = self.bounds.size.height
        maskLayer.frame = CGRect(x: 0, y: 0, width: w, height: h)
        
        let path = UIBezierPath(roundedRect: self.bounds, byRoundingCorners: rectCorner, cornerRadii: CGSize(width: radius, height: radius))
        maskLayer.path = path.cgPath
        
        self.layer.mask = maskLayer
    }
    
    /// This method is used to set the border.
    ///
    /// - Parameters:
    ///   - rectCorner: The corners of a rectangle.
    ///   - radius: The radius of each corner.
    ///   - lineWidth: Specifies the line width of the shape’s path.
    ///   - color: The color used to stroke the shape’s path.
    public func setBorder(rectCorner: UIRectCorner = UIRectCorner.allCorners, radius: CGFloat, lineWidth: CGFloat, color: UIColor?) {
        
        let maskLayer = CAShapeLayer()
        let w = self.bounds.size.width
        let h = self.bounds.size.height
        maskLayer.frame = CGRect(x: 0, y: 0, width: w, height: h)
        
        let borderLayer = CAShapeLayer()
        borderLayer.frame = CGRect(x: 0, y: 0, width: w, height: h)
        borderLayer.lineWidth = lineWidth
        borderLayer.strokeColor = color?.cgColor
        borderLayer.fillColor = UIColor.clear.cgColor
        
        let path = UIBezierPath(roundedRect: self.bounds, byRoundingCorners: rectCorner, cornerRadii: CGSize(width: radius, height: radius))
        borderLayer.path = path.cgPath
        maskLayer.path = path.cgPath
        
        self.layer.insertSublayer(borderLayer, at: 0)
        self.layer.mask = maskLayer
    }
    
}