threema-ios/Threema/ForwardMultipleURLActivity.m

//  _____ _
// |_   _| |_  _ _ ___ ___ _ __  __ _
//   | | | ' \| '_/ -_) -_) '  \/ _` |_
//   |_| |_||_|_| \___\___|_|_|_\__,_(_)
//
// Threema iOS Client
// Copyright (c) 2020 Threema GmbH
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License, version 3,
// as published by the Free Software Foundation.
//
// This program 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 Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.

#import "ForwardMultipleURLActivity.h"
#import "ContactGroupPickerViewController.h"
#import "MessageSender.h"
#import "ModalNavigationController.h"
#import "FeatureMaskChecker.h"

@interface ForwardMultipleURLActivity () <ContactGroupPickerDelegate, ModalNavigationControllerDelegate>

@property NSMutableArray *objects;

@end

@implementation ForwardMultipleURLActivity

+ (UIActivityCategory)activityCategory {
    return UIActivityCategoryAction;
}

- (NSString *)activityType {
    return APP_ID ".forwardMsg";
}

- (NSString *)activityTitle {
    return NSLocalizedString(@"forward", nil);
}

- (UIImage *)activityImage {
    return [UIImage imageNamed:@"ShareForward"];
}

- (BOOL)canPerformWithActivityItems:(NSArray *)activityItems {
    if (activityItems.count > 20) {
        return NO;
    }
    
    for (id item in activityItems) {
        if (![item isKindOfClass:[NSURL class]]) {
            if ([item isKindOfClass:[NSDictionary class]]) {
                NSDictionary *itemDict = (NSDictionary *)item;
                if (![itemDict[@"url"] isKindOfClass:[NSURL class]]) {
                    return false;
                }
            } else {
                return false;
            }
        }
    }
    
    return true;
}

- (void)prepareWithActivityItems:(NSArray *)activityItems {
    if (_objects == nil) {
        _objects = [NSMutableArray new];
    }
    [_objects removeAllObjects];
    for (id item in activityItems) {
        if ([item isKindOfClass:[NSDictionary class]]) {
            NSDictionary *itemDict = (NSDictionary *)item;
            if ([itemDict[@"url"] isKindOfClass:[NSURL class]]) {
                NSMutableDictionary *object = [NSMutableDictionary new];
                [object setObject:itemDict[@"url"] forKey:@"url"];
                [object setObject:itemDict[@"renderType"] forKey:@"renderType"];
                [_objects addObject:object];
            }
        } else {
            NSMutableDictionary *object = [NSMutableDictionary new];
            [object setObject:item forKey:@"url"];
            [object setObject:@0 forKey:@"renderType"];
            [_objects addObject:object];
        }
    }
}

- (UIViewController *)activityViewController {
    ModalNavigationController *navigationController = [ContactGroupPickerViewController pickerFromStoryboardWithDelegate:self];
    ContactGroupPickerViewController *picker = (ContactGroupPickerViewController *)navigationController.topViewController;
    picker.enableMulitSelection = true;
    picker.enableTextInput = true;
    picker.submitOnSelect = false;
    picker.renderType = @0;
    return navigationController;
}

#pragma mark - ContactPickerDelegate

- (void)contactPicker:(ContactGroupPickerViewController*)contactPicker didPickConversations:(NSSet *)conversations renderType:(NSNumber *)renderType sendAsFile:(BOOL)sendAsFile {
    FeatureMaskChecker *featureMaskChecker = [[FeatureMaskChecker alloc] init];
    
    for (NSDictionary *object in _objects) {
        NSURL *url = object[@"url"];
        
        [featureMaskChecker checkFileTransferFor:conversations presentAlarmOn:contactPicker onSuccess:^{
            for (Conversation *conversation in conversations) {
                [URLSender sendUrl:url asFile:sendAsFile caption:contactPicker.additionalTextToSend conversation:conversation];
            }
            [contactPicker dismissViewControllerAnimated:YES completion:nil];
        } onFailure:^{
            [contactPicker dismissViewControllerAnimated:YES completion:nil];
        }];
    }
}

- (void)contactPickerDidCancel:(ContactGroupPickerViewController*)contactPicker {
    [contactPicker dismissViewControllerAnimated:YES completion:nil];
}

#pragma mark - ModalNavigationControllerDelegate

- (void)willDismissModalNavigationController {
    
}

@end