@percy-io/percy-storybook/lib/getStories.js

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});

var _regenerator = require('babel-runtime/regenerator');

var _regenerator2 = _interopRequireDefault(_regenerator);

var _asyncToGenerator2 = require('babel-runtime/helpers/asyncToGenerator');

var _asyncToGenerator3 = _interopRequireDefault(_asyncToGenerator2);

var _promise = require('babel-runtime/core-js/promise');

var _promise2 = _interopRequireDefault(_promise);

var _jsdom = require('jsdom');

var _jsdom2 = _interopRequireDefault(_jsdom);

var _constants = require('./constants');

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

// jsdom doesn't support Web Workers yet.
// We use workerMock to allow the user's preview.js to interact with the Worker API.
var workerMock = '\n    function MockWorker(path) {\n      var api = this;\n\n      function addEventListener() {}\n      function removeEventListener() {}\n      function postMessage() {}\n      function terminate() {}\n\n      api.postMessage = postMessage;\n      api.addEventListener = addEventListener;\n      api.removeEventListener = removeEventListener;\n      api.terminate = terminate;\n\n      return api;\n    }\n    window.Worker = MockWorker;\n';

// jsdom doesn't support localStorage yet.
// We use localStorageMock to allow the user's preview.js to interact with localStorage.
var localStorageMock = '\n    var localStorageMock = (function() {\n      var store = {};\n      return {\n        getItem: function(key) {\n          return store[key];\n        },\n        setItem: function(key, value) {\n          store[key] = value.toString();\n        },\n        clear: function() {\n          store = {};\n        }\n      };\n    })();\n    Object.defineProperty(window, \'localStorage\', { value: localStorageMock });\n    ';

// jsdom doesn't support matchMedia yet.
var matchMediaMock = '\n    window.matchMedia = window.matchMedia || (() => {\n      return {\n        matches: false,\n        addListener: () => {},\n        removeListener: () => {},\n      };\n    });\n    ';

function getStoriesFromDom(previewJavascriptCode, options) {
  return new _promise2.default(function (resolve, reject) {
    var jsDomConfig = {
      html: '',
      url: 'https://example.com/iframe.js?selectedKind=none&selectedStory=none',
      src: [workerMock, localStorageMock, matchMediaMock, previewJavascriptCode],
      done: function done(err, window) {
        if (err) return reject(err.response.body);
        if (!window || !window[_constants.storiesKey]) {
          var message = 'Storybook object not found on window. ' + "Check your call to serializeStories in your Storybook's config.js.";
          reject(new Error(message));
        }
        resolve(window[_constants.storiesKey]);
      }
    };
    if (options.debug) {
      jsDomConfig.virtualConsole = _jsdom2.default.createVirtualConsole().sendTo(console);
    }
    _jsdom2.default.env(jsDomConfig);
  });
}

exports.default = function () {
  var _ref = (0, _asyncToGenerator3.default)( /*#__PURE__*/_regenerator2.default.mark(function _callee(storybookCode) {
    var options = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {};
    var stories;
    return _regenerator2.default.wrap(function _callee$(_context) {
      while (1) {
        switch (_context.prev = _context.next) {
          case 0:
            if (!(!storybookCode || storybookCode === '')) {
              _context.next = 2;
              break;
            }

            throw new Error('Storybook code was not received.');

          case 2:
            _context.next = 4;
            return getStoriesFromDom(storybookCode, options);

          case 4:
            stories = _context.sent;
            return _context.abrupt('return', stories);

          case 6:
          case 'end':
            return _context.stop();
        }
      }
    }, _callee, this);
  }));

  function getStories(_x2) {
    return _ref.apply(this, arguments);
  }

  return getStories;
}();