feat: throw an error if animated-subscribe-button don't receive two <…

…span> elements
magicuidesign · Jan 14, 2025 · 337655f · 337655f
1 parent 22102de
commit 337655f
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/registry/default/magicui/animated-subscribe-button.tsx b/registry/default/magicui/animated-subscribe-button.tsx
@@ -22,6 +22,17 @@ export const AnimatedSubscribeButton = React.forwardRef<
   ) => {
     const [isSubscribed, setIsSubscribed] = useState<boolean>(subscribeStatus);
 
+    if (
+      React.Children.count(children) !== 2 ||
+      !React.Children.toArray(children).every(
+        (child) => React.isValidElement(child) && child.type === "span",
+      )
+    ) {
+      throw new Error(
+        "AnimatedSubscribeButton expects two children, both of which must be <span> elements.",
+      );
+    }
+
     const childrenArray = React.Children.toArray(children);
     const initialChild = childrenArray[0];
     const changeChild = childrenArray[1];