mirror of
https://github.com/kokonect-link/cherrypick
synced 2024-11-01 15:45:58 +09:00
136 lines
3.4 KiB
TypeScript
136 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={`<${component.slice(0, -4)} />`} />}
|
||
|
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)));
|
||
|
}
|
||
|
);
|
||
|
})
|
||
|
));
|