fix: storybook addon

itkrt2y · Dec 5, 2023 · 5ec0be6 · 5ec0be6
1 parent 6e2b976
commit 5ec0be6
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/.changeset/dirty-carrots-check.md b/.changeset/dirty-carrots-check.md
@@ -0,0 +1,5 @@
+---
+"@chakra-ui/storybook-addon": patch
+---
+
+Fix issue in storybook addon where theme could be undefined.
diff --git a/tooling/storybook-addon/src/ChakraProviderDecorator.tsx b/tooling/storybook-addon/src/ChakraProviderDecorator.tsx
@@ -11,16 +11,17 @@ export const ChakraProviderDecorator = makeDecorator({
   parameterName: "chakra",
   skipIfNoParametersOrOptions: false,
   wrapper: (getStory, context, { parameters }) => {
-    const chakraTheme = parameters?.theme
+    const chakraTheme: Record<string, any> | undefined = parameters?.theme
 
     const direction = useDirection(
-      context.globals[DIRECTION_TOOL_ID] || chakraTheme.direction,
+      context.globals[DIRECTION_TOOL_ID] || chakraTheme?.direction,
     )
 
-    const themeWithDirectionOverride = React.useMemo(
-      () => extendTheme({ direction }, chakraTheme),
-      [chakraTheme, direction],
-    )
+    const themeWithDirectionOverride = React.useMemo(() => {
+      return chakraTheme
+        ? extendTheme({ direction }, chakraTheme)
+        : extendTheme({ direction })
+    }, [chakraTheme, direction])
 
     return (
       <ChakraBaseProvider {...parameters} theme={themeWithDirectionOverride}>