addDecorator question.

[PaulieScanlon]

Hey, i love this addon it's really great. My config is as follows

addDecorator(withThemesProvider(themes));

Which works great but i'd also like to add something like this

addDecorator(story => <Main>{story}</Main>)

and have main also respond to the theme changes when the UI buttons are clicked... but if i do it this way Main as a decorator has no access to the theme... any ideas?

As a work around i've done this

const stories = storiesOf("Buttons/ButtonAction", module).addDecorator(
  story => <Main>{story()}</Main>
);

stories.add("usage", () => <ButtonAction>Button Action</ButtonAction>, {
  info: "This is a buttonAction"
});

which works fine but Main will show up in my story source which is kinda not what i want.

[echoulen]

source code

export const withThemesProvider = (themes: Theme[]) => (story): JSX.Element => {
  return <ThemesProvider themes={List(themes)}>{story()}</ThemesProvider>;
};

try to custom your decorator

import ThemesProvider from "storybook-addon-styled-component-theme";
import {List} from "immutable"

export const withCustomThemesProvider = (themes: Theme[]) => (story): JSX.Element => {
  return <ThemesProvider themes={List(themes)}><!-- custom here --></ThemesProvider>;
};