JKSwiftExtension，测试用例在 UIButtonExtensionViewController.swift 里面
目录：
1、基本的扩展
2、链式调用
3、UIButton 图片 与 title 位置关系(提示：title和image要在设置布局关系之前设置)
4、自带倒计时功能的 Button
一、基本的扩展
// MARK:- 一、基本的扩展
public extension UIButton {
enum SmallButtonType {
case red
case pink
}
// MARK: 1.1、创建一个带颜色的 Button
/// 创建一个带颜色的 Button
/// - Parameters:
/// - type: 类型
/// - height: 高度
/// - Returns: 返回自身
@discardableResult
static func small(type: SmallButtonType = .red, height: CGFloat = 45) -> UIButton {
let normalColor: UIColor
let disabledColor: UIColor
let lineTypeNormal: LineType
let lineTypeDisable: LineType
let titleColorNormal: UIColor
let titleColorDisable: UIColor
switch type {
case .red:
normalColor = .hexStringColor(hexString: "#E54749")
disabledColor = .hexStringColor(hexString: "#CCCCCC")
lineTypeNormal = .none
lineTypeDisable = .none
titleColorNormal = .white
titleColorDisable = .white
case .pink:
normalColor = .hexStringColor(hexString: "#FFE8E8")
disabledColor = .hexStringColor(hexString: "#CCCCCC")
lineTypeNormal = .color(.hexStringColor(hexString: "#F6CDCD"))
lineTypeDisable = .color(.hexStringColor(hexString: "#9C9C9C"))
titleColorNormal = .hexStringColor(hexString: "#E54749")
titleColorDisable = .white
}
let btn = UIButton(type: .custom).font(.systemFont(ofSize: 16))
btn.setTitleColor(titleColorNormal, for: .normal)
btn.setTitleColor(titleColorDisable, for: .disabled)
btn.setBackgroundImage(drawSmallBtn(color: normalColor, height: height, lineType: lineTypeNormal), for: .normal)
btn.setBackgroundImage(drawSmallBtn(color: disabledColor, height: height, lineType: lineTypeDisable), for: .disabled)
btn.titleEdgeInsets = UIEdgeInsets(top: 0, left: 0, bottom: 13, right: 0)
return btn
}
// MARK: 1.2、创建一个常规的 Button
/// 创建一个常规的 Button
/// - Returns: 返回自身
static func normal() -> UIButton {
let btn = UIButton(type: .custom).font(.boldSystemFont(ofSize: 18))
btn.setTitleColor(.white, for: .normal)
btn.setTitleColor(.white, for: .disabled)
btn.setBackgroundImage(drawNormalBtn(color: .hexStringColor(hexString: "#E54749"))?.resizableImage(withCapInsets: UIEdgeInsets(top: 10, left: 15, bottom: 15, right: 15)), for: .normal)
btn.setBackgroundImage(drawNormalBtn(color: .hexStringColor(hexString: "#CCCCCC"))?.resizableImage(withCapInsets: UIEdgeInsets(top: 10, left: 15, bottom: 15, right: 15)), for: .disabled)
btn.titleEdgeInsets = UIEdgeInsets(top: 0, left: 0, bottom: 4, right: 0)
return btn
}
private static func drawSmallBtn(color: UIColor, height: CGFloat, lineType: LineType) -> UIImage? {
let rect = CGRect(x: 0, y: 0, width: 200, height: height + 20)
let path = UIBezierPath(roundedRect: CGRect(x: 10, y: 3, width: 180, height: height), cornerRadius: height / 2)
UIGraphicsBeginImageContextWithOptions(rect.size, false, UIScreen.main.scale)
let context = UIGraphicsGetCurrentContext()
context?.setFillColor(color.cgColor)
context?.addPath(path.cgPath)
context?.setShadow(offset: CGSize(width: 1, height: 4), blur: 10, color: color.withAlphaComponent(0.5).cgColor)
context?.fillPath()
switch lineType {
case let .color(color):
color.setStroke()
path.lineWidth = kPixel
path.stroke()
default:
break
}
let img = UIGraphicsGetImageFromCurrentImageContext()
UIGraphicsEndImageContext()
return img
}
private static func drawNormalBtn(color: UIColor) -> UIImage? {
let rect = CGRect(x: 0, y: 0, width: 260, height: 50)
let path = UIBezierPath(roundedRect: CGRect(x: 10, y: 3, width: 240, height: 40), cornerRadius: 3)
UIGraphicsBeginImageContextWithOptions(rect.size, false, UIScreen.main.scale)
let context = UIGraphicsGetCurrentContext()
context?.setFillColor(color.cgColor)
context?.addPath(path.cgPath)
context?.setShadow(offset: CGSize(width: 1, height: 2), blur: 6, color: color.withAlphaComponent(0.5).cgColor)
context?.fillPath()
let img = UIGraphicsGetImageFromCurrentImageContext()
UIGraphicsEndImageContext()
return img
}
}
二、链式调用
// MARK:- 二、链式调用
public extension UIButton {
// MARK: 2.1、设置title
/// 设置title
/// - Parameters:
/// - text: 文字
/// - state: 状态
/// - Returns: 返回自身
@discardableResult
func title(_ text: String, _ state: UIControl.State = .normal) -> Self {
setTitle(text, for: state)
return self
}
// MARK: 2.2、设置文字颜色
/// 设置文字颜色
/// - Parameters:
/// - color: 文字颜色
/// - state: 状态
/// - Returns: 返回自身
@discardableResult
func textColor(_ color: UIColor, _ state: UIControl.State = .normal) -> Self {
setTitleColor(color, for: state)
return self
}
// MARK: 2.3、设置字体大小(UIFont)
/// 设置字体大小
/// - Parameter font: 字体 UIFont
/// - Returns: 返回自身
@discardableResult
func font(_ font: UIFont) -> Self {
titleLabel?.font = font
return self
}
// MARK: 2.4、设置字体大小(CGFloat)
/// 设置字体大小(CGFloat)
/// - Parameter fontSize: 字体的大小
/// - Returns: 返回自身
@discardableResult
func font(_ fontSize: CGFloat) -> Self {
titleLabel?.font = UIFont.systemFont(ofSize: fontSize)
return self
}
// MARK: 2.5、设置字体粗体
/// 设置粗体
/// - Parameter fontSize: 设置字体粗体
/// - Returns: 返回自身
@discardableResult
func boldFont(_ fontSize: CGFloat) -> Self {
titleLabel?.font = UIFont.boldSystemFont(ofSize: fontSize)
return self
}
// MARK: 2.6、设置图片
/// 设置图片
/// - Parameters:
/// - image: 图片
/// - state: 状态
/// - Returns: 返回自身
@discardableResult
func image(_ image: UIImage?, _ state: UIControl.State = .normal) -> Self {
setImage(image, for: state)
return self
}
// MARK: 2.7、设置图片(通过Bundle加载)
/// 设置图片(通过Bundle加载)
/// - Parameters:
/// - bundle: Bundle
/// - imageName: 图片名字
/// - state: 状态
/// - Returns: 返回自身
@discardableResult
func image(in bundle: Bundle? = nil, _ imageName: String, _ state: UIControl.State = .normal) -> Self {
let image = UIImage(named: imageName, in: bundle, compatibleWith: nil)
setImage(image, for: state)
return self
}
// MARK: 2.8、设置图片(通过Bundle加载)
/// 设置图片(通过Bundle加载)
/// - Parameters:
/// - aClass: className bundle所在的类的类名
/// - bundleName: bundle 的名字
/// - imageName: 图片的名字
/// - state: 状态
/// - Returns: 返回自身
@discardableResult
func image(forParent aClass: AnyClass, bundleName: String, _ imageName: String, _ state: UIControl.State = .normal) -> Self {
guard let path = Bundle(for: aClass).path(forResource: bundleName, ofType: "bundle") else {
return self
}
let image = UIImage(named: imageName, in: Bundle(path: path), compatibleWith: nil)
setImage(image, for: state)
return self
}
// MARK: 2.9、设置图片(纯颜色的图片)
/// 设置图片(纯颜色的图片)
/// - Parameters:
/// - color: 图片颜色
/// - state: 状态
/// - Returns: 返回自身
@discardableResult
func image(_ color: UIColor, _ size: CGSize = CGSize(width: 20.0, height: 20.0), _ state: UIControl.State = .normal) -> Self {
let image = UIImage.image(color: color, size: size)
setImage(image, for: state)
return self
}
// MARK: 2.10、设置背景图片
/// 设置背景图片
/// - Parameters:
/// - image: 图片
/// - state: 状态
/// - Returns: 返回自身
@discardableResult
func bgImage(_ image: UIImage?, _ state: UIControl.State = .normal) -> Self {
setBackgroundImage(image, for: state)
return self
}
// MARK: 2.11、设置背景图片(通过Bundle加载)
/// 设置背景图片(通过Bundle加载)
/// - Parameters:
/// - aClass: className bundle所在的类的类名
/// - bundleName: bundle 的名字
/// - imageName: 图片的名字
/// - state: 状态
/// - Returns: 返回自身
@discardableResult
func bgImage(forParent aClass: AnyClass, bundleName: String, _ imageName: String, _: UIControl.State = .normal) -> Self {
guard let path = Bundle(for: aClass).path(forResource: bundleName, ofType: "bundle") else {
return self
}
let image = UIImage(named: imageName, in: Bundle(path: path), compatibleWith: nil)
setBackgroundImage(image, for: state)
return self
}
// MARK: 2.12、设置背景图片(通过Bundle加载)
/// 设置背景图片(通过Bundle加载)
/// - Parameters:
/// - bundle: Bundle
/// - imageName: 图片的名字
/// - state: 状态
/// - Returns: 返回自身
@discardableResult
func bgImage(in bundle: Bundle? = nil, _ imageName: String, _ state: UIControl.State = .normal) -> Self {
let image = UIImage(named: imageName, in: bundle, compatibleWith: nil)
setBackgroundImage(image, for: state)
return self
}
// MARK: 2.13、设置背景图片(纯颜色的图片)
/// 设置背景图片(纯颜色的图片)
/// - Parameters:
/// - color: 背景色
/// - state: 状态
/// - Returns: 返回自身
@discardableResult
func bgImage(_ color: UIColor, _ state: UIControl.State = .normal) -> Self {
let image = UIImage.image(color: color)
setBackgroundImage(image, for: state)
return self
}
// MARK: 2.14、按钮点击的变化
/// 按钮点击的变化
/// - Returns: 返回自身
@discardableResult
func confirmButton() -> Self {
let normalImage = UIImage.image(color: UIColor.hexStringColor(hexString: "#E54749"), size: CGSize(width: 10, height: 10), round: 4)?.resizableImage(withCapInsets: UIEdgeInsets(top: 5, left: 5, bottom: 5, right: 5))
let disableImg = UIImage.image(color: UIColor.hexStringColor(hexString: "#E6E6E6"), size: CGSize(width: 10, height: 10), round: 4)?.resizableImage(withCapInsets: UIEdgeInsets(top: 5, left: 5, bottom: 5, right: 5))
setBackgroundImage(normalImage, for: .normal)
setBackgroundImage(disableImg, for: .disabled)
return self
}
}
三、UIButton 图片 与 title 位置关系(提示：title和image要在设置布局关系之前设置)
// MARK:- 三、UIButton 图片 与 title 位置关系
/// UIButton 图片与title位置关系 https://www.jianshu.com/p/0f34c1b52604
public extension UIButton {
/// 图片 和 title 的布局样式
enum ImageTitleLayout {
case imgTop
case imgBottom
case imgLeft
case imgRight
}
// MARK: 3.1、设置图片和 title 的位置关系(提示：title和image要在设置布局关系之前设置)
/// 设置图片和 title 的位置关系(提示：title和image要在设置布局关系之前设置)
/// - Parameters:
/// - layout: 布局
/// - spacing: 间距
/// - Returns: 返回自身
@discardableResult
func setImageTitleLayout(_ layout: ImageTitleLayout, spacing: CGFloat = 0) -> Self {
switch layout {
case .imgLeft:
alignHorizontal(spacing: spacing, imageFirst: true)
case .imgRight:
alignHorizontal(spacing: spacing, imageFirst: false)
case .imgTop:
alignVertical(spacing: spacing, imageTop: true)
case .imgBottom:
alignVertical(spacing: spacing, imageTop: false)
}
return self
}
/// 水平方向
/// - Parameters:
/// - spacing: 间距
/// - imageFirst: 图片是否优先
private func alignHorizontal(spacing: CGFloat, imageFirst: Bool) {
let edgeOffset = spacing / 2
imageEdgeInsets = UIEdgeInsets(top: 0,
left: -edgeOffset,
bottom: 0,
right: edgeOffset)
titleEdgeInsets = UIEdgeInsets(top: 0,
left: edgeOffset,
bottom: 0,
right: -edgeOffset)
if !imageFirst {
self.transform = CGAffineTransform(scaleX: -1, y: 1)
imageView?.transform = CGAffineTransform(scaleX: -1, y: 1)
titleLabel?.transform = CGAffineTransform(scaleX: -1, y: 1)
}
contentEdgeInsets = UIEdgeInsets(top: 0, left: edgeOffset, bottom: 0, right: edgeOffset)
}
/// 垂直方向
/// - Parameters:
/// - spacing: 间距
/// - imageTop: 图片是不是在顶部
private func alignVertical(spacing: CGFloat, imageTop: Bool) {
guard let imageSize = self.imageView?.image?.size,
let text = self.titleLabel?.text,
let font = self.titleLabel?.font
else {
return
}
let labelString = NSString(string: text)
let titleSize = labelString.size(withAttributes: [NSAttributedString.Key.font: font])
let imageVerticalOffset = (titleSize.height + spacing) / 2
let titleVerticalOffset = (imageSize.height + spacing) / 2
let imageHorizontalOffset = (titleSize.width) / 2
let titleHorizontalOffset = (imageSize.width) / 2
let sign: CGFloat = imageTop ? 1 : -1
imageEdgeInsets = UIEdgeInsets(top: -imageVerticalOffset * sign,
left: imageHorizontalOffset,
bottom: imageVerticalOffset * sign,
right: -imageHorizontalOffset)
titleEdgeInsets = UIEdgeInsets(top: titleVerticalOffset * sign,
left: -titleHorizontalOffset,
bottom: -titleVerticalOffset * sign,
right: titleHorizontalOffset)
// increase content height to avoid clipping
let edgeOffset = (min(imageSize.height, titleSize.height) + spacing)/2
contentEdgeInsets = UIEdgeInsets(top: edgeOffset, left: 0, bottom: edgeOffset, right: 0)
}
}
四、自带倒计时功能的 Button"
// MARK:- 四、自带倒计时功能的 Button（有待改进）
/// 自带倒计时功能的Button
/// - 状态分为 [倒计时中，倒计时完成]，分别提供回调
/// - 需要和业务结合时，后期再考虑
public extension UIButton {
// MARK: 4.1、设置 Button 倒计时
/// 设置 Button 倒计时
/// - Parameters:
/// - count: 最初的倒计时数字
/// - timering: 倒计时中的 Block
/// - complete: 倒计时完成的 Block
/// - timeringPrefix: 倒计时文字的：前缀
/// - completeText: 倒计时完成后的文字
func countDown(_ count: Int, timering: TimeringBlock? = nil, complete: CompletionBlock? = nil, timeringPrefix: String = "再次获取", completeText: String = "重新获取") {
isEnabled = false
let begin = ProcessInfo().systemUptime
let c_default = UIColor.hexStringColor(hexString: "#2798fd")
let c_default_disable = UIColor.hexStringColor(hexString: "#999999")
self.textColor(titleColor(for: .normal) ?? c_default)
self.textColor(titleColor(for: .disabled) ?? c_default_disable, .disabled)
var remainingCount: Int = count {
willSet {
setTitle(timeringPrefix + "(\(newValue)s)", for: .normal)
if newValue <= 0 {
setTitle(completeText, for: .normal)
}
}
}
self.invalidate()
self.timer = DispatchSource.makeTimerSource(queue:DispatchQueue.global())
self.timer?.schedule(deadline: .now(), repeating: .seconds(1))
self.isTiming = true
self.timer?.setEventHandler(handler: {
DispatchQueue.main.async {
remainingCount = count - Int(ProcessInfo().systemUptime - begin)
if remainingCount <= 0 {
if let cb = complete {
cb()
}
// 计时结束后，enable的条件
self.isEnabled = self.reEnableCond ?? true
self.isTiming = false
self.invalidate()
} else {
if let tb = timering {
tb(remainingCount)
}
}
}
})
self.timer?.resume()
}
// MARK: 4.2、是否可以点击
/// 是否可以点击
var reEnableCond: Bool? {
get {
if let value = objc_getAssociatedObject(self, &TimerKey.reEnableCond_key) {
return value as? Bool
}
return nil
}
set {
objc_setAssociatedObject(self, &TimerKey.reEnableCond_key, newValue, .OBJC_ASSOCIATION_RETAIN_NONATOMIC)
}
}
// MARK: 4.3、是否正在倒计时
/// 是否正在倒计时
var isTiming: Bool {
get {
if let value = objc_getAssociatedObject(self, &TimerKey.running_key) {
return value as! Bool
}
// 默认状态
return false
}
set {
objc_setAssociatedObject(self, &TimerKey.running_key, newValue, .OBJC_ASSOCIATION_RETAIN_NONATOMIC)
}
}
// MARK: 4.3、处于倒计时时，前缀文案，如：「再次获取」 + (xxxs)
/// 处于倒计时时，前缀文案，如：「再次获取」 + (xxxs)
var timeringPrefix: String? {
get {
if let value = objc_getAssociatedObject(self, &TimerKey.timeringPrefix_key) {
return value as? String
}
return nil
}
set {
objc_setAssociatedObject(self, &TimerKey.timeringPrefix_key, newValue, .OBJC_ASSOCIATION_RETAIN_NONATOMIC)
}
}
// MARK: 销毁定时器
/// 销毁定时器
func invalidate() {
if self.timer != nil {
self.timer?.cancel()
self.timer = nil
}
}
// MARK: 时间对象
/// 时间对象
var timer: DispatchSourceTimer? {
get {
if let value = objc_getAssociatedObject(self, &TimerKey.timer_key) {
return value as? DispatchSourceTimer
}
return nil
}
set {
objc_setAssociatedObject(self, &TimerKey.timer_key, newValue, .OBJC_ASSOCIATION_RETAIN_NONATOMIC)
}
}
typealias TimeringBlock = (Int) -> ()
typealias CompletionBlock = () -> ()
private struct TimerKey {
static var timer_key = "timer_key"
static var running_key = "running_key"
static var timeringPrefix_key = "timering_prefix_key"
static var reEnableCond_key = "re_enable_cond_key"
}
}
