storybook-chrome-screenshot
Version:
A Storybook addon, Save the screenshot image of your stories! via puppeteer.
48 lines • 2.22 kB
JavaScript
;
var __assign = (this && this.__assign) || Object.assign || function(t) {
for (var s, i = 1, n = arguments.length; i < n; i++) {
s = arguments[i];
for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p))
t[p] = s[p];
}
return t;
};
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
var __importStar = (this && this.__importStar) || function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k in mod) if (Object.hasOwnProperty.call(mod, k)) result[k] = mod[k];
result["default"] = mod;
return result;
};
Object.defineProperty(exports, "__esModule", { value: true });
var addons_1 = __importDefault(require("@storybook/addons"));
var React = __importStar(require("react"));
var util_inspect_1 = __importDefault(require("util-inspect"));
var constants_1 = require("../../core/constants");
var screenshot_options_1 = require("../../screenshot-options");
var ScreenshotWrapper_1 = require("./components/ScreenshotWrapper");
exports.withScreenshot = function (options) {
if (options === void 0) { options = {}; }
return function (storyFn, ctx) {
var channel = addons_1.default.getChannel();
var wrapperWithContext = function (context) {
var props = __assign({}, screenshot_options_1.mergeScreenshotOptions(options), { channel: channel,
context: context });
return React.createElement(ScreenshotWrapper_1.ScreenshotWrapper, __assign({}, props), storyFn(context));
};
if (typeof storyFn !== 'function') {
var msg = "The story may not be correct, (storyFn = \"" + util_inspect_1.default(storyFn) + "\")";
channel.emit(constants_1.EventTypes.COMPONENT_ERROR, msg); // For puppeteer
throw new Error(msg); // For browser
}
if (ctx != null) {
return wrapperWithContext(ctx);
}
// tslint:disable-next-line: no-unnecessary-callback-wrapper
return function (context) { return wrapperWithContext(context); };
};
};
//# sourceMappingURL=withScreenshot.js.map