sizuku 0.0.6 → 0.1.0

package/dist/generator/mermaid-er/generator/{generate-relation-line.js → relation-line.js}

@@ -1,14 +1,11 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.generateRelationLine = generateRelationLine;
-const build_relation_line_1 = require("../relationship/build-relation-line");
+import { buildRelationLine } from '../relationship/build-relation-line.js';
 /**
  * Generate a relation line for a relation
  * @param relation - The relation to generate a line for
  * @returns The generated relation line
  */
-function generateRelationLine(relation) {
-    const cardinality = (0, build_relation_line_1.buildRelationLine)(relation.type);
+export function relationLine(relation) {
+    const cardinality = buildRelationLine(relation.type);
     if (!cardinality) {
         throw new Error(`Unknown relation type: ${relation.type}`);
     }

package/dist/generator/mermaid-er/index.d.ts

@@ -1,5 +1,6 @@
-#!/usr/bin/env node
-import type { Config } from './config';
-export declare const ER_HEADER: readonly ["```mermaid", "erDiagram"];
-export declare const ER_FOOTER: readonly ["```"];
-export declare function main(dev?: boolean, config?: Config): Promise<boolean>;
+/**
+ * Generate Mermaid ER diagram
+ * @param code - The code to generate Mermaid ER diagram from
+ * @param output - The output file path
+ */
+export declare function sizukuMermaidER(code: string[], output: string): Promise<import("neverthrow").Result<void, Error>>;

package/dist/generator/mermaid-er/index.js

@@ -1,77 +1,16 @@
-#!/usr/bin/env node
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.ER_FOOTER = exports.ER_HEADER = void 0;
-exports.main = main;
-const node_fs_1 = require("node:fs");
-const node_path_1 = __importDefault(require("node:path"));
-const generate_er_content_1 = require("./generator/generate-er-content");
-const parse_table_info_1 = require("./validator/parse-table-info");
-const extract_relations_1 = require("./core/extract-relations");
-const config_1 = require("./config");
-const node_process_1 = require("node:process");
-// ER diagram header
-exports.ER_HEADER = ['```mermaid', 'erDiagram'];
-// ER diagram footer
-exports.ER_FOOTER = ['```'];
-async function main(dev = false, config = (0, config_1.getConfig)()) {
-    // 1. argv ['**/bin/node', '**/dist/src/generator/mermaid-er/index.js', 'db/schema.ts', '-o', 'mermaid-er/ER.md']
-    if (config.output === undefined && !node_process_1.argv.includes('-o')) {
-        console.error('Error: -o is not found');
-        return false;
-    }
-    // 2. slice [ 'db/schema.ts', '-o', 'mermaid-er/ER.md']
-    const args = process.argv.slice(2);
-    // 3. input = args[0] = 'db/schema.ts'
-    const input = config.input ?? args[0];
-    config.input = input;
-    // 4. output = 'mermaid-er/ER.md'
-    const output = config.output ?? args[args.indexOf('-o') + 1];
-    config.output = output;
-    try {
-        // 5. read db/schema.ts
-        const content = (0, node_fs_1.readFileSync)(input, 'utf-8');
-        // 6. split lines
-        const lines = content.split('\n');
-        // 7. create output directory
-        const outputDir = node_path_1.default.dirname(output);
-        if (!(0, node_fs_1.existsSync)(outputDir)) {
-            (0, node_fs_1.mkdirSync)(outputDir, { recursive: true });
-        }
-        // 8. skip import section
-        const codeStart = lines.findIndex((line) => !line.trim().startsWith('import') && line.trim() !== '');
-        // 9. parse table info
-        const tables = (0, parse_table_info_1.parseTableInfo)(lines.slice(codeStart));
-        // 10. extract relations
-        const relations = (0, extract_relations_1.extractRelations)(lines.slice(codeStart));
-        // 11. generate ER content
-        const ERContent = (0, generate_er_content_1.generateERContent)(relations, tables);
-        // 12. write ER content to output file
-        (0, node_fs_1.writeFileSync)(output, ERContent);
-        console.log(`Generated ER at: ${output}`);
-        return true;
-    }
-    catch (e) {
-        if (e instanceof Error) {
-            console.error(e.message);
-            if (dev) {
-                throw e;
-            }
-            process.exit(1);
-        }
-        if (dev) {
-            throw new Error('Unknown error occurred');
-        }
-        return false;
-    }
-}
-if (require.main === module) {
-    main().then((success) => {
-        if (!success) {
-            process.exit(1);
-        }
-    });
+import { mkdir, writeFile } from '../../shared/fsp/index.js';
+import { extractRelations } from './core/extract-relations.js';
+import { erContent } from './generator/index.js';
+import { parseTableInfo } from './validator/parse-table-info.js';
+import path from 'node:path';
+/**
+ * Generate Mermaid ER diagram
+ * @param code - The code to generate Mermaid ER diagram from
+ * @param output - The output file path
+ */
+export async function sizukuMermaidER(code, output) {
+    const tables = parseTableInfo(code);
+    const relations = extractRelations(code);
+    const ERContent = erContent(relations, tables);
+    return await mkdir(path.dirname(output)).andThen(() => writeFile(output, ERContent));
 }

package/dist/generator/mermaid-er/relationship/build-relation-line.js

@@ -1,7 +1,4 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.buildRelationLine = buildRelationLine;
-const is_relationship_1 = require("../validator/is-relationship");
+import { isRelationship } from '../validator/is-relationship.js';
 const RELATIONSHIPS = {
     'zero-one': '|o',
     one: '||',
@@ -13,7 +10,7 @@ const RELATIONSHIPS = {
  * @param { string } input
  * @returns { string }
  */
-function buildRelationLine(input) {
+export function buildRelationLine(input) {
     const parts = input.split('-to-');
     if (parts.length !== 2) {
         throw new Error(`Invalid input format: ${input}`);
@@ -24,7 +21,7 @@ function buildRelationLine(input) {
     const from = parts[0];
     const to = toRaw;
     const isOptional = optionalFlag === 'optional';
-    if (!((0, is_relationship_1.isRelationship)(from) && (0, is_relationship_1.isRelationship)(to))) {
+    if (!(isRelationship(from) && isRelationship(to))) {
         throw new Error(`Invalid relationship string: ${input}`);
     }
     const fromSymbol = RELATIONSHIPS[from];

package/dist/generator/mermaid-er/types.js

@@ -0,0 +1 @@
+export {};

package/dist/generator/mermaid-er/validator/index.d.ts

@@ -0,0 +1,4 @@
+export { isRelationship } from './is-relationship.js';
+export { parseRelationLine } from './parse-relation-line.js';
+export { parseTableInfo } from './parse-table-info.js';
+export { removeDuplicateRelations } from './remove-duplicate-relations.js';

package/dist/generator/mermaid-er/validator/index.js

@@ -0,0 +1,4 @@
+export { isRelationship } from './is-relationship.js';
+export { parseRelationLine } from './parse-relation-line.js';
+export { parseTableInfo } from './parse-table-info.js';
+export { removeDuplicateRelations } from './remove-duplicate-relations.js';

package/dist/generator/mermaid-er/validator/is-relationship.d.ts

@@ -1,4 +1,4 @@
-import type { Relationship } from '../relationship/build-relation-line';
+import type { Relationship } from '../relationship/build-relation-line.js';
 /**
  * isRelationship
  * @param { string } key

package/dist/generator/mermaid-er/validator/is-relationship.js

@@ -1,11 +1,8 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.isRelationship = isRelationship;
 /**
  * isRelationship
  * @param { string } key
  * @returns { key is Relationship }
  */
-function isRelationship(key) {
+export function isRelationship(key) {
     return ['zero-one', 'one', 'zero-many', 'many'].includes(key);
 }

package/dist/generator/mermaid-er/validator/parse-relation-line.js

@@ -1,12 +1,9 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.parseRelationLine = parseRelationLine;
 /**
  * Parse a relation line
  * @param line - The line to parse
  * @returns The parsed relation
  */
-function parseRelationLine(line) {
+export function parseRelationLine(line) {
     // @relation <fromModel>.<fromField> <toModel>.<toField> <relationType>
     const relationMatch = line.match(/@relation\s+(\w+)\.(\w+)\s+(\w+)\.(\w+)\s+(\w+-to-\w+)/);
     if (relationMatch) {

package/dist/generator/mermaid-er/validator/parse-table-info.d.ts

@@ -1,8 +1,2 @@
-import type { TableInfo } from '../type';
-/**
- * Parse table info from code
- * @function parseTableInfo
- * @param code - The code to parse the table info from
- * @returns The parsed table info
- */
+import type { TableInfo } from '../types.js';
 export declare function parseTableInfo(code: string[]): TableInfo[];

package/dist/generator/mermaid-er/validator/parse-table-info.js

@@ -1,91 +1,71 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.parseTableInfo = parseTableInfo;
-/**
- * Parse table info from code
- * @function parseTableInfo
- * @param code - The code to parse the table info from
- * @returns The parsed table info
- */
-function parseTableInfo(code) {
-    const initialAccumulator = {
-        tables: [],
-        currentTable: null,
-        currentDescription: '',
-        tempFields: [],
-        currentTableStartLine: 0,
-    };
-    const result = code.reduce((acc, line, index) => {
-        // extract description
-        const descriptionMatch = line.match(/\/\/\/\s*([^@].*)/);
-        if (descriptionMatch &&
-            !(line.includes('@z.') || line.includes('@v.')) &&
-            !line.includes('@relation')) {
-            acc.currentDescription = descriptionMatch[1]?.trim() ?? null;
-            return acc;
-        }
-        // extract table name
-        const tableMatch = line.match(/export const (\w+)\s*=\s*mysqlTable/);
-        if (tableMatch) {
-            if (acc.currentTable && acc.tempFields.length > 0) {
-                acc.currentTable.fields = acc.tempFields.map((field) => ({
-                    name: field.name,
-                    type: field.type,
-                    description: field.description,
-                }));
-                acc.tables.push(acc.currentTable);
-            }
-            acc.currentTable = {
-                name: tableMatch[1],
-                fields: [],
+import { Project, Node } from 'ts-morph';
+const baseBuilderName = (expr) => {
+    if (Node.isIdentifier(expr))
+        return expr.getText();
+    if (Node.isCallExpression(expr) || Node.isPropertyAccessExpression(expr))
+        return baseBuilderName(expr.getExpression());
+    return '';
+};
+const isFieldInfo = (v) => v !== null;
+export function parseTableInfo(code) {
+    const source = code.join('\n');
+    const file = new Project({ useInMemoryFileSystem: true }).createSourceFile('temp.ts', source);
+    return file
+        .getVariableStatements()
+        .filter((stmt) => stmt.isExported())
+        .flatMap((stmt) => {
+        const decl = stmt.getDeclarations()[0];
+        if (!Node.isVariableDeclaration(decl))
+            return [];
+        const varName = decl.getName();
+        if (varName.toLowerCase().includes('relation'))
+            return [];
+        const init = decl.getInitializer();
+        if (!(init && Node.isCallExpression(init)))
+            return [];
+        const callee = init.getExpression().getText();
+        if (!callee.endsWith('Table') || callee === 'relations')
+            return [];
+        const objLit = init.getArguments()[1];
+        if (!(objLit && Node.isObjectLiteralExpression(objLit)))
+            return [];
+        const fields = objLit
+            .getProperties()
+            .filter(Node.isPropertyAssignment)
+            .map((prop) => {
+            const keyNode = prop.getNameNode();
+            if (!Node.isIdentifier(keyNode))
+                return null;
+            const fieldName = keyNode.getText();
+            const initExpr = prop.getInitializer();
+            if (!(initExpr && Node.isCallExpression(initExpr)))
+                return null;
+            const fieldType = baseBuilderName(initExpr);
+            const initText = initExpr.getText();
+            const lineIdx = prop.getStartLineNumber() - 1;
+            const baseDesc = code
+                .slice(0, lineIdx)
+                .reverse()
+                .find((line) => {
+                const t = line.trim();
+                return (t.startsWith('///') &&
+                    !t.includes('@z.') &&
+                    !t.includes('@v.') &&
+                    !t.includes('@relation'));
+            })
+                ?.replace(/^\s*\/\/\/\s*/, '') ?? '';
+            const prefix = initText.includes('.primaryKey()')
+                ? '(PK) '
+                : initText.includes('.references(')
+                    ? '(FK) '
+                    : '';
+            return {
+                name: fieldName,
+                type: fieldType,
+                description: `${prefix}${baseDesc}`.trim(),
             };
-            acc.tempFields = [];
-            acc.currentTableStartLine = index;
-            return acc;
-        }
-        // field info extraction
-        if (!acc.currentTable) {
-            return acc;
-        }
-        const fieldMatch = line.match(/^\s*([^:]+):\s*([^(]+)\(/);
-        if (fieldMatch) {
-            const [_, fieldName, fieldType] = fieldMatch;
-            const description = line.includes('.primaryKey()')
-                ? `(PK) ${acc.currentDescription || ''}`
-                : acc.currentDescription || '';
-            acc.tempFields.push({
-                name: fieldName.trim(),
-                type: fieldType.trim(),
-                description,
-                line: index,
-            });
-            acc.currentDescription = '';
-            return acc;
-        }
-        // detect foreign key constraint
-        if (line.includes('.references') || line.includes('relations')) {
-            const lastIndex = acc.tempFields.length - 1;
-            if (lastIndex >= 0) {
-                const lastField = acc.tempFields[lastIndex];
-                acc.tempFields[lastIndex] = {
-                    name: lastField.name,
-                    type: lastField.type,
-                    description: `(FK) ${lastField.description}`,
-                    line: lastField.line,
-                };
-            }
-            return acc;
-        }
-        return acc;
-    }, initialAccumulator);
-    // process fields of the last table
-    if (result.currentTable && result.tempFields.length > 0) {
-        result.currentTable.fields = result.tempFields.map((field) => ({
-            name: field.name,
-            type: field.type,
-            description: field.description,
-        }));
-        result.tables.push(result.currentTable);
-    }
-    return result.tables;
+        })
+            .filter(isFieldInfo);
+        return [{ name: varName, fields }];
+    });
 }

package/dist/generator/mermaid-er/validator/remove-duplicate-relations.js

@@ -1,12 +1,9 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.removeDuplicateRelations = removeDuplicateRelations;
 /**
  * Remove duplicate relations
  * @function removeDuplicateRelations
  * @param relations - The relations to remove duplicates from
  * @returns The relations without duplicates
  */
-function removeDuplicateRelations(relations) {
+export function removeDuplicateRelations(relations) {
     return [...new Set(relations)];
 }

package/dist/generator/valibot/config/index.d.ts

@@ -1,7 +1,7 @@
-import type { Config } from '../../../common/config';
+import type { Config } from '../../../shared/config/index.js'
 /**
  * Loads the configuration from the `sizuku.json` file or returns the default configuration.
  *
  * @returns The configuration object.
  */
-export declare function getConfig(): Config;
+export declare function getConfig(): Config

package/dist/generator/valibot/config/index.js

@@ -1,19 +1,13 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.getConfig = getConfig;
-const node_fs_1 = __importDefault(require("node:fs"));
-const config_1 = require("../../../common/config");
+import fs from 'node:fs';
+import { DEFAULT_CONFIG } from '../../../shared/config/index.js';
 /**
  * Loads the configuration from the `sizuku.json` file or returns the default configuration.
  *
  * @returns The configuration object.
  */
-function getConfig() {
-    const config = node_fs_1.default.existsSync('sizuku-valibot.json')
-        ? { ...config_1.DEFAULT_CONFIG, ...JSON.parse(node_fs_1.default.readFileSync('sizuku-valibot.json', 'utf-8')) }
-        : config_1.DEFAULT_CONFIG;
+export function getConfig() {
+    const config = fs.existsSync('sizuku-valibot.json')
+        ? { ...DEFAULT_CONFIG, ...JSON.parse(fs.readFileSync('sizuku-valibot.json', 'utf-8')) }
+        : DEFAULT_CONFIG;
     return config;
 }

package/dist/generator/valibot/core/extract-schema.d.ts

@@ -1,8 +1,5 @@
-import type { Schema } from '../../../common/type';
+import type { Schema } from '../../../shared/types.js';
 /**
- * Extract schemas from lines of code
- * @function extractSchemas
- * @param lines - Lines of code
- * @returns Schemas
+ * Public API: extract schemas from code lines
  */
 export declare function extractSchemas(lines: string[]): Schema[];