Fixed custom asset view controller class handling

davbeck · Apr 3, 2015 · 1127d80 · 1127d80
1 parent e136379
commit 1127d80
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/Pod/Classes/TNKAssetViewController.h b/Pod/Classes/TNKAssetViewController.h
@@ -14,6 +14,7 @@
 @interface TNKAssetViewController : UIViewController
 
 @property (nonatomic, strong) PHAsset *asset;
+@property (nonatomic, strong) NSIndexPath *assetIndexPath;
 
 
 @property (nonatomic, readonly) UIButton *selectButton;

diff --git a/Pod/Classes/TNKAssetsDetailViewController.m b/Pod/Classes/TNKAssetsDetailViewController.m
@@ -53,6 +53,15 @@ - (void)_refetch {
     }
 }
 
+- (void)setAssetViewControllerClass:(Class)assetViewControllerClass {
+    NSAssert([assetViewControllerClass isSubclassOfClass:[TNKAssetViewController class]], @"assetViewControllerClass must be a subclass of TNKAssetViewController");
+
+    _assetViewControllerClass = assetViewControllerClass;
+
+    NSIndexPath *indexPath = [self.viewControllers.firstObject assetIndexPath];
+    [self showAssetAtIndexPath:indexPath];
+}
+
 
 #pragma mark - Initialization
 
@@ -230,6 +239,7 @@ - (TNKAssetViewController *)_assetViewControllerWithAssetAtIndexPath:(NSIndexPat
     objc_setAssociatedObject(next.selectButton, @selector(indexPath), indexPath, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
 
     next.asset = asset;
+    next.assetIndexPath = indexPath;
 
     return next;
 }