iOS-一行代码搞定复杂的UIBarButtonItem创建（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)

/*
 *默认的左返回按钮
 */
+ (UIBarButtonItem *)backItemWithTarget:(id)target selector:(SEL)selector;

/**********UIBarButtonItem为文字的设置**********/
+ (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为图片的设置**********/
/*
 *通用leftBarItem imgName:图片名 target:目标对象 selector:相应方法
 */
+ (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:目标对象 selector:相应方法
 */
+ (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为文字的设置**********/

+ (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为图片的设置**********/

+ (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

    /// 默认的左返回按钮
    ///
    /// - Parameters:
    ///   - target: 目标
    ///   - 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为文字的设置**********/
    
    /// 左Item为纯文字
    ///
    /// - Parameters:
    ///   - title: 标题
    ///   - target: 目标
    ///   - 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: 标题
    ///   - target: 目标
    ///   - selector: 方法
    ///   - block: 按钮回调
    /// - 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: 标题
    ///   - target: 目标
    ///   - 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: 标题
    ///   - target: 目标
    ///   - selector: 方法
    ///   - block: 按钮回调
    /// - 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为图片的设置**********/
    
    
    /// 左Item为纯图片
    ///
    /// - Parameters:
    ///   - imgName: 图片名
    ///   - target: 目标
    ///   - 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: 目标
    ///   - selector: 方法
    ///   - block: 按钮回调
    /// - 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: 目标
    ///   - 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: 目标
    ///   - selector: 方法
    ///   - block: 按钮回调
    /// - 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
    }


    
}

最后编辑于：2017.12.07 06:23:29

人面猴
序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
沈念sama阅读 204,530评论 6赞 478
死咒
序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
沈念sama阅读 86,403评论 2赞 381
救了他两次的神仙让他今天三更去死
文/潘晓璐我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
开封第一讲书人阅读 151,120评论 0赞 337
道士缉凶录：失踪的卖姜人
文/不坏的土叔我叫张陵，是天一观的道长。经常有香客问我，道长，这世上最难降的妖魔是什么？我笑而不...
开封第一讲书人阅读 54,770评论 1赞 277
港岛之恋（遗憾婚礼）
正文为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
茶点故事阅读 63,758评论 5赞 367
恶毒庶女顶嫁案：这布局不是一般人想出来的
文/花漫我一把揭开白布。她就那样静静地躺着，像睡着了一般。火红的嫁衣衬着肌肤如雪。梳的纹丝不乱的头发上，一...
开封第一讲书人阅读 48,649评论 1赞 281
城市分裂传说
那天，我揣着相机与录音，去河边找鬼。笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。我是一名探鬼主播，决...
沈念sama阅读 38,021评论 3赞 398
双鸳鸯连环套：你想象不到人心有多黑
文/苍兰香墨我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
开封第一讲书人阅读 36,675评论 0赞 258
万荣杀人案实录
序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
沈念sama阅读 40,931评论 1赞 299
护林员之死
正文独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋以下内容为张勋视角年9月15日...
茶点故事阅读 35,659评论 2赞 321
白月光启示录
正文我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
茶点故事阅读 37,751评论 1赞 330
活死人
序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
沈念sama阅读 33,410评论 4赞 321
日本核电站爆炸内幕
正文年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
茶点故事阅读 39,004评论 3赞 307
男人毒药：我在死后第九天来索命
文/蒙蒙一、第九天我趴在偏房一处隐蔽的房顶上张望。院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
开封第一讲书人阅读 29,969评论 0赞 19
一桩弑父案，背后竟有这般阴谋
文/苍兰香墨我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。一阵脚步声响...
开封第一讲书人阅读 31,203评论 1赞 260
情欲美人皮
我被黑心中介骗来泰国打工，没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。一个月前我还...
沈念sama阅读 45,042评论 2赞 350
代替公主和亲
正文我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
茶点故事阅读 42,493评论 2赞 343

iOS-一行代码搞定复杂的UIBarButtonItem创建（OC+Swift）

推荐阅读更多精彩内容