Expose collection view supplementary node data source method

levi · Levi McCallum · commit e492770aed2c · 2015-10-15T11:38:48.000-07:00
diff --git a/AsyncDisplayKit/ASCollectionView.h b/AsyncDisplayKit/ASCollectionView.h
@@ -274,6 +274,8 @@
 
 @optional
 
+- (ASDisplayNode *)collectionView:(ASCollectionView *)collectionView nodeForSupplementaryElementOfKind:(NSString *)kind atIndexPath:(NSIndexPath *)indexPath;
+
 /**
  * Provides the constrained size range for measuring the node at the index path.
  *
diff --git a/examples/ASCollectionView/Sample/ViewController.m b/examples/ASCollectionView/Sample/ViewController.m
@@ -76,6 +76,17 @@ - (ASCellNode *)collectionView:(ASCollectionView *)collectionView nodeForItemAtI
   return node;
 }
 
+- (ASDisplayNode *)collectionView:(ASCollectionView *)collectionView nodeForSupplementaryElementOfKind:(NSString *)kind atIndexPath:(NSIndexPath *)indexPath
+{
+  ASDisplayNode *node = [[ASDisplayNode alloc] init];
+  if ([kind isEqualToString:UICollectionElementKindSectionHeader]) {
+    node.backgroundColor = [UIColor blueColor];
+  } else {
+    node.backgroundColor = [UIColor redColor];
+  }
+  return node;
+}
+
 - (NSInteger)collectionView:(UICollectionView *)collectionView numberOfItemsInSection:(NSInteger)section
 {
   return 300;

Original file line number	Diff line number	Diff line change
`@@ -274,6 +274,8 @@`
`274`	`274`
`275`	`275`	`@optional`
`276`	`276`
	`277`	`+- (ASDisplayNode )collectionView:(ASCollectionView )collectionView nodeForSupplementaryElementOfKind:(NSString )kind atIndexPath:(NSIndexPath )indexPath;`
	`278`	`+`
`277`	`279`	`/**`
`278`	`280`	`* Provides the constrained size range for measuring the node at the index path.`
`279`	`281`	`*`