//
//  UIPasteboard+GetImageInfo.m
//  Mattermost
//
//  Created by Tek Min Ewe on 05/08/2019.
//  Copyright © 2019 Facebook. All rights reserved.
//

#import "UIPasteboard+GetImageInfo.h"
#import "NSData+MimeType.h"

@implementation UIPasteboard (GetImageInfo)

-(NSArray<NSDictionary *> *)getCopiedImages {
  NSArray<UIImage *> *images = self.images;
  NSMutableArray<NSDictionary *> *imageInfos = [[NSMutableArray alloc] init];
  for (int i = 0; i < images.count; i++) {
    UIImage *image = images[i];
    NSString *uri = self.string;
    NSArray<NSString *> *types = self.pasteboardTypes;
    
    NSData *imageData;
    if ([types[0] isEqual:@"public.jpeg"]) {
      imageData = UIImageJPEGRepresentation(image, 1.0);
    } else if ([types[0] isEqual:@"public.png"]) {
      imageData = UIImagePNGRepresentation(image);
    } else {
      imageData = [self dataForPasteboardType:types[0]];
    }
    NSString *extension = [imageData extension];
    NSString *mimeType = [imageData mimeType];
    
    NSString *tempFilename = [NSString stringWithFormat:@"%@%@", [[NSProcessInfo processInfo] globallyUniqueString], extension];
    NSURL *tempFileURL = [NSURL fileURLWithPath:[NSTemporaryDirectory() stringByAppendingPathComponent:tempFilename]];
    BOOL success = [imageData writeToURL:tempFileURL atomically:YES];
    
    if (success) {
      uri = tempFileURL.absoluteString;
    }
    
    [imageInfos addObject:@{
                            @"fileName": tempFilename,
                            @"fileSize": @([imageData length]),
                            @"type": mimeType,
                            @"uri": uri,
                            }];
  }
  
  return imageInfos;
}

@end