@storybook/api
Version:
Core Storybook API & Context
547 lines (419 loc) • 22.5 kB
JavaScript
function _typeof(obj) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (obj) { return typeof obj; } : function (obj) { return obj && "function" == typeof Symbol && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }, _typeof(obj); }
require("core-js/modules/es.object.get-prototype-of.js");
require("core-js/modules/es.reflect.construct.js");
require("core-js/modules/es.symbol.js");
require("core-js/modules/es.symbol.description.js");
require("core-js/modules/es.symbol.iterator.js");
require("core-js/modules/es.array.iterator.js");
require("core-js/modules/es.string.iterator.js");
require("core-js/modules/web.dom-collections.iterator.js");
require("core-js/modules/es.array.from.js");
require("core-js/modules/es.array.slice.js");
require("core-js/modules/es.function.name.js");
require("core-js/modules/es.regexp.exec.js");
require("core-js/modules/es.weak-map.js");
require("core-js/modules/es.object.get-own-property-descriptor.js");
Object.defineProperty(exports, "__esModule", {
value: true
});
exports.ActiveTabs = void 0;
exports.Consumer = ManagerConsumer;
exports.combineParameters = exports.Provider = void 0;
Object.defineProperty(exports, "isGroup", {
enumerable: true,
get: function get() {
return _stories.isGroup;
}
});
Object.defineProperty(exports, "isRoot", {
enumerable: true,
get: function get() {
return _stories.isRoot;
}
});
Object.defineProperty(exports, "isStory", {
enumerable: true,
get: function get() {
return _stories.isStory;
}
});
Object.defineProperty(exports, "merge", {
enumerable: true,
get: function get() {
return _merge.default;
}
});
exports.useAddonState = useAddonState;
exports.useArgTypes = useArgTypes;
exports.useArgs = useArgs;
exports.useChannel = void 0;
exports.useGlobalTypes = useGlobalTypes;
exports.useGlobals = useGlobals;
exports.useParameter = useParameter;
exports.useSharedState = useSharedState;
exports.useStoryPrepared = useStoryPrepared;
exports.useStorybookApi = useStorybookApi;
exports.useStorybookState = useStorybookState;
require("core-js/modules/es.array.concat.js");
require("core-js/modules/es.object.to-string.js");
require("core-js/modules/web.dom-collections.for-each.js");
require("core-js/modules/es.object.assign.js");
require("core-js/modules/es.array.map.js");
require("core-js/modules/es.array.filter.js");
require("core-js/modules/es.object.entries.js");
var _react = _interopRequireWildcard(require("react"));
var _mergeWith = _interopRequireDefault(require("lodash/mergeWith"));
var _coreEvents = require("@storybook/core-events");
var _context = require("./context");
var _store = _interopRequireDefault(require("./store"));
var _initialState = _interopRequireDefault(require("./initial-state"));
var _stories = require("./lib/stories");
var provider = _interopRequireWildcard(require("./modules/provider"));
var addons = _interopRequireWildcard(require("./modules/addons"));
var channel = _interopRequireWildcard(require("./modules/channel"));
var notifications = _interopRequireWildcard(require("./modules/notifications"));
var settings = _interopRequireWildcard(require("./modules/settings"));
var releaseNotes = _interopRequireWildcard(require("./modules/release-notes"));
var stories = _interopRequireWildcard(require("./modules/stories"));
var refs = _interopRequireWildcard(require("./modules/refs"));
var layout = _interopRequireWildcard(require("./modules/layout"));
var shortcuts = _interopRequireWildcard(require("./modules/shortcuts"));
var url = _interopRequireWildcard(require("./modules/url"));
var version = _interopRequireWildcard(require("./modules/versions"));
var globals = _interopRequireWildcard(require("./modules/globals"));
var _merge = _interopRequireDefault(require("./lib/merge"));
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function _getRequireWildcardCache(nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); }
function _interopRequireWildcard(obj, nodeInterop) { if (!nodeInterop && obj && obj.__esModule) { return obj; } if (obj === null || _typeof(obj) !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(nodeInterop); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (key !== "default" && Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } return newObj; }
function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
function _iterableToArrayLimit(arr, i) { var _i = arr == null ? null : typeof Symbol !== "undefined" && arr[Symbol.iterator] || arr["@@iterator"]; if (_i == null) return; var _arr = []; var _n = true; var _d = false; var _s, _e; try { for (_i = _i.call(arr); !(_n = (_s = _i.next()).done); _n = true) { _arr.push(_s.value); if (i && _arr.length === i) break; } } catch (err) { _d = true; _e = err; } finally { try { if (!_n && _i["return"] != null) _i["return"](); } finally { if (_d) throw _e; } } return _arr; }
function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
function _toConsumableArray(arr) { return _arrayWithoutHoles(arr) || _iterableToArray(arr) || _unsupportedIterableToArray(arr) || _nonIterableSpread(); }
function _nonIterableSpread() { throw new TypeError("Invalid attempt to spread non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
function _iterableToArray(iter) { if (typeof Symbol !== "undefined" && iter[Symbol.iterator] != null || iter["@@iterator"] != null) return Array.from(iter); }
function _arrayWithoutHoles(arr) { if (Array.isArray(arr)) return _arrayLikeToArray(arr); }
function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) { arr2[i] = arr[i]; } return arr2; }
function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
function _defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, descriptor.key, descriptor); } }
function _createClass(Constructor, protoProps, staticProps) { if (protoProps) _defineProperties(Constructor.prototype, protoProps); if (staticProps) _defineProperties(Constructor, staticProps); Object.defineProperty(Constructor, "prototype", { writable: false }); return Constructor; }
function _inherits(subClass, superClass) { if (typeof superClass !== "function" && superClass !== null) { throw new TypeError("Super expression must either be null or a function"); } subClass.prototype = Object.create(superClass && superClass.prototype, { constructor: { value: subClass, writable: true, configurable: true } }); Object.defineProperty(subClass, "prototype", { writable: false }); if (superClass) _setPrototypeOf(subClass, superClass); }
function _setPrototypeOf(o, p) { _setPrototypeOf = Object.setPrototypeOf || function _setPrototypeOf(o, p) { o.__proto__ = p; return o; }; return _setPrototypeOf(o, p); }
function _createSuper(Derived) { var hasNativeReflectConstruct = _isNativeReflectConstruct(); return function _createSuperInternal() { var Super = _getPrototypeOf(Derived), result; if (hasNativeReflectConstruct) { var NewTarget = _getPrototypeOf(this).constructor; result = Reflect.construct(Super, arguments, NewTarget); } else { result = Super.apply(this, arguments); } return _possibleConstructorReturn(this, result); }; }
function _possibleConstructorReturn(self, call) { if (call && (_typeof(call) === "object" || typeof call === "function")) { return call; } else if (call !== void 0) { throw new TypeError("Derived constructors may only return object or undefined"); } return _assertThisInitialized(self); }
function _assertThisInitialized(self) { if (self === void 0) { throw new ReferenceError("this hasn't been initialised - super() hasn't been called"); } return self; }
function _isNativeReflectConstruct() { if (typeof Reflect === "undefined" || !Reflect.construct) return false; if (Reflect.construct.sham) return false; if (typeof Proxy === "function") return true; try { Boolean.prototype.valueOf.call(Reflect.construct(Boolean, [], function () {})); return true; } catch (e) { return false; } }
function _getPrototypeOf(o) { _getPrototypeOf = Object.setPrototypeOf ? Object.getPrototypeOf : function _getPrototypeOf(o) { return o.__proto__ || Object.getPrototypeOf(o); }; return _getPrototypeOf(o); }
var ActiveTabs = layout.ActiveTabs;
exports.ActiveTabs = ActiveTabs;
var ManagerContext = (0, _context.createContext)({
api: undefined,
state: (0, _initialState.default)({})
});
// This is duplicated from @storybook/client-api for the reasons mentioned in lib-addons/types.js
var combineParameters = function combineParameters() {
for (var _len = arguments.length, parameterSets = new Array(_len), _key = 0; _key < _len; _key++) {
parameterSets[_key] = arguments[_key];
}
return _mergeWith.default.apply(void 0, [{}].concat(parameterSets, [function (objValue, srcValue) {
// Treat arrays as scalars:
if (Array.isArray(srcValue)) return srcValue;
return undefined;
}]));
};
exports.combineParameters = combineParameters;
var ManagerProvider = /*#__PURE__*/function (_Component) {
_inherits(ManagerProvider, _Component);
var _super = _createSuper(ManagerProvider);
function ManagerProvider(props) {
var _this;
_classCallCheck(this, ManagerProvider);
_this = _super.call(this, props);
_this.api = {};
_this.modules = void 0;
_this.initModules = function () {
// Now every module has had a chance to set its API, call init on each module which gives it
// a chance to do things that call other modules' APIs.
_this.modules.forEach(function (_ref) {
var init = _ref.init;
if (init) {
init();
}
});
};
var location = props.location,
path = props.path,
refId = props.refId,
_props$viewMode = props.viewMode,
viewMode = _props$viewMode === void 0 ? props.docsMode ? 'docs' : 'story' : _props$viewMode,
singleStory = props.singleStory,
storyId = props.storyId,
docsMode = props.docsMode,
navigate = props.navigate;
var store = new _store.default({
getState: function getState() {
return _this.state;
},
setState: function setState(stateChange, callback) {
return _this.setState(stateChange, callback);
}
});
var routeData = {
location: location,
path: path,
viewMode: viewMode,
singleStory: singleStory,
storyId: storyId,
refId: refId
}; // Initialize the state to be the initial (persisted) state of the store.
// This gives the modules the chance to read the persisted state, apply their defaults
// and override if necessary
var docsModeState = {
layout: {
showToolbar: false,
showPanel: false
},
ui: {
docsMode: true
}
};
_this.state = store.getInitialState((0, _initialState.default)(Object.assign({}, routeData, docsMode ? docsModeState : null)));
var apiData = {
navigate: navigate,
store: store,
provider: props.provider
};
_this.modules = [provider, channel, addons, layout, notifications, settings, releaseNotes, shortcuts, stories, refs, globals, url, version].map(function (m) {
return m.init(Object.assign({}, routeData, apiData, {
state: _this.state,
fullAPI: _this.api
}));
}); // Create our initial state by combining the initial state of all modules, then overlaying any saved state
var state = _initialState.default.apply(void 0, [_this.state].concat(_toConsumableArray(_this.modules.map(function (m) {
return m.state;
})))); // Get our API by combining the APIs exported by each module
var api = Object.assign.apply(Object, [_this.api, {
navigate: navigate
}].concat(_toConsumableArray(_this.modules.map(function (m) {
return m.api;
}))));
_this.state = state;
_this.api = api;
return _this;
}
_createClass(ManagerProvider, [{
key: "shouldComponentUpdate",
value: function shouldComponentUpdate(nextProps, nextState) {
var prevState = this.state;
var prevProps = this.props;
if (prevState !== nextState) {
return true;
}
if (prevProps.path !== nextProps.path) {
return true;
}
return false;
}
}, {
key: "render",
value: function render() {
var children = this.props.children;
var value = {
state: this.state,
api: this.api
};
return /*#__PURE__*/_react.default.createElement(EffectOnMount, {
effect: this.initModules
}, /*#__PURE__*/_react.default.createElement(ManagerContext.Provider, {
value: value
}, /*#__PURE__*/_react.default.createElement(ManagerConsumer, null, children)));
}
}], [{
key: "getDerivedStateFromProps",
value: function getDerivedStateFromProps(props, state) {
if (state.path !== props.path) {
return Object.assign({}, state, {
location: props.location,
path: props.path,
refId: props.refId,
// if its a docsOnly page, even the 'story' view mode is considered 'docs'
viewMode: (props.docsMode && props.viewMode) === 'story' ? 'docs' : props.viewMode,
storyId: props.storyId
});
}
return null;
}
}]);
return ManagerProvider;
}(_react.Component);
exports.Provider = ManagerProvider;
ManagerProvider.displayName = "ManagerProvider";
ManagerProvider.displayName = 'Manager';
// EffectOnMount exists to work around a bug in Reach Router where calling
// navigate inside of componentDidMount (as could happen when we call init on any
// of our modules) does not cause Reach Router's LocationProvider to update with
// the correct path. Calling navigate inside on an effect does not have the
// same problem. See https://github.com/reach/router/issues/404
var EffectOnMount = function EffectOnMount(_ref2) {
var children = _ref2.children,
effect = _ref2.effect;
_react.default.useEffect(effect, []);
return children;
};
var defaultFilter = function defaultFilter(c) {
return c;
};
function ManagerConsumer(_ref3) {
var _ref3$filter = _ref3.filter,
filter = _ref3$filter === void 0 ? defaultFilter : _ref3$filter,
children = _ref3.children;
var c = (0, _react.useContext)(ManagerContext);
var renderer = (0, _react.useRef)(children);
var filterer = (0, _react.useRef)(filter);
if (typeof renderer.current !== 'function') {
return /*#__PURE__*/_react.default.createElement(_react.Fragment, null, renderer.current);
}
var data = filterer.current(c);
var l = (0, _react.useMemo)(function () {
return _toConsumableArray(Object.entries(data).reduce(function (acc, keyval) {
return acc.concat(keyval);
}, []));
}, [c.state]);
return (0, _react.useMemo)(function () {
var Child = renderer.current;
return /*#__PURE__*/_react.default.createElement(Child, data);
}, l);
}
function useStorybookState() {
var _useContext = (0, _react.useContext)(ManagerContext),
state = _useContext.state;
return state;
}
function useStorybookApi() {
var _useContext2 = (0, _react.useContext)(ManagerContext),
api = _useContext2.api;
return api;
}
function orDefault(fromStore, defaultState) {
if (typeof fromStore === 'undefined') {
return defaultState;
}
return fromStore;
}
var useChannel = function useChannel(eventMap) {
var deps = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : [];
var api = useStorybookApi();
(0, _react.useEffect)(function () {
Object.entries(eventMap).forEach(function (_ref4) {
var _ref5 = _slicedToArray(_ref4, 2),
type = _ref5[0],
listener = _ref5[1];
return api.on(type, listener);
});
return function () {
Object.entries(eventMap).forEach(function (_ref6) {
var _ref7 = _slicedToArray(_ref6, 2),
type = _ref7[0],
listener = _ref7[1];
return api.off(type, listener);
});
};
}, deps);
return api.emit;
};
exports.useChannel = useChannel;
function useStoryPrepared(storyId) {
var api = useStorybookApi();
return api.isPrepared(storyId);
}
function useParameter(parameterKey, defaultValue) {
var api = useStorybookApi();
var result = api.getCurrentParameter(parameterKey);
return orDefault(result, defaultValue);
}
// cache for taking care of HMR
var addonStateCache = {}; // shared state
function useSharedState(stateId, defaultState) {
var api = useStorybookApi();
var existingState = api.getAddonState(stateId);
var state = orDefault(existingState, addonStateCache[stateId] ? addonStateCache[stateId] : defaultState);
var setState = function setState(s, options) {
// set only after the stories are loaded
if (addonStateCache[stateId]) {
addonStateCache[stateId] = s;
}
api.setAddonState(stateId, s, options);
};
var allListeners = (0, _react.useMemo)(function () {
var _stateChangeHandlers, _stateInitializationH;
var stateChangeHandlers = (_stateChangeHandlers = {}, _defineProperty(_stateChangeHandlers, "".concat(_coreEvents.SHARED_STATE_CHANGED, "-client-").concat(stateId), function client(s) {
return setState(s);
}), _defineProperty(_stateChangeHandlers, "".concat(_coreEvents.SHARED_STATE_SET, "-client-").concat(stateId), function client(s) {
return setState(s);
}), _stateChangeHandlers);
var stateInitializationHandlers = (_stateInitializationH = {}, _defineProperty(_stateInitializationH, _coreEvents.SET_STORIES, function () {
var currentState = api.getAddonState(stateId);
if (currentState) {
addonStateCache[stateId] = currentState;
api.emit("".concat(_coreEvents.SHARED_STATE_SET, "-manager-").concat(stateId), currentState);
} else if (addonStateCache[stateId]) {
// this happens when HMR
setState(addonStateCache[stateId]);
api.emit("".concat(_coreEvents.SHARED_STATE_SET, "-manager-").concat(stateId), addonStateCache[stateId]);
} else if (defaultState !== undefined) {
// if not HMR, yet the defaults are from the manager
setState(defaultState); // initialize addonStateCache after first load, so its available for subsequent HMR
addonStateCache[stateId] = defaultState;
api.emit("".concat(_coreEvents.SHARED_STATE_SET, "-manager-").concat(stateId), defaultState);
}
}), _defineProperty(_stateInitializationH, _coreEvents.STORY_CHANGED, function () {
var currentState = api.getAddonState(stateId);
if (currentState !== undefined) {
api.emit("".concat(_coreEvents.SHARED_STATE_SET, "-manager-").concat(stateId), currentState);
}
}), _stateInitializationH);
return Object.assign({}, stateChangeHandlers, stateInitializationHandlers);
}, [stateId]);
var emit = useChannel(allListeners);
return [state, function (newStateOrMerger, options) {
setState(newStateOrMerger, options);
emit("".concat(_coreEvents.SHARED_STATE_CHANGED, "-manager-").concat(stateId), newStateOrMerger);
}];
}
function useAddonState(addonId, defaultState) {
return useSharedState(addonId, defaultState);
}
function useArgs() {
var _useStorybookApi = useStorybookApi(),
getCurrentStoryData = _useStorybookApi.getCurrentStoryData,
updateStoryArgs = _useStorybookApi.updateStoryArgs,
resetStoryArgs = _useStorybookApi.resetStoryArgs;
var data = getCurrentStoryData();
var args = (0, _stories.isStory)(data) ? data.args : {};
var updateArgs = (0, _react.useCallback)(function (newArgs) {
return updateStoryArgs(data, newArgs);
}, [data, updateStoryArgs]);
var resetArgs = (0, _react.useCallback)(function (argNames) {
return resetStoryArgs(data, argNames);
}, [data, resetStoryArgs]);
return [args, updateArgs, resetArgs];
}
function useGlobals() {
var api = useStorybookApi();
return [api.getGlobals(), api.updateGlobals];
}
function useGlobalTypes() {
return useStorybookApi().getGlobalTypes();
}
function useCurrentStory() {
var _useStorybookApi2 = useStorybookApi(),
getCurrentStoryData = _useStorybookApi2.getCurrentStoryData;
return getCurrentStoryData();
}
function useArgTypes() {
var _useCurrentStory;
return ((_useCurrentStory = useCurrentStory()) === null || _useCurrentStory === void 0 ? void 0 : _useCurrentStory.argTypes) || {};
}
;