2022-02-23 17:17:25 -05:00

73 lines
2.7 KiB
TypeScript

import path, { join } from 'path';
import { logger } from '@storybook/node-logger';
interface PresetOptions {
configDir?: string;
backgrounds?: any;
viewport?: any;
docs?: any;
}
const requireMain = (configDir: string) => {
let main = {};
const absoluteConfigDir = path.isAbsolute(configDir)
? configDir
: path.join(process.cwd(), configDir);
const mainFile = path.join(absoluteConfigDir, 'main');
try {
// eslint-disable-next-line global-require,import/no-dynamic-require
main = require(mainFile);
} catch (err) {
try {
// Try finding a .cjs version of the main file
const mainFileCjs = path.join(absoluteConfigDir, 'main.cjs');
// eslint-disable-next-line global-require,import/no-dynamic-require
main = require(mainFileCjs);
} catch (cjsErr) {
logger.warn(`Unable to find main.js or main.cjs: ${mainFile}`);
}
}
return main;
};
export function addons(options: PresetOptions = {}) {
const checkInstalled = (addon: string, main: any) => {
const existingAddon = main.addons?.find((entry: string | { name: string }) => {
const name = typeof entry === 'string' ? entry : entry.name;
return name?.startsWith(addon);
});
if (existingAddon) {
logger.info(`Found existing addon ${JSON.stringify(existingAddon)}, skipping.`);
}
return !!existingAddon;
};
const main = requireMain(options.configDir);
return (
['docs', 'controls', 'actions', 'backgrounds', 'viewport', 'toolbars', 'measure', 'outline']
.filter((key) => (options as any)[key] !== false)
.map((key) => `@storybook/addon-${key}`)
.filter((addon) => !checkInstalled(addon, main))
// Use `require.resolve` to ensure Yarn PnP compatibility
// Files of various addons should be resolved in the context of `addon-essentials` as they are listed as deps here
// and not in `@storybook/core` nor in SB user projects. If `@storybook/core` make the require itself Yarn 2 will
// throw an error saying that the package to require must be added as a dependency. Doing `require.resolve` will
// allow `@storybook/core` to work with absolute path directly, no more require of dep no more issue.
// File to load can be `preset.js`, `register.js`, or the package entry point, so we need to check all these cases
// as it's done in `lib/core/src/server/presets.js`.
.map((addon) => {
try {
return require.resolve(join(addon, 'preset'));
// eslint-disable-next-line no-empty
} catch (err) {}
try {
return require.resolve(join(addon, 'register'));
// eslint-disable-next-line no-empty
} catch (err) {}
return require.resolve(addon);
})
);
}