Fix an edge case where args/decorators could cause a React hook rules…

… error. (#235)
tajo · Aug 26, 2022 · 7aee108 · 7aee108 · vercel · Aug 26, 2022
1 parent 222ebcb
commit 7aee108
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/.changeset/flat-bears-rule.md b/.changeset/flat-bears-rule.md
@@ -0,0 +1,5 @@
+---
+"@ladle/react": patch
+---
+
+Fix an edge case where args/decorators could cause a React hook rules error.
diff --git a/packages/ladle/lib/app/src/args-provider.tsx b/packages/ladle/lib/app/src/args-provider.tsx
@@ -104,7 +104,6 @@ const ArgsProvider = ({
   ) {
     return null;
   }
-  if (!decorator) return component(props);
   return decorator(() => component(props), {
     globalState,
     dispatch,

diff --git a/packages/ladle/lib/app/src/compose-enhancers.tsx b/packages/ladle/lib/app/src/compose-enhancers.tsx
@@ -9,6 +9,7 @@ export default function composeEnhancers(module: any, storyName: string) {
     args: module[storyName].args,
     argTypes: module[storyName].argTypes,
     component: module[storyName],
+    decorator: (Component) => <Component />,
   };
   if (module[storyName] && Array.isArray(module[storyName].decorators)) {
     funcs = [...funcs, ...module[storyName].decorators];