//
//  RCTMultipleImagePicker.m
//  RCTMultipleImagePicker
//
//  Created by ShenTong on 16/6/20.
//
//

#import "RCTMultipleImagePicker.h"

#import <AssetsLibrary/AssetsLibrary.h>
#import <Photos/Photos.h>

@implementation RCTMultipleImagePicker

RCT_EXPORT_MODULE();

-(id)init {
    self = [super init];
    if(self) {
        self.assetsFromPath = [[NSMutableDictionary alloc] init];
        self.assetsToPath = [[NSMutableDictionary alloc] init];
    }
    return self;
}

RCT_EXPORT_METHOD(launchImageGallery:(NSDictionary *)options resolver:(RCTPromiseResolveBlock)resolve rejecter:(RCTPromiseRejectBlock)reject) {
    self.resolve = resolve;
    self.reject = reject;
    NSInteger maxImagesCount = [RCTConvert NSInteger:options[@"maxImagesCount"]];
    NSArray *selectedPaths = [RCTConvert NSArray:options[@"selectedPaths"]];
    NSMutableArray *selectedAssets = [[NSMutableArray alloc] init];
    [selectedPaths enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
        NSString *path = [RCTConvert NSString:obj];
        [selectedAssets addObject:[self.assetsFromPath objectForKey:path]];
    }];
  
    NSError * error = nil;
    
    [self checkPhotosPermissions:^(BOOL granted) {
        if (!granted) {
          self.reject(@"camera_permission_not_granted", @"Camera permissions not granted", error);
          return;
        } else {
          [self showImagePickerController:maxImagesCount selectedAssets:selectedAssets];
        }
    }];
}

RCT_EXPORT_METHOD(close) {
    if(self.imagePickerController !=nil) {
        [self.imagePickerController cancelButtonClick];
    }
}

RCT_EXPORT_METHOD(getBase64StringFromArray:(NSArray *)inputs callback:(RCTResponseSenderBlock)callback){
    NSMutableArray *result = [[NSMutableArray alloc] init];
    
    for (id object in inputs) {
        NSString *input = object;
        
        NSURL *url = [[NSURL alloc] initWithString:input];
        NSData *zipFileData = [NSData dataWithContentsOfFile:url];
        NSString *base64String = [zipFileData base64EncodedStringWithOptions:0];
        
        [result addObject:base64String];
    }
    
    callback(@[[NSNull null], result]);
}

RCT_EXPORT_METHOD(getBase64String:(NSString *)input callback:(RCTResponseSenderBlock)callback){
    NSURL *url = [[NSURL alloc] initWithString:input];
    NSData *zipFileData = [NSData dataWithContentsOfFile:url];
    NSString *base64String = [zipFileData base64EncodedStringWithOptions:0];
    
    callback(@[[NSNull null], base64String]);
}

#pragma mark TZImagePickerControllerDelegate

- (void)showImagePickerController:(NSInteger)maxImagesCount selectedAssets:(NSMutableArray *)selectedAssets {
  self.imagePickerController = [[TZImagePickerController alloc] initWithMaxImagesCount:maxImagesCount delegate:self];
  self.imagePickerController.allowPickingOriginalPhoto = NO;
  self.imagePickerController.allowPickingVideo = NO;
  self.imagePickerController.selectedAssets = selectedAssets;
  dispatch_async(dispatch_get_main_queue(), ^{
    UIViewController *root = [[[[UIApplication sharedApplication] delegate] window] rootViewController];
    [root presentViewController:self.imagePickerController
          animated:YES
          completion:NULL];
  });
}

- (void)imagePickerControllerDidCancel:(TZImagePickerController *)picker {
    self.reject(@"user_cancelled", @"User has cancelled.", nil);
}

- (void)imagePickerController:(TZImagePickerController *)picker didFinishPickingPhotos:(NSArray<UIImage *> *)photos sourceAssets:(NSArray *)assets isSelectOriginalPhoto:(BOOL)isSelectOriginalPhoto {
    NSMutableArray *result = [[NSMutableArray alloc] init];
    NSArray *paths = NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES);
    NSString *filePathBase = [[paths objectAtIndex:0] stringByAppendingPathComponent:[[NSProcessInfo processInfo] globallyUniqueString]];

    NSError * error = nil;
    [[NSFileManager defaultManager] createDirectoryAtPath:filePathBase
                              withIntermediateDirectories:YES
                                               attributes:nil
                                                    error:&error];
    if (error != nil) {
        self.reject(@"create_directory_failed", @"Fail to create directory.", nil);
    }
    
    [photos enumerateObjectsUsingBlock:^(UIImage * _Nonnull image, NSUInteger index, BOOL * _Nonnull stop) {
        PHAsset *asset = [assets objectAtIndex:index];
        NSString *path = [self.assetsToPath objectForKey:asset];
        if(path == nil) {
            path = [filePathBase stringByAppendingPathComponent:[NSString stringWithFormat: @"%d.jpg", (int)index]];
            [UIImageJPEGRepresentation(image, 1.0) writeToFile:path atomically:YES];
            [self.assetsFromPath setObject:asset forKey:path];
            [self.assetsToPath setObject:path forKey:asset];
        }
        [result addObject:path];
    }];
    
    self.resolve(result);
}

- (void)checkPhotosPermissions:(void(^)(BOOL granted))callback
{
    PHAuthorizationStatus status = [PHPhotoLibrary authorizationStatus];
    if (status == PHAuthorizationStatusAuthorized) {
        callback(YES);
        return;
    } else if (status == PHAuthorizationStatusNotDetermined) {
        [PHPhotoLibrary requestAuthorization:^(PHAuthorizationStatus status) {
            if (status == PHAuthorizationStatusAuthorized) {
                callback(YES);
                return;
            }
            else {
                callback(NO);
                return;
            }
        }];
    }
    else {
        callback(NO);
    }
}

@end