limitation for video size

Author: parveshneedhoo

src/ios/GMImagePicker/GMGridViewController.h

@@ -17,6 +17,7 @@
 @interface GMGridViewController : UICollectionViewController
 
 @property (strong) PHFetchResult *assetsFetchResults;
+@property (nonatomic, strong) NSArray *filteredAssets;
 @property (nonatomic, weak) NSMutableDictionary * dic_asset_fetches;
 
 -(id)initWithPicker:(GMImagePickerController *)picker;

src/ios/GMImagePicker/GMGridViewController.m

@@ -82,6 +82,12 @@ @implementation GMGridViewController
 
 @synthesize dic_asset_fetches;
 
+- (void)setAssetsFetchResults:(PHFetchResult *)assetsFetchResults
+{
+    _assetsFetchResults = assetsFetchResults;
+    [self filterAssets];
+}
+
 -(id)initWithPicker:(GMImagePickerController *)picker
 {
     //Custom init. The picker contains custom information to create the FlowLayout
@@ -147,6 +153,7 @@ - (void)viewDidLoad
     }
 
     self.imageManager = [[PHCachingImageManager alloc] init];
+    [self filterAssets];
     [self resetCachedAssets];
     [[PHPhotoLibrary sharedPhotoLibrary] registerChangeObserver:self];
 
@@ -222,6 +229,34 @@ - (void)setupViews
     self.collectionView.backgroundColor = [UIColor whiteColor];
 }
 
+- (void)filterAssets
+{
+    if (!self.assetsFetchResults) {
+        self.filteredAssets = @[];
+        return;
+    }
+    
+    NSMutableArray *filtered = [NSMutableArray array];
+    
+    // Check if delegate implements shouldShowAsset method
+    BOOL implementsFilter = [self.picker.delegate respondsToSelector:@selector(assetsPickerController:shouldShowAsset:)];
+    
+    for (NSInteger i = 0; i < self.assetsFetchResults.count; i++) {
+        PHAsset *asset = self.assetsFetchResults[i];
+        
+        if (implementsFilter) {
+            if ([self.picker.delegate assetsPickerController:self.picker shouldShowAsset:asset]) {
+                [filtered addObject:asset];
+            }
+        } else {
+            // If delegate doesn't implement the method, show all assets
+            [filtered addObject:asset];
+        }
+    }
+    
+    self.filteredAssets = [filtered copy];
+}
+
 - (void)setupButtons
 {
     self.navigationItem.rightBarButtonItem =
@@ -310,7 +345,7 @@ - (UICollectionViewCell *)collectionView:(UICollectionView *)collectionView cell
     NSInteger currentTag = cell.tag + 1;
     cell.tag = currentTag;
 
-    PHAsset *asset = self.assetsFetchResults[indexPath.item];
+    PHAsset *asset = self.filteredAssets[indexPath.item];
     [cell bind:asset];
 
     //GMFetchItem * fetch_item = [dic_asset_fetches objectForKey:[NSNumber numberWithLong:indexPath.item] ];
@@ -438,7 +473,7 @@ - (BOOL)collectionView:(UICollectionView *)collectionView shouldSelectItemAtInde
         return NO;
     }
 
-    PHAsset *asset = self.assetsFetchResults[indexPath.item];
+    PHAsset *asset = self.filteredAssets[indexPath.item];
     //GMFetchItem * fetch_item = [dic_asset_fetches objectForKey:[ NSNumber numberWithLong:indexPath.item ]];
     GMFetchItem * fetch_item = [dic_asset_fetches objectForKey:asset];
 
@@ -580,7 +615,7 @@ - (BOOL)collectionView:(UICollectionView *)collectionView shouldSelectItemAtInde
 
 - (void)collectionView:(UICollectionView *)collectionView didSelectItemAtIndexPath:(NSIndexPath *)indexPath
 {
-    PHAsset *asset = self.assetsFetchResults[indexPath.item];
+    PHAsset *asset = self.filteredAssets[indexPath.item];
     //GMFetchItem * fetch_item = [dic_asset_fetches objectForKey:[ NSNumber numberWithLong:indexPath.item ]];
     GMFetchItem * fetch_item = [dic_asset_fetches objectForKey:asset];
 
@@ -593,7 +628,7 @@ - (void)collectionView:(UICollectionView *)collectionView didSelectItemAtIndexPa
 
 - (BOOL)collectionView:(UICollectionView *)collectionView shouldDeselectItemAtIndexPath:(NSIndexPath *)indexPath
 {
-    PHAsset *asset = self.assetsFetchResults[indexPath.item];
+    PHAsset *asset = self.filteredAssets[indexPath.item];
 
     if ([self.picker.delegate respondsToSelector:@selector(assetsPickerController:shouldDeselectAsset:)])
         return [self.picker.delegate assetsPickerController:self.picker shouldDeselectAsset:asset];
@@ -603,7 +638,7 @@ - (BOOL)collectionView:(UICollectionView *)collectionView shouldDeselectItemAtIn
 
 - (void)collectionView:(UICollectionView *)collectionView didDeselectItemAtIndexPath:(NSIndexPath *)indexPath
 {
-    PHAsset *asset = self.assetsFetchResults[indexPath.item];
+    PHAsset *asset = self.filteredAssets[indexPath.item];
     //GMFetchItem * fetch_item = [dic_asset_fetches objectForKey:[ NSNumber numberWithLong:indexPath.item ]];
     GMFetchItem * fetch_item = [dic_asset_fetches objectForKey:asset];
 
@@ -616,7 +651,7 @@ - (void)collectionView:(UICollectionView *)collectionView didDeselectItemAtIndex
 
 - (BOOL)collectionView:(UICollectionView *)collectionView shouldHighlightItemAtIndexPath:(NSIndexPath *)indexPath
 {
-    PHAsset *asset = self.assetsFetchResults[indexPath.item];
+    PHAsset *asset = self.filteredAssets[indexPath.item];
 
     if ([self.picker.delegate respondsToSelector:@selector(assetsPickerController:shouldHighlightAsset:)])
         return [self.picker.delegate assetsPickerController:self.picker shouldHighlightAsset:asset];
@@ -626,15 +661,15 @@ - (BOOL)collectionView:(UICollectionView *)collectionView shouldHighlightItemAtI
 
 - (void)collectionView:(UICollectionView *)collectionView didHighlightItemAtIndexPath:(NSIndexPath *)indexPath
 {
-    PHAsset *asset = self.assetsFetchResults[indexPath.item];
+    PHAsset *asset = self.filteredAssets[indexPath.item];
 
     if ([self.picker.delegate respondsToSelector:@selector(assetsPickerController:didHighlightAsset:)])
         [self.picker.delegate assetsPickerController:self.picker didHighlightAsset:asset];
 }
 
 - (void)collectionView:(UICollectionView *)collectionView didUnhighlightItemAtIndexPath:(NSIndexPath *)indexPath
 {
-    PHAsset *asset = self.assetsFetchResults[indexPath.item];
+    PHAsset *asset = self.filteredAssets[indexPath.item];
 
     if ([self.picker.delegate respondsToSelector:@selector(assetsPickerController:didUnhighlightAsset:)])
         [self.picker.delegate assetsPickerController:self.picker didUnhighlightAsset:asset];
@@ -646,7 +681,7 @@ - (void)collectionView:(UICollectionView *)collectionView didUnhighlightItemAtIn
 
 - (NSInteger)collectionView:(UICollectionView *)collectionView numberOfItemsInSection:(NSInteger)section
 {
-    NSInteger count = self.assetsFetchResults.count;
+    NSInteger count = self.filteredAssets.count;
     return count;
 }
 
@@ -664,6 +699,7 @@ - (void)photoLibraryDidChange:(PHChange *)changeInstance
 
             // get the new fetch result
             self.assetsFetchResults = [collectionChanges fetchResultAfterChanges];
+            [self filterAssets];
 
             //NSLog( @"reset all" );
 
@@ -788,7 +824,7 @@ - (NSArray *)assetsAtIndexPaths:(NSArray *)indexPaths
 
     NSMutableArray *assets = [NSMutableArray arrayWithCapacity:indexPaths.count];
     for (NSIndexPath *indexPath in indexPaths) {
-        PHAsset *asset = self.assetsFetchResults[indexPath.item];
+        PHAsset *asset = self.filteredAssets[indexPath.item];
         [assets addObject:asset];
     }
     return assets;

src/ios/SOSPicker.h

@@ -25,5 +25,6 @@
 @property (nonatomic, assign) NSInteger height;
 @property (nonatomic, assign) NSInteger quality;
 @property (nonatomic, assign) NSInteger outputType;
+@property (nonatomic, assign) NSInteger maxMB;
 
 @end

src/ios/SOSPicker.m

@@ -7,6 +7,8 @@
 //
 
 #import "SOSPicker.h"
+#import <Photos/Photos.h>
+#import <AVFoundation/AVFoundation.h>
 #import "GMImagePickerController.h"
 #import "GMFetchItem.h"
 
@@ -65,6 +67,7 @@ - (void) getPictures:(CDVInvokedUrlCommand *)command {
     self.width = [[options objectForKey:@"width"] integerValue];
     self.height = [[options objectForKey:@"height"] integerValue];
     self.quality = [[options objectForKey:@"quality"] integerValue];
+    self.maxMB = [[options objectForKey:@"maxFileSize"] integerValue];
 
     self.callbackId = command.callbackId;
     [self launchGMImagePicker:allow_video title:title message:message disable_popover:disable_popover maximumImagesCount:maximumImagesCount];
@@ -435,4 +438,42 @@ - (void) closeImagePicker:(CDVInvokedUrlCommand *)command {
     [self.commandDelegate sendPluginResult:pluginResult callbackId:command.callbackId];
 }
 
+#pragma mark - GMImagePickerControllerDelegate - Asset Filtering
+
+- (BOOL)assetsPickerController:(GMImagePickerController *)picker shouldShowAsset:(PHAsset *)asset {
+    if (asset.mediaType != PHAssetMediaTypeVideo) {
+        return YES;
+    }
+
+    NSArray *resources = [PHAssetResource assetResourcesForAsset:asset];
+    long long fileSize = 0;
+    
+    for (PHAssetResource *resource in resources) {
+        if (resource.type == PHAssetResourceTypeVideo ||
+            resource.type == PHAssetResourceTypeFullSizeVideo ||
+            resource.type == PHAssetResourceTypePairedVideo) {
+            
+            @try {
+                id fileSizeValue = [resource valueForKey:@"fileSize"];
+                if ([fileSizeValue isKindOfClass:[NSNumber class]]) {
+                    fileSize = [fileSizeValue longLongValue];
+                    break;
+                }
+            } @catch (NSException *exception) {
+                // Continue to next resource if fileSize is not available
+            }
+        }
+    }
+    
+    if (fileSize > 0) {
+        CGFloat mb = fileSize / (1024.0 * 1024.0);
+        
+        if (mb > self.maxMB) {
+            return NO;
+        }
+        return YES;
+    }
+    return YES;
+}
+
 @end