PPBadgeView
1、前言
最近項(xiàng)目的一個(gè)需求是在UIView, UITabBarItem, UIBarButtonItem 這三種類型的控件上添加消息提醒小圓點(diǎn)(Badge),一開始找了一個(gè)星星很多的Badge庫(kù) RKNotificationHub,但發(fā)現(xiàn)其不能很好的滿足項(xiàng)目需求(在UITabBarItem表現(xiàn)不是很好), 于是就自己動(dòng)手寫了PPBadgeView,可以很方便的為UIView, UITabBarItem, UIBarButtonItem及其子類添加Badge, 支持** Objective-C與Swift **.

2、原理
PPBadgeView的實(shí)現(xiàn)原理很簡(jiǎn)單: ** 將一個(gè)UILabel控件(Badge,也可為其它)添加到UIView(及子類)的對(duì)象上 **, 所以我們的關(guān)鍵點(diǎn)是要找到這個(gè)UIView對(duì)象! 像UIControl、UILabel、UIButton、UIImageView...這些控件就不用說(shuō)了,都是繼承UIView,可直接在它們的身上添加Badge,但是 ** UITabBarItem **, ** UIBarButtonItem **就不同了,它們兩個(gè)并不是繼承的UIView,該怎么辦?
2.1、給UITabBarItem添加Badge
通過(guò)系統(tǒng)的API可以看到** UITabBarItem **繼承關(guān)系:
** UIBarButtonItem --> UIBarItem --> NSObject **,
并沒有看見UIView(及子類) , UITabBarItem/UIBarItem的公開屬性里也沒發(fā)現(xiàn)可用的UIView...不急,還記得Xcode自帶的UI視圖調(diào)試神器嗎?
運(yùn)行Demo后點(diǎn)擊打開,可清楚的看到底部欄的UITabBarItem內(nèi)有一個(gè)UITabBarButton,其下屬還有一個(gè)UITabBarSwappableImageView的圖片控件,我們要找的就是這個(gè)UITabBarSwappableImageView
接下來(lái)就是要獲取這個(gè)UITabBarSwappableImageView,我們可以使用Runtime + KVC 的方式:
- 先利用runtime獲的UITabBarButton的對(duì)象名稱(具體方法可以查看這篇博客:三分鐘教會(huì)你runtime獲取屬性和成員變量), 最后打印的結(jié)果為(因打印的內(nèi)容太多,這里只貼出最關(guān)鍵的結(jié)果):
打印結(jié)果: UITabBarItem內(nèi)的成員變量類型為: @"UITabBarButton",名字為: _view - 再使用KVC取出這個(gè)UITabBarButton對(duì)象,遍歷出UITabBarSwappableImageView對(duì)象
UIView *tabBarButton = [tabBarItem valueForKey:@"_view"];
for (UIView *subView in tabBarButton.subviews) {
if ([subView isKindOfClass:NSClassFromString(@"UITabBarSwappableImageView")]) {
return subView;
}
}
找到了Badge可以依靠的UIView, 剩下添加Badge的工作是不是變得容易很多了,這里不多說(shuō), 在PPBadgeView里有詳細(xì)代碼.
2.2、給UIBarButtonItem添加Badge
通過(guò)系統(tǒng)的API可以看到** UIBarButtonItem **繼承關(guān)系:
** UIBarButtonItem --> UIBarItem --> NSObject **
和UITabBarItem一樣其公開的屬性也是沒有可用的UIView對(duì)象的,不過(guò)有了上面的??經(jīng)驗(yàn),獲取UIBarButtonItem中的UIView(及子類)也是同樣的做法, 看圖:
我們要找的就是UINavigationButton中的UIImageView屬性了
很奇怪,利用runtime獲取出來(lái)UINavigationButton類名稱為"UIView"
打印結(jié)果: UIBarButtonItem內(nèi)的成員變量類型為: @"UIView",名字為: _view
這里需要注意的是, UINavigationButton中的UIImageView對(duì)象的layer默認(rèn)masksToBounds = YES,在取出的時(shí)候需要設(shè)置masksToBounds = NO:
UIView *navigationButton = [barButtonItem valueForKey:@"_view"];
for (UIView *subView in navigationButton.subviews) {
if ([subView isKindOfClass:NSClassFromString(@"UIImageView")]) {
subView.layer.masksToBounds = NO;
return subView;
}
}
當(dāng)然,你也可以直接使用navigationButton來(lái)做Badge的父視圖...
** ========================= **
** 2017-06-27更新:** UITabBarItem分類/拓展移除掉直接比對(duì)"私有屬性API:UITabBarSwappableImageView"的代碼,規(guī)避上架審核被拒的風(fēng)險(xiǎn).非常感謝@iOS程序犭袁 大大的指出的這個(gè)問題
** ========================= **
結(jié)束
PPBadgeView現(xiàn)已托管到GitHub維護(hù),有Objective-C和Swift雙版本,支持CocoaPods導(dǎo)入,地址: https://github.com/jkpang/PPBadgeView
我的GitHub主頁(yè): https://github.com/jkpang