Center badges 2

Tweak.xm

/*%hook SBIconImageContainerView

@interface SBIconImageContainerView : UIView
@end

-(void)addSubview:(UIView *)view{
    //Class _icon = objc_getClass("SBIconBadgeImage");
    Class _icon = objc_getClass("UIImageView");
    if( [view isKindOfClass:_icon] )
    {
        [view setCenter:CGPointMake(self.bounds.size.width/2,view.bounds.origin.y)];
    }
    %orig;
}
%end*/

%hook SBIconView

@interface SBIconView : UIView
-(UIImageView*)_badgeView;
//-(void)_updateBadgePosition;
@end

-(void)updateBadge{
    %orig;
    NSLog(@"__________________________");
    NSLog(@"__________________________");
    NSLog(@"__________________________");
    NSLog(@"__________________________");
    NSLog(@"__________________________");
    NSLog(@"__________________________");
    NSLog(@"__________________________");
    NSLog(@"__________________________");
    NSLog(@"__________________________");
    NSLog(@"__________________________");
    NSLog(@"__________________________");
    NSLog(@"__________________________");
    NSLog(@"__________________________");
    NSLog(@"%@", [self _badgeView]);
    [[self _badgeView] setCenter:CGPointMake(self.bounds.size.width/2,self.bounds.origin.y)];
    //[self _updateBadgePosition];
}
%end