'use strict';

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

var _keys = require('babel-runtime/core-js/object/keys');

var _keys2 = _interopRequireDefault(_keys);

exports.storiesOf = storiesOf;
exports.getStories = getStories;
exports.configure = configure;
exports.renderMain = renderMain;

var _data = require('./data');

var _storybook2 = require('./storybook');

var _storybook3 = _interopRequireDefault(_storybook2);

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

function storiesOf(kind, m) {
  // XXX: Add a better way to create stories and mutate them.
  m.hot.dispose(function () {
    delete _storybook3.default[kind];
  });

  _storybook3.default[kind] = {};
  function add(storyName, fn) {
    _storybook3.default[kind][storyName] = fn;

    var _storybook = {};
    (0, _keys2.default)(_storybook3.default).forEach(function (kind) {
      var stories = _storybook3.default[kind];
      _storybook[kind] = (0, _keys2.default)(stories);
    });

    (0, _data.setData)({ storybook: _storybook });
    return { add: add };
  }

  return { add: add };
}

function getStories() {
  return _storybook3.default;
}

function configure(loaders, module) {
  var render = function render() {
    function renderApp() {
      loaders();
      renderMain(getStories());
    }

    try {
      renderApp();
    } catch (error) {
      renderError(error);
    }
  };

  if (module.hot) {
    module.hot.accept(function () {
      setTimeout(render);
    });
  }

  render();
}

function renderMain(stories) {
  var data = (0, _data.getData)();
  data.error = null;

  data.selectedKind = _storybook3.default[data.selectedKind] ? data.selectedKind : (0, _keys2.default)(_storybook3.default)[0];

  if (data.selectedKind) {
    var _stories = _storybook3.default[data.selectedKind];
    data.selectedStory = _stories[data.selectedStory] ? data.selectedStory : (0, _keys2.default)(_stories)[0];
  }

  (0, _data.setData)(data);
};

var renderError = exports.renderError = function renderError(e) {
  var data = (0, _data.getData)();
  data.error = e;

  (0, _data.setData)(data);
};