adds handling of stories wrapped with info addon

Author: flipace

example/stories/index.js

@@ -2,18 +2,25 @@ import React from 'react'
 import { storiesOf } from '@storybook/react'
 import { withSmartKnobs } from '../../src'
 import { withKnobs, select } from '@storybook/addon-knobs'
+import { withInfo } from '@storybook/addon-info'
 
 import SmartKnobedComponent from './SmartKnobedComponent'
 import SmartKnobedComponentMissingProps from './SmartKnobedComponentMissingProps'
 
 const stub = fn => fn()
 
+  
+storiesOf("Example of smart Knobs", module)
+  .addDecorator(withSmartKnobs)
+  .addDecorator(withKnobs)
+  .add("full example", () => <SmartKnobedComponent />);
+
 storiesOf('Example of smart Knobs', module)
   .addDecorator(withSmartKnobs)
   .addDecorator(withKnobs)
-  .add('full example', () => (
+  .add('full example with info', withInfo('test')(() => (
     <SmartKnobedComponent />
-  ))
+  )))
 
 storiesOf('Smart Knobs missing props', module)
   .addDecorator(withSmartKnobs)

package.json

@@ -19,6 +19,7 @@
     "react": "^0.14.7 || ^15.5.4"
   },
   "devDependencies": {
+    "@storybook/addon-info": "^3.2.10",
     "@storybook/addon-knobs": "^3.0.0",
     "@storybook/addon-options": "^3.0.0",
     "@storybook/react": "^3.0.0",

src/index.js

@@ -56,10 +56,15 @@ addKnobResolver({
 
 export const withSmartKnobs = (story, context) => {
   const component = story(context)
-  const { __docgenInfo: { props } = { props: { } } } = component.type
+
+  let target = component.props.marksyConf && component.props.propTables && component.props.propTablesExclude
+    ? component.props.children
+    : component
+
+  const { __docgenInfo: { props } = { props: {} } } = target.type
   const defaultProps = {
-    ...(component.type.defaultProps || {}),
-    ...component.props
+    ...(target.type.defaultProps || {}),
+    ...target.props
   }
 
   const finalProps = props ? Object.keys(props).reduce((acc, n) => {
@@ -76,7 +81,13 @@ export const withSmartKnobs = (story, context) => {
     }
   }, {}) : {}
 
-  return cloneElement(component, resolvePropValues(finalProps, defaultProps))
+  const newProps = resolvePropValues(finalProps, defaultProps)
+
+  if (component.props.marksyConf) {
+    return cloneElement(component, { ...component.props, children: cloneElement(component.props.children, newProps) })
+  }
+
+  return cloneElement(component, newProps)
 }
 
 const resolvePropValues = (propTypes, defaultProps) => {