GitHub/storybook
1
0
Fork 0
mirror of https://github.com/storybookjs/storybook.git synced 2025-04-05 01:31:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'master' of https://github.com/466023746/storybook

Browse Source
This commit is contained in:
娄涛 2018-08-23 16:01:33 +08:00
commit a328a4a3d5
2 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
addons/storysource/src/loader/index.js
View File

@ -1,13 +1,16 @@
import { getOptions } from 'loader-utils';
import injectDecorator from './inject-decorator';
const ADD_DECORATOR_STATEMENT = '.addDecorator(withStorySource(__STORY__, __ADDS_MAP__))';
let ADD_DECORATOR_STATEMENT = '.addDecorator(withStorySource(__STORY__, __ADDS_MAP__))';
function transform(source) {
const options = getOptions(this) || {};
if (options.noStoriesOf) ADD_DECORATOR_STATEMENT = '';
const result = injectDecorator(source, ADD_DECORATOR_STATEMENT, this.resourcePath, options);
if (!result.changed) {
if (!options.noStoriesOf && !result.changed) {
return source;
}
@ -17,13 +20,22 @@ function transform(source) {
const addsMap = JSON.stringify(result.addsMap);
return `
if (!options.noStoriesOf) {
return `
var withStorySource = require('@storybook/addon-storysource').withStorySource;
var __STORY__ = ${sourceJson};
var __ADDS_MAP__ = ${addsMap};
${result.source}
`;
} else {
return `
export var __STORY__ = ${sourceJson};
export var __ADDS_MAP__ = ${addsMap};
${result.source}
`;
}
}
export default transform;

2
addons/storysource/src/loader/inject-decorator.js
View File

@ -23,7 +23,7 @@ function inject(source, decorator, filepath, options = {}) {
? generateSourceWithDecorators(source, decorator, options.parser)
: generateSourceWithoutDecorators(source, options.parser);
if (!changed) {
if (!options.noStoriesOf && !changed) {
return {
source: newSource,
addsMap: {},