creevey 0.8.0-beta.0 → 0.8.0-beta.1

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

@@ -3,8 +3,8 @@
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
-exports.registerCreeveyPanels = registerCreeveyPanels;
 exports.ADDON_ID = void 0;
+exports.registerCreeveyPanels = registerCreeveyPanels;
 
 var _addons = require("@storybook/addons");
 
@@ -18,79 +18,55 @@ var _Manager = require("./Manager");
 
 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 ADDON_ID = 'creevey';
+const ADDON_ID = 'creevey';
 exports.ADDON_ID = ADDON_ID;
 
-_addons.addons.register(ADDON_ID, function (api) {
+_addons.addons.register(ADDON_ID, api => {
   void registerCreeveyPanels(api);
 });
 
-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 _Manager.CreeveyManager(storybookApi);
-
-            _addons.addons.addPanel("".concat(ADDON_ID, "/panel/run"), {
-              title: "Creevey/Run",
-              match: function match(_ref) {
-                var viewMode = _ref.viewMode;
-                return !!(viewMode && /^story$/.exec(viewMode));
-              },
-              type: _addons.types.TOOL,
-              // eslint-disable-next-line react/display-name
-              render: function render() {
-                return /*#__PURE__*/_react.default.createElement(_Tools.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.addons.addPanel(panelId, {
-                title: title,
-                type: _addons.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.default.createElement(_Addon.Addon, {
-                    active: active,
-                    key: key,
-                    manager: manager,
-                    browser: browser
-                  });
-                }
-              });
-            });
-
-          case 6:
-          case "end":
-            return _context.stop();
-        }
+async function registerCreeveyPanels(storybookApi) {
+  const manager = new _Manager.CreeveyManager(storybookApi);
+
+  _addons.addons.addPanel("".concat(ADDON_ID, "/panel/run"), {
+    title: "Creevey/Run",
+    match: _ref => {
+      let {
+        viewMode
+      } = _ref;
+      return !!(viewMode && /^story$/.exec(viewMode));
+    },
+    type: _addons.types.TOOL,
+    // eslint-disable-next-line react/display-name
+    render: () => /*#__PURE__*/_react.default.createElement(_Tools.Tools, {
+      manager
+    })
+  });
+
+  await manager.initAll();
+  const browsers = manager.getBrowsers();
+  browsers.forEach(browser => {
+    const panelId = "".concat(ADDON_ID, "/panel/").concat(browser);
+    const title = manager.getTabTitle(browser);
+
+    _addons.addons.addPanel(panelId, {
+      title,
+      type: _addons.types.PANEL,
+      paramKey: browser,
+      // NOTE key = PANEL_ID needs to correct render button in addons panel
+      // eslint-disable-next-line react/display-name
+      render: _ref2 => {
+        let {
+          active,
+          key
+        } = _ref2;
+        return /*#__PURE__*/_react.default.createElement(_Addon.Addon, {
+          active,
+          key,
+          manager,
+          browser
+        });
       }
-    }, _callee);
-  }));
-  return _registerCreeveyPanels.apply(this, arguments);
+    });
+  });
 }

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

@@ -4,9 +4,10 @@ Object.defineProperty(exports, "__esModule", {
   value: true
 });
 exports.getEmojiByTestStatus = getEmojiByTestStatus;
+exports.isInternetExplorer = void 0;
 
 function getEmojiByTestStatus(status) {
-  var skip = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : false;
+  let skip = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : false;
 
   switch (status) {
     case 'failed':
@@ -35,4 +36,7 @@ function getEmojiByTestStatus(status) {
         return '';
       }
   }
-}
+}
+
+const isInternetExplorer = navigator.userAgent.includes('Trident/');
+exports.isInternetExplorer = isInternetExplorer;

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

@@ -3,79 +3,62 @@
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
+exports.capture = capture;
 exports.withCreevey = withCreevey;
 
 var Events = _interopRequireWildcard(require("@storybook/core-events"));
 
 var polyfill = _interopRequireWildcard(require("event-source-polyfill"));
 
+var _testingLibrary = require("@storybook/testing-library");
+
 var _addons = require("@storybook/addons");
 
 var _types = require("../../types");
 
-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); }
+var _shared = require("../../shared");
 
-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; }
+var _helpers = require("../shared/helpers");
 
-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); } }
+var _utils = require("./utils");
 
-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 _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function (nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); }
 
-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); }
+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; }
 
-if ((typeof process === "undefined" ? "undefined" : _typeof(process)) != 'object' || typeof process.version != 'string') {
+if (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
+  const {
+    NativeEventSource,
+    EventSourcePolyfill
+  } = polyfill; // 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);
-}
+const 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() {
-  _resetCurrentStory = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee3(channel) {
-    return regeneratorRuntime.wrap(function _callee3$(_context3) {
-      while (1) {
-        switch (_context3.prev = _context3.next) {
-          case 0:
-            setTimeout(function () {
-              return channel.emit(Events.SET_CURRENT_STORY, {
-                storyId: true,
-                name: '',
-                kind: ''
-              });
-            }, 0);
-            return _context3.abrupt("return", new Promise(function (resolve) {
-              return channel.once(Events.STORY_MISSING, resolve);
-            }));
-
-          case 2:
-          case "end":
-            return _context3.stop();
-        }
-      }
-    }, _callee3);
-  }));
-  return _resetCurrentStory.apply(this, arguments);
+async function resetCurrentStory(channel) {
+  setTimeout(() => channel.emit(Events.SET_CURRENT_STORY, {
+    storyId: true,
+    name: '',
+    kind: ''
+  }), 0);
+  return new Promise(resolve => channel.once(Events.STORY_MISSING, resolve));
 }
 
 function catchRenderError(channel) {
-  var rejectCallback;
-  var promise = new Promise(function (_resolve, reject) {
-    return rejectCallback = reject;
-  });
+  let rejectCallback;
+  const promise = new Promise((_resolve, reject) => rejectCallback = reject);
 
   function errorHandler(_ref) {
-    var title = _ref.title,
-        description = _ref.description;
+    let {
+      title,
+      description
+    } = _ref;
     rejectCallback({
       message: title,
       stack: description
@@ -99,10 +82,8 @@ function catchRenderError(channel) {
 }
 
 function waitForStoryRendered(channel) {
-  var resolveCallback;
-  var promise = new Promise(function (resolve) {
-    return resolveCallback = resolve;
-  });
+  let resolveCallback;
+  const promise = new Promise(resolve => resolveCallback = resolve);
 
   function renderHandler() {
     resolveCallback();
@@ -120,13 +101,11 @@ function waitForStoryRendered(channel) {
 
 function waitForFontsLoaded() {
   if (!document.fonts) return;
-  var areFontsLoading = Array.from(document.fonts).some(function (font) {
-    return font.status == 'loading';
-  });
+  const areFontsLoading = Array.from(document.fonts).some(font => font.status == 'loading');
 
   if (areFontsLoading) {
-    return new Promise(function (resolve) {
-      var fontsLoadedHandler = function fontsLoadedHandler() {
+    return new Promise(resolve => {
+      const fontsLoadedHandler = () => {
         document.fonts.removeEventListener('loadingdone', fontsLoadedHandler);
         resolve();
       };
@@ -136,128 +115,134 @@ function waitForFontsLoaded() {
   }
 }
 
+function waitForCaptureCall() {
+  return new Promise(resolve => captureResolver = resolve);
+}
+
+function initCreeveyState() {
+  var _window$localStorage$;
+
+  const 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', () => {
+    window.localStorage.setItem('Creevey_Tests', JSON.stringify({
+      creeveyHost: window.__CREEVEY_SERVER_HOST__,
+      creeveyPort: window.__CREEVEY_SERVER_PORT__,
+      setStoriesCounter,
+      isTestBrowser
+    }));
+  });
+}
+
+let isTestBrowser = false;
+let captureResolver;
+let waitForCreevey;
+let creeveyReady;
+let setStoriesCounter = 0;
+
 function withCreevey() {
-  var currentStory = '';
-  var isAnimationDisabled = false;
+  let currentStory = '';
+  let isAnimationDisabled = false;
+  initCreeveyState();
 
   function disableAnimation() {
     isAnimationDisabled = true;
-    var style = document.createElement('style');
-    var textNode = document.createTextNode(disableAnimationsStyles);
+    const style = document.createElement('style');
+    const textNode = document.createTextNode(disableAnimationsStyles);
     style.setAttribute('type', 'text/css');
     style.appendChild(textNode);
     document.head.appendChild(style);
   }
 
-  function selectStory(_x2, _x3, _x4, _x5, _x6) {
-    return _selectStory.apply(this, arguments);
+  async function getStories() {
+    var _window$__STORYBOOK_S;
+
+    const storiesPromise = new Promise(resolve => _addons.addons.getChannel().once(Events.SET_STORIES, data => resolve((0, _shared.serializeRawStories)((0, _shared.denormalizeStoryParameters)(data)))));
+    const store = (_window$__STORYBOOK_S = window.__STORYBOOK_STORY_STORE__) !== null && _window$__STORYBOOK_S !== void 0 ? _window$__STORYBOOK_S : {};
+
+    if (store.cacheAllCSFFiles) {
+      await store.cacheAllCSFFiles();
+
+      _addons.addons.getChannel().emit(Events.SET_STORIES, store.getSetStoriesPayload());
+    } else return;
+
+    _addons.addons.getChannel().on(Events.SET_STORIES, data => {
+      // TODO Figure out how to get only updated stories
+      // TODO Subscribe on hmr? like use dummy-hmr
+      setStoriesCounter += 1;
+      const stories = (0, _shared.serializeRawStories)((0, _shared.denormalizeStoryParameters)(data));
+      const storiesByFiles = new Map();
+      Object.values(stories).forEach(story => {
+        const storiesFromFile = storiesByFiles.get(story.parameters.fileName);
+        if (storiesFromFile) storiesFromFile.push(story);else storiesByFiles.set(story.parameters.fileName, [story]);
+      });
+      void fetch("http://".concat((0, _helpers.getConnectionUrl)(), "/stories"), {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json'
+        },
+        body: JSON.stringify({
+          setStoriesCounter,
+          stories: [...storiesByFiles.entries()]
+        })
+      });
+    });
+
+    return storiesPromise;
   }
 
-  function _selectStory() {
-    _selectStory = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee2(storyId, kind, name, shouldWaitForReady, callback) {
-      var channel, waitForReady, renderPromise, errorPromise, _reason$stack, errorMessage;
-
-      return regeneratorRuntime.wrap(function _callee2$(_context2) {
-        while (1) {
-          switch (_context2.prev = _context2.next) {
-            case 0:
-              if (!isAnimationDisabled) disableAnimation();
-              channel = _addons.addons.getChannel();
-              waitForReady = shouldWaitForReady ? new Promise(function (resolve) {
-                return window.__CREEVEY_SET_READY_FOR_CAPTURE__ = resolve;
-              }) : Promise.resolve();
-
-              if (!(storyId == currentStory)) {
-                _context2.next = 8;
-                break;
-              }
-
-              _context2.next = 6;
-              return resetCurrentStory(channel);
-
-            case 6:
-              _context2.next = 9;
-              break;
-
-            case 8:
-              currentStory = storyId;
-
-            case 9:
-              renderPromise = waitForStoryRendered(channel);
-              errorPromise = catchRenderError(channel);
-              setTimeout(function () {
-                return channel.emit(Events.SET_CURRENT_STORY, {
-                  storyId: storyId,
-                  name: name,
-                  kind: kind
-                });
-              }, 0);
-              _context2.prev = 12;
-              _context2.next = 15;
-              return Promise.race([_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 renderPromise;
-
-                      case 2:
-                        _context.next = 4;
-                        return waitForFontsLoaded();
-
-                      case 4:
-                        _context.next = 6;
-                        return waitForReady;
-
-                      case 6:
-                      case "end":
-                        return _context.stop();
-                    }
-                  }
-                }, _callee);
-              }))(), errorPromise]);
-
-            case 15:
-              callback();
-              _context2.next = 22;
-              break;
-
-            case 18:
-              _context2.prev = 18;
-              _context2.t0 = _context2["catch"](12);
-              // 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 = _context2.t0 instanceof Error ? (_reason$stack = _context2.t0.stack) !== null && _reason$stack !== void 0 ? _reason$stack : _context2.t0.message : (0, _types.isObject)(_context2.t0) ? "".concat(_context2.t0.message, "\n    ").concat(_context2.t0.stack) : _context2.t0;
-              callback(errorMessage);
-
-            case 22:
-              _context2.prev = 22;
-              renderPromise.cancel();
-              errorPromise.cancel();
-              return _context2.finish(22);
-
-            case 26:
-            case "end":
-              return _context2.stop();
-          }
-        }
-      }, _callee2, null, [[12, 18, 22, 26]]);
-    }));
-    return _selectStory.apply(this, arguments);
+  async function selectStory(storyId, kind, name, shouldWaitForReady, callback) {
+    if (!isAnimationDisabled) disableAnimation();
+    isTestBrowser = true;
+
+    const channel = _addons.addons.getChannel();
+
+    const waitForReady = shouldWaitForReady ? new Promise(resolve => window.__CREEVEY_SET_READY_FOR_CAPTURE__ = resolve) : Promise.resolve();
+    if (storyId == currentStory) await resetCurrentStory(channel);else currentStory = storyId;
+    let isCaptureCalled = false;
+    const renderPromise = waitForStoryRendered(channel);
+    const errorPromise = catchRenderError(channel);
+    const capturePromise = waitForCaptureCall().then(() => isCaptureCalled = true);
+    setTimeout(() => channel.emit(Events.SET_CURRENT_STORY, {
+      storyId,
+      name,
+      kind
+    }), 0);
+
+    try {
+      await Promise.race([(async () => {
+        await Promise.race([renderPromise, capturePromise]);
+        await waitForFontsLoaded();
+        await waitForReady;
+      })(), errorPromise]);
+      callback([null, isCaptureCalled]);
+    } catch (reason) {
+      var _reason$stack;
+
+      // 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
+      const errorMessage = reason instanceof Error ? (_reason$stack = reason.stack) !== null && _reason$stack !== void 0 ? _reason$stack : reason.message : (0, _types.isObject)(reason) ? "".concat(reason.message, "\n    ").concat(reason.stack) : reason;
+      callback([errorMessage]);
+    } finally {
+      renderPromise.cancel();
+      errorPromise.cancel();
+    }
   }
 
   function updateGlobals(globals) {
     _addons.addons.getChannel().emit(Events.UPDATE_GLOBALS, {
-      globals: globals
+      globals
     });
   }
 
   function insertIgnoreStyles(ignoreSelectors) {
-    var stylesElement = document.createElement('style');
+    const stylesElement = document.createElement('style');
     stylesElement.setAttribute('type', 'text/css');
     document.head.appendChild(stylesElement);
-    ignoreSelectors.forEach(function (selector) {
+    ignoreSelectors.forEach(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;
@@ -269,16 +254,97 @@ function withCreevey() {
     (_ignoreStyles$parentN = ignoreStyles.parentNode) === null || _ignoreStyles$parentN === void 0 ? void 0 : _ignoreStyles$parentN.removeChild(ignoreStyles);
   }
 
+  function hasPlayCompletedYet(callback) {
+    creeveyReady();
+    let isCaptureCalled = false;
+    let isPlayCompleted = false;
+
+    const channel = _addons.addons.getChannel();
+
+    void waitForStoryRendered(channel).then(() => {
+      if (isCaptureCalled) return;
+      isPlayCompleted = true;
+      callback(true);
+    });
+    void waitForCaptureCall().then(() => {
+      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__ = _types.noop;
+
+  const queryAllByQuery = (container, query) => [...container.querySelectorAll(query)].filter(e => e instanceof HTMLElement);
+
+  const getMultipleError = (_, query) => "Found multiple elements by query: ".concat(query);
+
+  const getMissingError = (_, query) => "Unable to find an element by query: ".concat(query);
+
+  const [queryByQuery, getAllByQuery, getByQuery, findAllByQuery, findByQuery] = (0, _testingLibrary.buildQueries)(queryAllByQuery, getMultipleError, getMissingError);
+  const queries = {
+    queryByQuery,
+    getAllByQuery,
+    getByQuery,
+    findAllByQuery,
+    findByQuery
+  };
   return (0, _addons.makeDecorator)({
     name: 'withCreevey',
     parameterName: 'creevey',
-    wrapper: function wrapper(getStory, context) {
+    wrapper: (getStory, context) => {
+      var _ref2;
+
+      // TODO Define proper types, like captureElement is a promise
+      const {
+        captureElement
+      } = context.parameters.creevey = (_ref2 = context.parameters.creevey) !== null && _ref2 !== void 0 ? _ref2 : {};
+      Object.defineProperty(context.parameters.creevey, 'captureElement', {
+        get() {
+          switch (true) {
+            case captureElement === undefined:
+              return Promise.resolve(context.canvasElement);
+
+            case captureElement === null:
+              return Promise.resolve(document.documentElement);
+
+            case typeof captureElement == 'string':
+              return _utils.isInternetExplorer // some code from testing-library makes IE hang
+              ? Promise.resolve(context.canvasElement.querySelector(captureElement)) : (0, _testingLibrary.within)(context.canvasElement, queries).findByQuery(captureElement);
+
+            case typeof captureElement == 'function':
+              // TODO Define type for it
+              return Promise.resolve(captureElement(context));
+          }
+        },
+
+        enumerable: true,
+        configurable: true
+      });
       return getStory(context);
     }
   });
+}
+
+async function capture(options) {
+  if (!isTestBrowser) return;
+  captureResolver();
+  waitForCreevey = new Promise(resolve => creeveyReady = resolve);
+  await fetch("http://".concat((0, _helpers.getConnectionUrl)(), "/capture"), {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/json'
+    },
+    body: JSON.stringify({
+      workerId: window.__CREEVEY_WORKER_ID__,
+      options
+    })
+  });
+  await waitForCreevey;
 }