Use cell mode while wrapping supplementary nodes (#1282)

TextureGroup · Jan 10, 2019 · 52b84ba · 52b84ba
1 parent 22acb98
commit 52b84ba
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/Source/ASCollectionView.mm b/Source/ASCollectionView.mm
@@ -2044,13 +2044,17 @@ - (ASCellNodeBlock)dataController:(ASDataController *)dataController supplementa
   }
 
   // Wrap the node block
+  BOOL disableRangeController = ASCellLayoutModeIncludes(ASCellLayoutModeDisableRangeController);
   __weak __typeof__(self) weakSelf = self;
   return ^{
     __typeof__(self) strongSelf = weakSelf;
     ASCellNode *node = block();
     ASDisplayNodeAssert([node isKindOfClass:[ASCellNode class]],
                         @"ASCollectionNode provided a non-ASCellNode! %@, %@", node, strongSelf);
 
+    if (!disableRangeController) {
+      [node enterHierarchyState:ASHierarchyStateRangeManaged];
+    }
     if (node.interactionDelegate == nil) {
       node.interactionDelegate = strongSelf;
     }