use ts for stories

Author: jscottsmith

stories/Container.js renamed to stories/Container.tsx

@@ -1,6 +1,6 @@
 import React from 'react';
-import styles from './Image.module.scss';
 import { useController } from '../../src';
+import styles from './Image.module.scss';
 
 export const Image = (props) => {
   const parallaxController = useController();

stories/Element/Element.js renamed to stories/Element/Element.tsx

@@ -0,0 +1,26 @@
+import React, { PropsWithChildren } from 'react';
+import { ParallaxProvider } from '../src';
+
+export const ScrollContainer = (
+  props: PropsWithChildren<{
+    scrollAxis: 'vertical' | 'horizontal';
+  }>
+) => {
+  const [scrollEl, setScrollElement] = React.useState(null);
+  const ref = React.useRef();
+
+  React.useEffect(() => {
+    setScrollElement(ref.current);
+  });
+
+  return (
+    <div className="scroll-container" ref={ref}>
+      <ParallaxProvider
+        scrollContainer={scrollEl}
+        scrollAxis={props.scrollAxis}
+      >
+        {props.children}
+      </ParallaxProvider>
+    </div>
+  );
+};