Save changes to edited tokens and refresh the token list

Author: mattrubin

Authenticator/Classes/OTPTokenEditViewController.h

@@ -25,8 +25,17 @@
 #import "OTPTokenEntryViewController.h"
 
 
+@protocol OTPTokenEditorDelegate <OTPTokenSourceDelegate>
+
+- (void)tokenSource:(id)tokenSource didEditToken:(OTPToken *)token;
+
+@end
+
+
 @interface OTPTokenEditViewController : OTPTokenEntryViewController
 
+@property (nonatomic, weak) id <OTPTokenEditorDelegate> delegate;
+
 @property (nonatomic, strong) OTPToken *token;
 
 @end

Authenticator/Classes/OTPTokenEditViewController.m

@@ -24,7 +24,7 @@
 
 #import "OTPTokenEditViewController.h"
 #import "OTPTextFieldCell.h"
-#import "OTPToken.h"
+#import "OTPToken+Persistence.h"
 
 
 @interface OTPTokenEntryViewController () <UITextFieldDelegate>
@@ -71,8 +71,17 @@ - (void)setToken:(OTPToken *)token
 
 - (void)updateToken
 {
-    NSLog(@"!!!");
+    if (!self.formIsValid) return;
 
+    if (![self.token.name isEqualToString:self.accountNameCell.textField.text] ||
+        ![self.token.issuer isEqualToString:self.issuerCell.textField.text]) {
+        self.token.name = self.accountNameCell.textField.text;
+        self.token.issuer = self.issuerCell.textField.text;
+        [self.token saveToKeychain];
+    }
+
+    id <OTPTokenEditorDelegate> delegate = self.delegate;
+    [delegate tokenSource:self didEditToken:self.token];
 }
 
 

Authenticator/Classes/OTPTokenListViewController.m

@@ -33,7 +33,7 @@
 #import "OTPTokenEditViewController.h"
 
 
-@interface OTPTokenListViewController ()
+@interface OTPTokenListViewController () <OTPTokenEditorDelegate>
 
 @property (nonatomic, strong) OTPTokenManager *tokenManager;
 @property (nonatomic, strong) OTPProgressRing *ring;
@@ -144,6 +144,8 @@ - (CGFloat)tableView:(UITableView *)tableView heightForRowAtIndexPath:(NSIndexPa
 - (void)tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath
 {
     if (self.isEditing) {
+        self.editing = NO;
+
         OTPTokenEditViewController *editController = [OTPTokenEditViewController new];
         editController.token = self.tokenManager.tokens[(NSUInteger)indexPath.row];
         editController.delegate = self;
@@ -191,4 +193,13 @@ - (void)tokenSource:(id)tokenSource didCreateToken:(OTPToken *)token
     }
 }
 
+
+#pragma mark - OTPTokenEditorDelegate
+
+- (void)tokenSource:(id)tokenSource didEditToken:(OTPToken *)token
+{
+    [self dismissViewControllerAnimated:YES completion:nil];
+    [self.tableView reloadData];
+}
+
 @end