packages/image_picker/ios/Classes/ImagePickerPlugin.m - mirrors/plugins - Git at Google

 // Copyright 2017 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.

 #import "ImagePickerPlugin.h"

 #import <AVFoundation/AVFoundation.h>
 #import <MobileCoreServices/MobileCoreServices.h>
 #import <Photos/Photos.h>
 #import <UIKit/UIKit.h>

 @interface FLTImagePickerPlugin () <UINavigationControllerDelegate, UIImagePickerControllerDelegate>
 @end

 static const int SOURCE_CAMERA = 0;
 static const int SOURCE_GALLERY = 1;

 @implementation FLTImagePickerPlugin {
   FlutterResult _result;
   NSDictionary *_arguments;
   UIImagePickerController *_imagePickerController;
   UIViewController *_viewController;
 }

 + (void)registerWithRegistrar:(NSObject<FlutterPluginRegistrar> *)registrar {
   FlutterMethodChannel *channel =
       [FlutterMethodChannel methodChannelWithName:@"plugins.flutter.io/image_picker"
                                   binaryMessenger:[registrar messenger]];
   UIViewController *viewController =
       [UIApplication sharedApplication].delegate.window.rootViewController;
   FLTImagePickerPlugin *instance =
       [[FLTImagePickerPlugin alloc] initWithViewController:viewController];
   [registrar addMethodCallDelegate:instance channel:channel];
 }

 - (instancetype)initWithViewController:(UIViewController *)viewController {
   self = [super init];
   if (self) {
     _viewController = viewController;
     _imagePickerController = [[UIImagePickerController alloc] init];
   }
   return self;
 }

 - (void)handleMethodCall:(FlutterMethodCall *)call result:(FlutterResult)result {
   if (_result) {
     _result([FlutterError errorWithCode:@"multiple_request"
                                 message:@"Cancelled by a second request"
                                 details:nil]);
     _result = nil;
   }

   if ([@"pickImage" isEqualToString:call.method]) {
     _imagePickerController.modalPresentationStyle = UIModalPresentationCurrentContext;
     _imagePickerController.delegate = self;
     _imagePickerController.mediaTypes = @[ (NSString *)kUTTypeImage ];

     _result = result;
     _arguments = call.arguments;

     int imageSource = [[_arguments objectForKey:@"source"] intValue];

     switch (imageSource) {
       case SOURCE_CAMERA:
         [self checkAuthorization];
         break;
       case SOURCE_GALLERY:
         [self showPhotoLibrary];
         break;
       default:
         result([FlutterError errorWithCode:@"invalid_source"
                                    message:@"Invalid image source."
                                    details:nil]);
         break;
     }
   } else if ([@"pickVideo" isEqualToString:call.method]) {
     _imagePickerController.modalPresentationStyle = UIModalPresentationCurrentContext;
     _imagePickerController.delegate = self;
     _imagePickerController.mediaTypes = @[
       (NSString *)kUTTypeMovie, (NSString *)kUTTypeAVIMovie, (NSString *)kUTTypeVideo,
       (NSString *)kUTTypeMPEG4
     ];
     _imagePickerController.videoQuality = UIImagePickerControllerQualityTypeHigh;

     _result = result;
     _arguments = call.arguments;

     int imageSource = [[_arguments objectForKey:@"source"] intValue];

     switch (imageSource) {
       case SOURCE_CAMERA:
         [self checkAuthorization];
         break;
       case SOURCE_GALLERY:
         [self showPhotoLibrary];
         break;
       default:
         result([FlutterError errorWithCode:@"invalid_source"
                                    message:@"Invalid video source."
                                    details:nil]);
         break;
     }
   } else {
     result(FlutterMethodNotImplemented);
   }
 }

 - (void)showCamera {
   @synchronized(self) {
     if (_imagePickerController.beingPresented) {
       return;
     }
   }
   // Camera is not available on simulators
   if ([UIImagePickerController isSourceTypeAvailable:UIImagePickerControllerSourceTypeCamera]) {
     _imagePickerController.sourceType = UIImagePickerControllerSourceTypeCamera;
     [_viewController presentViewController:_imagePickerController animated:YES completion:nil];
   } else {
     [[[UIAlertView alloc] initWithTitle:@"Error"
                                 message:@"Camera not available."
                                delegate:nil
                       cancelButtonTitle:@"OK"
                       otherButtonTitles:nil] show];
   }
 }

 - (void)checkAuthorization {
   AVAuthorizationStatus status = [AVCaptureDevice authorizationStatusForMediaType:AVMediaTypeVideo];

   switch (status) {
     case AVAuthorizationStatusAuthorized:
       [self showCamera];
       break;
     case AVAuthorizationStatusNotDetermined: {
       [AVCaptureDevice requestAccessForMediaType:AVMediaTypeVideo
                                completionHandler:^(BOOL granted) {
                                  if (granted) {
                                    dispatch_async(dispatch_get_main_queue(), ^{
                                      if (granted) {
                                        [self showCamera];
                                      }
                                    });
                                  } else {
                                    dispatch_async(dispatch_get_main_queue(), ^{
                                      [self errorNoAccess:AVAuthorizationStatusDenied];
                                    });
                                  }
                                }];
     }; break;
     case AVAuthorizationStatusDenied:
     case AVAuthorizationStatusRestricted:
     default:
       [self errorNoAccess:status];
       break;
   }
 }

 - (void)errorNoAccess:(AVAuthorizationStatus)status {
   switch (status) {
     case AVAuthorizationStatusRestricted:
       _result([FlutterError errorWithCode:@"camera_access_restricted"
                                   message:@"The user is not allowed to use the camera."
                                   details:nil]);
       break;
     case AVAuthorizationStatusDenied:
     default:
       _result([FlutterError errorWithCode:@"camera_access_denied"
                                   message:@"The user did not allow camera access."
                                   details:nil]);
       break;
   }
 }

 - (void)showPhotoLibrary {
   // No need to check if SourceType is available. It always is.
   _imagePickerController.sourceType = UIImagePickerControllerSourceTypePhotoLibrary;
   [_viewController presentViewController:_imagePickerController animated:YES completion:nil];
 }

 - (void)imagePickerController:(UIImagePickerController *)picker
     didFinishPickingMediaWithInfo:(NSDictionary<NSString *, id> *)info {
   NSURL *videoURL = [info objectForKey:UIImagePickerControllerMediaURL];
   UIImage *image = [info objectForKey:UIImagePickerControllerEditedImage];
   [_imagePickerController dismissViewControllerAnimated:YES completion:nil];
   // The method dismissViewControllerAnimated does not immediately prevent
   // further didFinishPickingMediaWithInfo invocations. A nil check is necessary
   // to prevent below code to be unwantly executed multiple times and cause a
   // crash.
   if (!_result) {
     return;
   }
   if (videoURL != nil) {
     _result(videoURL.description);
   } else {
     if (image == nil) {
       image = [info objectForKey:UIImagePickerControllerOriginalImage];
     }
     image = [self normalizedImage:image];

     NSNumber *maxWidth = [_arguments objectForKey:@"maxWidth"];
     NSNumber *maxHeight = [_arguments objectForKey:@"maxHeight"];

     if (maxWidth != (id)[NSNull null] || maxHeight != (id)[NSNull null]) {
       image = [self scaledImage:image maxWidth:maxWidth maxHeight:maxHeight];
     }

     BOOL saveAsPNG = [self hasAlpha:image];
     NSData *data =
         saveAsPNG ? UIImagePNGRepresentation(image) : UIImageJPEGRepresentation(image, 1.0);
     NSString *fileExtension = saveAsPNG ? @"image_picker_%@.png" : @"image_picker_%@.jpg";
     NSString *guid = [[NSProcessInfo processInfo] globallyUniqueString];
     NSString *tmpFile = [NSString stringWithFormat:fileExtension, guid];
     NSString *tmpDirectory = NSTemporaryDirectory();
     NSString *tmpPath = [tmpDirectory stringByAppendingPathComponent:tmpFile];

     if ([[NSFileManager defaultManager] createFileAtPath:tmpPath contents:data attributes:nil]) {
       _result(tmpPath);
     } else {
       _result([FlutterError errorWithCode:@"create_error"
                                   message:@"Temporary file could not be created"
                                   details:nil]);
     }
   }

   _result = nil;
   _arguments = nil;
 }

 - (void)imagePickerControllerDidCancel:(UIImagePickerController *)picker {
   [_imagePickerController dismissViewControllerAnimated:YES completion:nil];
   _result(nil);

   _result = nil;
   _arguments = nil;
 }

 // The way we save images to the tmp dir currently throws away all EXIF data
 // (including the orientation of the image). That means, pics taken in portrait
 // will not be orientated correctly as is. To avoid that, we rotate the actual
 // image data.
 // TODO(goderbauer): investigate how to preserve EXIF data.
 - (UIImage *)normalizedImage:(UIImage *)image {
   if (image.imageOrientation == UIImageOrientationUp) return image;

   UIGraphicsBeginImageContextWithOptions(image.size, NO, image.scale);
   [image drawInRect:(CGRect){0, 0, image.size}];
   UIImage *normalizedImage = UIGraphicsGetImageFromCurrentImageContext();
   UIGraphicsEndImageContext();
   return normalizedImage;
 }

 - (UIImage *)scaledImage:(UIImage *)image
                 maxWidth:(NSNumber *)maxWidth
                maxHeight:(NSNumber *)maxHeight {
   double originalWidth = image.size.width;
   double originalHeight = image.size.height;

   bool hasMaxWidth = maxWidth != (id)[NSNull null];
   bool hasMaxHeight = maxHeight != (id)[NSNull null];

   double width = hasMaxWidth ? MIN([maxWidth doubleValue], originalWidth) : originalWidth;
   double height = hasMaxHeight ? MIN([maxHeight doubleValue], originalHeight) : originalHeight;

   bool shouldDownscaleWidth = hasMaxWidth && [maxWidth doubleValue] < originalWidth;
   bool shouldDownscaleHeight = hasMaxHeight && [maxHeight doubleValue] < originalHeight;
   bool shouldDownscale = shouldDownscaleWidth || shouldDownscaleHeight;

   if (shouldDownscale) {
     double downscaledWidth = floor((height / originalHeight) * originalWidth);
     double downscaledHeight = floor((width / originalWidth) * originalHeight);

     if (width < height) {
       if (!hasMaxWidth) {
         width = downscaledWidth;
       } else {
         height = downscaledHeight;
       }
     } else if (height < width) {
       if (!hasMaxHeight) {
         height = downscaledHeight;
       } else {
         width = downscaledWidth;
       }
     } else {
       if (originalWidth < originalHeight) {
         width = downscaledWidth;
       } else if (originalHeight < originalWidth) {
         height = downscaledHeight;
       }
     }
   }

   UIGraphicsBeginImageContextWithOptions(CGSizeMake(width, height), NO, 1.0);
   [image drawInRect:CGRectMake(0, 0, width, height)];

   UIImage *scaledImage = UIGraphicsGetImageFromCurrentImageContext();
   UIGraphicsEndImageContext();

   return scaledImage;
 }

 // Returns true if the image has an alpha layer
 - (BOOL)hasAlpha:(UIImage *)image {
   CGImageAlphaInfo alpha = CGImageGetAlphaInfo(image.CGImage);
   return (alpha == kCGImageAlphaFirst || alpha == kCGImageAlphaLast ||
           alpha == kCGImageAlphaPremultipliedFirst || alpha == kCGImageAlphaPremultipliedLast);
 }

 @end
	// Copyright 2017 The Chromium Authors. All rights reserved.
	// Use of this source code is governed by a BSD-style license that can be
	// found in the LICENSE file.

	#import "ImagePickerPlugin.h"

	#import <AVFoundation/AVFoundation.h>
	#import <MobileCoreServices/MobileCoreServices.h>
	#import <Photos/Photos.h>
	#import <UIKit/UIKit.h>

	@interface FLTImagePickerPlugin () <UINavigationControllerDelegate, UIImagePickerControllerDelegate>
	@end

	static const int SOURCE_CAMERA = 0;
	static const int SOURCE_GALLERY = 1;

	@implementation FLTImagePickerPlugin {
	FlutterResult _result;
	NSDictionary *_arguments;
	UIImagePickerController *_imagePickerController;
	UIViewController *_viewController;
	}

	+ (void)registerWithRegistrar:(NSObject<FlutterPluginRegistrar> *)registrar {
	FlutterMethodChannel *channel =
	[FlutterMethodChannel methodChannelWithName:@"plugins.flutter.io/image_picker"
	binaryMessenger:[registrar messenger]];
	UIViewController *viewController =
	[UIApplication sharedApplication].delegate.window.rootViewController;
	FLTImagePickerPlugin *instance =
	[[FLTImagePickerPlugin alloc] initWithViewController:viewController];
	[registrar addMethodCallDelegate:instance channel:channel];
	}

	- (instancetype)initWithViewController:(UIViewController *)viewController {
	self = [super init];
	if (self) {
	_viewController = viewController;
	_imagePickerController = [[UIImagePickerController alloc] init];
	}
	return self;
	}

	- (void)handleMethodCall:(FlutterMethodCall *)call result:(FlutterResult)result {
	if (_result) {
	_result([FlutterError errorWithCode:@"multiple_request"
	message:@"Cancelled by a second request"
	details:nil]);
	_result = nil;
	}

	if ([@"pickImage" isEqualToString:call.method]) {
	_imagePickerController.modalPresentationStyle = UIModalPresentationCurrentContext;
	_imagePickerController.delegate = self;
	_imagePickerController.mediaTypes = @[ (NSString *)kUTTypeImage ];

	_result = result;
	_arguments = call.arguments;

	int imageSource = [[_arguments objectForKey:@"source"] intValue];

	switch (imageSource) {
	case SOURCE_CAMERA:
	[self checkAuthorization];
	break;
	case SOURCE_GALLERY:
	[self showPhotoLibrary];
	break;
	default:
	result([FlutterError errorWithCode:@"invalid_source"
	message:@"Invalid image source."
	details:nil]);
	break;
	}
	} else if ([@"pickVideo" isEqualToString:call.method]) {
	_imagePickerController.modalPresentationStyle = UIModalPresentationCurrentContext;
	_imagePickerController.delegate = self;
	_imagePickerController.mediaTypes = @[
	(NSString )kUTTypeMovie, (NSString )kUTTypeAVIMovie, (NSString *)kUTTypeVideo,
	(NSString *)kUTTypeMPEG4
	];
	_imagePickerController.videoQuality = UIImagePickerControllerQualityTypeHigh;

	_result = result;
	_arguments = call.arguments;

	int imageSource = [[_arguments objectForKey:@"source"] intValue];

	switch (imageSource) {
	case SOURCE_CAMERA:
	[self checkAuthorization];
	break;
	case SOURCE_GALLERY:
	[self showPhotoLibrary];
	break;
	default:
	result([FlutterError errorWithCode:@"invalid_source"
	message:@"Invalid video source."
	details:nil]);
	break;
	}
	} else {
	result(FlutterMethodNotImplemented);
	}
	}

	- (void)showCamera {
	@synchronized(self) {
	if (_imagePickerController.beingPresented) {
	return;
	}
	}
	// Camera is not available on simulators
	if ([UIImagePickerController isSourceTypeAvailable:UIImagePickerControllerSourceTypeCamera]) {
	_imagePickerController.sourceType = UIImagePickerControllerSourceTypeCamera;
	[_viewController presentViewController:_imagePickerController animated:YES completion:nil];
	} else {
	[[[UIAlertView alloc] initWithTitle:@"Error"
	message:@"Camera not available."
	delegate:nil
	cancelButtonTitle:@"OK"
	otherButtonTitles:nil] show];
	}
	}

	- (void)checkAuthorization {
	AVAuthorizationStatus status = [AVCaptureDevice authorizationStatusForMediaType:AVMediaTypeVideo];

	switch (status) {
	case AVAuthorizationStatusAuthorized:
	[self showCamera];
	break;
	case AVAuthorizationStatusNotDetermined: {
	[AVCaptureDevice requestAccessForMediaType:AVMediaTypeVideo
	completionHandler:^(BOOL granted) {
	if (granted) {
	dispatch_async(dispatch_get_main_queue(), ^{
	if (granted) {
	[self showCamera];
	}
	});
	} else {
	dispatch_async(dispatch_get_main_queue(), ^{
	[self errorNoAccess:AVAuthorizationStatusDenied];
	});
	}
	}];
	}; break;
	case AVAuthorizationStatusDenied:
	case AVAuthorizationStatusRestricted:
	default:
	[self errorNoAccess:status];
	break;
	}
	}

	- (void)errorNoAccess:(AVAuthorizationStatus)status {
	switch (status) {
	case AVAuthorizationStatusRestricted:
	_result([FlutterError errorWithCode:@"camera_access_restricted"
	message:@"The user is not allowed to use the camera."
	details:nil]);
	break;
	case AVAuthorizationStatusDenied:
	default:
	_result([FlutterError errorWithCode:@"camera_access_denied"
	message:@"The user did not allow camera access."
	details:nil]);
	break;
	}
	}

	- (void)showPhotoLibrary {
	// No need to check if SourceType is available. It always is.
	_imagePickerController.sourceType = UIImagePickerControllerSourceTypePhotoLibrary;
	[_viewController presentViewController:_imagePickerController animated:YES completion:nil];
	}

	- (void)imagePickerController:(UIImagePickerController *)picker
	didFinishPickingMediaWithInfo:(NSDictionary<NSString , id> )info {
	NSURL *videoURL = [info objectForKey:UIImagePickerControllerMediaURL];
	UIImage *image = [info objectForKey:UIImagePickerControllerEditedImage];
	[_imagePickerController dismissViewControllerAnimated:YES completion:nil];
	// The method dismissViewControllerAnimated does not immediately prevent
	// further didFinishPickingMediaWithInfo invocations. A nil check is necessary
	// to prevent below code to be unwantly executed multiple times and cause a
	// crash.
	if (!_result) {
	return;
	}
	if (videoURL != nil) {
	_result(videoURL.description);
	} else {
	if (image == nil) {
	image = [info objectForKey:UIImagePickerControllerOriginalImage];
	}
	image = [self normalizedImage:image];

	NSNumber *maxWidth = [_arguments objectForKey:@"maxWidth"];
	NSNumber *maxHeight = [_arguments objectForKey:@"maxHeight"];

	if (maxWidth != (id)[NSNull null] \|\| maxHeight != (id)[NSNull null]) {
	image = [self scaledImage:image maxWidth:maxWidth maxHeight:maxHeight];
	}

	BOOL saveAsPNG = [self hasAlpha:image];
	NSData *data =
	saveAsPNG ? UIImagePNGRepresentation(image) : UIImageJPEGRepresentation(image, 1.0);
	NSString *fileExtension = saveAsPNG ? @"image_picker_%@.png" : @"image_picker_%@.jpg";
	NSString *guid = [[NSProcessInfo processInfo] globallyUniqueString];
	NSString *tmpFile = [NSString stringWithFormat:fileExtension, guid];
	NSString *tmpDirectory = NSTemporaryDirectory();
	NSString *tmpPath = [tmpDirectory stringByAppendingPathComponent:tmpFile];

	if ([[NSFileManager defaultManager] createFileAtPath:tmpPath contents:data attributes:nil]) {
	_result(tmpPath);
	} else {
	_result([FlutterError errorWithCode:@"create_error"
	message:@"Temporary file could not be created"
	details:nil]);
	}
	}

	_result = nil;
	_arguments = nil;
	}

	- (void)imagePickerControllerDidCancel:(UIImagePickerController *)picker {
	[_imagePickerController dismissViewControllerAnimated:YES completion:nil];
	_result(nil);

	_result = nil;
	_arguments = nil;
	}

	// The way we save images to the tmp dir currently throws away all EXIF data
	// (including the orientation of the image). That means, pics taken in portrait
	// will not be orientated correctly as is. To avoid that, we rotate the actual
	// image data.
	// TODO(goderbauer): investigate how to preserve EXIF data.
	- (UIImage )normalizedImage:(UIImage )image {
	if (image.imageOrientation == UIImageOrientationUp) return image;

	UIGraphicsBeginImageContextWithOptions(image.size, NO, image.scale);
	[image drawInRect:(CGRect){0, 0, image.size}];
	UIImage *normalizedImage = UIGraphicsGetImageFromCurrentImageContext();
	UIGraphicsEndImageContext();
	return normalizedImage;
	}

	- (UIImage )scaledImage:(UIImage )image
	maxWidth:(NSNumber *)maxWidth
	maxHeight:(NSNumber *)maxHeight {
	double originalWidth = image.size.width;
	double originalHeight = image.size.height;

	bool hasMaxWidth = maxWidth != (id)[NSNull null];
	bool hasMaxHeight = maxHeight != (id)[NSNull null];

	double width = hasMaxWidth ? MIN([maxWidth doubleValue], originalWidth) : originalWidth;
	double height = hasMaxHeight ? MIN([maxHeight doubleValue], originalHeight) : originalHeight;

	bool shouldDownscaleWidth = hasMaxWidth && [maxWidth doubleValue] < originalWidth;
	bool shouldDownscaleHeight = hasMaxHeight && [maxHeight doubleValue] < originalHeight;
	bool shouldDownscale = shouldDownscaleWidth \|\| shouldDownscaleHeight;

	if (shouldDownscale) {
	double downscaledWidth = floor((height / originalHeight) * originalWidth);
	double downscaledHeight = floor((width / originalWidth) * originalHeight);

	if (width < height) {
	if (!hasMaxWidth) {
	width = downscaledWidth;
	} else {
	height = downscaledHeight;
	}
	} else if (height < width) {
	if (!hasMaxHeight) {
	height = downscaledHeight;
	} else {
	width = downscaledWidth;
	}
	} else {
	if (originalWidth < originalHeight) {
	width = downscaledWidth;
	} else if (originalHeight < originalWidth) {
	height = downscaledHeight;
	}
	}
	}

	UIGraphicsBeginImageContextWithOptions(CGSizeMake(width, height), NO, 1.0);
	[image drawInRect:CGRectMake(0, 0, width, height)];

	UIImage *scaledImage = UIGraphicsGetImageFromCurrentImageContext();
	UIGraphicsEndImageContext();

	return scaledImage;
	}

	// Returns true if the image has an alpha layer
	- (BOOL)hasAlpha:(UIImage *)image {
	CGImageAlphaInfo alpha = CGImageGetAlphaInfo(image.CGImage);
	return (alpha == kCGImageAlphaFirst \|\| alpha == kCGImageAlphaLast \|\|
	alpha == kCGImageAlphaPremultipliedFirst \|\| alpha == kCGImageAlphaPremultipliedLast);
	}

	@end