wrap a single child into an array for getSteps

jcmcneal · Sep 5, 2020 · b6f5c87 · b6f5c87
1 parent 17e80df
commit b6f5c87
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/index.d.ts b/src/index.d.ts
@@ -23,7 +23,7 @@ export type StepWizardProps = Partial<{
     exitLeft?: string
   }
 
-  children: JSX.Element[]
+  children: JSX.Element | JSX.Element[]
 }>
 
 export type StepWizardChildProps<T extends Record<string, any> = {}> = {

diff --git a/src/index.js b/src/index.js
@@ -136,7 +136,9 @@ export default class StepWizard extends PureComponent {
         return this.getSteps().length;
     }
 
-    getSteps = () => this.props.children.filter(el => el);
+    getSteps = () => (
+        Array.isArray(this.props.children) ? this.props.children : [this.props.children]
+    ).filter(el => el);
 
     /** Go to first step */
     firstStep = () => this.goToStep(1)