iOS-一行代碼搞定復(fù)雜的UIBarButtonItem創(chuàng)建（OC+Swift）

MYCSDN:samuelandkevin

//  OC版
//  UIBarButtonItem+Custom.h
//  samuelandkevin
//
//  Created by samuelandkevin on 16/11/30.
//  Copyright ? 2016年 samuelandkevin. All rights reserved.
//

#import <UIKit/UIKit.h>

@interface UIBarButtonItem (Custom)

/*
 *默認(rèn)的左返回按鈕
 */
+ (UIBarButtonItem *)backItemWithTarget:(id)target selector:(SEL)selector;

/**********UIBarButtonItem為文字的設(shè)置**********/
+ (UIBarButtonItem *)leftItemWithTitle:(NSString *)title target:(id)target selector:(SEL)selector;
+ (UIBarButtonItem *)leftItemWithTitle:(NSString *)title target:(id)target selector:(SEL)selector block:(void(^)(UIButton *btn))block;

+ (UIBarButtonItem *)rightItemWithTitle:(NSString *)title target:(id)target selector:(SEL)selector;
+ (UIBarButtonItem *)rightItemWithTitle:(NSString *)title target:(id)target selector:(SEL)selector block:(void(^)(UIButton *btn))block;


/**********UIBarButtonItem為圖片的設(shè)置**********/
/*
 *通用leftBarItem imgName:圖片名 target:目標(biāo)對(duì)象 selector:相應(yīng)方法
 */
+ (UIBarButtonItem *)leftItemWithImgName:(NSString *)imgName target:(id)target selector:(SEL)selector;

+ (UIBarButtonItem *)leftItemWithImgName:(NSString *)imgName target:(id)target selector:(SEL)selector block:(void(^)(UIButton *btn))block;

/*
 *通用rightBarItem imgName:圖片名 target:目標(biāo)對(duì)象 selector:相應(yīng)方法
 */
+ (UIBarButtonItem *)rightItemWithImgName:(NSString *)imgName target:(id)target selector:(SEL)selector;

+ (UIBarButtonItem *)rightItemWithImgName:(NSString *)imgName target:(id)target selector:(SEL)selector block:(void(^)(UIButton *btn))block;

@end

//
//  UIBarButtonItem+Custom.m
//  samuelandkevin
//
//  Created by samuelandkevin on 16/11/30.
//  Copyright ? 2016年 samuelandkevin. All rights reserved.
//

#import "UIBarButtonItem+Custom.h"
#import <objc/runtime.h>

static const char *btnBlockKey = "btnBlockKey";
@interface UIBarButtonItem()
@property (nonatomic,copy) void(^btnBlock)();
@end

@implementation UIBarButtonItem (Custom)

- (void (^)())btnBlock{
    return objc_getAssociatedObject(self, btnBlockKey);
}

- (void)setBtnBlock:(void (^)())btnBlock{
    objc_setAssociatedObject(self, btnBlockKey, btnBlock, OBJC_ASSOCIATION_COPY);
}

#pragma mark - Public
+ (UIBarButtonItem *)backItemWithTarget:(id)target selector:(SEL)selector{

    return  [self barButtonItemWithFrame:CGRectMake(0, 0, 40, 40) imgName:@"leftarrow" imageEdgeInsets:UIEdgeInsetsMake(0, -30, 0, 0) target:target selector:selector];
}

/**********UIBarButtonItem為文字的設(shè)置**********/

+ (UIBarButtonItem *)leftItemWithTitle:(NSString *)title target:(id)target selector:(SEL)selector{
     return [self barButtonItemWithFrame:CGRectMake(0, 0, 40, 40) title:title target:target selector:selector block:nil];
}

+ (UIBarButtonItem *)leftItemWithTitle:(NSString *)title target:(id)target selector:(SEL)selector block:(void(^)(UIButton *btn))block{
     return  [self barButtonItemWithFrame:CGRectMake(0, 0, 40, 40) title:title target:target selector:selector block:block];
}

+ (UIBarButtonItem *)rightItemWithTitle:(NSString *)title target:(id)target selector:(SEL)selector{
    return [self barButtonItemWithFrame:CGRectMake(0, 0, 40, 40) title:title target:target selector:selector block:nil];
}

+ (UIBarButtonItem *)rightItemWithTitle:(NSString *)title target:(id)target selector:(SEL)selector block:(void(^)(UIButton *btn))block{
    return  [self barButtonItemWithFrame:CGRectMake(0, 0, 40, 40) title:title target:target selector:selector block:block];
}


/**********UIBarButtonItem為圖片的設(shè)置**********/

+ (UIBarButtonItem *)leftItemWithImgName:(NSString *)imgName target:(id)target selector:(SEL)selector{
    return  [self barButtonItemWithFrame:CGRectMake(0, 0, 44, 44) imgName:imgName imageEdgeInsets:UIEdgeInsetsMake(0, 0, 0, 20) target:target selector:selector];
}

+ (UIBarButtonItem *)leftItemWithImgName:(NSString *)imgName target:(id)target selector:(SEL)selector block:(void(^)(UIButton *btn))block{
     return  [self barButtonItemWithFrame:CGRectMake(0, 0, 44, 44) imgName:imgName imageEdgeInsets:UIEdgeInsetsMake(0, 0, 0, 20) target:target selector:selector block:block];
}

+ (UIBarButtonItem *)rightItemWithImgName:(NSString *)imgName target:(id)target selector:(SEL)selector{
    
    return  [self barButtonItemWithFrame:CGRectMake(0, 0, 44, 44) imgName:imgName imageEdgeInsets:UIEdgeInsetsMake(0, 0, 0, -20) target:target selector:selector];
    
}

+ (UIBarButtonItem *)rightItemWithImgName:(NSString *)imgName target:(id)target selector:(SEL)selector block:(void(^)(UIButton *btn))block{
    return  [self barButtonItemWithFrame:CGRectMake(0, 0, 44, 44) imgName:imgName imageEdgeInsets:UIEdgeInsetsMake(0, 0, 0, -20) target:target selector:selector block:block];
}

#pragma mark - Private

+ (UIButton *)barButtonWithFrame:(CGRect)frame imgName:(NSString *)imgName imageEdgeInsets:(UIEdgeInsets)imageEdgeInsets target:(id)target selector:(SEL)selector{
    UIButton *btn = [UIButton buttonWithType:UIButtonTypeSystem];
    btn.frame = frame;
    [btn setImage:[UIImage imageNamed:imgName] forState:UIControlStateNormal];
    btn.titleLabel.font = [UIFont systemFontOfSize:18];
    btn.titleLabel.textColor = [UIColor whiteColor];
    [btn addTarget:target action:selector forControlEvents:UIControlEventTouchUpInside];
    btn.imageEdgeInsets = imageEdgeInsets;
    return btn;
}

+ (UIBarButtonItem *)barButtonItemWithFrame:(CGRect)frame imgName:(NSString *)imgName imageEdgeInsets:(UIEdgeInsets)imageEdgeInsets target:(id)target selector:(SEL)selector {

     UIButton *btn = [self barButtonWithFrame:frame imgName:imgName imageEdgeInsets:imageEdgeInsets target:target selector:selector];
    
    return  [[UIBarButtonItem alloc] initWithCustomView:btn];
}

+ (UIBarButtonItem *)barButtonItemWithFrame:(CGRect)frame imgName:(NSString *)imgName imageEdgeInsets:(UIEdgeInsets)imageEdgeInsets target:(id)target selector:(SEL)selector block:(void(^)(UIButton *btn))block{
    
   UIButton *btn = [self barButtonWithFrame:frame imgName:imgName imageEdgeInsets:imageEdgeInsets target:target selector:selector];
    
    UIBarButtonItem *item = [[UIBarButtonItem alloc] initWithCustomView:btn];
    if (block) {
        item.btnBlock = block;
        if (item.btnBlock) {
            item.btnBlock(btn);
        }
    }
    return  item;
}


+ (UIBarButtonItem *)barButtonItemWithFrame:(CGRect)frame title:(NSString *)title target:(id)target selector:(SEL)selector block:(void(^)(UIButton *btn))block{
    
    UIButton *btn = [UIButton buttonWithType:UIButtonTypeSystem];
    btn.frame = frame;
    [btn setTitle:title forState:UIControlStateNormal];
    btn.titleLabel.font = [UIFont systemFontOfSize:18];
    btn.titleLabel.textColor = [UIColor whiteColor];
    [btn addTarget:target action:selector forControlEvents:UIControlEventTouchUpInside];
    UIBarButtonItem *item = [[UIBarButtonItem alloc] initWithCustomView:btn];
    if (block) {
        item.btnBlock = block;
        if (item.btnBlock) {
            item.btnBlock(btn);
        }
    }
    return  item;
}



@end


//  Swift版
//  UIBarButtonItem+Extension.swift
//  samuelandkevin
//
//  Created by samuelandkevin on 2017/5/8.
//  Copyright ? 2017年 samuelandkevin. All rights reserved.
//

import Foundation


let btnBlockKey = "btnBlockKey"
typealias BtnBlock = ((_ btn:UIButton) -> Void)?

extension UIBarButtonItem {
    
    //定義按鈕閉包
    var btnBlock: BtnBlock {
        get{
            return objc_getAssociatedObject(self, btnBlockKey) as? (UIButton)->Void
        }
        set{
            objc_setAssociatedObject(self, btnBlockKey, newValue, .OBJC_ASSOCIATION_COPY_NONATOMIC)
        }
    
    }
    
    // MARK: - Public Method

    /// 默認(rèn)的左返回按鈕
    ///
    /// - Parameters:
    ///   - target: 目標(biāo)
    ///   - selector: 方法
    /// - Returns: UIBarButtonItem
    internal class func backItem(target:Any,selector:Selector) -> UIBarButtonItem{
    
        return _barButtonItem(frame: CGRect(x: 0, y: 0, width: 40, height: 40), imgName: "leftarrow", imageEdgeInsets: UIEdgeInsetsMake(0, -30, 0, 0), target: target, selector: selector)
    }
    
    /**********UIBarButtonItem為文字的設(shè)置**********/
    
    /// 左Item為純文字
    ///
    /// - Parameters:
    ///   - title: 標(biāo)題
    ///   - target: 目標(biāo)
    ///   - selector: 方法
    /// - Returns: UIBarButtonItem
    internal class func leftItem(title:String,target:Any,selector:Selector) -> UIBarButtonItem{
       
        return  _barButtonItem(frame: CGRect(x: 0, y: 0, width: 40, height: 40), title: title, target: target, selector: selector, block: nil)
    }
    
    
    /// 左Item為純文字
    ///
    /// - Parameters:
    ///   - title: 標(biāo)題
    ///   - target: 目標(biāo)
    ///   - selector: 方法
    ///   - block: 按鈕回調(diào)
    /// - Returns: UIBarButtonItem
    internal class func leftItem(title:String,target:Any,selector:Selector,block:BtnBlock) -> UIBarButtonItem{
        
        return  _barButtonItem(frame: CGRect(x: 0, y: 0, width: 40, height: 40), title: title, target: target, selector: selector, block: block)
    }
    
    
    /// 右Item為純文字
    ///
    /// - Parameters:
    ///   - title: 標(biāo)題
    ///   - target: 目標(biāo)
    ///   - selector: 方法
    /// - Returns: UIBarButtonItem
    internal class func rightItem(title:String,target:Any,selector:Selector) -> UIBarButtonItem{
        
        return  _barButtonItem(frame: CGRect(x: 0, y: 0, width: 40, height: 40), title: title, target: target, selector: selector, block: nil)
    }

    
    /// 右Item為純文字
    ///
    /// - Parameters:
    ///   - title: 標(biāo)題
    ///   - target: 目標(biāo)
    ///   - selector: 方法
    ///   - block: 按鈕回調(diào)
    /// - Returns: UIBarButtonItem
    internal class func rightItem(title:String,target:Any,selector:Selector,block:BtnBlock) -> UIBarButtonItem{
        
        return  _barButtonItem(frame: CGRect(x: 0, y: 0, width: 40, height: 40), title: title, target: target, selector: selector, block: block)
    }
    
    /**********UIBarButtonItem為圖片的設(shè)置**********/
    
    
    /// 左Item為純圖片
    ///
    /// - Parameters:
    ///   - imgName: 圖片名
    ///   - target: 目標(biāo)
    ///   - selector: 方法
    /// - Returns: UIBarButtonItem
    internal class func leftItem(imgName:String,target:Any,selector:Selector) -> UIBarButtonItem{
        
        return _barButtonItem(frame: CGRect(x: 0, y: 0, width: 44, height: 44), imgName: imgName, imageEdgeInsets: UIEdgeInsetsMake(0, 0, 0, 20), target: target, selector: selector)
    }
    
    
    /// 左Item為純圖片
    ///
    /// - Parameters:
    ///   - imgName: 圖片名
    ///   - target: 目標(biāo)
    ///   - selector: 方法
    ///   - block: 按鈕回調(diào)
    /// - Returns: UIBarButtonItem
    internal class func leftItem(imgName:String,target:Any,selector:Selector,block:BtnBlock) -> UIBarButtonItem{
        return _barButtonItem(frame: CGRect(x: 0, y: 0, width: 44, height: 44), imgName: imgName, imageEdgeInsets: UIEdgeInsetsMake(0, 0, 0, 20), target: target, selector: selector,block:block)
    }
    
    
    /// 右Item為純圖片
    ///
    /// - Parameters:
    ///   - imgName: 圖片名
    ///   - target: 目標(biāo)
    ///   - selector: 方法
    /// - Returns: UIBarButtonItem
    internal class func rightItem(imgName:String,target:Any,selector:Selector) -> UIBarButtonItem{
        
        return _barButtonItem(frame: CGRect(x: 0, y: 0, width: 44, height: 44), imgName: imgName, imageEdgeInsets: UIEdgeInsetsMake(0, 0, 0, -20), target: target, selector: selector)
    }
    
    
    /// 右Item為純圖片
    ///
    /// - Parameters:
    ///   - imgName: 圖片名
    ///   - target: 目標(biāo)
    ///   - selector: 方法
    ///   - block: 按鈕回調(diào)
    /// - Returns: UIBarButtonItem
    internal class func rightItem(imgName:String,target:Any,selector:Selector,block:BtnBlock) -> UIBarButtonItem{
        
        return _barButtonItem(frame: CGRect(x: 0, y: 0, width: 44, height: 44), imgName: imgName, imageEdgeInsets: UIEdgeInsetsMake(0, 0, 0, -20), target: target, selector: selector,block:block)
    }
    
    
    
   // MARK: - Private Method
    
    internal class func _barButton(frame:CGRect,imgName:String,imageEdgeInsets:UIEdgeInsets,target:Any,selector:Selector) -> UIButton{
        let btn = UIButton(type: .system)
        btn.frame = frame
        btn.setImage(UIImage(named:imgName), for: .normal)
        btn.titleLabel?.font = UIFont.systemFont(ofSize: 18)
        btn.titleLabel?.textColor = UIColor.white
        btn.addTarget(target, action: selector, for: .touchUpInside)
        btn.imageEdgeInsets = imageEdgeInsets
        return btn
    }
    
    internal class func _barButtonItem(frame:CGRect,imgName:String,imageEdgeInsets:UIEdgeInsets,target:Any,selector:Selector) -> UIBarButtonItem {
    
        let btn = _barButton(frame: frame, imgName: imgName, imageEdgeInsets: imageEdgeInsets, target: target, selector: selector)
        let barButtonItem = UIBarButtonItem(customView: btn)
        return barButtonItem
    }
    
    
    internal class func _barButtonItem(frame:CGRect,imgName:String,imageEdgeInsets:UIEdgeInsets,target:Any,selector:Selector,block:BtnBlock )-> UIBarButtonItem{
    
        let aBtn = _barButton(frame: frame, imgName: imgName, imageEdgeInsets: imageEdgeInsets, target: target, selector: selector)
        let item = UIBarButtonItem(customView: aBtn)
    
        if ((block) != nil) {
            item.btnBlock = block
            if ((item.btnBlock) != nil) {
                item.btnBlock!(aBtn)
            }
        }
        return  item;
    }

    
    
    internal class func _barButtonItem(frame:CGRect,title:String,target:Any,selector:Selector,block:BtnBlock ) -> UIBarButtonItem{
    
        let btn = UIButton(type: .system)
        btn.frame = frame
        btn.setTitle(title, for: .normal)
        btn.titleLabel?.font = UIFont.systemFont(ofSize: 18)
        btn.titleLabel?.textColor = UIColor.white
        btn.addTarget(target, action: selector, for: .touchUpInside)
        let item = UIBarButtonItem(customView: btn)
        if ((block) != nil) {
            item.btnBlock = block
            if ((item.btnBlock) != nil) {
                item.btnBlock!(btn)
            }
        }
        return  item
    }


    
}
最后編輯于 ：
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成，瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點(diǎn)，簡書系信息發(fā)布平臺(tái)，僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容