fix: workarounds sb issues with current story on mdx

ccontrols · Nov 13, 2020 · 38684a3 · 38684a3
1 parent 53f2e75
commit 38684a3
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/integrations/storybook/src/panel/AddonPanel.tsx b/integrations/storybook/src/panel/AddonPanel.tsx
@@ -21,13 +21,23 @@ export const AddonPanel: React.FC<AddonPanelProps> = ({
       setStoryId(props.storyId);
     };
     const { id } = api.getCurrentStoryData() || {};
+    console.log(api.storyId);
     setStoryId(id);
     channel.on(SET_CURRENT_STORY, onChangeStory);
     return () => channel.off(SET_CURRENT_STORY, onChangeStory);
   }, [api, channel]);
-  const docId = storyId && store ? store.stories[storyId].doc : undefined;
-  return active && storyId ? (
-    <BlockContextProvider store={store} storyId={storyId} docId={docId}>
+  const docId =
+    storyId && store
+      ? store.stories[storyId]
+        ? store.stories[storyId].doc
+        : storyId
+      : undefined;
+  return active && docId ? (
+    <BlockContextProvider
+      store={store}
+      storyId={storyId !== docId ? storyId : undefined}
+      docId={docId}
+    >
       {children}
     </BlockContextProvider>
   ) : null;