creevey/lib/client/addon/register.js

Cross-browser screenshot testing tool for Storybook with fancy UI Runner

"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.registerCreeveyPanels = registerCreeveyPanels;
exports.ADDON_ID = void 0;

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

var _react = _interopRequireDefault(require("react"));

var _Addon = require("./components/Addon");

var _Tools = require("./components/Tools");

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';
exports.ADDON_ID = ADDON_ID;

_addons.addons.register(ADDON_ID, function (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();
        }
      }
    }, _callee);
  }));
  return _registerCreeveyPanels.apply(this, arguments);
}