storybook/scripts/run-registry.ts
2022-07-23 19:40:03 +02:00

156 lines
4.2 KiB
TypeScript
Executable File

import { exec } from 'child_process';
import { remove, pathExists } from 'fs-extra';
import chalk from 'chalk';
import path from 'path';
import program from 'commander';
import detectFreePort from 'detect-port';
import fs from 'fs';
import yaml from 'js-yaml';
import startVerdaccioServer from 'verdaccio';
import pLimit from 'p-limit';
// @ts-ignore
import { maxConcurrentTasks } from './utils/concurrency';
import { listOfPackages } from './utils/list-packages';
program
.option('-O, --open', 'keep process open')
.option('-P, --publish', 'should publish packages')
.option('-p, --port <port>', 'port to run https server on');
program.parse(process.argv);
const logger = console;
const freePort = (port?: number) => port || detectFreePort(port);
const startVerdaccio = (port: number): Promise<any> => {
let resolved = false;
return Promise.race([
new Promise((resolve) => {
const cache = path.join(__dirname, '..', '.verdaccio-cache');
const config = {
...(yaml.safeLoad(
fs.readFileSync(path.join(__dirname, 'verdaccio.yaml'), 'utf8')
) as Record<string, any>),
self_path: cache,
};
const onReady = (webServer: any) => {
webServer.listen(port, () => {
resolved = true;
resolve(webServer);
});
};
startVerdaccioServer(config, 6000, cache, '1.0.0', 'verdaccio', onReady);
}),
new Promise((_, rej) => {
setTimeout(() => {
if (!resolved) {
resolved = true;
rej(new Error(`TIMEOUT - verdaccio didn't start within 10s`));
}
}, 10000);
}),
]);
};
const currentVersion = async () => {
const { version } = (await import('../code/lerna.json')).default;
return version;
};
const publish = (packages: { name: string; location: string }[], url: string) => {
logger.log(`Publishing packages with a concurrency of ${maxConcurrentTasks}`);
const limit = pLimit(maxConcurrentTasks);
let i = 0;
return Promise.all(
packages.map(({ name, location }) =>
limit(
() =>
new Promise((res, rej) => {
logger.log(
`🛫 publishing ${name} (${location.replace(
path.resolve(path.join(__dirname, '..')),
'.'
)})`
);
const command = `cd ${location} && npm publish --registry ${url} --force --access restricted --ignore-scripts`;
exec(command, (e) => {
if (e) {
rej(e);
} else {
i += 1;
logger.log(`${i}/${packages.length} 🛬 successful publish of ${name}!`);
res(undefined);
}
});
})
)
)
);
};
// const addUser = (url: string) =>
// new Promise((res, rej) => {
// logger.log(`👤 add temp user to verdaccio`);
// exec(`npx npm-cli-adduser -r "${url}" -a -u user -p password -e user@example.com`, (e) => {
// if (e) {
// rej(e);
// } else {
// res();
// }
// });
// });
const run = async () => {
const port = await freePort(program.port);
logger.log(`🌏 found a open port: ${port}`);
const verdaccioUrl = `http://localhost:${port}`;
logger.log(`📐 reading version of storybook`);
logger.log(`🚛 listing storybook packages`);
if (!process.env.CI) {
// when running e2e locally, clear cache to avoid EPUBLISHCONFLICT errors
const verdaccioCache = path.resolve(__dirname, '..', '.verdaccio-cache');
if (await pathExists(verdaccioCache)) {
logger.log(`🗑 cleaning up cache`);
await remove(verdaccioCache);
}
}
logger.log(`🎬 starting verdaccio (this takes ±5 seconds, so be patient)`);
const [verdaccioServer, packages, version] = await Promise.all([
startVerdaccio(port),
listOfPackages(),
currentVersion(),
]);
logger.log(`🌿 verdaccio running on ${verdaccioUrl}`);
// first time running, you might need to enable this
// await addUser(verdaccioUrl);
logger.log(`📦 found ${packages.length} storybook packages at version ${chalk.blue(version)}`);
if (program.publish) {
await publish(packages, verdaccioUrl);
}
if (!program.open) {
verdaccioServer.close();
}
};
run().catch((e) => {
logger.error(e);
process.exit(1);
});