Added switch statement

software-mansion · Jul 19, 2024 · b21c8da · b21c8da
1 parent ab1f949
commit b21c8da
Showing 1 changed file with 16 additions and 6 deletions.
diff --git a/ios/RNSScreenStackHeaderConfig.mm b/ios/RNSScreenStackHeaderConfig.mm
@@ -763,12 +763,22 @@ - (void)unmountChildComponentView:(UIView<RCTComponentViewProtocol> *)childCompo
 - (void)setViewToSnapshot:(RNSScreenStackHeaderSubview *)childComponentView
 {
   UINavigationItem *navitem = _screenView.controller.navigationItem;
-  if (childComponentView.type == RNSScreenStackHeaderSubviewTypeLeft) {
-    navitem.leftBarButtonItem.customView = [navitem.leftBarButtonItem.customView snapshotViewAfterScreenUpdates:NO];
-  } else if (childComponentView.type == RNSScreenStackHeaderSubviewTypeCenter) {
-    navitem.titleView = [navitem.titleView snapshotViewAfterScreenUpdates:NO];
-  } else if (childComponentView.type == RNSScreenStackHeaderSubviewTypeRight) {
-    navitem.rightBarButtonItem.customView = [navitem.rightBarButtonItem.customView snapshotViewAfterScreenUpdates:NO];
+  UIView *snapshot = [childComponentView snapshotViewAfterScreenUpdates:NO];
+
+  switch (childComponentView.type) {
+    case RNSScreenStackHeaderSubviewTypeLeft:
+      navitem.leftBarButtonItem.customView = snapshot;
+      break;
+    case RNSScreenStackHeaderSubviewTypeCenter:
+    case RNSScreenStackHeaderSubviewTypeTitle:
+      navitem.titleView = snapshot;
+      break;
+    case RNSScreenStackHeaderSubviewTypeRight:
+      navitem.rightBarButtonItem.customView = snapshot;
+      break;
+    case RNSScreenStackHeaderSubviewTypeSearchBar:
+    case RNSScreenStackHeaderSubviewTypeBackButton:
+      break;
   }
 }