Fixed undefined error.

Author: lucasconstantino

dist/index.js

@@ -31,75 +31,83 @@ var addKnobResolver = exports.addKnobResolver = function addKnobResolver(_ref) {
  * --------------------------------
  */
 
-var propTypeKnobResolver = exports.propTypeKnobResolver = function propTypeKnobResolver(test, knob) {
+var propTypeKnobResolver = exports.propTypeKnobResolver = function propTypeKnobResolver(name, knob) {
   for (var _len = arguments.length, args = Array(_len > 2 ? _len - 2 : 0), _key = 2; _key < _len; _key++) {
     args[_key - 2] = arguments[_key];
   }
 
-  return function (name, validate, value, propTypes, defaultProps) {
-    return !validate({ 'prop': test }, 'prop') ? knob.apply(undefined, [name, value].concat(args)) : undefined;
+  return function (propName, propType, value, propTypes, defaultProps) {
+    return propType.type.name === name ? knob.apply(undefined, [propName, value].concat(args)) : undefined;
   };
 };
 
 /* eslint-disable no-multi-spaces */
 // Default simple PropType based knob map.
-var propTypeKnobsMap = [{ name: 'string', test: '', knob: _storybookAddonKnobs.text }, { name: 'number', test: 0, knob: _storybookAddonKnobs.number }, { name: 'bool', test: true, knob: _storybookAddonKnobs.boolean }, { name: 'func', test: function test() {}, knob: function knob(name, value) {
+var propTypeKnobsMap = [{ name: 'string', knob: _storybookAddonKnobs.text }, { name: 'number', knob: _storybookAddonKnobs.number }, { name: 'bool', knob: _storybookAddonKnobs.boolean }, { name: 'func', knob: function knob(name, value) {
     return value || (0, _storybook.action)(name);
-  } }, { name: 'object', test: {}, knob: _storybookAddonKnobs.object }];
+  } }, { name: 'object', knob: _storybookAddonKnobs.object }, { name: 'node', knob: _storybookAddonKnobs.text }, { name: 'element', knob: _storybookAddonKnobs.text }];
 
-propTypeKnobsMap.forEach(function (_ref2) {
+propTypeKnobsMap.forEach(function (_ref2, weight) {
   var name = _ref2.name;
-  var test = _ref2.test;
   var knob = _ref2.knob;
   var _ref2$args = _ref2.args;
   var args = _ref2$args === undefined ? [] : _ref2$args;
   return addKnobResolver({
-    name: name,
-    resolver: propTypeKnobResolver.apply(undefined, [test, knob].concat(_toConsumableArray(args)))
+    weight: weight * 10,
+    name: 'PropTypes.' + name,
+    resolver: propTypeKnobResolver.apply(undefined, [name, knob].concat(_toConsumableArray(args)))
   });
 });
 
-// Defalt oneOf PropType knob resolver.
+// Register 'oneOf' PropType knob resolver.
 addKnobResolver({
-  name: 'oneOf',
-  resolver: function resolver(name, validate, value, propTypes, defaultProps) {
-    var error = validate({ 'prop': '' }, 'prop');
-    var match = error ? /expected one of (\[.*\])/.exec(error.message) : null;
-
-    if (match && match[1]) {
-      var parsedOptions = JSON.parse(match[1]);
-      var options = parsedOptions.reduce(function (res, value) {
-        return _extends({}, res, _defineProperty({}, value, value));
-      }, {});
-
-      return (0, _storybookAddonKnobs.select)(name, _extends({
-        '': '--'
-      }, options), defaultProps[name]);
+  name: 'PropTypes.oneOf',
+  resolver: function resolver(propName, propType, value, propTypes, defaultProps) {
+    /* eslint-disable quotes */
+    if (propType.type.name === 'enum' && propType.type.value.length) {
+      try {
+        var options = propType.type.value.map(function (value) {
+          return value.value;
+        })
+        // Cleanup string quotes, if any.
+        .map(function (value) {
+          return value[0] === "'" && value[value.length - 1] === "'" ? '"' + value.replace(/'"'/g, '\\"').slice(1, value.length - 1) + '"' : value;
+        }).map(JSON.parse).reduce(function (res, value) {
+          return _extends({}, res, _defineProperty({}, value, value));
+        }, {});
+
+        return (0, _storybookAddonKnobs.select)(propName, _extends({ '': '--' }, options), defaultProps[propName]);
+      } catch (e) {}
     }
   }
 });
 
 var withSmartKnobs = exports.withSmartKnobs = function withSmartKnobs(story, context) {
-  var component = story();
-  var propTypes = component.type.propTypes;
-  var defaultProps = _extends({}, component.type.defaultProps, component.props);
+  var component = story(context);
+  var _component$type$__doc = component.type.__docgenInfo;
+  _component$type$__doc = _component$type$__doc === undefined ? { props: {} } : _component$type$__doc;
+  var props = _component$type$__doc.props;
 
-  return (0, _react.cloneElement)(component, resolvePropValues(propTypes, defaultProps));
+  var defaultProps = _extends({}, component.type.defaultProps || {}, component.props);
+
+  return (0, _storybookAddonKnobs.withKnobs)(function () {
+    return (0, _react.cloneElement)(component, resolvePropValues(props, defaultProps));
+  }, context);
 };
 
 var resolvePropValues = function resolvePropValues(propTypes, defaultProps) {
-  var propKeys = Object.keys(propTypes);
+  var propNames = Object.keys(propTypes);
   var resolvers = Object.keys(knobResolvers).sort(function (a, b) {
     return knobResolvers[a].weight < knobResolvers[b].weight;
   }).map(function (name) {
     return knobResolvers[name].resolver;
   });
 
-  return propKeys.map(function (prop) {
+  return propNames.map(function (propName) {
     return resolvers.reduce(function (value, resolver) {
-      return value !== undefined ? value : resolver(prop, propTypes[prop], defaultProps[prop], propTypes, defaultProps);
+      return value !== undefined ? value : resolver(propName, propTypes[propName], defaultProps[propName], propTypes, defaultProps);
     }, undefined);
   }).reduce(function (props, value, i) {
-    return _extends({}, props, _defineProperty({}, propKeys[i], value));
+    return _extends({}, props, _defineProperty({}, propNames[i], value));
   }, defaultProps);
 };