//==============================================================================
// This file is part of Master Password.
// Copyright (c) 2011-2017, Maarten Billemont.
//
// Master Password is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Master Password is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You can find a copy of the GNU General Public License in the
// LICENSE file.  Alternatively, see <http://www.gnu.org/licenses/>.
//==============================================================================

#import "MPiOSAppDelegate.h"
#import "MPAppDelegate_Key.h"
#import "MPAppDelegate_Store.h"
#import "MPStoreViewController.h"
#import "mpw-marshal.h"
#import "MPSecrets.h"

#import <Sentry/Sentry.h>
#import <Countly/Countly.h>

@interface CountlyPushNotifications
@end

@interface CountlyPushNotifications(MPNotifications)
@end

@implementation CountlyPushNotifications(MPNotifications)

- (void)openURL:(NSString *)URLString {
    [[MPiOSAppDelegate get].window.rootViewController performSegueWithIdentifier:@"web" sender:[NSURL URLWithString:URLString]];
}

@end

@interface MPiOSAppDelegate()<UIDocumentInteractionControllerDelegate>

@property(nonatomic, strong) UIDocumentInteractionController *interactionController;
@property(nonatomic, strong) PearlHangDetector *hangDetector;
@end

@implementation MPiOSAppDelegate

@synthesize window;

+ (void)initialize {

    [MPiOSConfig get];
}

- (BOOL)application:(UIApplication *)application didFinishLaunchingWithOptions:(NSDictionary *)launchOptions {

    @try {
        // Sentry
        [SentrySDK startWithOptions:@{
                @"dsn"                      : NilToNSNull( decrypt( sentryDSN ) ),
#ifdef DEBUG
                @"debug"                    : @(NO), //@(YES),
                @"environment"              : @"Development",
#elif PUBLIC
                @"debug"                    : @(NO),
                @"environment"              : @"Public",
#else
                @"debug"                    : @(NO),
                @"environment"              : @"Private",
#endif
                @"enabled"                  : @([[MPiOSConfig get].sendInfo boolValue] || ![[MPiOSConfig get].sendInfoDecided boolValue]),
                @"enableAutoSessionTracking": @(YES),
        }];
        [[PearlLogger get] registerListener:^BOOL(PearlLogMessage *message) {
            PearlLogLevel level = PearlLogLevelWarn;
            if ([[MPConfig get].sendInfo boolValue])
                level = PearlLogLevelDebug;

            if (message.level >= level) {
                SentryLevel sentryLevel = kSentryLevelInfo;
                switch (message.level) {
                    case PearlLogLevelTrace:
                        sentryLevel = kSentryLevelNone;
                        break;
                    case PearlLogLevelDebug:
                        sentryLevel = kSentryLevelDebug;
                        break;
                    case PearlLogLevelInfo:
                        sentryLevel = kSentryLevelInfo;
                        break;
                    case PearlLogLevelWarn:
                        sentryLevel = kSentryLevelWarning;
                        break;
                    case PearlLogLevelError:
                        sentryLevel = kSentryLevelError;
                        break;
                    case PearlLogLevelFatal:
                        sentryLevel = kSentryLevelFatal;
                        break;
                }
                SentryBreadcrumb *breadcrumb = [[SentryBreadcrumb alloc] initWithLevel:sentryLevel category:@"Pearl"];
                breadcrumb.type = @"log";
                breadcrumb.message = message.message;
                breadcrumb.timestamp = message.occurrence;
                breadcrumb.data = @{ @"file": message.fileName, @"line": @(message.lineNumber), @"function": message.function };
                [SentrySDK addBreadcrumb:breadcrumb];
            }

            return YES;
        }];

        // Countly
        CountlyConfig *countlyConfig = [CountlyConfig new];
        countlyConfig.host = @"https://countly.lyndir.com";
        countlyConfig.appKey = decrypt( countlyKey );
        countlyConfig.features = @[ CLYPushNotifications, CLYAutoViewTracking ];
        countlyConfig.requiresConsent = YES;
        countlyConfig.alwaysUsePOST = YES;
        countlyConfig.deviceID = [PearlKeyChain deviceIdentifier];
        countlyConfig.secretSalt = decrypt( countlySalt );
#if DEBUG
        //countlyConfig.enableDebug = YES;
        countlyConfig.pushTestMode = CLYPushTestModeDevelopment;
#elif ! PUBLIC
        countlyConfig.enableDebug = NO;
        countlyConfig.pushTestMode = CLYPushTestModeTestFlightOrAdHoc;
#endif
        [Countly.sharedInstance startWithConfig:countlyConfig];

#if ! DEBUG
        [self.hangDetector = [[PearlHangDetector alloc] initWithHangAction:^(NSTimeInterval hangTime) {
            MPError( [NSError errorWithDomain:MPErrorDomain code:MPErrorHangCode userInfo:@{
                    @"time": @(hangTime)
            }], @"Timeout waiting for main thread after %fs.", hangTime );
        }] start];
#endif
    }
    @catch (id exception) {
        err( @"During Analytics Setup: %@", exception );
    }
    @try {
        PearlAddNotificationObserver( MPCheckConfigNotification, nil, [NSOperationQueue mainQueue], ^(id self, NSNotification *note) {
            [self updateConfigKey:note.object];
        } );
        PearlAddNotificationObserver( NSUserDefaultsDidChangeNotification, nil, nil, ^(id self, NSNotification *note) {
            [[NSNotificationCenter defaultCenter] postNotificationName:MPCheckConfigNotification object:nil];
        } );
    }
    @catch (id exception) {
        err( @"During Config Test: %@", exception );
    }
    @try {
        inf( @"Started up with device identifier: %@", [PearlKeyChain deviceIdentifier] );

        PearlAddNotificationObserver(
                MPFoundInconsistenciesNotification, nil, [NSOperationQueue mainQueue], ^(id self, NSNotification *note) {
            switch ((MPFixableResult)[note.userInfo[MPInconsistenciesFixResultUserKey] unsignedIntegerValue]) {

                case MPFixableResultNoProblems:
                    break;
                case MPFixableResultProblemsFixed: {
                    UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Inconsistencies Fixed" message:
                                    @"Some inconsistencies were detected in your sites.\n"
                                    @"All issues were fixed."
                                                                            preferredStyle:UIAlertControllerStyleAlert];
                    [alert addAction:[UIAlertAction actionWithTitle:@"Okay" style:UIAlertActionStyleCancel handler:nil]];
                    [self presentViewController:alert animated:YES completion:nil];
                    break;
                }
                case MPFixableResultProblemsNotFixed: {
                    UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Inconsistencies Found" message:
                                    @"Some inconsistencies were detected in your sites.\n"
                                    @"Not all issues could be fixed.  Try signing in to each user or checking the logs."
                                                                            preferredStyle:UIAlertControllerStyleAlert];
                    [alert addAction:[UIAlertAction actionWithTitle:@"Okay" style:UIAlertActionStyleCancel handler:nil]];
                    [self presentViewController:alert animated:YES completion:nil];
                    break;
                }
            }
        } );

        SKStoreProductViewController *migrateVC = [SKStoreProductViewController new];
        [migrateVC loadProductWithParameters:@{
                SKStoreProductParameterCampaignToken       : @"app-masterpassword.ios", /* Campaign:    From MasterPassword iOS */
                SKStoreProductParameterProviderToken       : @153897, /*                   Provider:    Maarten Billemont */
//              SKStoreProductParameterITunesItemIdentifier: @510296984, /*                Application: MasterPassword iOS */
                SKStoreProductParameterITunesItemIdentifier: @1500430196, /*               Application: Spectre iOS */
        }                                   completionBlock:^(BOOL result, NSError *error) {
            if (error)
                err( @"Failed loading Spectre product information: %@", error );

                if (result) {
                    self.spectreViewController = migrateVC;
                    self.spectreViewController.delegate = self;
                } else {
                    self.spectreViewController = nil;
                }
        }];

        PearlMainQueueOperation( ^{
            if ([[MPiOSConfig get].showSetup boolValue])
                [self.window.rootViewController performSegueWithIdentifier:@"setup" sender:self];

            [self consentFeatures];
        } );
    }
    @catch (id exception) {
        err( @"During Post-Startup: %@", exception );
    }

    return YES;
}

- (BOOL)application:(UIApplication *)application openURL:(NSURL *)url
  sourceApplication:(NSString *)sourceApplication annotation:(id)annotation {

    // No URL?
    if (!url)
        return NO;

    // masterpassword: URLs.
    if ([url.scheme isEqualToString:@"masterpassword"]) {
        [self openURL:url];
        return YES;
    }

    // Arbitrary URL to mpsites data.
    [[[NSURLSession sharedSession] dataTaskWithURL:url completionHandler:
            ^(NSData *importedSitesData, NSURLResponse *response, NSError *error) {
                if (error)
                    MPError( error, @"While reading imported sites from %@.", url );

                if (!importedSitesData) {
                    PearlMainQueue( ^{
                        UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Error" message:
                                        strf( @"Master Password couldn't read the import sites.\n\n%@",
                                                (id)[error localizedDescription]?: error )
                                                                                preferredStyle:UIAlertControllerStyleAlert];
                        [alert addAction:[UIAlertAction actionWithTitle:@"Continue" style:UIAlertActionStyleCancel handler:nil]];
                        [self.window.rootViewController presentViewController:alert animated:YES completion:nil];
                    } );
                    return;
                }

                NSString *importedSitesString = [[NSString alloc] initWithData:importedSitesData encoding:NSUTF8StringEncoding];
                if (!importedSitesString) {
                    PearlMainQueue( ^{
                        UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Error" message:
                                        @"Master Password couldn't understand the import file."
                                                                                preferredStyle:UIAlertControllerStyleAlert];
                        [alert addAction:[UIAlertAction actionWithTitle:@"Continue" style:UIAlertActionStyleCancel handler:nil]];
                        [self.window.rootViewController presentViewController:alert animated:YES completion:nil];
                    } );
                    return;
                }

                [self importSites:importedSitesString];
            }] resume];

    return YES;
}

- (void)consentFeatures {
    if ([self askDiagnostics])
        return;

    [self tryNotifications];
}

- (BOOL)askDiagnostics {

    if ([[MPiOSConfig get].sendInfoDecided boolValue])
        return NO;

    PearlMainQueue( ^{
        UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Diagnostics" message:
                        @"We look for bugs, sudden crashes, runtime issues & statistics.\n\n"
                        @"Diagnostics are scrubbed and personal details will never leave your device."
                                                                preferredStyle:UIAlertControllerStyleAlert];

        [alert addAction:[UIAlertAction actionWithTitle:@"Disable" style:UIAlertActionStyleCancel handler:^(UIAlertAction *action) {
            [MPiOSConfig get].sendInfo = @(NO);
            [MPiOSConfig get].sendInfoDecided = @(YES);
            [self consentFeatures];
        }]];
        [alert addAction:[UIAlertAction actionWithTitle:@"Engage" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
            [MPiOSConfig get].sendInfo = @(YES);
            [MPiOSConfig get].sendInfoDecided = @(YES);
            [self consentFeatures];
        }]];

        [self.window.rootViewController presentViewController:alert animated:YES completion:nil];
    } );

    return YES;
}

- (void)tryNotifications {

    [Countly.sharedInstance giveConsentForFeature:CLYConsentPushNotifications];
    if (@available( iOS 12, * )) {
        [Countly.sharedInstance askForNotificationPermissionWithOptions:UNAuthorizationOptionProvisional | UNAuthorizationOptionAlert
                                                      completionHandler:^(BOOL granted, NSError *error) {
                                                          if (!granted)
                                                              err( @"No provisional notification permission: %@", error );

                                                          [self askNotifications];
                                                      }];
        return;
    }

    [self askNotifications];
}

- (void)askNotifications {

    if ([[MPiOSConfig get].notificationsDecided boolValue])
        return;

    PearlMainQueue( ^{
        UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Coming Soon" message:
                        @"Master Password is rolling out a brand new, updated version and we're excited to bring you along.\n\n"
                        @"When it's time, we'll send you a notification to help you make an effortless transition."
                                                                preferredStyle:UIAlertControllerStyleAlert];
        [alert addAction:[UIAlertAction actionWithTitle:@"Thanks" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
            if (@available( iOS 12, * )) {
                [Countly.sharedInstance askForNotificationPermissionWithOptions:UNAuthorizationOptionAlert completionHandler:
                        ^(BOOL granted, NSError *error) {
                            [MPiOSConfig get].notificationsDecided = @(YES);
                        }];
            }
            else {
                [Countly.sharedInstance askForNotificationPermission];
                [MPiOSConfig get].notificationsDecided = @(YES);
            }
        }]];
        [self.window.rootViewController presentViewController:alert animated:YES completion:nil];
    } );
}

- (void)importSites:(NSString *)importData {

    if ([NSThread isMainThread]) {
        PearlNotMainQueue( ^{
            [self importSites:importData];
        } );
        return;
    }

    PearlOverlay *activityOverlay = [PearlOverlay showProgressOverlayWithTitle:@"Importing"];
    [self importSites:importData askImportPassword:^NSString *(NSString *userName) {
        return PearlAwait( ^(void (^setResult)(id)) {
            PearlMainQueue( ^{
                UIAlertController *alert = [UIAlertController alertControllerWithTitle:strf( @"Importing Sites For\n%@", userName ) message:
                                @"Enter the master password used to create this export file."
                                                                        preferredStyle:UIAlertControllerStyleAlert];
                [alert addTextFieldWithConfigurationHandler:^(UITextField *textField) {
                    textField.secureTextEntry = YES;
                }];
                [alert addAction:[UIAlertAction actionWithTitle:@"Import" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
                    setResult( alert.textFields.firstObject.text );
                }]];
                [alert addAction:[UIAlertAction actionWithTitle:@"Cancel" style:UIAlertActionStyleCancel handler:^(UIAlertAction *action) {
                    setResult( nil );
                }]];
                [self.window.rootViewController presentViewController:alert animated:YES completion:nil];
            } );
        } );
    } askUserPassword:^NSString *(NSString *userName) {
        return PearlAwait( (id)^(void (^setResult)(id)) {
            PearlMainQueue( ^{
                UIAlertController *alert = [UIAlertController alertControllerWithTitle:strf( @"Master Password For\n%@", userName ) message:
                                @"Enter the current master password for this user."
                                                                        preferredStyle:UIAlertControllerStyleAlert];
                [alert addTextFieldWithConfigurationHandler:^(UITextField *textField) {
                    textField.secureTextEntry = YES;
                }];
                [alert addAction:[UIAlertAction actionWithTitle:@"Import" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
                    setResult( alert.textFields.firstObject.text );
                }]];
                [alert addAction:[UIAlertAction actionWithTitle:@"Cancel" style:UIAlertActionStyleCancel handler:^(UIAlertAction *action) {
                    setResult( nil );
                }]];
                [self.window.rootViewController presentViewController:alert animated:YES completion:nil];
            } );
        } );
    }          result:^(NSError *error) {
        PearlMainQueue( ^{
            [activityOverlay cancelOverlayAnimated:YES];

            if (error && !(error.domain == NSCocoaErrorDomain && error.code == NSUserCancelledError)) {
                UIAlertController *controller = [UIAlertController alertControllerWithTitle:@"Error" message:[error localizedDescription]
                                                                             preferredStyle:UIAlertControllerStyleAlert];
                [controller addAction:[UIAlertAction actionWithTitle:@"Continue" style:UIAlertActionStyleCancel handler:nil]];
                [self.window.rootViewController presentViewController:controller animated:YES completion:nil];
            }
        } );
    }];
}

- (void)applicationWillEnterForeground:(UIApplication *)application {

    inf( @"Will foreground" );

    [self.hangDetector start];

    PearlNotMainQueue( ^{
        NSString *importData = [UIPasteboard generalPasteboard].string;
        MPMarshalledFile *importFile = mpw_marshal_read( NULL, importData.UTF8String );
        if (importFile && importFile->error.type == MPMarshalSuccess && importFile->info->format != MPMarshalFormatNone) {
            PearlMainQueue( ^{
                UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Import Sites?" message:
                                @"We've detected Master Password import sites on your pasteboard, would you like to import them?"
                                                                        preferredStyle:UIAlertControllerStyleAlert];
                [alert addAction:[UIAlertAction actionWithTitle:@"Import Sites" style:UIAlertActionStyleDefault handler:
                        ^(UIAlertAction *action) {
                            [self importSites:importData];
                            [UIPasteboard generalPasteboard].string = @"";
                        }]];
                [alert addAction:[UIAlertAction actionWithTitle:@"No" style:UIAlertActionStyleCancel handler:nil]];
                [self.window.rootViewController presentViewController:alert animated:YES completion:nil];
            } );
        }
        mpw_marshal_file_free( &importFile );
    } );
}

- (void)applicationDidBecomeActive:(UIApplication *)application {

    inf( @"Re-activated" );
    [[NSNotificationCenter defaultCenter] postNotificationName:MPCheckConfigNotification object:nil];
}

- (void)applicationDidReceiveMemoryWarning:(UIApplication *)application {

    inf( @"Received memory warning." );
}

- (void)applicationDidEnterBackground:(UIApplication *)application {

    inf( @"Did background" );
    if (![[MPiOSConfig get].rememberLogin boolValue]) {
        [UIView setAnimationsEnabled:NO];
        [self signOut];
        [UIView setAnimationsEnabled:YES];
    }

    [self.hangDetector stop];
}

#pragma mark - Behavior

- (void)openURL:(NSURL *)url {
    if ([url.scheme isEqualToString:@"masterpassword"]) {
        if ([url.host isEqualToString:@"open-url"]) {
            for (NSURLQueryItem *item in [NSURLComponents componentsWithString:[url absoluteString]].queryItems)
                if ([item.name isEqualToString:@"url"]) {
                    [UIApp openURL:[NSURL URLWithString:item.value]];
                    return;
                }
        }
        else if ([url.host isEqualToString:@"show-url"]) {
            for (NSURLQueryItem *item in [NSURLComponents componentsWithString:[url absoluteString]].queryItems)
                if ([item.name isEqualToString:@"url"]) {
                    [self.window.rootViewController performSegueWithIdentifier:@"web" sender:[NSURL URLWithString:item.value]];
                    return;
                }
        }
        else if ([url.host isEqualToString:@"migrate"]) {
            for (NSURLQueryItem *item in [NSURLComponents componentsWithString:[url absoluteString]].queryItems)
                if ([item.name isEqualToString:@"fullName"]) {
                    [MPiOSAppDelegate managedObjectContextPerformBlock:^(NSManagedObjectContext *context) {
                        NSFetchRequest
                                *fetchRequest = [NSFetchRequest fetchRequestWithEntityName:NSStringFromClass( [MPUserEntity class] )];
                        fetchRequest.predicate = [NSPredicate predicateWithFormat:@"name == %@", item.value];
                        NSArray *users = [context executeFetchRequest:fetchRequest error:nil];
                        [self migrateFor:users.firstObject];
                    }];
                    return;
                }

            [self migrateFor:nil];
            return;
        }
    } else
        [UIApp openURL:url];
}

- (void)showFeedbackWithLogs:(BOOL)logs forVC:(UIViewController *)viewController {

    if (![PearlEMail canSendMail]) {
        UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Feedback" message:
                        @"Have a question, comment, issue or just saying thanks?\n\n"
                        @"We'd love to hear what you think!\n"
                        @"help@masterpassword.app"
                                                                preferredStyle:UIAlertControllerStyleAlert];
        [alert addAction:[UIAlertAction actionWithTitle:@"Okay" style:UIAlertActionStyleCancel handler:nil]];
        [self.window.rootViewController presentViewController:alert animated:YES completion:nil];
    }
    else if (logs) {
        UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Feedback" message:
                        @"Have a question, comment, issue or just saying thanks?\n\n"
                        @"If you're having trouble, it may help us if you can first reproduce the problem "
                        @"and then include log files in your message."
                                                                preferredStyle:UIAlertControllerStyleAlert];
        [alert addAction:[UIAlertAction actionWithTitle:@"Include Logs" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
            [self openFeedbackWithLogs:YES forVC:viewController];
        }]];
        [alert addAction:[UIAlertAction actionWithTitle:@"No Logs" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
            [self openFeedbackWithLogs:NO forVC:viewController];
        }]];
        [self.window.rootViewController presentViewController:alert animated:YES completion:nil];
    }
    else
        [self openFeedbackWithLogs:NO forVC:viewController];
}

- (void)openFeedbackWithLogs:(BOOL)logs forVC:(UIViewController *)viewController {

    NSString *userName = [[MPiOSAppDelegate get] activeUserForMainThread].name;
    PearlLogLevel logLevel = PearlLogLevelInfo;
    if (logs && ([[MPConfig get].sendInfo boolValue] || [[MPiOSConfig get].traceMode boolValue]))
        logLevel = PearlLogLevelDebug;

    [[[PearlEMail alloc] initForEMailTo:@"Master Password Development <help@masterpassword.app"
                                subject:strf( @"Feedback for Master Password [%@]",
                                        [[PearlKeyChain deviceIdentifier] stringByDeletingMatchesOf:@"-.*"] )
                                   body:strf( @"\n\n\n"
                                              @"--\n"
                                              @"%@"
                                              @"Master Password %@, build %@",
                                           userName? ([userName stringByAppendingString:@"\n"]): @"",
                                           [PearlInfoPlist get].CFBundleShortVersionString,
                                           [PearlInfoPlist get].CFBundleVersion )

                            attachments:(logs
                                         ? [[PearlEMailAttachment alloc]
                                                 initWithContent:[[[PearlLogger get] formatMessagesWithLevel:logLevel]
                                                         dataUsingEncoding:NSUTF8StringEncoding]
                                                        mimeType:@"text/plain"
                                                        fileName:strf( @"%@-%@.log",
                                                                [[NSDateFormatter rfc3339DateFormatter] stringFromDate:[NSDate date]],
                                                                [PearlKeyChain deviceIdentifier] )]
                                         : nil), nil]
            showComposerForVC:viewController];
}

- (void)handleCoordinatorError:(NSError *)error {

    static dispatch_once_t once = 0;
    dispatch_once( &once, ^{
        PearlMainQueue( ^{
            UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Failed To Load Sites" message:
                            @"Master Password was unable to open your sites history.\n"
                            @"This may be due to corruption.  You can either reset Master Password and "
                            @"recreate your user, or E-Mail us your logs and leave your corrupt store as-is for now."
                                                                    preferredStyle:UIAlertControllerStyleAlert];
            [alert addAction:[UIAlertAction actionWithTitle:@"Try Again" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
                [self retryCorruptStore];
            }]];
            [alert addAction:[UIAlertAction actionWithTitle:@"Send Logs" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
                [self openFeedbackWithLogs:YES forVC:nil];
            }]];
            [alert addAction:[UIAlertAction actionWithTitle:@"Reset" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
                [self deleteAndResetStore];
            }]];
            [self.window.rootViewController presentViewController:alert animated:YES completion:nil];
        } );
    } );
}

- (void)showExportForVC:(UIViewController *)viewController {

    UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Exporting Your Sites" message:
                    @"An export is great for keeping a backup list of your accounts.\n\n"
                    @"When the file is ready, you will be able to mail it to yourself.\n"
                    @"You can open it with a text editor or with Master Password if you need to restore your list of sites."
                                                            preferredStyle:UIAlertControllerStyleAlert];
    [alert addAction:[UIAlertAction actionWithTitle:@"Export Sites" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
        UIAlertController *sheet = [UIAlertController alertControllerWithTitle:@"Show Passwords?" message:
                        @"Would you like to make all your passwords visible in the export file?\n\n"
                        @"A safe export will include all sites but make their passwords invisible.\n"
                        @"It is great as a backup and remains safe when fallen in the wrong hands."
                                                                preferredStyle:UIAlertControllerStyleAlert];
        [sheet addAction:[UIAlertAction actionWithTitle:@"Safe Export" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
            [self showExportRevealPasswords:NO forVC:viewController];
        }]];
        [sheet addAction:[UIAlertAction actionWithTitle:@"Show Passwords" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
            [self showExportRevealPasswords:YES forVC:viewController];
        }]];
        [sheet addAction:[UIAlertAction actionWithTitle:@"Cancel" style:UIAlertActionStyleCancel handler:nil]];
        [self.window.rootViewController presentViewController:sheet animated:YES completion:nil];
    }]];
    [alert addAction:[UIAlertAction actionWithTitle:@"Cancel" style:UIAlertActionStyleCancel handler:nil]];
    [self.window.rootViewController presentViewController:alert animated:YES completion:nil];
}

- (void)showExportRevealPasswords:(BOOL)revealPasswords forVC:(UIViewController *)viewController {

    if (![PearlEMail canSendMail]) {
        UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Cannot Send Mail" message:
                        @"Your device is not yet set up for sending mail.\n"
                        @"Close Master Password, go into Settings and add a Mail account."
                                                                preferredStyle:UIAlertControllerStyleAlert];
        [alert addAction:[UIAlertAction actionWithTitle:@"Okay" style:UIAlertActionStyleCancel handler:nil]];
        [self.window.rootViewController presentViewController:alert animated:YES completion:nil];
        return;
    }

    [MPiOSAppDelegate managedObjectContextPerformBlock:^(NSManagedObjectContext *context) {
        NSError *error = nil;
        NSString *exportedUser = [self exportSitesFor:[self activeUserInContext:context] revealPasswords:revealPasswords askExportPassword:
                ^NSString *(NSString *userName) {
                    return PearlAwait( ^(void (^setResult)(id)) {
                        PearlMainQueue( ^{
                            UIAlertController *alert = [UIAlertController alertControllerWithTitle:strf(
                                            @"Master Password For:\n%@", userName )
                                                                                           message:@"Enter your master password to export the user."
                                                                                    preferredStyle:UIAlertControllerStyleAlert];
                            [alert addTextFieldWithConfigurationHandler:^(UITextField *textField) {
                                textField.secureTextEntry = YES;
                            }];
                            [alert addAction:[UIAlertAction actionWithTitle:@"Export" style:UIAlertActionStyleDefault
                                                                    handler:^(UIAlertAction *action) {
                                                                        setResult( alert.textFields.firstObject.text );
                                                                    }]];
                            [alert addAction:[UIAlertAction actionWithTitle:@"Cancel" style:UIAlertActionStyleCancel
                                                                    handler:^(UIAlertAction *action) {
                                                                        setResult( nil );
                                                                    }]];
                            [self.window.rootViewController presentViewController:alert animated:YES completion:nil];
                        } );
                    } );
                }                               error:&error];

        PearlMainQueue( ^{
            if (error) {
                MPError( error, @"Failed to export mpsites." );
                UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Export Error" message:[error localizedDescription]
                                                                        preferredStyle:UIAlertControllerStyleAlert];
                [alert addAction:[UIAlertAction actionWithTitle:@"Okay" style:UIAlertActionStyleCancel handler:nil]];
                [self.window.rootViewController presentViewController:alert animated:YES completion:nil];
            }
            if (!exportedUser)
                return;

            NSDateFormatter *exportDateFormatter = [NSDateFormatter new];
            [exportDateFormatter setDateFormat:@"yyyy'-'MM'-'dd"];
            NSString *exportFileName = strf( @"%@ (%@).mpsites",
                    [self activeUserForMainThread].name, [exportDateFormatter stringFromDate:[NSDate date]] );

            UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Export Destination" message:nil
                                                                    preferredStyle:UIAlertControllerStyleAlert];
            [alert addAction:[UIAlertAction actionWithTitle:@"Send As E-Mail" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
                NSString *message;
                if (revealPasswords)
                    message = strf( @"Export of Master Password sites with passwords included.\n\n"
                                    @"REMINDER: Make sure nobody else sees this file!  Passwords are visible!\n\n\n"
                                    @"--\n"
                                    @"%@\n"
                                    @"Master Password %@, build %@",
                            [self activeUserForMainThread].name,
                            [PearlInfoPlist get].CFBundleShortVersionString,
                            [PearlInfoPlist get].CFBundleVersion );
                else
                    message = strf( @"Backup of Master Password sites.\n\n\n"
                                    @"--\n"
                                    @"%@\n"
                                    @"Master Password %@, build %@",
                            [self activeUserForMainThread].name,
                            [PearlInfoPlist get].CFBundleShortVersionString,
                            [PearlInfoPlist get].CFBundleVersion );

                [PearlEMail sendEMailTo:nil fromVC:viewController subject:@"Master Password Export" body:message
                            attachments:[[PearlEMailAttachment alloc] initWithContent:[exportedUser dataUsingEncoding:NSUTF8StringEncoding]
                                                                             mimeType:@"text/plain"
                                                                             fileName:exportFileName], nil];
                return;
            }]];
            [alert addAction:[UIAlertAction actionWithTitle:@"Share / Export" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
                NSURL *applicationSupportURL = [[[NSFileManager defaultManager] URLsForDirectory:NSApplicationSupportDirectory
                                                                                       inDomains:NSUserDomainMask] lastObject];
                NSURL *exportURL = [[applicationSupportURL
                        URLByAppendingPathComponent:[NSBundle mainBundle].bundleIdentifier isDirectory:YES]
                        URLByAppendingPathComponent:exportFileName isDirectory:NO];
                NSError *writeError = nil;
                if (![[exportedUser dataUsingEncoding:NSUTF8StringEncoding]
                        writeToURL:exportURL options:NSDataWritingFileProtectionComplete error:&writeError])
                    MPError( writeError, @"Failed to write export data to URL %@.", exportURL );
                else {
                    self.interactionController = [UIDocumentInteractionController interactionControllerWithURL:exportURL];
                    self.interactionController.UTI = @"com.lyndir.masterpassword.sites";
                    self.interactionController.delegate = self;
                    [self.interactionController presentOpenInMenuFromRect:CGRectZero inView:viewController.view animated:YES];
                }
            }]];
            [alert addAction:[UIAlertAction actionWithTitle:@"Continue" style:UIAlertActionStyleCancel handler:nil]];
            [self.window.rootViewController presentViewController:alert animated:YES completion:nil];
        } );
    }];
}

- (void)migrateFor:(MPUserEntity *)user {

    if ([UIApp canOpenURL:[[NSURL alloc] initWithString:@"spectre:"]]) {
        if (!user) {
            [MPiOSAppDelegate managedObjectContextPerformBlock:^(NSManagedObjectContext *context) {
                NSFetchRequest *fetchRequest = [NSFetchRequest fetchRequestWithEntityName:NSStringFromClass( [MPUserEntity class] )];
                NSArray *users = [context executeFetchRequest:fetchRequest error:nil];
                if (![users count])
                    return;

                UIAlertController *usersSheet = [UIAlertController alertControllerWithTitle:@"Migrate User"
                                                                                    message:@"Choose a user to migrate out to Spectre."
                                                                             preferredStyle:UIAlertControllerStyleAlert];
                [usersSheet addAction:[UIAlertAction actionWithTitle:@"Cancel" style:UIAlertActionStyleCancel handler:nil]];
                for (MPUserEntity *user_ in users)
                    [usersSheet addAction:[UIAlertAction actionWithTitle:user_.name style:UIAlertActionStyleDefault handler:
                            ^(UIAlertAction *action) { [self migrateFor:user_]; }]];

                PearlMainQueue( ^{
                    [self.window.rootViewController presentViewController:usersSheet animated:YES completion:nil];
                } );
            }];
            return;
        }

        [MPAppDelegate_Shared managedObjectContextPerformBlock:^(NSManagedObjectContext *context) {
            NSError *error = nil;
            NSString *exportedUser = [[MPAppDelegate_Shared get] exportSitesFor:[MPUserEntity existingObjectWithID:user.objectID inContext:context]
                                                                revealPasswords:NO askExportPassword:^NSString *(NSString *userName) {
                        return PearlAwait( ^(void (^setResult)(id)) {
                            PearlMainQueue( ^{
                                UIAlertController *alert = [UIAlertController alertControllerWithTitle:strf( @"Master Password For:\n%@", userName )
                                                                                               message:@"Enter your master password to export the user."
                                                                                        preferredStyle:UIAlertControllerStyleAlert];
                                [alert addTextFieldWithConfigurationHandler:^(UITextField *textField) {
                                    textField.secureTextEntry = YES;
                                }];
                                [alert addAction:[UIAlertAction actionWithTitle:@"Export" style:UIAlertActionStyleDefault handler:
                                        ^(UIAlertAction *action) { setResult( alert.textFields.firstObject.text ); }]];
                                [alert addAction:[UIAlertAction actionWithTitle:@"Cancel" style:UIAlertActionStyleCancel handler:
                                        ^(UIAlertAction *action) { setResult( nil ); }]];
                                [self.window.rootViewController presentViewController:alert animated:YES completion:nil];
                            } );
                        } );
                    } error:&error];

            PearlMainQueue( ^{
                if (error) {
                    MPError( error, @"Failed to export user." );
                    UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Export Error"
                                                                                   message:[error localizedDescription]
                                                                            preferredStyle:UIAlertControllerStyleAlert];
                    [alert addAction:[UIAlertAction actionWithTitle:@"Okay" style:UIAlertActionStyleCancel handler:nil]];
                    [self.window.rootViewController presentViewController:alert animated:YES completion:nil];
                }
                if (!exportedUser)
                    return;

                NSURLComponents *components = [NSURLComponents new];
                components.scheme = @"spectre";
                components.path = @"import";
                components.queryItems = @[ [[NSURLQueryItem alloc] initWithName:@"data" value:exportedUser] ];
                [UIApp openURL:components.URL];
            } );
        }];
    }

    else if (self.spectreViewController)
        [self.window.rootViewController presentViewController:self.spectreViewController animated:YES completion:nil];
}

- (void)changeMasterPasswordFor:(MPUserEntity *)user saveInContext:(NSManagedObjectContext *)moc didResetBlock:(void ( ^ )(void))didReset {

    PearlMainQueue( ^{
        UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Changing Master Password" message:
                        @"If you continue, you'll be able to set a new master password.\n\n"
                        @"Changing your master password will cause all your generated passwords to change!\n"
                        @"Changing the master password back to the old one will cause your passwords to revert as well."
                                                                preferredStyle:UIAlertControllerStyleAlert];
        [alert addAction:[UIAlertAction actionWithTitle:@"Continue" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
            [moc performBlockAndWait:^{
                inf( @"Clearing keyID for user: %@.", user.userID );
                user.keyID = nil;
                [self forgetSavedKeyFor:user];
                [moc saveToStore];
            }];

            [self signOut];
            if (didReset)
                didReset();
        }]];
        [alert addAction:[UIAlertAction actionWithTitle:@"Abort" style:UIAlertActionStyleCancel handler:nil]];
        [self.window.rootViewController presentViewController:alert animated:YES completion:nil];
    } );
}

#pragma mark - SKStoreProductViewControllerDelegate

- (void)productViewControllerDidFinish:(SKStoreProductViewController *)viewController {

    [viewController dismissViewControllerAnimated:YES completion:nil];
}

#pragma mark - UIDocumentInteractionControllerDelegate

- (void)documentInteractionController:(UIDocumentInteractionController *)controller didEndSendingToApplication:(NSString *)application {

//    self.interactionController = nil;
}

#pragma mark - PearlConfigDelegate

- (void)didUpdateConfigForKey:(SEL)configKey fromValue:(id)value {

    [[NSNotificationCenter defaultCenter] postNotificationName:MPCheckConfigNotification object:NSStringFromSelector( configKey )];
}

- (void)updateConfigKey:(NSString *)key {

    // Trace mode
    [PearlLogger get].historyLevel = [[MPiOSConfig get].traceMode boolValue]? PearlLogLevelTrace: PearlLogLevelInfo;

    // Send info
    NSArray *countlyFeatures = @[
            CLYConsentSessions, CLYConsentEvents, CLYConsentUserDetails, CLYConsentCrashReporting, CLYConsentViewTracking, CLYConsentStarRating
    ];
    if ([[MPiOSConfig get].sendInfo boolValue] || ![[MPiOSConfig get].sendInfoDecided boolValue]) {
        if ([PearlLogger get].printLevel > PearlLogLevelInfo)
            [PearlLogger get].printLevel = PearlLogLevelInfo;

        [SentrySDK.currentHub getClient].options.enabled = @YES;
        [SentrySDK configureScope:^(SentryScope *scope) {
            [scope setExtraValue:[MPiOSConfig get].rememberLogin forKey:@"rememberLogin"];
            [scope setExtraValue:[MPiOSConfig get].sendInfo forKey:@"sendInfo"];
            [scope setExtraValue:[MPiOSConfig get].helpHidden forKey:@"helpHidden"];
            [scope setExtraValue:[MPiOSConfig get].showSetup forKey:@"showQuickStart"];
            [scope setExtraValue:[PearlConfig get].firstRun forKey:@"firstRun"];
            [scope setExtraValue:[PearlConfig get].launchCount forKey:@"launchCount"];
            [scope setExtraValue:[PearlConfig get].askForReviews forKey:@"askForReviews"];
            [scope setExtraValue:[PearlConfig get].reviewAfterLaunches forKey:@"reviewAfterLaunches"];
            [scope setExtraValue:[PearlConfig get].reviewedVersion forKey:@"reviewedVersion"];
            [scope setExtraValue:@([PearlDeviceUtils isSimulator]) forKey:@"simulator"];
            [scope setExtraValue:@([PearlDeviceUtils isAppEncrypted]) forKey:@"encrypted"];
            [scope setExtraValue:@([PearlDeviceUtils isJailbroken]) forKey:@"jailbroken"];
            [scope setExtraValue:[PearlDeviceUtils platform] forKey:@"platform"];
#ifdef APPSTORE
            [scope setExtraValue:@([PearlDeviceUtils isAppEncrypted]) forKey:@"reviewedVersion"];
#else
            [scope setExtraValue:@(NO) forKey:@"reviewedVersion"];
#endif
        }];
        [Countly.sharedInstance giveConsentForFeatures:countlyFeatures];
    }
    else {
        [Countly.sharedInstance cancelConsentForFeatures:countlyFeatures];
        [SentrySDK.currentHub getClient].options.enabled = @NO;
    }
}

@end