fix(component-stage): fix and improve component-stage

softconstruct · Dec 22, 2023 · b244edb · b244edb
1 parent d3849b2
commit b244edb
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 3 deletions.
diff --git a/.storybook/decorators.js b/.storybook/decorators.js
@@ -8,7 +8,7 @@ import { componentStage } from '../stories/assets/storybook.globals';
 const ComponentStageMessage = ({ stage, currentVersion }) => (
     <>
         This component is <b>{stage}</b>, to learn more about the <b>{stage}</b> stage{' '}
-        <a href={`${currentVersion}/?path=/docs/introduction--page#component-stages`}>read here</a>.
+        <a href={`${currentVersion}/?path=/docs/introduction--docs#component-stages`}>read here</a>.
     </>
 );
 
@@ -18,7 +18,19 @@ const CustomDecorator = ({ children }) => {
     const [allowRenderChildren, setAllowRenderChildren] = useState(false);
     const isDarkMode = useDarkMode();
     let type;
-    const componentStageProp = children?.props?.componentStage || children?.props?.children?.props?.componentStage;
+    const componentStageGetter = (children, num) => {
+        return (
+            Array.isArray(children?.props?.children) &&
+            children?.props?.children.length >= num &&
+            children?.props?.children[num]?.props?.componentStage
+        );
+    };
+
+    const componentStageProp =
+        children?.props?.componentStage ||
+        children?.props?.children?.props?.componentStage ||
+        componentStageGetter(children, 1) ||
+        componentStageGetter(children, 0);
 
     const currentVersion = useMemo(
         () =>

diff --git a/stories/organisms/Table/Table.stories.jsx b/stories/organisms/Table/Table.stories.jsx
@@ -58,7 +58,10 @@ export default {
             <div>
                 <Button>headerActions</Button>
             </div>
-        )
+        ),
+        componentStage: {
+            type: componentStage.deprecated
+        }
     }
 };
 
@@ -611,3 +614,9 @@ export const ControlledComboTable = ({ ...args }) => {
         </>
     );
 };
+
+ControlledComboTable.args = {
+    componentStage: {
+        type: componentStage.deprecated
+    }
+};