* When persisting controllers, allow them to say they are "important"…

…, which means they must always be restored even if the persistence expiration date has passed. Crucial to avoid losing user-created content.
almakdad · Aug 7, 2009 · f0e544c · f0e544c
1 parent 6118081
commit f0e544c
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 1 deletion.
diff --git a/src/TTMessageController.m b/src/TTMessageController.m
@@ -461,6 +461,7 @@ - (BOOL)persistView:(NSMutableDictionary*)state {
 
   NSInteger firstResponder = [self fieldIndexOfFirstResponder];
   [state setObject:[NSNumber numberWithInt:firstResponder] forKey:@"firstResponder"];
+  [state setObject:[NSNumber numberWithBool:YES] forKey:@"__important__"];
   return [super persistView:state];
 }
 

diff --git a/src/TTNavigator.m b/src/TTNavigator.m
@@ -509,13 +509,24 @@ - (void)persistViewControllers {
   [self persistController:_rootViewController path:path];
   TTLOG(@"DEBUG PERSIST %@", path);
 
+  // Check if any of the paths were "important", and therefore unable to expire
+  BOOL important = NO;
+  for (NSDictionary* state in path) {
+    if ([state objectForKey:@"__important__"]) {
+      important = YES;
+      break;
+    }
+  }
+
   NSUserDefaults* defaults = [NSUserDefaults standardUserDefaults];
   if (path.count) {
     [defaults setObject:path forKey:@"TTNavigatorHistory"];
     [defaults setObject:[NSDate date] forKey:@"TTNavigatorHistoryTime"];
+    [defaults setObject:[NSNumber numberWithInt:important] forKey:@"TTNavigatorHistoryImportant"];
   } else {
     [defaults removeObjectForKey:@"TTNavigatorHistory"];
     [defaults removeObjectForKey:@"TTNavigatorHistoryTime"];
+    [defaults removeObjectForKey:@"TTNavigatorHistoryImportant"];
   }
   [defaults synchronize];
 }
@@ -524,9 +535,12 @@ - (UIViewController*)restoreViewControllers {
   NSUserDefaults* defaults = [NSUserDefaults standardUserDefaults];
   NSDate* timestamp = [defaults objectForKey:@"TTNavigatorHistoryTime"];
   NSArray* path = [defaults objectForKey:@"TTNavigatorHistory"];
+  BOOL important = [[defaults objectForKey:@"TTNavigatorHistoryImportant"] boolValue];
   TTLOG(@"DEBUG RESTORE %@ FROM %@", path, [timestamp formatRelativeTime]);
 
-  if (_persistenceExpirationAge && -timestamp.timeIntervalSinceNow > _persistenceExpirationAge) {
+  BOOL expired = _persistenceExpirationAge
+                 && -timestamp.timeIntervalSinceNow > _persistenceExpirationAge;
+  if (expired && !important) {
     return nil;
   }
 
@@ -618,6 +632,7 @@ - (void)resetDefaults {
   NSUserDefaults* defaults = [NSUserDefaults standardUserDefaults];
   [defaults removeObjectForKey:@"TTNavigatorHistory"];
   [defaults removeObjectForKey:@"TTNavigatorHistoryTime"];
+  [defaults removeObjectForKey:@"TTNavigatorHistoryImportant"];
   [defaults synchronize];
 }
 

diff --git a/src/TTPostController.m b/src/TTPostController.m
@@ -205,6 +205,8 @@ - (void)loadView {
 // UIViewController (TTCategory)
 
 - (BOOL)persistView:(NSMutableDictionary*)state {
+  [state setObject:[NSNumber numberWithBool:YES] forKey:@"__important__"];
+
   NSString* delegate = [[TTNavigator navigator] pathForObject:_delegate];
   if (delegate) {
     [state setObject:delegate forKey:@"delegate"];