added class check in debug

software-mansion · Oct 7, 2024 · ea77267 · ea77267
1 parent 282b86d
commit ea77267
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/ios/RNSScreenStackHeaderSubview.mm b/ios/RNSScreenStackHeaderSubview.mm
@@ -23,9 +23,15 @@ @implementation RNSScreenStackHeaderSubview
 
 #pragma mark - Common
 
-- (RNSScreenStackHeaderConfig *)getHeaderConfig
+- (nullable RNSScreenStackHeaderConfig *)getHeaderConfig
 {
-  RNSScreenStackHeaderConfig *headerConfig = (RNSScreenStackHeaderConfig *)self.reactSuperview;
+  RNSScreenStackHeaderConfig *headerConfig = (RNSScreenStackHeaderConfig *_Nullable)self.reactSuperview;
+#ifdef DEBUG
+  if (headerConfig != nil && ![headerConfig isKindOfClass:[RNSScreenStackHeaderConfig class]]) {
+    RCTLogError(@"[RNScreens] Invalid view type, expecting RNSScreenStackHeaderConfig, got: %@", headerConfig);
+    return nil;
+  }
+#endif
   return headerConfig;
 }