79 lines
2.7 KiB
JavaScript
79 lines
2.7 KiB
JavaScript
'use strict';
|
|
|
|
Object.defineProperty(exports, "__esModule", {
|
|
value: true
|
|
});
|
|
|
|
var _extends3 = require('babel-runtime/helpers/extends');
|
|
|
|
var _extends4 = _interopRequireDefault(_extends3);
|
|
|
|
var _classCallCheck2 = require('babel-runtime/helpers/classCallCheck');
|
|
|
|
var _classCallCheck3 = _interopRequireDefault(_classCallCheck2);
|
|
|
|
var _possibleConstructorReturn2 = require('babel-runtime/helpers/possibleConstructorReturn');
|
|
|
|
var _possibleConstructorReturn3 = _interopRequireDefault(_possibleConstructorReturn2);
|
|
|
|
var _inherits2 = require('babel-runtime/helpers/inherits');
|
|
|
|
var _inherits3 = _interopRequireDefault(_inherits2);
|
|
|
|
var _react = require('react');
|
|
|
|
var _setDisplayName = require('./setDisplayName');
|
|
|
|
var _setDisplayName2 = _interopRequireDefault(_setDisplayName);
|
|
|
|
var _wrapDisplayName = require('./wrapDisplayName');
|
|
|
|
var _wrapDisplayName2 = _interopRequireDefault(_wrapDisplayName);
|
|
|
|
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
|
|
|
|
var withState = function withState(stateName, stateUpdaterName, initialState) {
|
|
return function (BaseComponent) {
|
|
var factory = (0, _react.createFactory)(BaseComponent);
|
|
|
|
var WithState = function (_Component) {
|
|
(0, _inherits3.default)(WithState, _Component);
|
|
|
|
function WithState() {
|
|
var _temp, _this, _ret;
|
|
|
|
(0, _classCallCheck3.default)(this, WithState);
|
|
|
|
for (var _len = arguments.length, args = Array(_len), _key = 0; _key < _len; _key++) {
|
|
args[_key] = arguments[_key];
|
|
}
|
|
|
|
return _ret = (_temp = (_this = (0, _possibleConstructorReturn3.default)(this, _Component.call.apply(_Component, [this].concat(args))), _this), _this.state = {
|
|
stateValue: typeof initialState === 'function' ? initialState(_this.props) : initialState
|
|
}, _this.updateStateValue = function (updateFn, callback) {
|
|
return _this.setState(function (_ref) {
|
|
var stateValue = _ref.stateValue;
|
|
return {
|
|
stateValue: typeof updateFn === 'function' ? updateFn(stateValue) : updateFn
|
|
};
|
|
}, callback);
|
|
}, _temp), (0, _possibleConstructorReturn3.default)(_this, _ret);
|
|
}
|
|
|
|
WithState.prototype.render = function render() {
|
|
var _extends2;
|
|
|
|
return factory((0, _extends4.default)({}, this.props, (_extends2 = {}, _extends2[stateName] = this.state.stateValue, _extends2[stateUpdaterName] = this.updateStateValue, _extends2)));
|
|
};
|
|
|
|
return WithState;
|
|
}(_react.Component);
|
|
|
|
if (process.env.NODE_ENV !== 'production') {
|
|
return (0, _setDisplayName2.default)((0, _wrapDisplayName2.default)(BaseComponent, 'withState'))(WithState);
|
|
}
|
|
return WithState;
|
|
};
|
|
};
|
|
|
|
exports.default = withState; |