Avoid background reset from UITableViewCell

Author: D-32

BitStore/Cells/TransactionCell.m

@@ -101,4 +101,10 @@ - (void)updateValues {
     }
 }
 
+- (void)setHighlighted:(BOOL)highlighted animated:(BOOL)animated {
+    // hack to avoid the background reset the UITableView does on all subviews
+    [super setHighlighted:highlighted animated:animated];
+    [_circleIndicator setHighlighted:highlighted animated:animated];
+}
+
 @end

BitStore/Views/CircleIndicator.h

@@ -12,6 +12,7 @@
 
 - (id)initWithCircles:(int)numberOfCircles;
 - (void)setFilledCircles:(int)filledCircles;
+- (void)setHighlighted:(BOOL)highlighted animated:(BOOL)animated;
 
 - (id)init UNAVAILABLE_ATTRIBUTE;
 - (instancetype)initWithFrame:(CGRect)frame UNAVAILABLE_ATTRIBUTE;

BitStore/Views/CircleIndicator.m

@@ -12,6 +12,7 @@
 
 @implementation CircleIndicator {
     NSMutableArray* _circleViews;
+    int _filledCircles;
 }
 
 - (id)initWithCircles:(int)numberOfCircles {
@@ -33,12 +34,20 @@ - (id)initWithCircles:(int)numberOfCircles {
 - (void)setFilledCircles:(int)filledCircles {
     NSAssert(filledCircles <= _circleViews.count, @"Can't fill more circles than provided.");
     NSAssert(filledCircles >= 0, @"Can't fill < 0 circles.");
-    
-    for (int i = 0; i < filledCircles; i++) {
+    _filledCircles = filledCircles;
+    [self updateViews];
+}
+
+- (void)setHighlighted:(BOOL)highlighted animated:(BOOL)animated {
+    [self updateViews]; // table view cell fix
+}
+
+- (void)updateViews {
+    for (int i = 0; i < _filledCircles; i++) {
         UIView* circle = [_circleViews objectAtIndex:_circleViews.count - i - 1];
         circle.backgroundColor = [UIColor colorWithWhite:0.8 alpha:1.0];
     }
-    for (int i = filledCircles; i < _circleViews.count; i++) {
+    for (int i = _filledCircles; i < _circleViews.count; i++) {
         UIView* circle = [_circleViews objectAtIndex:_circleViews.count - i - 1];
         circle.backgroundColor = [UIColor clearColor];
     }
@@ -53,4 +62,6 @@ - (void)layoutSubviews {
     }
 }
 
+
+
 @end