creevey 0.9.0-beta.1 → 0.9.0-beta.2

package/lib/esm/client/addon/preset.js

@@ -0,0 +1,56 @@
+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; }
+
+import path from 'path';
+import { exec } from 'shelljs';
+import { DefinePlugin as FallbackDefinePlugin } from 'webpack';
+import { isStorybookVersionLessThan, resolveFromStorybookCore, resolveFromStorybookCoreServer } from '../../server/storybook/helpers';
+export function config() {
+  var entry = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : [];
+  return [].concat(_toConsumableArray(entry), [require.resolve('./decorator')]);
+}
+export function managerEntries() {
+  var entry = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : [];
+  return [].concat(_toConsumableArray(entry), [require.resolve('./register')]);
+}
+export function managerWebpack(config, options) {
+  var _options$presets$appl, _options$presets;
+
+  // TODO How to execute with non-webpack bundlers
+  if (options.configType == 'PRODUCTION' && !isStorybookVersionLessThan(6, 2) && options.skipExtract != true) {
+    var args = [];
+    if (options.creeveyPreExtract) args.push("--require \"".concat(options.creeveyPreExtract, "\""));
+    args.push(path.join(__dirname, '../../cli'));
+    args.push("--extract \"".concat(options.outputDir, "\""));
+    if (options.creeveyConfigPath) args.push("--config \"".concat(options.creeveyConfigPath, "\""));
+    exec("node ".concat(args.join(' ')), {
+      async: true
+    });
+  }
+
+  return ((_options$presets$appl = (_options$presets = options.presets) === null || _options$presets === void 0 ? void 0 : _options$presets.apply('webpackInstance')) !== null && _options$presets$appl !== void 0 ? _options$presets$appl : Promise.resolve(undefined)).then(function (webpack) {
+    return webpack !== null && webpack !== void 0 ? webpack : import(isStorybookVersionLessThan(6, 2) ? resolveFromStorybookCore('webpack') : resolveFromStorybookCoreServer('webpack'));
+  }).then(function (webpack) {
+    var _config$plugins, _options$creeveyPort;
+
+    var _ref = webpack !== null && webpack !== void 0 ? webpack : {},
+        _ref$DefinePlugin = _ref.DefinePlugin,
+        DefinePlugin = _ref$DefinePlugin === void 0 ? FallbackDefinePlugin : _ref$DefinePlugin;
+
+    (_config$plugins = config.plugins) === null || _config$plugins === void 0 ? void 0 : _config$plugins.push(new DefinePlugin({
+      __CREEVEY_SERVER_HOST__: options.creeveyHost,
+      __CREEVEY_SERVER_PORT__: (_options$creeveyPort = options.creeveyPort) !== null && _options$creeveyPort !== void 0 ? _options$creeveyPort : 3000,
+      __CREEVEY_CLIENT_PORT__: options.clientPort
+    }));
+    return config;
+  });
+}

package/lib/esm/client/addon/readyForCapture.js

@@ -0,0 +1,5 @@
+export function readyForCapture() {
+  var _window$__CREEVEY_SET, _window;
+
+  (_window$__CREEVEY_SET = (_window = window).__CREEVEY_SET_READY_FOR_CAPTURE__) === null || _window$__CREEVEY_SET === void 0 ? void 0 : _window$__CREEVEY_SET.call(_window);
+}

package/lib/esm/client/addon/register.js

@@ -0,0 +1,75 @@
+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); }); }; }
+
+import { addons, types } from '@storybook/addons';
+import React from 'react';
+import { Addon } from './components/Addon';
+import { Tools } from './components/Tools';
+import { CreeveyManager } from './Manager';
+export var ADDON_ID = 'creevey';
+addons.register(ADDON_ID, function (api) {
+  void registerCreeveyPanels(api);
+});
+export function registerCreeveyPanels(_x) {
+  return _registerCreeveyPanels.apply(this, arguments);
+}
+
+function _registerCreeveyPanels() {
+  _registerCreeveyPanels = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee(storybookApi) {
+    var manager, browsers;
+    return regeneratorRuntime.wrap(function _callee$(_context) {
+      while (1) {
+        switch (_context.prev = _context.next) {
+          case 0:
+            manager = new CreeveyManager(storybookApi);
+            addons.addPanel("".concat(ADDON_ID, "/panel/run"), {
+              title: "Creevey/Run",
+              match: function match(_ref) {
+                var viewMode = _ref.viewMode;
+                return !!(viewMode && /^story$/.exec(viewMode));
+              },
+              type: types.TOOL,
+              // eslint-disable-next-line react/display-name
+              render: function render() {
+                return /*#__PURE__*/React.createElement(Tools, {
+                  manager: manager
+                });
+              }
+            });
+            _context.next = 4;
+            return manager.initAll();
+
+          case 4:
+            browsers = manager.getBrowsers();
+            browsers.forEach(function (browser) {
+              var panelId = "".concat(ADDON_ID, "/panel/").concat(browser);
+              var title = manager.getTabTitle(browser);
+              addons.addPanel(panelId, {
+                title: title,
+                type: types.PANEL,
+                paramKey: browser,
+                // NOTE key = PANEL_ID needs to correct render button in addons panel
+                // eslint-disable-next-line react/display-name
+                render: function render(_ref2) {
+                  var active = _ref2.active,
+                      key = _ref2.key;
+                  return /*#__PURE__*/React.createElement(Addon, {
+                    active: active,
+                    key: key,
+                    manager: manager,
+                    browser: browser
+                  });
+                }
+              });
+            });
+
+          case 6:
+          case "end":
+            return _context.stop();
+        }
+      }
+    }, _callee);
+  }));
+  return _registerCreeveyPanels.apply(this, arguments);
+}

package/lib/esm/client/addon/utils.js

@@ -0,0 +1,31 @@
+export function getEmojiByTestStatus(status) {
+  var skip = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : false;
+
+  switch (status) {
+    case 'failed':
+      {
+        return '❌';
+      }
+
+    case 'success':
+      {
+        return '✔';
+      }
+
+    case 'running':
+      {
+        return '🟡';
+      }
+
+    case 'pending':
+      {
+        return '🕗';
+      }
+
+    default:
+      {
+        if (skip) return '⏸';
+        return '';
+      }
+  }
+}

package/lib/esm/client/addon/withCreevey.js

@@ -0,0 +1,509 @@
+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 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); }); }; }
+
+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); }
+
+import * as Events from '@storybook/core-events';
+import * as polyfill from 'event-source-polyfill';
+// import { buildQueries, within } from '@storybook/testing-library';
+import { addons, makeDecorator } from '@storybook/addons';
+import { isObject, noop } from '../../types';
+import { denormalizeStoryParameters, serializeRawStories } from '../../shared';
+import { getConnectionUrl } from '../shared/helpers';
+
+if ((typeof process === "undefined" ? "undefined" : _typeof(process)) != 'object' || typeof process.version != 'string') {
+  // NOTE If you don't use babel-polyfill or any other polyfills that add EventSource for IE11
+  // You don't get hot reload in IE11. So put polyfill for that to better UX
+  // Don't load in nodejs environment
+  // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
+  var NativeEventSource = polyfill.NativeEventSource,
+      EventSourcePolyfill = polyfill.EventSourcePolyfill; // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
+
+  window.EventSource = NativeEventSource || EventSourcePolyfill;
+}
+
+var disableAnimationsStyles = "\n*,\n*:hover,\n*::before,\n*::after {\n  animation-delay: -0.0001ms !important;\n  animation-duration: 0s !important;\n  animation-play-state: paused !important;\n  cursor: none !important;\n  caret-color: transparent !important;\n  transition: 0s !important;\n}\n";
+
+function resetCurrentStory(_x) {
+  return _resetCurrentStory.apply(this, arguments);
+}
+
+function _resetCurrentStory() {
+  _resetCurrentStory = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee4(channel) {
+    return regeneratorRuntime.wrap(function _callee4$(_context4) {
+      while (1) {
+        switch (_context4.prev = _context4.next) {
+          case 0:
+            setTimeout(function () {
+              return channel.emit(Events.SET_CURRENT_STORY, {
+                storyId: true,
+                name: '',
+                kind: ''
+              });
+            }, 0);
+            return _context4.abrupt("return", new Promise(function (resolve) {
+              return channel.once(Events.STORY_MISSING, resolve);
+            }));
+
+          case 2:
+          case "end":
+            return _context4.stop();
+        }
+      }
+    }, _callee4);
+  }));
+  return _resetCurrentStory.apply(this, arguments);
+}
+
+function catchRenderError(channel) {
+  var rejectCallback;
+  var promise = new Promise(function (_resolve, reject) {
+    return rejectCallback = reject;
+  });
+
+  function errorHandler(_ref) {
+    var title = _ref.title,
+        description = _ref.description;
+    rejectCallback({
+      message: title,
+      stack: description
+    });
+  }
+
+  function exceptionHandler(exception) {
+    rejectCallback(exception);
+  }
+
+  function removeHandlers() {
+    channel.off(Events.STORY_ERRORED, errorHandler);
+    channel.off(Events.STORY_THREW_EXCEPTION, errorHandler);
+  }
+
+  channel.once(Events.STORY_ERRORED, errorHandler);
+  channel.once(Events.STORY_THREW_EXCEPTION, exceptionHandler);
+  return Object.assign(promise, {
+    cancel: removeHandlers
+  });
+}
+
+function waitForStoryRendered(channel) {
+  var resolveCallback;
+  var promise = new Promise(function (resolve) {
+    return resolveCallback = resolve;
+  });
+
+  function renderHandler() {
+    resolveCallback();
+  }
+
+  function removeHandlers() {
+    channel.off(Events.STORY_RENDERED, renderHandler);
+  }
+
+  channel.once(Events.STORY_RENDERED, renderHandler);
+  return Object.assign(promise, {
+    cancel: removeHandlers
+  });
+}
+
+function waitForFontsLoaded() {
+  if (!document.fonts) return;
+  var areFontsLoading = Array.from(document.fonts).some(function (font) {
+    return font.status == 'loading';
+  });
+
+  if (areFontsLoading) {
+    return new Promise(function (resolve) {
+      var fontsLoadedHandler = function fontsLoadedHandler() {
+        document.fonts.removeEventListener('loadingdone', fontsLoadedHandler);
+        resolve();
+      };
+
+      document.fonts.addEventListener('loadingdone', fontsLoadedHandler);
+    });
+  }
+}
+
+function waitForCaptureCall() {
+  return new Promise(function (resolve) {
+    return captureResolver = resolve;
+  });
+}
+
+function initCreeveyState() {
+  var _window$localStorage$;
+
+  var prevState = JSON.parse((_window$localStorage$ = window.localStorage.getItem('Creevey_Tests')) !== null && _window$localStorage$ !== void 0 ? _window$localStorage$ : '{}');
+  if (prevState.creeveyHost) window.__CREEVEY_SERVER_HOST__ = prevState.creeveyHost;
+  if (prevState.creeveyPort) window.__CREEVEY_SERVER_PORT__ = prevState.creeveyPort;
+  if (prevState.setStoriesCounter) setStoriesCounter = prevState.setStoriesCounter;
+  if (prevState.isTestBrowser) isTestBrowser = prevState.isTestBrowser;
+  window.addEventListener('beforeunload', function () {
+    window.localStorage.setItem('Creevey_Tests', JSON.stringify({
+      creeveyHost: window.__CREEVEY_SERVER_HOST__,
+      creeveyPort: window.__CREEVEY_SERVER_PORT__,
+      setStoriesCounter: setStoriesCounter,
+      isTestBrowser: isTestBrowser
+    }));
+  });
+}
+
+var isTestBrowser = false;
+var captureResolver;
+var waitForCreevey;
+var creeveyReady;
+var setStoriesCounter = 0;
+export function withCreevey() {
+  var currentStory = '';
+  var isAnimationDisabled = false;
+  initCreeveyState();
+
+  function disableAnimation() {
+    isAnimationDisabled = true;
+    var style = document.createElement('style');
+    var textNode = document.createTextNode(disableAnimationsStyles);
+    style.setAttribute('type', 'text/css');
+    style.appendChild(textNode);
+    document.head.appendChild(style);
+  }
+
+  function getStories() {
+    return _getStories.apply(this, arguments);
+  }
+
+  function _getStories() {
+    _getStories = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee() {
+      var _window$__STORYBOOK_S;
+
+      var storiesPromise, store;
+      return regeneratorRuntime.wrap(function _callee$(_context) {
+        while (1) {
+          switch (_context.prev = _context.next) {
+            case 0:
+              storiesPromise = new Promise(function (resolve) {
+                return addons.getChannel().once(Events.SET_STORIES, function (data) {
+                  return resolve(serializeRawStories(denormalizeStoryParameters(data)));
+                });
+              });
+              store = (_window$__STORYBOOK_S = window.__STORYBOOK_STORY_STORE__) !== null && _window$__STORYBOOK_S !== void 0 ? _window$__STORYBOOK_S : {}; // @ts-expect-error `pushToManager` exists only in Storybook 6.0 - 6.3
+
+              if (!store.pushToManager) {
+                _context.next = 6;
+                break;
+              }
+
+              // @ts-expect-error `pushToManager` exists only in Storybook 6.0 - 6.3
+              // eslint-disable-next-line @typescript-eslint/no-unsafe-call
+              store.pushToManager();
+              _context.next = 13;
+              break;
+
+            case 6:
+              if (!store.cacheAllCSFFiles) {
+                _context.next = 12;
+                break;
+              }
+
+              _context.next = 9;
+              return store.cacheAllCSFFiles();
+
+            case 9:
+              addons.getChannel().emit(Events.SET_STORIES, store.getSetStoriesPayload());
+              _context.next = 13;
+              break;
+
+            case 12:
+              return _context.abrupt("return");
+
+            case 13:
+              addons.getChannel().on(Events.SET_STORIES, function (data) {
+                // TODO Figure out how to get only updated stories
+                // TODO Subscribe on hmr? like use dummy-hmr
+                setStoriesCounter += 1;
+                var stories = serializeRawStories(denormalizeStoryParameters(data));
+                var storiesByFiles = new Map();
+                Object.values(stories).forEach(function (story) {
+                  var storiesFromFile = storiesByFiles.get(story.parameters.fileName);
+                  if (storiesFromFile) storiesFromFile.push(story);else storiesByFiles.set(story.parameters.fileName, [story]);
+                });
+                void fetch("http://".concat(getConnectionUrl(), "/stories"), {
+                  method: 'POST',
+                  headers: {
+                    'Content-Type': 'application/json'
+                  },
+                  body: JSON.stringify({
+                    setStoriesCounter: setStoriesCounter,
+                    stories: _toConsumableArray(storiesByFiles.entries())
+                  })
+                });
+              });
+              return _context.abrupt("return", storiesPromise);
+
+            case 15:
+            case "end":
+              return _context.stop();
+          }
+        }
+      }, _callee);
+    }));
+    return _getStories.apply(this, arguments);
+  }
+
+  function selectStory(_x2, _x3, _x4, _x5, _x6) {
+    return _selectStory.apply(this, arguments);
+  }
+
+  function _selectStory() {
+    _selectStory = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee3(storyId, kind, name, shouldWaitForReady, callback) {
+      var channel, waitForReady, isCaptureCalled, renderPromise, errorPromise, capturePromise, _reason$stack, errorMessage;
+
+      return regeneratorRuntime.wrap(function _callee3$(_context3) {
+        while (1) {
+          switch (_context3.prev = _context3.next) {
+            case 0:
+              if (!isAnimationDisabled) disableAnimation();
+              isTestBrowser = true;
+              channel = addons.getChannel();
+              waitForReady = shouldWaitForReady ? new Promise(function (resolve) {
+                return window.__CREEVEY_SET_READY_FOR_CAPTURE__ = resolve;
+              }) : Promise.resolve();
+
+              if (!(storyId == currentStory)) {
+                _context3.next = 9;
+                break;
+              }
+
+              _context3.next = 7;
+              return resetCurrentStory(channel);
+
+            case 7:
+              _context3.next = 10;
+              break;
+
+            case 9:
+              currentStory = storyId;
+
+            case 10:
+              isCaptureCalled = false;
+              renderPromise = waitForStoryRendered(channel);
+              errorPromise = catchRenderError(channel);
+              capturePromise = waitForCaptureCall().then(function () {
+                return isCaptureCalled = true;
+              });
+              setTimeout(function () {
+                return channel.emit(Events.SET_CURRENT_STORY, {
+                  storyId: storyId,
+                  name: name,
+                  kind: kind
+                });
+              }, 0);
+              _context3.prev = 15;
+              _context3.next = 18;
+              return Promise.race([_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.race([renderPromise, capturePromise]);
+
+                      case 2:
+                        _context2.next = 4;
+                        return waitForFontsLoaded();
+
+                      case 4:
+                        _context2.next = 6;
+                        return waitForReady;
+
+                      case 6:
+                      case "end":
+                        return _context2.stop();
+                    }
+                  }
+                }, _callee2);
+              }))(), errorPromise]);
+
+            case 18:
+              callback([null, isCaptureCalled]);
+              _context3.next = 25;
+              break;
+
+            case 21:
+              _context3.prev = 21;
+              _context3.t0 = _context3["catch"](15);
+              // NOTE Event `STORY_THREW_EXCEPTION` triggered only in react and vue frameworks and return Error instance
+              // NOTE Event `STORY_ERRORED` return error-like object without `name` field
+              errorMessage = _context3.t0 instanceof Error ? (_reason$stack = _context3.t0.stack) !== null && _reason$stack !== void 0 ? _reason$stack : _context3.t0.message : isObject(_context3.t0) ? "".concat(_context3.t0.message, "\n    ").concat(_context3.t0.stack) : _context3.t0;
+              callback([errorMessage]);
+
+            case 25:
+              _context3.prev = 25;
+              renderPromise.cancel();
+              errorPromise.cancel();
+              return _context3.finish(25);
+
+            case 29:
+            case "end":
+              return _context3.stop();
+          }
+        }
+      }, _callee3, null, [[15, 21, 25, 29]]);
+    }));
+    return _selectStory.apply(this, arguments);
+  }
+
+  function updateGlobals(globals) {
+    addons.getChannel().emit(Events.UPDATE_GLOBALS, {
+      globals: globals
+    });
+  }
+
+  function insertIgnoreStyles(ignoreSelectors) {
+    var stylesElement = document.createElement('style');
+    stylesElement.setAttribute('type', 'text/css');
+    document.head.appendChild(stylesElement);
+    ignoreSelectors.forEach(function (selector) {
+      stylesElement.innerHTML += "\n        ".concat(selector, " {\n          background: #000 !important;\n          box-shadow: none !important;\n          text-shadow: none !important;\n          outline: 0 !important;\n          color: rgba(0,0,0,0) !important;\n        }\n        ").concat(selector, " *, ").concat(selector, "::before, ").concat(selector, "::after {\n          visibility: hidden !important;\n        }\n      ");
+    });
+    return stylesElement;
+  }
+
+  function removeIgnoreStyles(ignoreStyles) {
+    var _ignoreStyles$parentN;
+
+    (_ignoreStyles$parentN = ignoreStyles.parentNode) === null || _ignoreStyles$parentN === void 0 ? void 0 : _ignoreStyles$parentN.removeChild(ignoreStyles);
+  }
+
+  function hasPlayCompletedYet(callback) {
+    creeveyReady();
+    var isCaptureCalled = false;
+    var isPlayCompleted = false;
+    var channel = addons.getChannel();
+    void waitForStoryRendered(channel).then(function () {
+      if (isCaptureCalled) return;
+      isPlayCompleted = true;
+      callback(true);
+    });
+    void waitForCaptureCall().then(function () {
+      if (isPlayCompleted) return;
+      isCaptureCalled = true;
+      callback(false);
+    });
+  }
+
+  window.__CREEVEY_GET_STORIES__ = getStories;
+  window.__CREEVEY_SELECT_STORY__ = selectStory;
+  window.__CREEVEY_UPDATE_GLOBALS__ = updateGlobals;
+  window.__CREEVEY_INSERT_IGNORE_STYLES__ = insertIgnoreStyles;
+  window.__CREEVEY_REMOVE_IGNORE_STYLES__ = removeIgnoreStyles;
+  window.__CREEVEY_HAS_PLAY_COMPLETED_YET__ = hasPlayCompletedYet;
+  window.__CREEVEY_SET_READY_FOR_CAPTURE__ = noop; // const queryAllByQuery = (container: HTMLElement, query: string): HTMLElement[] =>
+  //   [...container.querySelectorAll(query)].filter((e) => e instanceof HTMLElement) as HTMLElement[];
+  // const getMultipleError = (_: Element | null, query: string): string => `Found multiple elements by query: ${query}`;
+  // const getMissingError = (_: Element | null, query: string): string => `Unable to find an element by query: ${query}`;
+  // const [queryByQuery, getAllByQuery, getByQuery, findAllByQuery, findByQuery] = buildQueries(
+  //   queryAllByQuery,
+  //   getMultipleError,
+  //   getMissingError,
+  // );
+  // const queries = {
+  //   queryByQuery,
+  //   getAllByQuery,
+  //   getByQuery,
+  //   findAllByQuery,
+  //   findByQuery,
+  // };
+
+  return makeDecorator({
+    name: 'withCreevey',
+    parameterName: 'creevey',
+    wrapper: function wrapper(getStory, context) {
+      var _ref2;
+
+      // TODO Define proper types, like captureElement is a promise
+      var _context$parameters$c = context.parameters.creevey = (_ref2 = context.parameters.creevey) !== null && _ref2 !== void 0 ? _ref2 : {},
+          captureElement = _context$parameters$c.captureElement;
+
+      Object.defineProperty(context.parameters.creevey, 'captureElement', {
+        get: function get() {
+          switch (true) {
+            case captureElement === undefined:
+              return Promise.resolve(context.canvasElement);
+
+            case captureElement === null:
+              return Promise.resolve(document.documentElement);
+            // case typeof captureElement == 'string':
+            //   return within<typeof queries>(context.canvasElement, queries).findByQuery(captureElement as string);
+
+            case typeof captureElement == 'function':
+              // TODO Define type for it
+              return Promise.resolve(captureElement(context));
+          }
+        },
+        enumerable: true,
+        configurable: true
+      });
+      return getStory(context);
+    }
+  });
+}
+export function capture(_x7) {
+  return _capture.apply(this, arguments);
+}
+
+function _capture() {
+  _capture = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee5(options) {
+    return regeneratorRuntime.wrap(function _callee5$(_context5) {
+      while (1) {
+        switch (_context5.prev = _context5.next) {
+          case 0:
+            if (isTestBrowser) {
+              _context5.next = 2;
+              break;
+            }
+
+            return _context5.abrupt("return");
+
+          case 2:
+            captureResolver();
+            waitForCreevey = new Promise(function (resolve) {
+              return creeveyReady = resolve;
+            });
+            _context5.next = 6;
+            return fetch("http://".concat(getConnectionUrl(), "/capture"), {
+              method: 'POST',
+              headers: {
+                'Content-Type': 'application/json'
+              },
+              body: JSON.stringify({
+                workerId: window.__CREEVEY_WORKER_ID__,
+                options: options
+              })
+            });
+
+          case 6:
+            _context5.next = 8;
+            return waitForCreevey;
+
+          case 8:
+          case "end":
+            return _context5.stop();
+        }
+      }
+    }, _callee5);
+  }));
+  return _capture.apply(this, arguments);
+}