monkeeShark/packages/frontend/.storybook/generate.tsx
2023-03-19 22:31:18 +09:00

135 lines
3.4 KiB
TypeScript

import * as fs from 'node:fs/promises';
import { basename, dirname, join } from 'node:path/posix';
import { promisify } from 'node:util';
import { generate } from 'astring';
import type * as estree from 'estree';
import * as glob from 'glob';
import { format } from 'prettier';
function h<T extends estree.Node>(component: T['type'], props: Omit<T, 'type'>): T {
const type = component.replace(/(?:^|-)([a-z])/g, (_, c) => c.toUpperCase());
return Object.assign(props, { type }) as T;
}
function toStories(component: string, location: string): string {
const literal = (
<literal value={join(location, component).slice(4, -4)} />
) as unknown as estree.Literal;
const identifier = (
<identifier name={component.slice(0, -4).replace(/[-.]|^(?=\d)/g, '_')} />
) as unknown as estree.Identifier;
const program = (
<program
body={[
<import-declaration
source={<literal value="@storybook/vue3" />}
specifiers={[
<import-specifier
local={<identifier name="Meta" />}
imported={<identifier name="Meta" />}
/>,
<import-specifier
local={<identifier name="Story" />}
imported={<identifier name="Story" />}
/>,
]}
/>,
<import-declaration
source={<literal value={`./${component}`} />}
specifiers={[
<import-default-specifier
local={identifier}
imported={identifier}
/>,
]}
/>,
<variable-declaration
kind="const"
declarations={[
<variable-declarator
id={<identifier name="meta" />}
init={
<object-expression
properties={[
<property
key={<identifier name="title" />}
value={literal}
kind="init"
/>,
<property
key={<identifier name="component" />}
value={identifier}
kind="init"
/>,
]}
/>
}
/>,
]}
/>,
<export-named-declaration
declaration={
<variable-declaration
kind="const"
declarations={[
<variable-declarator
id={<identifier name="Default" />}
init={
<object-expression
properties={[
<property
key={<identifier name="components" />}
value={
<object-expression
properties={[
<property
key={identifier}
value={identifier}
kind="init"
shorthand
/>,
]}
/>
}
kind="init"
/>,
<property
key={<identifier name="template" />}
value={<literal value={`<${identifier.name} />`} />}
kind="init"
/>,
]}
/>
}
/>,
]}
/>
}
/>,
<export-default-declaration
declaration={<identifier name="meta" />}
/>,
]}
/>
) as unknown as estree.Program;
return format(
generate(program),
{
parser: 'babel-ts',
singleQuote: true,
useTabs: true,
}
);
}
promisify(glob)('src/{components,pages,ui,widgets}/**/*.vue').then((components) => Promise.all(
components.map((component) => {
const stories = component.replace(/\.vue$/, '.stories.ts');
fs.stat(stories).then(
() => {},
() => {
fs.writeFile(stories, toStories(basename(component), dirname(component)));
}
);
})
));