1
2
3
|
-
(
void
)
saveImage
:
(
UIImage
*
)
image
toAlbum
:
(
NSString
*
)
albumName
withCompletionBlock
:
(
SaveImageCompletion
)
completionBlock
;
|
1
|
#import <AssetsLibrary/AssetsLibrary.h>
|
1
|
@property
(
strong
,
atomic
)
ALAssetsLibrary
*
library
;
|
#import <UIKit/UIKit.h>
#import <AssetsLibrary/AssetsLibrary.h>
typedefvoid(^SaveImageCompletion)(NSError* error);
@interface ALAssetsLibrary (CustomPhotoAlbum)
-(void)saveImage:(UIImage *)image toAlbum:(NSString*)albumName withCompletionBlock:(SaveImageCompletion)completionBlock;
-(void)addAssetURL:(NSURL*)assetURL toAlbum:(NSString*)albumName withCompletionBlock:(SaveImageCompletion)completionBlock;
@end
#import "ALAssetsLibrary+CustomPhotoAlbum.h"
@implementation ALAssetsLibrary (CustomPhotoAlbum)
-(void)saveImage:(UIImage*)image toAlbum:(NSString*)albumName withCompletionBlock:(SaveImageCompletion)completionBlock
{
//write the image data to the assets library (camera roll)
[selfwriteImageToSavedPhotosAlbum:image.CGImageorientation:(ALAssetOrientation)image.imageOrientation
completionBlock:^(NSURL* assetURL,NSError* error) {
//error handling
if (error!=nil) {
completionBlock(error);
return;
}
//add the asset to the custom photo album
[selfaddAssetURL: assetURL
toAlbum:albumName
withCompletionBlock:completionBlock];
}];
}
-(void)addAssetURL:(NSURL*)assetURL toAlbum:(NSString*)albumName withCompletionBlock:(SaveImageCompletion)completionBlock
{
__block BOOL albumWasFound =NO;
//search all photo albums in the library
[selfenumerateGroupsWithTypes:ALAssetsGroupAlbum
usingBlock:^(ALAssetsGroup *group,BOOL *stop) {
//compare the names of the albums
if ([albumName compare: [groupvalueForProperty:ALAssetsGroupPropertyName]]==NSOrderedSame) {
//target album is found
albumWasFound =YES;
//get a hold of the photo's asset instance
[selfassetForURL: assetURL
resultBlock:^(ALAsset *asset) {
//add photo to the target album
[groupaddAsset: asset];
//run the completion block
completionBlock(nil);
}failureBlock: completionBlock];
//album was found, bail out of the method
return;
}
if (group==nil && albumWasFound==NO) {
//photo albums are over, target album does not exist, thus create it
__weak ALAssetsLibrary* weakSelf =self;
//create new assets album
[selfaddAssetsGroupAlbumWithName:albumName
resultBlock:^(ALAssetsGroup *group) {
//get the photo's instance
[weakSelfassetForURL: assetURL
resultBlock:^(ALAsset *asset) {
//add photo to the newly created album
[groupaddAsset: asset];
//call the completion block
completionBlock(nil);
}failureBlock: completionBlock];
}failureBlock: completionBlock];
//should be the last iteration anyway, but just in case
return;
}
}failureBlock: completionBlock];
}
@end