MacPass/MacPass/MPTOTPViewController.m

//
//  MPOTPViewController.m
//  MacPass
//
//  Created by Michael Starke on 27.11.20.
//  Copyright © 2020 HicknHack Software GmbH. All rights reserved.
//

#import "MPTOTPViewController.h"

#import <KeePassKit/KeePassKit.h>

@interface MPTOTPViewController ()

@property (strong) KPKTimeOTPGenerator *generator;

@end

@implementation MPTOTPViewController

- (void)viewDidLoad {
  self.remainingTimeButton.title = @"";
}

- (void)setRepresentedObject:(id)representedObject {
  NSArray *notificationNames = @[KPKWillAddAttributeNotification, KPKDidAddAttributeNotification, KPKWillChangeAttributeNotification, KPKDidChangeAttributeNotification, KPKWillRemoveAttributeNotification, KPKDidRemoveAttributeNotification];
  if(self.representedObject) {
    for(NSString *notificationName in notificationNames) {
      [NSNotificationCenter.defaultCenter removeObserver:self name:notificationName object:self.representedObject];
    }
  }
  super.representedObject = representedObject;
  if(representedObject) {
    for(NSString *notificationName in notificationNames) {
      [NSNotificationCenter.defaultCenter addObserver:self selector:@selector(_didChangeAttribute:) name:notificationName object:representedObject];
    }
  }
  [self _didChangeAttribute:nil];
}

- (void)_didChangeAttribute:(NSNotification *)notification {
  [self _updateDisplay];
}

- (void)_updateDisplay {
  KPKEntry *entry = (KPKEntry *)self.representedObject;
  BOOL showTOTP = entry.hasTimeOTP;
  self.view.hidden = !showTOTP;
  if(showTOTP) {
    self.generator = [[KPKTimeOTPGenerator alloc] initWithEntry:entry];
    self.generator.time = NSDate.date.timeIntervalSince1970;
    self.toptValueTextField.stringValue = self.generator.string;
    
    NSString *template = NSLocalizedString(@"TOTP_REMAINING_TIME_%ld_SECONDS", @"Time in seconds remaining for a valid TOTP string, format should be %ld");
    
    self.remainingTimeButton.title = [NSString stringWithFormat:template, (NSUInteger)self.generator.remainingTime];
    [self performSelector:@selector(_updateDisplay) withObject:nil afterDelay:0.5];
  }
  else {
    self.remainingTimeButton.title = @"";
    self.toptValueTextField.stringValue = @"";
  }
}
@end