Allow ASCollectionView to register for supplementary container

AsyncDisplayKit/ASCollectionView.h

@@ -119,6 +119,8 @@
  */
 - (void)reloadData;
 
+- (void)registerSupplementaryViewOfKind:(NSString *)elementKind;
+
 /**
  * Inserts one or more sections.
  *

AsyncDisplayKit/ASCollectionView.mm

@@ -379,6 +379,12 @@ static BOOL _isInterceptedSelector(SEL sel)
   [self performBatchAnimated:YES updates:updates completion:completion];
 }
 
+- (void)registerSupplementaryViewOfKind:(NSString *)elementKind
+{
+  NSString *identifier = [NSString stringWithFormat:@"_ASCollectionSupplementaryView_%@", elementKind];
+  [self registerClass:[UIView class] forSupplementaryViewOfKind:elementKind withReuseIdentifier:identifier];
+}
+
 - (void)insertSections:(NSIndexSet *)sections
 {
   ASDisplayNodeAssertMainThread();

examples/ASCollectionView/Sample/ViewController.m

@@ -39,6 +39,9 @@
   _collectionView.asyncDelegate = self;
   _collectionView.backgroundColor = [UIColor whiteColor];
   
+  [_collectionView registerSupplementaryViewOfKind:UICollectionElementKindSectionHeader];
+  [_collectionView registerSupplementaryViewOfKind:UICollectionElementKindSectionFooter];
+  
   return self;
 }