session-ios/Signal/src/view controllers/CodeVerificationViewController.m

//
//  CodeVerificationViewController.m
//  Signal
//
//  Created by Dylan Bourgeois on 13/11/14.
//  Copyright (c) 2014 Open Whisper Systems. All rights reserved.
//

#import "CodeVerificationViewController.h"

#import "Environment.h"
#import "PhoneNumberDirectoryFilterManager.h"
#import "RPServerRequestsManager.h"
#import "LocalizableText.h"
#import "PushManager.h"
#import "SignalKeyingStorage.h"
#import "TSAccountManager.h"

@interface CodeVerificationViewController ()

@end

@implementation CodeVerificationViewController

- (void)viewDidLoad {
    [super viewDidLoad];
    [self initializeKeyboardHandlers];
    _headerLabel.text = NSLocalizedString(@"VERIFICATION_HEADER", @"");
    _challengeTextField.placeholder = NSLocalizedString(@"VERIFICATION_CHALLENGE_DEFAULT_TEXT", @"");
    [_challengeButton setTitle:NSLocalizedString(@"VERIFICATION_CHALLENGE_SUBMIT_CODE", @"") forState:UIControlStateNormal];
    
    [_sendCodeViaSMSAgainButton setTitle:NSLocalizedString(@"VERIFICATION_CHALLENGE_SUBMIT_AGAIN", @"") forState:UIControlStateNormal];
    [_sendCodeViaVoiceButton setTitle:[@"     " stringByAppendingString:NSLocalizedString(@"VERIFICATION_CHALLENGE_SEND_VIAVOICE", @"")] forState:UIControlStateNormal];
    [_changeNumberButton setTitle:[@"     " stringByAppendingString:NSLocalizedString(@"VERIFICATION_CHALLENGE_CHANGE_NUMBER", @"")] forState:UIControlStateNormal];
}

-(void)viewWillAppear:(BOOL)animated
{
    [super viewWillAppear:animated];
    [self enableServerActions:YES];
    [_phoneNumberEntered setText:_formattedPhoneNumber];
    [self adjustScreenSizes];
}

- (void)didReceiveMemoryWarning {
    [super didReceiveMemoryWarning];
}


- (IBAction)verifyChallengeAction:(id)sender {
    [self enableServerActions:NO];
    [_challengeTextField resignFirstResponder];
    
    [self registerWithSuccess:^{
        [_submitCodeSpinner stopAnimating];
        [Environment.getCurrent.phoneDirectoryManager forceUpdate];
        [self.navigationController dismissViewControllerAnimated:YES completion:nil];
    } failure:^(NSError *error) {
        [self showAlertForError:error];
        [self enableServerActions:YES];
        [_submitCodeSpinner stopAnimating];
    }];
}


- (void)registerWithSuccess:(void(^)())success failure:(void(^)(NSError *))failure{
    //TODO: Refactor this to use futures? Better error handling needed. Good enough for PoC
    
    [_submitCodeSpinner startAnimating];
    [[RPServerRequestsManager sharedInstance] performRequest:[RPAPICall verifyVerificationCode:_challengeTextField.text] success:^(NSURLSessionDataTask *task, id responseObject) {
        
        [PushManager.sharedManager registrationAndRedPhoneTokenRequestWithSuccess:^(NSData *pushToken, NSString *signupToken) {
            
            [TSAccountManager registerWithRedPhoneToken:signupToken pushToken:pushToken success:^{
                success();
            } failure:^(NSError *error) {
                failure(error);
            }];
        } failure:^{
            // PushManager shows its own error alerts, so we don't want to show a second one
            failure(nil);
            [_submitCodeSpinner stopAnimating];
        }];
    } failure:^(NSURLSessionDataTask *task, NSError *error) {
        NSHTTPURLResponse* badResponse = (NSHTTPURLResponse*)task.response;
        NSError *responseError = [self errorForResponse:badResponse];
        
        failure(responseError);
        [_submitCodeSpinner stopAnimating];

    }];
}


// TODO: If useful, this could possibly go in a less-specific class
- (void)showAlertForError:(NSError *)error {
    
    if (error == nil) {
        DDLogCError(@"%@: Error condition, but no NSError to display", self.class);
        return;
    } else if (error.localizedDescription.length == 0) {
        DDLogCError(@"%@: Unable to display error because localizedDescription was not set: %@", self.class, error);
        return;
    }
    
    NSString *alertBody = nil;
    if (error.localizedFailureReason.length > 0) {
        alertBody = error.localizedFailureReason;
    } else if (error.localizedRecoverySuggestion.length > 0) {
        alertBody = error.localizedRecoverySuggestion;
    }
    
    SignalAlertView(error.localizedDescription, alertBody);
}


- (NSError *)errorForResponse:(NSHTTPURLResponse *)badResponse {
    
    NSString *description = NSLocalizedString(@"REGISTRATION_ERROR", @"");
    NSString *failureReason = nil;
    TSRegistrationFailure failureType;
    
    if (badResponse.statusCode == 401) {
        failureReason = REGISTER_CHALLENGE_ALERT_VIEW_BODY;
        failureType = kTSRegistrationFailureAuthentication;
    } else if (badResponse.statusCode == 413){
        failureReason = NSLocalizedString(@"REGISTER_RATE_LIMITING_BODY", @"");
        failureType = kTSRegistrationFailureRateLimit;
    } else {
        failureReason = [NSString stringWithFormat:@"%@ %lu", NSLocalizedString(@"SERVER_CODE", @""),(unsigned long)badResponse.statusCode];
        failureType = kTSRegistrationFailureNetwork;
    }
    
    NSDictionary *userInfo = @{NSLocalizedDescriptionKey: description, NSLocalizedFailureReasonErrorKey: failureReason};
    NSError *error = [NSError errorWithDomain:TSRegistrationErrorDomain code:failureType userInfo:userInfo];
    
    return error;
}

#pragma mark - Send codes again
- (IBAction)sendCodeSMSAction:(id)sender {
  
    [self enableServerActions:NO];
    
    [_requestCodeAgainSpinner startAnimating];
    [[RPServerRequestsManager sharedInstance]performRequest:[RPAPICall requestVerificationCode] success:^(NSURLSessionDataTask *task, id responseObject) {
        [self enableServerActions:YES];
        [_requestCodeAgainSpinner stopAnimating];

    } failure:^(NSURLSessionDataTask *task, NSError *error) {
    
        DDLogError(@"Registration failed with information %@", error.description);
    
        UIAlertView *registrationErrorAV = [[UIAlertView alloc]initWithTitle:REGISTER_ERROR_ALERT_VIEW_TITLE
                                                                 message:REGISTER_ERROR_ALERT_VIEW_BODY
                                                                delegate:nil
                                                       cancelButtonTitle:REGISTER_ERROR_ALERT_VIEW_DISMISS
                                                       otherButtonTitles:nil, nil];
    
        [registrationErrorAV show];
    
        [self enableServerActions:YES];
        [_requestCodeAgainSpinner stopAnimating];   
    }];
}

- (IBAction)sendCodeVoiceAction:(id)sender {
    
    [self enableServerActions:NO];
    
    [_requestCallSpinner startAnimating];
    [[RPServerRequestsManager sharedInstance]performRequest:[RPAPICall requestVerificationCodeWithVoice] success:^(NSURLSessionDataTask *task, id responseObject) {
        
        [self enableServerActions:YES];
        [_requestCallSpinner stopAnimating];

    } failure:^(NSURLSessionDataTask *task, NSError *error) {
        
        DDLogError(@"Registration failed with information %@", error.description);
        
        UIAlertView *registrationErrorAV = [[UIAlertView alloc]initWithTitle:REGISTER_ERROR_ALERT_VIEW_TITLE
                                                                     message:REGISTER_ERROR_ALERT_VIEW_BODY
                                                                    delegate:nil
                                                           cancelButtonTitle:REGISTER_ERROR_ALERT_VIEW_DISMISS
                                                           otherButtonTitles:nil, nil];
        
        [registrationErrorAV show];
        [self enableServerActions:YES];
        [_requestCallSpinner stopAnimating];
    }];
}

-(void)enableServerActions:(BOOL)enabled {
    [_challengeButton setEnabled:enabled];
    [_sendCodeViaSMSAgainButton setEnabled:enabled];
    [_sendCodeViaVoiceButton setEnabled:enabled];    
}


#pragma mark - Keyboard notifications

- (void)initializeKeyboardHandlers{
    UITapGestureRecognizer *outsideTabRecognizer = [[UITapGestureRecognizer alloc] initWithTarget:self action:@selector(dismissKeyboardFromAppropriateSubView)];
    [self.view addGestureRecognizer:outsideTabRecognizer];
}

- (void)dismissKeyboardFromAppropriateSubView {
    [self.view endEditing:NO];
}

- (void)adjustScreenSizes {
    CGFloat screenHeight = [UIScreen mainScreen].bounds.size.height;
    CGFloat blueHeaderHeight;
    
    if (screenHeight < 667) {
        self.signalLogo.hidden = YES;
        blueHeaderHeight = screenHeight - 400;
    } else {
        blueHeaderHeight = screenHeight - 410;
    }
    
    _headerConstraint.constant = blueHeaderHeight;
}

@end