@storybook/addon-storysource/dist/preview.js

"use strict";

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

var _addons = _interopRequireDefault(require("@storybook/addons"));

var _events = require("./events");

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

var getLocation = function getLocation(context, locationsMap) {
  return locationsMap[context.id];
};

function setStorySource(context, source, locationsMap) {
  var currentLocation = getLocation(context, locationsMap);

  _addons["default"].getChannel().emit(_events.EVENT_ID, {
    source: source,
    currentLocation: currentLocation,
    locationsMap: locationsMap
  });
}

function withStorySource(source) {
  var locationsMap = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {};
  return function (storyFn, context) {
    setStorySource(context, source, locationsMap);
    return storyFn();
  };
}