Setup TKCurrentTimeLabel to correctly update time when its added to a…

… view.
devinross · Feb 23, 2015 · 62e434d · 62e434d
1 parent eaaf8c2
commit 62e434d
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/src/TapkuLibrary/TKCurrentTimeLabel.m b/src/TapkuLibrary/TKCurrentTimeLabel.m
@@ -47,6 +47,16 @@ - (instancetype) initWithFrame:(CGRect)frame {
     return self;
 }
 
+- (void) willMoveToWindow:(UIWindow *)newWindow{
+    if(newWindow==nil){
+        [self.timer invalidate];
+        self.timer = nil;
+        return;
+    }
+    [self _updateTime];
+    [self _startTimer];
+}
+
 - (void) _updateTime{
     self.attributedText = [self attributeStringWithTime:[self currentTime]];
 }
@@ -63,7 +73,8 @@ - (void) handleEnteredBackground:(id)sender{
 }
 - (void) handleEnteredForeground:(id)sender{
     [self _updateTime];
-    [self _startTimer];
+    if(self.window)
+        [self _startTimer];
 }
 
 #pragma mark Functions for Subclassing