@storybook/addon-docs 6.4.0-beta.6 → 6.4.0-rc.0

package/dist/cjs/blocks/ArgsTable.js

@@ -233,10 +233,9 @@ var StoryTable = function StoryTable(props) {
         updateArgs = _useArgs2[1],
         resetArgs = _useArgs2[2];
 
-    if (!story) {
-      return /*#__PURE__*/_react.default.createElement("div", null, "Loading...");
-    }
-
+    if (!story) return /*#__PURE__*/_react.default.createElement(_components.ArgsTable, {
+      isLoading: true
+    });
     var argTypes = (0, _store.filterArgTypes)(story.argTypes, include, exclude);
     var mainLabel = (0, _utils.getComponentName)(component) || 'Story';
 

package/dist/cjs/blocks/Canvas.js

@@ -128,7 +128,8 @@ var Canvas = function Canvas(props) {
       previewProps = _getPreviewProps.previewProps;
 
   var children = props.children;
-  return isLoading ? null : /*#__PURE__*/_react.default.createElement(_react2.MDXProvider, {
+  if (isLoading) return /*#__PURE__*/_react.default.createElement(_components.PreviewSkeleton, null);
+  return /*#__PURE__*/_react.default.createElement(_react2.MDXProvider, {
     components: _components.resetComponents
   }, /*#__PURE__*/_react.default.createElement(_components.Preview, previewProps, children));
 };

package/dist/cjs/blocks/DocsStory.js

@@ -70,7 +70,8 @@ var DocsStory = function DocsStory(_ref) {
   }), /*#__PURE__*/_react.default.createElement(_Canvas.Canvas, {
     withToolbar: withToolbar
   }, /*#__PURE__*/_react.default.createElement(_Story.Story, {
-    id: id
+    id: id,
+    parameters: parameters
   })));
 };
 

package/dist/cjs/blocks/Source.js

@@ -92,7 +92,7 @@ var getSnippet = function getSnippet(snippet, story) {
 
   var userCode = (_parameters$docs2 = parameters.docs) === null || _parameters$docs2 === void 0 ? void 0 : (_parameters$docs2$sou = _parameters$docs2.source) === null || _parameters$docs2$sou === void 0 ? void 0 : _parameters$docs2$sou.code;
 
-  if (userCode) {
+  if (userCode !== undefined) {
     return userCode;
   } // if user has explicitly set this as dynamic, use snippet
 

package/dist/cjs/blocks/Story.js

@@ -2,23 +2,27 @@
 
 function _typeof(obj) { "@babel/helpers - typeof"; if (typeof Symbol === "function" && typeof Symbol.iterator === "symbol") { _typeof = function _typeof(obj) { return typeof obj; }; } else { _typeof = function _typeof(obj) { return obj && typeof Symbol === "function" && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }; } return _typeof(obj); }
 
-require("core-js/modules/es.weak-map.js");
+require("core-js/modules/es.symbol.js");
+
+require("core-js/modules/es.symbol.description.js");
 
 require("core-js/modules/es.object.to-string.js");
 
+require("core-js/modules/es.symbol.iterator.js");
+
 require("core-js/modules/es.string.iterator.js");
 
 require("core-js/modules/es.array.iterator.js");
 
 require("core-js/modules/web.dom-collections.iterator.js");
 
-require("core-js/modules/es.object.get-own-property-descriptor.js");
+require("core-js/modules/es.array.slice.js");
 
-require("core-js/modules/es.symbol.js");
+require("core-js/modules/es.array.from.js");
 
-require("core-js/modules/es.symbol.description.js");
+require("core-js/modules/es.weak-map.js");
 
-require("core-js/modules/es.symbol.iterator.js");
+require("core-js/modules/es.object.get-own-property-descriptor.js");
 
 Object.defineProperty(exports, "__esModule", {
   value: true
@@ -35,11 +39,15 @@ var _react = _interopRequireWildcard(require("react"));
 
 var _react2 = require("@mdx-js/react");
 
+var _global = _interopRequireDefault(require("global"));
+
 var _components = require("@storybook/components");
 
 var _csf = require("@storybook/csf");
 
-var _global = _interopRequireDefault(require("global"));
+var _addons = require("@storybook/addons");
+
+var _coreEvents = _interopRequireDefault(require("@storybook/core-events"));
 
 var _types = require("./types");
 
@@ -53,6 +61,18 @@ function _getRequireWildcardCache() { if (typeof WeakMap !== "function") return
 
 function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } if (obj === null || _typeof(obj) !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (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 _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 _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 _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 _iterableToArrayLimit(arr, i) { if (typeof Symbol === "undefined" || !(Symbol.iterator in Object(arr))) return; var _arr = []; var _n = true; var _d = false; var _e = undefined; try { for (var _i = arr[Symbol.iterator](), _s; !(_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; }
+
 var storyBlockIdFromId = function storyBlockIdFromId(storyId) {
   return "story--".concat(storyId);
 };
@@ -78,7 +98,7 @@ var getStoryId = function getStoryId(props, context) {
 
 exports.getStoryId = getStoryId;
 
-var getStoryProps = function getStoryProps(_ref4, story, context) {
+var getStoryProps = function getStoryProps(_ref4, story, context, onStoryFnCalled) {
   var height = _ref4.height,
       inline = _ref4.inline;
   var storyName = story.name,
@@ -103,9 +123,17 @@ var getStoryProps = function getStoryProps(_ref4, story, context) {
   }
 
   var boundStoryFn = function boundStoryFn() {
-    return story.unboundStoryFn(Object.assign({}, context.getStoryContext(story), {
-      loaded: {}
-    }));
+    var storyResult = story.unboundStoryFn(Object.assign({}, context.getStoryContext(story), {
+      loaded: {},
+      abortSignal: undefined,
+      canvasElement: undefined
+    })); // We need to wait until the bound story function has actually been called before we
+    // consider the story rendered. Certain frameworks (i.e. angular) don't actually render
+    // the component in the very first react render cycle, and so we can't just wait until the
+    // `PureStory` component has been rendered to consider the underlying story "rendered".
+
+    onStoryFnCalled();
+    return storyResult;
   };
 
   return Object.assign({
@@ -116,7 +144,7 @@ var getStoryProps = function getStoryProps(_ref4, story, context) {
   }, storyIsInline && {
     parameters: parameters,
     storyFn: function storyFn() {
-      return prepareForInline(boundStoryFn, story);
+      return prepareForInline(boundStoryFn, context.getStoryContext(story));
     }
   });
 };
@@ -127,21 +155,22 @@ var Story = function Story(props) {
   var _global$FEATURES;
 
   var context = (0, _react.useContext)(_DocsContext.DocsContext);
-  var ref = (0, _react.useRef)();
-  var story = (0, _useStory.useStory)(getStoryId(props, context), context); // Ensure we wait until this story is properly rendered in the docs context.
-  // The purpose of this is to ensure that that the `DOCS_RENDERED` event isn't emitted
-  // until all stories on the page have rendered.
-
-  var storyId = context.id,
-      registerRenderingStory = context.registerRenderingStory;
-  var storyRendered = (0, _react.useMemo)(registerRenderingStory, [storyId]);
-  (0, _react.useEffect)(function () {
-    if (story) storyRendered();
-  }, [story]);
+
+  var channel = _addons.addons.getChannel();
+
+  var storyRef = (0, _react.useRef)();
+  var storyId = getStoryId(props, context);
+  var story = (0, _useStory.useStory)(storyId, context);
+
+  var _useState = (0, _react.useState)(true),
+      _useState2 = _slicedToArray(_useState, 2),
+      showLoader = _useState2[0],
+      setShowLoader = _useState2[1];
+
   (0, _react.useEffect)(function () {
     var cleanup;
 
-    if (story && ref.current) {
+    if (story && storyRef.current) {
       var componentId = story.componentId,
           id = story.id,
           title = story.title,
@@ -161,8 +190,9 @@ var Story = function Story(props) {
       cleanup = context.renderStoryToElement({
         story: story,
         renderContext: renderContext,
-        element: ref.current
+        element: storyRef.current
       });
+      setShowLoader(false);
     }
 
     return function () {
@@ -171,10 +201,15 @@ var Story = function Story(props) {
   }, [story]);
 
   if (!story) {
-    return /*#__PURE__*/_react.default.createElement("div", null, "Loading...");
-  }
+    return /*#__PURE__*/_react.default.createElement(_components.StorySkeleton, null);
+  } // If we are rendering a old-style inline Story via `PureStory` below, we want to emit
+  // the `STORY_RENDERED` event when it renders. The modern mode below calls out to
+  // `Preview.renderStoryToDom()` which itself emits the event.
+
 
-  var storyProps = getStoryProps(props, story, context);
+  var storyProps = getStoryProps(props, story, context, function () {
+    return channel.emit(_coreEvents.default.STORY_RENDERED, storyId);
+  });
 
   if (!storyProps) {
     return null;
@@ -182,15 +217,15 @@ var Story = function Story(props) {
 
   if (_global.default !== null && _global.default !== void 0 && (_global$FEATURES = _global.default.FEATURES) !== null && _global$FEATURES !== void 0 && _global$FEATURES.modernInlineRender) {
     // We do this so React doesn't complain when we replace the span in a secondary render
-    var htmlContents = "<span data-is-loading-indicator=\"true\">loading story...</span>"; // FIXME: height/style/etc. lifted from PureStory
+    var htmlContents = "<span></span>"; // FIXME: height/style/etc. lifted from PureStory
 
     var height = storyProps.height;
     return /*#__PURE__*/_react.default.createElement("div", {
       id: storyBlockIdFromId(story.id)
     }, /*#__PURE__*/_react.default.createElement(_react2.MDXProvider, {
       components: _components.resetComponents
-    }, height ? /*#__PURE__*/_react.default.createElement("style", null, "#story--".concat(story.id, " { min-height: ").concat(height, "; transform: translateZ(0); overflow: auto }")) : null, /*#__PURE__*/_react.default.createElement("div", {
-      ref: ref,
+    }, height ? /*#__PURE__*/_react.default.createElement("style", null, "#story--".concat(story.id, " { min-height: ").concat(height, "; transform: translateZ(0); overflow: auto }")) : null, showLoader && /*#__PURE__*/_react.default.createElement(_components.StorySkeleton, null), /*#__PURE__*/_react.default.createElement("div", {
+      ref: storyRef,
       "data-name": story.name,
       dangerouslySetInnerHTML: {
         __html: htmlContents

package/dist/cjs/blocks/enhanceSource.js

@@ -35,7 +35,7 @@ require("core-js/modules/es.regexp.exec.js");
 
 require("core-js/modules/es.string.split.js");
 
-var _clientApi = require("@storybook/client-api");
+var _store = require("@storybook/store");
 
 function _toConsumableArray(arr) { return _arrayWithoutHoles(arr) || _iterableToArray(arr) || _unsupportedIterableToArray(arr) || _nonIterableSpread(); }
 
@@ -118,7 +118,7 @@ var enhanceSource = function enhanceSource(story) {
   var input = extract(id, storySource);
   var code = transformSource ? transformSource(input, story) : input;
   return {
-    docs: (0, _clientApi.combineParameters)(docs, {
+    docs: (0, _store.combineParameters)(docs, {
       source: {
         code: code
       }

package/dist/cjs/blocks/useStory.js

@@ -60,7 +60,12 @@ function useStory(storyId, context) {
 }
 
 function useStories(storyIds, context) {
-  var _useState = (0, _react.useState)({}),
+  var initialStoriesById = context.componentStories().reduce(function (acc, story) {
+    acc[story.id] = story;
+    return acc;
+  }, {});
+
+  var _useState = (0, _react.useState)(initialStoriesById),
       _useState2 = _slicedToArray(_useState, 2),
       storiesById = _useState2[0],
       setStories = _useState2[1];

package/dist/cjs/frameworks/angular/compodoc.js

@@ -316,8 +316,10 @@ var extractDefaultValue = function extractDefaultValue(property) {
 };
 
 var resolveTypealias = function resolveTypealias(compodocType) {
+  var _compodocJson$miscell, _compodocJson$miscell2;
+
   var compodocJson = getCompodocJson();
-  var typeAlias = compodocJson === null || compodocJson === void 0 ? void 0 : compodocJson.miscellaneous.typealiases.find(function (x) {
+  var typeAlias = compodocJson === null || compodocJson === void 0 ? void 0 : (_compodocJson$miscell = compodocJson.miscellaneous) === null || _compodocJson$miscell === void 0 ? void 0 : (_compodocJson$miscell2 = _compodocJson$miscell.typealiases) === null || _compodocJson$miscell2 === void 0 ? void 0 : _compodocJson$miscell2.find(function (x) {
     return x.name === compodocType;
   });
   return typeAlias ? resolveTypealias(typeAlias.rawtype) : compodocType;

package/dist/cjs/frameworks/angular/sourceDecorator.js

@@ -1,21 +1,47 @@
 "use strict";
 
+function _typeof(obj) { "@babel/helpers - typeof"; if (typeof Symbol === "function" && typeof Symbol.iterator === "symbol") { _typeof = function _typeof(obj) { return typeof obj; }; } else { _typeof = function _typeof(obj) { return obj && typeof Symbol === "function" && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }; } return _typeof(obj); }
+
+require("core-js/modules/es.weak-map.js");
+
+require("core-js/modules/es.string.iterator.js");
+
+require("core-js/modules/es.array.iterator.js");
+
+require("core-js/modules/web.dom-collections.iterator.js");
+
+require("core-js/modules/es.object.get-own-property-descriptor.js");
+
+require("core-js/modules/es.symbol.js");
+
+require("core-js/modules/es.symbol.description.js");
+
+require("core-js/modules/es.symbol.iterator.js");
+
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
 exports.sourceDecorator = exports.skipSourceRender = void 0;
 
+require("regenerator-runtime/runtime.js");
+
+require("core-js/modules/es.promise.js");
+
+require("core-js/modules/es.object.to-string.js");
+
 var _addons = require("@storybook/addons");
 
 var _renderer = require("@storybook/angular/renderer");
 
-var _parserHtml = _interopRequireDefault(require("prettier/parser-html"));
+var _shared = require("../../shared");
 
-var _standalone = _interopRequireDefault(require("prettier/standalone"));
+function _getRequireWildcardCache() { if (typeof WeakMap !== "function") return null; var cache = new WeakMap(); _getRequireWildcardCache = function _getRequireWildcardCache() { return cache; }; return cache; }
 
-var _shared = require("../../shared");
+function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } if (obj === null || _typeof(obj) !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (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 _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }
+
+function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
 
 var skipSourceRender = function skipSourceRender(context) {
   var _context$parameters$d;
@@ -32,16 +58,62 @@ var skipSourceRender = function skipSourceRender(context) {
 };
 
 exports.skipSourceRender = skipSourceRender;
-
-var prettyUp = function prettyUp(source) {
-  return _standalone.default.format(source, {
-    parser: 'angular',
-    plugins: [_parserHtml.default],
-    htmlWhitespaceSensitivity: 'ignore'
-  });
-};
+var prettyUpInternal;
+
+var makePrettyUp = /*#__PURE__*/function () {
+  var _ref = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee() {
+    var prettierHtml, prettier;
+    return regeneratorRuntime.wrap(function _callee$(_context) {
+      while (1) {
+        switch (_context.prev = _context.next) {
+          case 0:
+            if (!prettyUpInternal) {
+              _context.next = 2;
+              break;
+            }
+
+            return _context.abrupt("return", prettyUpInternal);
+
+          case 2:
+            _context.next = 4;
+            return Promise.resolve().then(function () {
+              return _interopRequireWildcard(require('prettier/parser-html'));
+            });
+
+          case 4:
+            prettierHtml = _context.sent;
+            _context.next = 7;
+            return Promise.resolve().then(function () {
+              return _interopRequireWildcard(require('prettier/standalone'));
+            });
+
+          case 7:
+            prettier = _context.sent;
+
+            prettyUpInternal = function prettyUpInternal(source) {
+              return prettier.format(source, {
+                parser: 'angular',
+                plugins: [prettierHtml],
+                htmlWhitespaceSensitivity: 'ignore'
+              });
+            };
+
+            return _context.abrupt("return", prettyUpInternal);
+
+          case 10:
+          case "end":
+            return _context.stop();
+        }
+      }
+    }, _callee);
+  }));
+
+  return function makePrettyUp() {
+    return _ref.apply(this, arguments);
+  };
+}();
 /**
- * Svelte source decorator.
+ * Angular source decorator.
  * @param storyFn Fn
  * @param context  StoryContext
  */
@@ -62,21 +134,25 @@ var sourceDecorator = function sourceDecorator(storyFn, context) {
   var component = context.component,
       argTypes = context.argTypes;
   var toEmit;
+  var prettyUpPromise = makePrettyUp();
   (0, _addons.useEffect)(function () {
-    if (toEmit) channel.emit(_shared.SNIPPET_RENDERED, context.id, prettyUp(template));
+    prettyUpPromise.then(function (prettyUp) {
+      if (toEmit) channel.emit(_shared.SNIPPET_RENDERED, context.id, prettyUp(toEmit));
+    });
   });
+  prettyUpPromise.then(function (prettyUp) {
+    if (component && !userDefinedTemplate) {
+      var _source = (0, _renderer.computesTemplateSourceFromComponent)(component, props, argTypes); // We might have a story with a Directive or Service defined as the component
+      // In these cases there might exist a template, even if we aren't able to create source from component
 
-  if (component && !userDefinedTemplate) {
-    var source = (0, _renderer.computesTemplateSourceFromComponent)(component, props, argTypes); // We might have a story with a Directive or Service defined as the component
-    // In these cases there might exist a template, even if we aren't able to create source from component
 
-    if (source || template) {
-      toEmit = prettyUp(source || template);
+      if (_source || template) {
+        toEmit = prettyUp(_source || template);
+      }
+    } else if (template) {
+      toEmit = prettyUp(template);
     }
-  } else if (template) {
-    toEmit = prettyUp(template);
-  }
-
+  });
   return story;
 };
 

package/dist/cjs/frameworks/common/config.js

@@ -1,19 +1,103 @@
 "use strict";
 
+function _typeof(obj) { "@babel/helpers - typeof"; if (typeof Symbol === "function" && typeof Symbol.iterator === "symbol") { _typeof = function _typeof(obj) { return typeof obj; }; } else { _typeof = function _typeof(obj) { return obj && typeof Symbol === "function" && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }; } return _typeof(obj); }
+
+require("core-js/modules/es.weak-map.js");
+
+require("core-js/modules/es.string.iterator.js");
+
+require("core-js/modules/es.array.iterator.js");
+
+require("core-js/modules/web.dom-collections.iterator.js");
+
+require("core-js/modules/es.object.get-own-property-descriptor.js");
+
+require("core-js/modules/es.symbol.js");
+
+require("core-js/modules/es.symbol.description.js");
+
+require("core-js/modules/es.symbol.iterator.js");
+
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
 exports.argTypesEnhancers = exports.parameters = void 0;
 
-var _blocks = require("../../blocks");
+require("regenerator-runtime/runtime.js");
+
+require("core-js/modules/es.promise.js");
+
+require("core-js/modules/es.object.to-string.js");
 
 var _enhanceArgTypes = require("./enhanceArgTypes");
 
+function _getRequireWildcardCache() { if (typeof WeakMap !== "function") return null; var cache = new WeakMap(); _getRequireWildcardCache = function _getRequireWildcardCache() { return cache; }; return cache; }
+
+function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } if (obj === null || _typeof(obj) !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (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 asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }
+
+function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
+
 var parameters = {
   docs: {
     inlineStories: false,
-    container: _blocks.DocsContainer,
-    page: _blocks.DocsPage,
+    getContainer: function () {
+      var _getContainer = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee() {
+        return regeneratorRuntime.wrap(function _callee$(_context) {
+          while (1) {
+            switch (_context.prev = _context.next) {
+              case 0:
+                _context.next = 2;
+                return Promise.resolve().then(function () {
+                  return _interopRequireWildcard(require('../../blocks'));
+                });
+
+              case 2:
+                return _context.abrupt("return", _context.sent.DocsContainer);
+
+              case 3:
+              case "end":
+                return _context.stop();
+            }
+          }
+        }, _callee);
+      }));
+
+      function getContainer() {
+        return _getContainer.apply(this, arguments);
+      }
+
+      return getContainer;
+    }(),
+    getPage: function () {
+      var _getPage = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee2() {
+        return regeneratorRuntime.wrap(function _callee2$(_context2) {
+          while (1) {
+            switch (_context2.prev = _context2.next) {
+              case 0:
+                _context2.next = 2;
+                return Promise.resolve().then(function () {
+                  return _interopRequireWildcard(require('../../blocks'));
+                });
+
+              case 2:
+                return _context2.abrupt("return", _context2.sent.DocsPage);
+
+              case 3:
+              case "end":
+                return _context2.stop();
+            }
+          }
+        }, _callee2);
+      }));
+
+      function getPage() {
+        return _getPage.apply(this, arguments);
+      }
+
+      return getPage;
+    }(),
     iframeHeight: 100
   }
 };

package/dist/cjs/frameworks/html/sourceDecorator.js

@@ -49,8 +49,15 @@ function sourceDecorator(storyFn, context) {
   var story = context !== null && context !== void 0 && (_context$parameters$d3 = context.parameters.docs) !== null && _context$parameters$d3 !== void 0 && (_context$parameters$d4 = _context$parameters$d3.source) !== null && _context$parameters$d4 !== void 0 && _context$parameters$d4.excludeDecorators ? context.originalStoryFn(context.args, context) : storyFn();
   var source;
 
-  if (typeof story === 'string' && !skipSourceRender(context)) {
-    source = applyTransformSource(story, context);
+  if (!skipSourceRender(context)) {
+    if (typeof story === 'string') {
+      source = story;
+    } // eslint-disable-next-line no-undef
+    else if (story instanceof Element) {
+        source = story.outerHTML;
+      }
+
+    if (source) source = applyTransformSource(source, context);
   }
 
   (0, _addons.useEffect)(function () {

package/dist/cjs/frameworks/react/react-argtypes.stories.js

@@ -40,7 +40,7 @@ var _react2 = require("@storybook/react");
 
 var _components = require("@storybook/components");
 
-var _clientApi = require("@storybook/client-api");
+var _store = require("@storybook/store");
 
 var _extractArgTypes = require("./extractArgTypes");
 
@@ -67,7 +67,7 @@ var argsTableProps = function argsTableProps(component) {
   var parameters = {
     __isArgsStory: true
   };
-  var rows = (0, _clientApi.inferControls)({
+  var rows = (0, _store.inferControls)({
     argTypes: argTypes,
     parameters: parameters
   });