@draftbit/core 48.3.1-264395.2 → 48.3.2-fab3b7.2

package/src/utilities.ts

@@ -1,3 +1,4 @@
+import React from "react";
 import { StyleSheet, StyleProp, TextStyle } from "react-native";
 import { isString, isNumber, pick, pickBy, identity } from "lodash";
 
@@ -228,3 +229,21 @@ export function getValueForRadioButton(value: string | number) {
     throw new Error(`Invalid value: ${value}`);
   }
 }
+
+// This is done to ensure that operations that depend on a particular child type still work even when wrapped in a react fragment (ex: .type checks or prop extraction of child)
+// Wrapping in a fragment can happen in Draftbit when a component is wrapped in a conditional for example or inside a Fetch component
+export function extractIfNestedInFragment(
+  component: React.ReactElement
+): React.ReactElement {
+  if (component.type === React.Fragment) {
+    const children = React.Children.toArray(
+      (component.props as any)?.children
+    ) as React.ReactElement[];
+
+    if (children.length === 1) {
+      return children[0];
+    }
+  }
+
+  return component;
+}