Use cell tint color if available

zengchanghuan · Oct 4, 2013 · 5a39d1c · 5a39d1c
1 parent 78f2276
commit 5a39d1c
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/quickdialog/QBadgeElement.m b/quickdialog/QBadgeElement.m
@@ -25,7 +25,7 @@ @implementation QBadgeElement {
 
 - (QBadgeElement *)init {
     self = [super init];
-    _badgeColor = [UIColor colorWithRed:0.530f green:0.600f blue:0.738f alpha:1.000f];
+    _badgeColor = nil;
     _badgeTextColor = [UIColor whiteColor];
     return self;
 }
@@ -43,8 +43,13 @@ - (UITableViewCell *)getCellForTableView:(QuickDialogTableView *)tableView contr
     QBadgeTableCell *cell = [[QBadgeTableCell alloc] init];
     cell.textLabel.text = _title;
     [cell applyAppearanceForElement:self];
-    cell.badgeLabel.badgeColor = _badgeColor;
     cell.badgeLabel.textColor = _badgeTextColor;
+
+    if ([cell respondsToSelector:@selector(tintColor)])
+        cell.badgeLabel.badgeColor = _badgeColor == nil ? cell.tintColor : _badgeTextColor;
+    else
+       cell.badgeLabel.badgeColor = _badgeColor;
+
     cell.badgeLabel.text = _badge;
     cell.accessoryType = UITableViewCellAccessoryNone;
     cell.selectionStyle = UITableViewCellSelectionStyleNone;