fix: update docId for new state

ccontrols · Jul 30, 2020 · 7930fb7 · 7930fb7
1 parent 46354d3
commit 7930fb7
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
diff --git a/integrations/storybook/src/context/BlockContext.tsx b/integrations/storybook/src/context/BlockContext.tsx
@@ -12,8 +12,9 @@ export const BlockContextProvider: React.FC<BlockContextProviderProps> = ({
 }) => {
   const defaultStoyId = useStoryId();
   const storyId = id ? id : defaultStoyId;
+  const docId = storyId && store ? store.stories[storyId].doc : undefined;
   return (
-    <BlocksContextProvider store={store} storyId={storyId}>
+    <BlocksContextProvider store={store} storyId={storyId} docId={docId}>
       {children}
     </BlocksContextProvider>
   );

diff --git a/integrations/storybook/src/docs-page/DocsContainer.tsx b/integrations/storybook/src/docs-page/DocsContainer.tsx
@@ -13,9 +13,16 @@ import { store } from '@component-controls/store/live_store';
 export const PageContextContainer: FC = ({ children }) => {
   const options = React.useMemo(() => getGlobalOptions(), []);
   const storyId = useStoryId();
+  const docId = storyId && store ? store.stories[storyId].doc : undefined;
+  console.group('HERE', docId);
   return (
     <ThemeProvider theme={store.config.theme}>
-      <BlockContextProvider storyId={storyId} store={store} options={options}>
+      <BlockContextProvider
+        storyId={storyId}
+        store={store}
+        docId={docId}
+        options={options}
+      >
         <BlockPageContainer variant="pagecontainer.storybook">
           {children}
         </BlockPageContainer>

diff --git a/integrations/storybook/src/panel/AddonPanel.tsx b/integrations/storybook/src/panel/AddonPanel.tsx
@@ -26,10 +26,10 @@ export const AddonPanel: React.FC<AddonPanelProps> = ({
     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 ? (
     <ThemeProvider>
-      <BlockContextProvider store={store} storyId={storyId}>
+      <BlockContextProvider store={store} storyId={storyId} docId={docId}>
         {children}
       </BlockContextProvider>
     </ThemeProvider>