mirror of
https://github.com/storybookjs/storybook.git
synced 2025-04-06 00:21:07 +08:00
37 lines
1.2 KiB
TypeScript
37 lines
1.2 KiB
TypeScript
import type { UserConfig } from 'vite';
|
|
import type { Builder_EnvsRaw } from '@storybook/types';
|
|
|
|
// Allowed env variables on the client
|
|
const allowedEnvVariables = [
|
|
'STORYBOOK',
|
|
// Vite `import.meta.env` default variables
|
|
// @see https://github.com/vitejs/vite/blob/6b8d94dca2a1a8b4952e3e3fcd0aed1aedb94215/packages/vite/types/importMeta.d.ts#L68-L75
|
|
'BASE_URL',
|
|
'MODE',
|
|
'DEV',
|
|
'PROD',
|
|
'SSR',
|
|
];
|
|
|
|
/**
|
|
* Customized version of stringifyProcessEnvs from @storybook/core-common which
|
|
* uses import.meta.env instead of process.env and checks for allowed variables.
|
|
*/
|
|
export function stringifyProcessEnvs(raw: Builder_EnvsRaw, envPrefix: UserConfig['envPrefix']) {
|
|
const updatedRaw: Builder_EnvsRaw = {};
|
|
const envs = Object.entries(raw).reduce((acc: Builder_EnvsRaw, [key, value]) => {
|
|
// Only add allowed values OR values from array OR string started with allowed prefixes
|
|
if (
|
|
allowedEnvVariables.includes(key) ||
|
|
(Array.isArray(envPrefix) && !!envPrefix.find((prefix) => key.startsWith(prefix))) ||
|
|
(typeof envPrefix === 'string' && key.startsWith(envPrefix))
|
|
) {
|
|
acc[`import.meta.env.${key}`] = JSON.stringify(value);
|
|
updatedRaw[key] = value;
|
|
}
|
|
return acc;
|
|
}, {});
|
|
|
|
return envs;
|
|
}
|