iOS 怎麼修改UITabBarItem角標badge顏色

溫特兒發表於2019-03-16

廢話不多說,直接上程式碼!

Paste_Image.png

為了不影響系統自帶功能,所以新增一個UITabBarItem的分類:

// .h 檔案
@interface UITabBarItem (Custom)
// 設定角標值時,替換系統的 'setBadgeValue:'方法
- (void)my_setBadgeValue:(NSString *)badgeValue;
@end
複製程式碼
// .m 檔案
@implementation UITabBarItem (Custom)

- (void)my_setBadgeValue:(NSString *)badgeValue
{
    // 先設定角標值
    [self setBadgeValue:badgeValue];
    
#if __IPHONE_OS_VERSION_MIN_REQUIRED >= __IPHONE_10_0
    // 如果系統是iOS10 以上的就使用系統方法修改
    UIColor *badgeColor = [UIColor blueColor];
    [self setBadgeColor:badgeColor];
#else
    // 這裡替換角標顏色的圖片,需要注意的時:這個圖片size=(36px,36px),圓的
    UIImage *badgeImage = [UIImage imageNamed:@"blueBadge"];
    [self customBadgeColorWith:badgeImage];
#endif
}

- (void)customBadgeColorWith:(UIImage *)badgeImage
{
    UIView *tabBarButton = (UIView *)[self performSelector:@selector(view)];
    
    // iOS10以下的版本 角標其實是一張圖片,所以我們一直找下去這個圖片,然後替換他
    for(UIView *subview in tabBarButton.subviews) {
        NSString *classString = NSStringFromClass([subview class]);
        if([classString rangeOfString:@"UIBadgeView"].location != NSNotFound) {
            for(UIView *badgeSubview in subview.subviews) {
                NSString *badgeSubviewClassString = NSStringFromClass([badgeSubview class]);
                if([badgeSubviewClassString rangeOfString:@"BadgeBackground"].location != NSNotFound) {
                    [badgeSubview setValue:badgeImage forKey:@"image"];
                }
            }
        }
    }
}
複製程式碼

相關文章