From 53a96a9896732ffeb2683953aa6bd23642743ec8 Mon Sep 17 00:00:00 2001
From: Sash Zats <zats@meta.com>
Date: Mon, 23 Oct 2023 10:41:50 -0700
Subject: [PATCH] expose trait collection

Summary: this can be helpful when section controllers calculate size depending on the size class

Differential Revision: D50425042

fbshipit-source-id: 5fbafe1b0662ffb428c42bf99593868915dfc6c7
---
 Source/IGListKit/IGListAdapter.m           | 4 ++++
 Source/IGListKit/IGListCollectionContext.h | 2 ++
 2 files changed, 6 insertions(+)

diff --git a/Source/IGListKit/IGListAdapter.m b/Source/IGListKit/IGListAdapter.m
index dd4a565e0..fedb02dea 100644
--- a/Source/IGListKit/IGListAdapter.m
+++ b/Source/IGListKit/IGListAdapter.m
@@ -987,6 +987,10 @@ - (CGSize)containerSize {
     return self.collectionView.bounds.size;
 }
 
+- (UITraitCollection *_Nullable)traitCollection {
+    return self.collectionView.traitCollection;
+}
+
 - (UIEdgeInsets)containerInset {
     return self.collectionView.contentInset;
 }
diff --git a/Source/IGListKit/IGListCollectionContext.h b/Source/IGListKit/IGListCollectionContext.h
index af3a69bbf..86569768b 100644
--- a/Source/IGListKit/IGListCollectionContext.h
+++ b/Source/IGListKit/IGListCollectionContext.h
@@ -26,6 +26,8 @@ NS_ASSUME_NONNULL_BEGIN
 NS_SWIFT_NAME(ListCollectionContext)
 @protocol IGListCollectionContext <NSObject>
 
+@property (nonatomic, nullable, readonly) UITraitCollection *traitCollection;
+
 /**
  The size of the collection view. You can use this for sizing cells.
  */