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

'use strict';

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

var _typeof2 = require('babel-runtime/helpers/typeof');

var _typeof3 = _interopRequireDefault(_typeof2);

var _map = require('babel-runtime/core-js/map');

var _map2 = _interopRequireDefault(_map);

var _getPrototypeOf = require('babel-runtime/core-js/object/get-prototype-of');

var _getPrototypeOf2 = _interopRequireDefault(_getPrototypeOf);

var _classCallCheck2 = require('babel-runtime/helpers/classCallCheck');

var _classCallCheck3 = _interopRequireDefault(_classCallCheck2);

var _possibleConstructorReturn2 = require('babel-runtime/helpers/possibleConstructorReturn');

var _possibleConstructorReturn3 = _interopRequireDefault(_possibleConstructorReturn2);

var _inherits2 = require('babel-runtime/helpers/inherits');

var _inherits3 = _interopRequireDefault(_inherits2);

exports.default = createPercyAddon;

var _es6Error = require('es6-error');

var _es6Error2 = _interopRequireDefault(_es6Error);

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

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

var InvalidOptionError = exports.InvalidOptionError = function (_ExtendableError) {
  (0, _inherits3.default)(InvalidOptionError, _ExtendableError);

  function InvalidOptionError() {
    (0, _classCallCheck3.default)(this, InvalidOptionError);
    return (0, _possibleConstructorReturn3.default)(this, (InvalidOptionError.__proto__ || (0, _getPrototypeOf2.default)(InvalidOptionError)).apply(this, arguments));
  }

  return InvalidOptionError;
}(_es6Error2.default);

function assertWidths(widths) {
  if (typeof widths === 'undefined') {
    return;
  }
  if (!Array.isArray(widths)) {
    throw new InvalidOptionError('Given widths option is not an array');
  }
  if (!widths.length) {
    throw new InvalidOptionError('Need at least one valid width');
  }
  widths.forEach(function (width) {
    if (isNaN(width) || width !== ~~width) {
      throw new InvalidOptionError("Given width '" + width + "' is invalid");
    }
  });
}

function assertRtl(rtl) {
  if (typeof rtl === 'undefined') {
    return;
  }
  if (typeof rtl !== 'boolean') {
    throw new InvalidOptionError("Given rtl setting '" + rtl + "' is invalid");
  }
}

function createPercyAddon() {
  /**
   * This map contains a mapping from stories for a kind to
   * options of the respective story,
   * looking like this:
   * {
   *   kindA: { story1: options, storyX: options },
   *   kindB: { story...}
   *   ...
   * }
   */
  var optionsTuples = new _map2.default();

  var percyAddon = {
    addWithPercyOptions: function addWithPercyOptions(storyName, options, storyFn) {
      if (typeof options === 'function') {
        storyFn = options;
        options = undefined;
      }

      if (options) {
        assertWidths(options.widths);
        assertRtl(options.rtl);

        optionsTuples.set(this.kind, optionsTuples.get(this.kind) || new _map2.default());
        var tuplesForKind = optionsTuples.get(this.kind);
        tuplesForKind.set(storyName, options);
      }
      return this.add(storyName, storyFn);
    }
  };

  var serializeStories = function serializeStories(getStorybook) {
    var storybook = getStorybook();
    storybook.forEach(function (storyBucket) {
      if (!optionsTuples.has(storyBucket.kind)) {
        return;
      }
      var tuplesForKind = optionsTuples.get(storyBucket.kind);
      storyBucket.stories.forEach(function (story) {
        if (tuplesForKind.has(story.name)) {
          story.options = tuplesForKind.get(story.name);
        }
      });
    });
    if ((typeof window === 'undefined' ? 'undefined' : (0, _typeof3.default)(window)) === 'object') window[_constants.storiesKey] = storybook;
    return storybook;
  };

  return { percyAddon: percyAddon, serializeStories: serializeStories };
}