fix(TabSwitcher): check for child displayName (#518)

tillersystems · May 6, 2019 · ff36178 · ff36178
1 parent 5e8a8ef
commit ff36178
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/src/TabSwitcher/index.js b/src/TabSwitcher/index.js
@@ -27,19 +27,24 @@ const TabSwitcher = ({ children, defaultIndex, index, isCompacted, onChange }) =
   const [activeIndex, setActiveIndex] = useState(defaultIndex);
 
   const content = Children.map(children, child => {
-    // ignore random <div/>s etc.
-    if (child) {
-      if (typeof child.type === 'string') return child;
+    if (child.type.displayName === 'Panes') {
       return cloneElement(child, {
         activeIndex: isControlled ? index : activeIndex,
-        isCompacted,
+        isCompacted: isCompacted,
+      });
+    } else if (child.type.displayName === 'Tabs') {
+      return cloneElement(child, {
+        activeIndex: isControlled ? index : activeIndex,
+        isCompacted: isCompacted,
         onActiveTab: index => {
           onChange && onChange(index);
           if (!isControlled) {
             setActiveIndex(index);
           }
         },
       });
+    } else {
+      return child;
     }
   });