//
//  Copyright (c) 2020 Open Whisper Systems. All rights reserved.
//

#import "SharingThreadPickerViewController.h"
#import "Environment.h"
#import "SignalApp.h"
#import "ThreadUtil.h"
#import "UIFont+OWS.h"
#import "UIView+OWS.h"
#import <SignalCoreKit/NSString+OWS.h>
#import <SignalMessaging/SignalMessaging-Swift.h>
#import <SignalServiceKit/OWSError.h>
#import <SignalServiceKit/OWSMessageSender.h>
#import <SignalServiceKit/TSThread.h>

NS_ASSUME_NONNULL_BEGIN

typedef void (^SendCompletionBlock)(NSError *_Nullable, TSOutgoingMessage *);
typedef void (^SendMessageBlock)(SendCompletionBlock completion);

@interface SharingThreadPickerViewController () <SelectThreadViewControllerDelegate,
    AttachmentApprovalViewControllerDelegate,
    TextApprovalViewControllerDelegate,
    ContactShareApprovalViewControllerDelegate>

@property (nonatomic, readonly) OWSContactsManager *contactsManager;
@property (nonatomic, readonly) OWSMessageSender *messageSender;
@property (nonatomic) TSThread *thread;
@property (nonatomic, readonly, weak) id<ShareViewDelegate> shareViewDelegate;
@property (nonatomic, readonly) UIProgressView *progressView;
@property (atomic, nullable) TSOutgoingMessage *outgoingMessage;

@end

#pragma mark -

@implementation SharingThreadPickerViewController

- (instancetype)initWithShareViewDelegate:(id<ShareViewDelegate>)shareViewDelegate
{
    self = [super init];
    if (!self) {
        return self;
    }

    _shareViewDelegate = shareViewDelegate;
    self.selectThreadViewDelegate = self;

    return self;
}

#pragma mark - Dependencies

- (SDSDatabaseStorage *)databaseStorage
{
    return SDSDatabaseStorage.shared;
}

#pragma mark - UIViewController overrides

- (void)loadView
{
    [super loadView];

    _contactsManager = Environment.shared.contactsManager;
    _messageSender = SSKEnvironment.shared.messageSender;

    _progressView = [[UIProgressView alloc] initWithProgressViewStyle:UIProgressViewStyleDefault];
    self.title = NSLocalizedString(@"SHARE_EXTENSION_VIEW_TITLE", @"Title for the 'share extension' view.");
}

- (void)viewDidLoad
{
    [super viewDidLoad];

    self.view.backgroundColor = Theme.backgroundColor;

    [[NSNotificationCenter defaultCenter] addObserver:self
                                             selector:@selector(attachmentUploadProgress:)
                                                 name:kAttachmentUploadProgressNotification
                                               object:nil];
}

- (BOOL)canSelectBlockedContact
{
    return NO;
}

- (nullable UIView *)createHeaderWithSearchBar:(UISearchBar *)searchBar
{
    OWSAssertDebug(searchBar);

    const CGFloat contentVMargin = 0;

    UIView *header = [UIView new];
    header.backgroundColor = Theme.backgroundColor;

    [header addSubview:searchBar];
    [searchBar autoPinEdgesToSuperviewEdges];

    UIView *borderView = [UIView new];
    [header addSubview:borderView];

    borderView.backgroundColor = Theme.cellSeparatorColor;
    [borderView autoSetDimension:ALDimensionHeight toSize:0.5];
    [borderView autoPinWidthToSuperview];
    [borderView autoPinEdgeToSuperviewEdge:ALEdgeBottom];

    // UITableViewController.tableHeaderView must have its height set.
    header.frame = CGRectMake(0, 0, 0, (contentVMargin * 2 + searchBar.frame.size.height));

    return header;
}

#pragma mark - SelectThreadViewControllerDelegate

- (nullable NSString *)convertAttachmentToMessageTextIfPossible
{
    if (self.attachments.count > 1) {
        return nil;
    }
    OWSAssertDebug(self.attachments.count == 1);
    SignalAttachment *attachment = self.attachments.firstObject;
    if (!attachment.isConvertibleToTextMessage) {
        return nil;
    }
    if (attachment.dataLength >= kOversizeTextMessageSizeThreshold) {
        return nil;
    }
    NSData *data = attachment.data;
    OWSAssertDebug(data.length < kOversizeTextMessageSizeThreshold);
    NSString *_Nullable messageText = [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding];
    OWSLogVerbose(@"messageTextForAttachment: %@", messageText);
    return [messageText filterStringForDisplay];
}

- (void)threadWasSelected:(TSThread *)thread
{
    OWSAssertDebug(self.attachments.count > 0);
    OWSAssertDebug(thread);

    self.thread = thread;

    if ([self tryToShareAsMessageText]) {
        return;
    }

    if ([self tryToShareAsContactShare]) {
        return;
    }

    OWSNavigationController *approvalModal =
        [AttachmentApprovalViewController wrappedInNavControllerWithAttachments:self.attachments
                                                             initialMessageText:nil
                                                               approvalDelegate:self];
    [self presentViewController:approvalModal animated:YES completion:nil];
}

- (BOOL)tryToShareAsMessageText
{
    OWSAssertDebug(self.attachments.count > 0);

    NSString *_Nullable messageText = [self convertAttachmentToMessageTextIfPossible];
    if (!messageText) {
        return NO;
    }

    TextApprovalViewController *approvalVC = [[TextApprovalViewController alloc] initWithMessageText:messageText];
    approvalVC.delegate = self;

    [self.navigationController pushViewController:approvalVC animated:YES];
    return YES;
}

- (BOOL)tryToShareAsContactShare
{
    OWSAssertDebug(self.attachments.count > 0);

    if (self.attachments.count > 1) {
        return NO;
    }
    OWSAssertDebug(self.attachments.count == 1);
    SignalAttachment *attachment = self.attachments.firstObject;
    if (!attachment.isConvertibleToContactShare) {
        return NO;
    }

    [self showContactShareApproval:attachment];
    return YES;
}

- (void)showContactShareApproval:(SignalAttachment *)attachment
{
    OWSAssertDebug(attachment);
    OWSAssertDebug(self.thread);
    OWSAssertDebug(attachment.isConvertibleToContactShare);

    NSData *data = attachment.data;

    CNContact *_Nullable cnContact = [Contact cnContactWithVCardData:data];
    Contact *_Nullable contact = [[Contact alloc] initWithSystemContact:cnContact];
    OWSContact *_Nullable contactShareRecord = [OWSContacts contactForSystemContact:cnContact];
    if (!contactShareRecord) {
        OWSLogError(@"Could not convert system contact.");
        return;
    }

    BOOL isProfileAvatar = NO;
    NSData *_Nullable avatarImageData = [self.contactsManager avatarDataForCNContactId:contact.cnContactId];
    for (SignalServiceAddress *address in contact.registeredAddresses) {
        if (avatarImageData) {
            break;
        }
        avatarImageData = [self.contactsManager profileImageDataForAddressWithSneakyTransaction:address];
        if (avatarImageData) {
            isProfileAvatar = YES;
        }
    }
    contactShareRecord.isProfileAvatar = isProfileAvatar;

    ContactShareViewModel *contactShare = [[ContactShareViewModel alloc] initWithContactShareRecord:contactShareRecord
                                                                                    avatarImageData:avatarImageData];

    ContactShareApprovalViewController *approvalVC =
        [[ContactShareApprovalViewController alloc] initWithContactShare:contactShare];
    approvalVC.delegate = self;
    [self.navigationController pushViewController:approvalVC animated:YES];
}

// override
- (void)dismissPressed:(id)sender
{
    OWSLogDebug(@"tapped dismiss share button");
    [self cancelShareExperience];
}

- (void)cancelShareExperience
{
    [self.shareViewDelegate shareViewWasCancelled];
}

#pragma mark - AttachmentApprovalViewControllerDelegate

- (void)attachmentApprovalDidAppear:(AttachmentApprovalViewController *_Nonnull)attachmentApproval
{
    // no-op
}

- (void)attachmentApproval:(AttachmentApprovalViewController *_Nonnull)attachmentApproval
     didApproveAttachments:(NSArray<SignalAttachment *> *_Nonnull)attachments
               messageText:(NSString *_Nullable)messageText
{
    [ThreadUtil addThreadToProfileWhitelistIfEmptyOrPendingRequestWithSneakyTransaction:self.thread];
    [self
        tryToSendMessageWithBlock:^(SendCompletionBlock sendCompletion) {
            OWSAssertIsOnMainThread();

            __block TSOutgoingMessage *outgoingMessage = nil;
            // DURABLE CLEANUP - SAE uses non-durable sending to make sure the app is running long enough to complete
            // the sending operation. Alternatively, we could use a durable send, but do more to make sure the
            // SAE runs as long as it needs.
            // TODO ALBUMS - send album via SAE

            [self.databaseStorage readWithBlock:^(SDSAnyReadTransaction *transaction) {
                outgoingMessage = [ThreadUtil sendMessageNonDurablyWithText:messageText
                                                           mediaAttachments:attachments
                                                                     thread:self.thread
                                                           quotedReplyModel:nil
                                                                transaction:transaction
                                                              messageSender:self.messageSender
                                                                 completion:^(NSError *_Nullable error) {
                                                                     sendCompletion(error, outgoingMessage);
                                                                 }];
            }];

            // This is necessary to show progress.
            self.outgoingMessage = outgoingMessage;
        }
               fromViewController:attachmentApproval];
}

- (void)attachmentApprovalDidCancel:(AttachmentApprovalViewController *)attachmentApproval
{
    [self cancelShareExperience];
}

- (void)attachmentApproval:(AttachmentApprovalViewController *)attachmentApproval
      didChangeMessageText:(nullable NSString *)newMessageText
{
    // no-op
}

- (nullable NSString *)attachmentApprovalTextInputContextIdentifier
{
    return nil;
}

- (NSArray<NSString *> *)attachmentApprovalRecipientNames
{
    return @[ [self.contactsManager displayNameForThreadWithSneakyTransaction:self.thread] ];
}

#pragma mark - TextApprovalViewControllerDelegate

- (void)textApproval:(TextApprovalViewController *)approvalViewController didApproveMessage:(NSString *)messageText
{
    OWSAssertDebug(messageText.length > 0);

    [ThreadUtil addThreadToProfileWhitelistIfEmptyOrPendingRequestWithSneakyTransaction:self.thread];
    [self tryToSendMessageWithBlock:^(SendCompletionBlock sendCompletion) {
        OWSAssertIsOnMainThread();

        __block TSOutgoingMessage *outgoingMessage = nil;
        // DURABLE CLEANUP - SAE uses non-durable sending to make sure the app is running long enough to complete
        // the sending operation. Alternatively, we could use a durable send, but do more to make sure the
        // SAE runs as long as it needs.
        [self.databaseStorage readWithBlock:^(SDSAnyReadTransaction *transaction) {
            outgoingMessage = [ThreadUtil sendMessageNonDurablyWithText:messageText
                                                                 thread:self.thread
                                                       quotedReplyModel:nil
                                                            transaction:transaction
                                                          messageSender:self.messageSender
                                                             completion:^(NSError *_Nullable error) {
                                                                 if (error) {
                                                                     sendCompletion(error, outgoingMessage);
                                                                 } else {
                                                                     sendCompletion(nil, outgoingMessage);
                                                                 }
                                                             }];
            // This is necessary to show progress.
            self.outgoingMessage = outgoingMessage;
        }];
    }
                 fromViewController:approvalViewController];
}

- (void)textApprovalDidCancel:(TextApprovalViewController *)approvalViewController
{
    [self cancelShareExperience];
}

- (nullable NSString *)textApprovalCustomTitle:(TextApprovalViewController *)approvalViewController
{
    return NSLocalizedString(@"MESSAGE_APPROVAL_DIALOG_TITLE", @"Title for the 'message approval' dialog.");
}

- (nullable NSString *)textApprovalRecipientsDescription:(TextApprovalViewController *)approvalViewController
{
    __block NSString *result;
    [self.databaseStorage uiReadWithBlock:^(SDSAnyReadTransaction *transaction) {
        result = [self.contactsManager displayNameForThread:self.thread transaction:transaction];
    }];
    return result;
}

- (ApprovalMode)textApprovalMode:(TextApprovalViewController *)approvalViewController
{
    return ApprovalModeSend;
}

#pragma mark - ContactShareApprovalViewControllerDelegate

- (void)approveContactShare:(ContactShareApprovalViewController *)approvalViewController
     didApproveContactShare:(ContactShareViewModel *)contactShare
{
    OWSLogInfo(@"");

    [ThreadUtil addThreadToProfileWhitelistIfEmptyOrPendingRequestWithSneakyTransaction:self.thread];
    [self tryToSendMessageWithBlock:^(SendCompletionBlock sendCompletion) {
        OWSAssertIsOnMainThread();
        // TODO - in line with QuotedReply and other message attachments, saving should happen as part of sending
        // preparation rather than duplicated here and in the SAE
        DatabaseStorageAsyncWrite(SDSDatabaseStorage.shared, ^(SDSAnyWriteTransaction *transaction) {
            if (contactShare.avatarImage) {
                [contactShare.dbRecord saveAvatarImage:contactShare.avatarImage transaction:transaction];
            }

            [transaction addAsyncCompletion:^{
                __block TSOutgoingMessage *outgoingMessage = nil;
                outgoingMessage = [ThreadUtil sendMessageNonDurablyWithContactShare:contactShare.dbRecord
                                                                             thread:self.thread
                                                                         completion:^(NSError *_Nullable error) {
                                                                             sendCompletion(error, outgoingMessage);
                                                                         }];
                // This is necessary to show progress.
                self.outgoingMessage = outgoingMessage;
            }];
        });
    }
                 fromViewController:approvalViewController];
}

- (void)approveContactShare:(ContactShareApprovalViewController *)approvalViewController
      didCancelContactShare:(ContactShareViewModel *)contactShare
{
    OWSLogInfo(@"");

    [self cancelShareExperience];
}

- (nullable NSString *)contactApprovalCustomTitle:(ContactShareApprovalViewController *)contactApproval
{
    return nil;
}

- (nullable NSString *)contactApprovalRecipientsDescription:(ContactShareApprovalViewController *)contactApproval
{
    OWSLogInfo(@"");

    __block NSString *result;
    [self.databaseStorage uiReadWithBlock:^(SDSAnyReadTransaction *transaction) {
        result = [self.contactsManager displayNameForThread:self.thread transaction:transaction];
    }];
    return result;
}

- (ApprovalMode)contactApprovalMode:(ContactShareApprovalViewController *)contactApproval
{
    OWSLogInfo(@"");

    return ApprovalModeSend;
}

#pragma mark - Helpers

- (void)tryToSendMessageWithBlock:(SendMessageBlock)sendMessageBlock
               fromViewController:(UIViewController *)fromViewController
{
    // Reset progress in case we're retrying
    self.progressView.progress = 0;

    ActionSheetController *progressActionSheet = [ActionSheetController new];

    UIView *headerWithProgress = [UIView new];
    headerWithProgress.backgroundColor = Theme.backgroundColor;
    headerWithProgress.layoutMargins = UIEdgeInsetsMake(16, 16, 16, 16);

    UILabel *titleLabel = [UILabel new];
    titleLabel.textAlignment = NSTextAlignmentCenter;
    titleLabel.numberOfLines = 0;
    titleLabel.lineBreakMode = NSLineBreakByWordWrapping;
    titleLabel.font = UIFont.ows_dynamicTypeSubheadlineClampedFont.ows_semibold;
    titleLabel.textColor = Theme.primaryTextColor;
    titleLabel.text = NSLocalizedString(@"SHARE_EXTENSION_SENDING_IN_PROGRESS_TITLE", @"Alert title");

    [headerWithProgress addSubview:titleLabel];
    [titleLabel autoPinWidthToSuperviewMargins];
    [titleLabel autoPinTopToSuperviewMargin];

    [headerWithProgress addSubview:self.progressView];
    [self.progressView autoPinWidthToSuperviewMargins];
    [self.progressView autoPinEdge:ALEdgeTop toEdge:ALEdgeBottom ofView:titleLabel withOffset:8];
    [self.progressView autoPinBottomToSuperviewMargin];

    progressActionSheet.customHeader = headerWithProgress;

    ActionSheetAction *progressCancelAction =
        [[ActionSheetAction alloc] initWithTitle:[CommonStrings cancelButton]
                                           style:ActionSheetActionStyleCancel
                                         handler:^(ActionSheetAction *_Nonnull action) {
                                             [self.shareViewDelegate shareViewWasCancelled];
                                         }];
    [progressActionSheet addAction:progressCancelAction];

    SendCompletionBlock sendCompletion = ^(NSError *_Nullable error, TSOutgoingMessage *message) {
        dispatch_async(dispatch_get_main_queue(), ^{
            if (error) {
                [fromViewController dismissViewControllerAnimated:YES
                                                       completion:^{
                                                           OWSLogInfo(@"Sending message failed with error: %@", error);
                                                           [self showSendFailureAlertWithError:error
                                                                                       message:message
                                                                            fromViewController:fromViewController];
                                                       }];
                return;
            }

            OWSLogInfo(@"Sending message succeeded.");
            [self.shareViewDelegate shareViewWasCompleted];
        });
    };

    [fromViewController presentActionSheet:progressActionSheet
                                completion:^{
                                    sendMessageBlock(sendCompletion);
                                }];
}

- (void)showSendFailureAlertWithError:(NSError *)error
                              message:(TSOutgoingMessage *)message
                   fromViewController:(UIViewController *)fromViewController
{
    OWSAssertIsOnMainThread();
    OWSAssertDebug(error);
    OWSAssertDebug(message);
    OWSAssertDebug(fromViewController);

    NSString *failureTitle = NSLocalizedString(@"SHARE_EXTENSION_SENDING_FAILURE_TITLE", @"Alert title");

    if ([error.domain isEqual:OWSSignalServiceKitErrorDomain] && error.code == OWSErrorCodeUntrustedIdentity) {
        SignalServiceAddress *_Nullable untrustedAddress = error.userInfo[OWSErrorRecipientAddressKey];

        NSString *failureFormat = NSLocalizedString(@"SHARE_EXTENSION_FAILED_SENDING_BECAUSE_UNTRUSTED_IDENTITY_FORMAT",
            @"alert body when sharing file failed because of untrusted/changed identity keys");

        NSString *displayName = [self.contactsManager displayNameForAddress:untrustedAddress];
        NSString *failureMessage = [NSString stringWithFormat:failureFormat, displayName];

        ActionSheetController *failureAlert = [[ActionSheetController alloc] initWithTitle:failureTitle
                                                                                   message:failureMessage];

        ActionSheetAction *failureCancelAction =
            [[ActionSheetAction alloc] initWithTitle:[CommonStrings cancelButton]
                                               style:ActionSheetActionStyleCancel
                                             handler:^(ActionSheetAction *_Nonnull action) {
                                                 [self.shareViewDelegate shareViewWasCancelled];
                                             }];
        [failureAlert addAction:failureCancelAction];

        if (untrustedAddress.isValid) {
            ActionSheetAction *confirmAction =
                [[ActionSheetAction alloc] initWithTitle:[SafetyNumberStrings confirmSendButton]
                                                   style:ActionSheetActionStyleDefault
                                                 handler:^(ActionSheetAction *action) {
                                                     [self confirmIdentityAndResendMessage:message
                                                                                   address:untrustedAddress
                                                                        fromViewController:fromViewController];
                                                 }];

            [failureAlert addAction:confirmAction];
        } else {
            // This shouldn't happen, but if it does we won't offer the user the ability to confirm.
            // They may have to return to the main app to accept the identity change.
            OWSFailDebug(@"Untrusted recipient error is missing recipient id.");
        }

        [fromViewController presentActionSheet:failureAlert];
    } else {
        // Non-identity failure, e.g. network offline, rate limit

        ActionSheetController *failureAlert = [[ActionSheetController alloc] initWithTitle:failureTitle
                                                                                   message:error.localizedDescription];

        ActionSheetAction *failureCancelAction =
            [[ActionSheetAction alloc] initWithTitle:[CommonStrings cancelButton]
                                               style:ActionSheetActionStyleCancel
                                             handler:^(ActionSheetAction *_Nonnull action) {
                                                 [self.shareViewDelegate shareViewWasCancelled];
                                             }];
        [failureAlert addAction:failureCancelAction];

        ActionSheetAction *retryAction =
            [[ActionSheetAction alloc] initWithTitle:[CommonStrings retryButton]
                                               style:ActionSheetActionStyleDefault
                                             handler:^(ActionSheetAction *action) {
                                                 [self resendMessage:message fromViewController:fromViewController];
                                             }];

        [failureAlert addAction:retryAction];
        [fromViewController presentActionSheet:failureAlert];
    }
}

- (void)confirmIdentityAndResendMessage:(TSOutgoingMessage *)message
                                address:(SignalServiceAddress *)address
                     fromViewController:(UIViewController *)fromViewController
{
    OWSAssertIsOnMainThread();
    OWSAssertDebug(message);
    OWSAssertDebug(address.isValid);
    OWSAssertDebug(fromViewController);

    OWSLogDebug(@"Confirming identity for recipient: %@", address);

    DatabaseStorageAsyncWrite(
        SDSDatabaseStorage.shared, ^(SDSAnyWriteTransaction *transaction) {
            OWSVerificationState verificationState =
                [[OWSIdentityManager sharedManager] verificationStateForAddress:address transaction:transaction];
            switch (verificationState) {
                case OWSVerificationStateVerified: {
                    OWSFailDebug(@"Shouldn't need to confirm identity if it was already verified");
                    break;
                }
                case OWSVerificationStateDefault: {
                    // If we learned of a changed SN during send, then we've already recorded the new identity
                    // and there's nothing else we need to do for the resend to succeed.
                    // We don't want to redundantly set status to "default" because we would create a
                    // "You marked Alice as unverified" notice, which wouldn't make sense if Alice was never
                    // marked as "Verified".
                    OWSLogInfo(@"recipient has acceptable verification status. Next send will succeed.");
                    break;
                }
                case OWSVerificationStateNoLongerVerified: {
                    OWSLogInfo(@"marked recipient: %@ as default verification status.", address);
                    NSData *identityKey = [[OWSIdentityManager sharedManager] identityKeyForAddress:address
                                                                                        transaction:transaction];
                    OWSAssertDebug(identityKey);
                    [[OWSIdentityManager sharedManager] setVerificationState:OWSVerificationStateDefault
                                                                 identityKey:identityKey
                                                                     address:address
                                                       isUserInitiatedChange:YES
                                                                 transaction:transaction];
                    break;
                }
            }

            [transaction addAsyncCompletion:^{
                [self resendMessage:message fromViewController:fromViewController];
            }];
        });
}

- (void)resendMessage:(TSOutgoingMessage *)message fromViewController:(UIViewController *)fromViewController
{
    OWSAssertIsOnMainThread();
    OWSAssertDebug(message);
    OWSAssertDebug(fromViewController);

    NSString *progressTitle = NSLocalizedString(@"SHARE_EXTENSION_SENDING_IN_PROGRESS_TITLE", @"Alert title");
    ActionSheetController *progressAlert = [[ActionSheetController alloc] initWithTitle:progressTitle message:nil];

    ActionSheetAction *progressCancelAction =
        [[ActionSheetAction alloc] initWithTitle:[CommonStrings cancelButton]
                                           style:ActionSheetActionStyleCancel
                                         handler:^(ActionSheetAction *_Nonnull action) {
                                             [self.shareViewDelegate shareViewWasCancelled];
                                         }];
    [progressAlert addAction:progressCancelAction];

    [fromViewController
        presentActionSheet:progressAlert
                completion:^{
                    [self.messageSender sendMessage:message.asPreparer
                        success:^{
                            OWSLogInfo(@"Resending attachment succeeded.");
                            dispatch_async(dispatch_get_main_queue(), ^{
                                [self.shareViewDelegate shareViewWasCompleted];
                            });
                        }
                        failure:^(NSError *error) {
                            dispatch_async(dispatch_get_main_queue(), ^{
                                [fromViewController
                                    dismissViewControllerAnimated:YES
                                                       completion:^{
                                                           OWSLogInfo(
                                                               @"Sending attachment failed with error: %@", error);
                                                           [self showSendFailureAlertWithError:error
                                                                                       message:message
                                                                            fromViewController:fromViewController];
                                                       }];
                            });
                        }];
                }];
}

- (void)attachmentUploadProgress:(NSNotification *)notification
{
    OWSLogDebug(@"upload progress.");
    OWSAssertIsOnMainThread();
    OWSAssertDebug(self.progressView);

    if (!self.outgoingMessage) {
        OWSLogDebug(@"Ignoring upload progress until there is an outgoing message.");
        return;
    }

    // TODO: Support multi-image messages.
    NSString *_Nullable attachmentRecordId = self.outgoingMessage.attachmentIds.firstObject;
    if (!attachmentRecordId) {
        OWSLogDebug(@"Ignoring upload progress until outgoing message has an attachment record id");
        return;
    }

    NSDictionary *userinfo = [notification userInfo];
    float progress = [[userinfo objectForKey:kAttachmentUploadProgressKey] floatValue];
    NSString *attachmentID = [userinfo objectForKey:kAttachmentUploadAttachmentIDKey];

    if ([attachmentRecordId isEqual:attachmentID]) {
        if (!isnan(progress)) {
            [self.progressView setProgress:progress animated:YES];
        } else {
            OWSFailDebug(@"Invalid attachment progress.");
        }
    }
}

@end

NS_ASSUME_NONNULL_END