1 files changed, 59 insertions, 0 deletions

diff --git a/.storybook/withTheme/index.tsx b/.storybook/withTheme/index.tsx
new file mode 100644
index 000000000..b357aa658
--- /dev/null
+++ b/.storybook/withTheme/index.tsx
@@ -0,0 +1,59 @@
+import React from 'react';
+import { Classes } from 'jss';
+import injectSheet, { ThemeProvider } from 'react-jss';
+import addons, { makeDecorator } from '@storybook/addons';
+import theme, { Theme, ThemeType } from '../../packages/theme/src';
+
+const defaultTheme = {
+  name: 'Default',
+  variables: theme(ThemeType.default),
+};
+
+const darkTheme = {
+  name: 'Dark Mode',
+  variables: theme(ThemeType.dark),
+};
+
+const themes = [defaultTheme, darkTheme];
+
+const styles = (theme: Theme) => ({
+  title: {
+    fontSize: 14,
+  },
+  container: {
+    border: theme.inputBorder,
+    borderRadius: theme.borderRadiusSmall,
+    marginBottom: 20,
+    padding: 20,
+    background: theme.colorContentBackground,
+  },
+});
+
+const Container = injectSheet(styles)(({ name, classes, story }: { name: string, classes: Classes, story: React.ReactNode }) => (
+  <article>
+    <h1 className={classes.title}>{name}</h1>
+    <div className={classes.container}>
+      {story}
+    </div>
+  </article>
+));
+
+export default makeDecorator({
+  name: 'withTheme',
+  parameterName: 'theme',
+  // This means don't run this decorator if the notes decorator is not set
+  // skipIfNoParametersOrOptions: true,
+  wrapper: (getStory: Function, context: any, { options }: any) => {
+    const channel = addons.getChannel();
+
+    return (
+      <>
+        {themes.map(theme => (
+          <ThemeProvider theme={theme.variables}>
+            <Container story={getStory(context)} name={theme.name} />
+          </ThemeProvider>
+        ))}
+      </>
+    );
+  }
+})