@an5/orm 1.0.4 → 1.0.6

package/generator/src/index.ts

@@ -1,92 +1,95 @@
-import path from 'path';
-import { SchemaParser } from './parser';
-import { CodeGenerator } from './code-generator';
-import { MetadataGenerator } from './metadata-generator';
-import { PythonGenerator } from './python-generator';
-import { DotnetGenerator } from './dotnet-generator';
-
-import fs from 'fs';
-
-function clearGeneratedFiles(outputDir: string, extension: string) {
-  if (!fs.existsSync(outputDir)) {
-    fs.mkdirSync(outputDir, { recursive: true });
-    return;
-  }
-
-  for (const entry of fs.readdirSync(outputDir)) {
-    const fullPath = path.join(outputDir, entry);
-    const stat = fs.statSync(fullPath);
-    if (stat.isDirectory()) continue;
-    if (entry.endsWith(extension)) {
-      fs.unlinkSync(fullPath);
-    }
-  }
-}
-
-async function main() {
+import path from 'path';
+import { SchemaParser } from './parser';
+import { CodeGenerator } from './code-generator';
+import { MetadataGenerator } from './metadata-generator';
+import { PythonGenerator } from './python-generator';
+import { DotnetGenerator } from './dotnet-generator';
+
+import fs from 'fs';
+
+function clearGeneratedFiles(outputDir: string, extension: string) {
+  if (!fs.existsSync(outputDir)) {
+    fs.mkdirSync(outputDir, { recursive: true });
+    return;
+  }
+
+  for (const entry of fs.readdirSync(outputDir)) {
+    const fullPath = path.join(outputDir, entry);
+    const stat = fs.statSync(fullPath);
+    if (stat.isDirectory()) continue;
+    if (entry.endsWith(extension)) {
+      fs.unlinkSync(fullPath);
+    }
+  }
+}
+
+async function main() {
   const rootDir = process.cwd();
-  let config: any = {};
-  
-  try {
-    const configPath = path.join(rootDir, 'an5Orm.config.js');
-    if (fs.existsSync(configPath)) {
-      config = require(configPath);
-    }
-  } catch (err) {
-    console.warn('⚠️ Could not load an5Orm.config.js, using defaults.', err);
-  }
-
-  const schemaDir = path.resolve(rootDir, config.schemaDir || 'an5Schema');
-  const outputTypesDir = path.resolve(rootDir, config.outputs?.typescript?.outputDir || 'an5Client/typescript');
-  const outputMetadataPath = path.resolve(rootDir, config.outputs?.typescript?.metadataFile || 'an5Client/typescript/an5Metadata.ts');
-  const outputPythonMetadataPath = path.resolve(rootDir, config.outputs?.python?.metadataFile || 'an5Client/python/an5_metadata.py');
-  const outputDotnetDir = path.resolve(rootDir, config.outputs?.dotnet?.outputDir || 'an5Client/dotnet');
-
-  console.log('🚀 Starting ORM generation...');
-
-  try {
-    clearGeneratedFiles(outputTypesDir, '.ts');
-    clearGeneratedFiles(outputDotnetDir, '.cs');
-    if (fs.existsSync(outputMetadataPath)) {
-      fs.unlinkSync(outputMetadataPath);
-    }
-    if (fs.existsSync(outputPythonMetadataPath)) {
-      fs.unlinkSync(outputPythonMetadataPath);
-    }
-
-    const parser = new SchemaParser(schemaDir);
-    const models = await parser.parse();
-    console.log(`📦 Parsed ${models.length} models from schema.`);
-
-    const codeGen = new CodeGenerator(outputTypesDir);
-    codeGen.generate(models);
-    console.log(`✨ Generated modular types in ${outputTypesDir}`);
-
-    const metadataDir = path.dirname(outputMetadataPath);
-    if (!fs.existsSync(metadataDir)) {
-      fs.mkdirSync(metadataDir, { recursive: true });
-    }
-    const metadataGen = new MetadataGenerator(outputMetadataPath);
-    metadataGen.generate(models);
-    console.log(`✨ Generated metadata in ${outputMetadataPath}`);
-
-    const pythonDir = path.dirname(outputPythonMetadataPath);
-    if (!fs.existsSync(pythonDir)) {
-      fs.mkdirSync(pythonDir, { recursive: true });
-    }
-    const pythonGen = new PythonGenerator(outputPythonMetadataPath);
-    pythonGen.generate(models);
-    console.log(`✨ Generated Python metadata in ${outputPythonMetadataPath}`);
-
-    const dotnetGen = new DotnetGenerator(outputDotnetDir);
-    dotnetGen.generate(models);
-    console.log(`✨ Generated .NET models in ${outputDotnetDir}`);
-
-    console.log('✅ ORM generation completed successfully.');
-  } catch (error) {
-    console.error('❌ ORM generation failed:', error);
-    process.exit(1);
-  }
-}
-
-main();
+  let config: any = {};
+  
+  try {
+    let configPath = path.join(rootDir, 'an5Orm.config.js');
+    if (!fs.existsSync(configPath)) {
+      configPath = path.join(rootDir, 'an5Orm.config.cjs');
+    }
+    if (fs.existsSync(configPath)) {
+      config = require(configPath);
+    }
+  } catch (err) {
+    console.warn('⚠️ Could not load an5Orm.config.js/.cjs, using defaults.', err);
+  }
+
+  const schemaDir = path.resolve(rootDir, config.schemaDir || 'an5Schema');
+  const outputTypesDir = path.resolve(rootDir, config.outputs?.typescript?.outputDir || 'an5Client/typescript');
+  const outputMetadataPath = path.resolve(rootDir, config.outputs?.typescript?.metadataFile || 'an5Client/typescript/an5Metadata.ts');
+  const outputPythonMetadataPath = path.resolve(rootDir, config.outputs?.python?.metadataFile || 'an5Client/python/an5_metadata.py');
+  const outputDotnetDir = path.resolve(rootDir, config.outputs?.dotnet?.outputDir || 'an5Client/dotnet');
+
+  console.log('🚀 Starting ORM generation...');
+
+  try {
+    clearGeneratedFiles(outputTypesDir, '.ts');
+    clearGeneratedFiles(outputDotnetDir, '.cs');
+    if (fs.existsSync(outputMetadataPath)) {
+      fs.unlinkSync(outputMetadataPath);
+    }
+    if (fs.existsSync(outputPythonMetadataPath)) {
+      fs.unlinkSync(outputPythonMetadataPath);
+    }
+
+    const parser = new SchemaParser(schemaDir);
+    const models = await parser.parse();
+    console.log(`📦 Parsed ${models.length} models from schema.`);
+
+    const codeGen = new CodeGenerator(outputTypesDir);
+    codeGen.generate(models);
+    console.log(`✨ Generated modular types in ${outputTypesDir}`);
+
+    const metadataDir = path.dirname(outputMetadataPath);
+    if (!fs.existsSync(metadataDir)) {
+      fs.mkdirSync(metadataDir, { recursive: true });
+    }
+    const metadataGen = new MetadataGenerator(outputMetadataPath);
+    metadataGen.generate(models);
+    console.log(`✨ Generated metadata in ${outputMetadataPath}`);
+
+    const pythonDir = path.dirname(outputPythonMetadataPath);
+    if (!fs.existsSync(pythonDir)) {
+      fs.mkdirSync(pythonDir, { recursive: true });
+    }
+    const pythonGen = new PythonGenerator(outputPythonMetadataPath);
+    pythonGen.generate(models);
+    console.log(`✨ Generated Python metadata in ${outputPythonMetadataPath}`);
+
+    const dotnetGen = new DotnetGenerator(outputDotnetDir);
+    dotnetGen.generate(models);
+    console.log(`✨ Generated .NET models in ${outputDotnetDir}`);
+
+    console.log('✅ ORM generation completed successfully.');
+  } catch (error) {
+    console.error('❌ ORM generation failed:', error);
+    process.exit(1);
+  }
+}
+
+main();

package/generator/src/metadata-generator.ts

@@ -1,79 +1,99 @@
-import fs from 'fs';
-import path from 'path';
-import { Model } from './types';
-
-export class MetadataGenerator {
-  constructor(private outputPath: string) {}
-
-  public generate(models: Model[]) {
-    let metaContent = '// This file is auto-generated. Do not edit directly.\n\n';
-
-    metaContent += 'export const modelToTable: Record<string, string> = {\n';
-    for (const model of models) {
-      const props = this.getAllPropertyVariations(model.name);
-      const fullTableName = `[${model.schemaName}].[${model.tableName}]`;
-      for (const prop of props) {
-        metaContent += `  ${prop}: "${fullTableName}",\n`;
-      }
-    }
-    metaContent += '};\n\n';
-
-    metaContent += 'export const modelFields: Record<string, Record<string, { ts: string; sql: string }>> = {\n';
-    for (const model of models) {
-      const props = this.getAllPropertyVariations(model.name);
-      const fieldsStr = `{ ${model.fields.map(f => `${f.name}: { ts: "${f.type}${f.isOptional ? '?' : ''}", sql: "${f.sqlType}" }`).join(', ')} }`;
-      for (const prop of props) {
-        metaContent += `  ${prop}: ${fieldsStr},\n`;
-      }
-    }
-    metaContent += '};\n\n';
-
-    metaContent += 'export interface RelationDef {\n  modelName: string;\n  relationType: "many" | "one";\n  foreignKey: string;\n  localKey: string;\n}\n\n';
-
-    metaContent += 'export const relationMap: Record<string, Record<string, RelationDef>> = {\n';
-    for (const model of models) {
-      const props = this.getAllPropertyVariations(model.name);
-      let relationsContent = `  {\n`;
-      for (const rel of model.relations) {
-        relationsContent += `    ${rel.name}: { modelName: "${this.toCamelCase(rel.type)}", relationType: "${rel.isArray ? 'many' : 'one'}", foreignKey: "${rel.foreignKey || 'id'}", localKey: "${rel.localKey || 'id'}" },\n`;
-      }
-      relationsContent += `  }`;
-
-      for (const prop of props) {
-        metaContent += `  ${prop}: ${relationsContent},\n`;
-      }
-    }
-    metaContent += '};\n';
-
-    fs.writeFileSync(this.outputPath, metaContent);
-  }
-
-  private getAllPropertyVariations(modelName: string): string[] {
-    const variations = new Set<string>();
-    
-    // 1. Standard camelCase (e.g. User -> user, McpServer -> mcpServer)
-    variations.add(this.toCamelCase(modelName));
-
-    // 2. Handle known acronyms at start (e.g. LLMProvider -> lLMProvider)
-    // This is already handled by toCamelCase if it only lowercases the first letter.
-    // But we might want 'llmProvider' as well.
-    const acronyms = ['LLM', 'AI', 'MCP', 'IT', 'QC', 'HR', 'MR', 'WH', 'SSIS', 'API', 'URL', 'ID', 'JSON'];
-    
-    for (const acronym of acronyms) {
-      if (modelName.startsWith(acronym)) {
-        // e.g. LLMProvider -> llmProvider
-        variations.add(acronym.toLowerCase() + modelName.slice(acronym.length));
-        
-        // e.g. LLMProvider -> lLMProvider (Prisma style)
-        variations.add(acronym[0].toLowerCase() + acronym.slice(1) + modelName.slice(acronym.length));
-      }
-    }
-
-    return Array.from(variations);
-  }
-
-  private toCamelCase(str: string): string {
-    if (!str) return '';
-    return str[0].toLowerCase() + str.slice(1);
-  }
-}
+import fs from 'fs';
+import path from 'path';
+import { Model } from './types';
+
+export class MetadataGenerator {
+  constructor(private outputPath: string) {}
+
+  public generate(models: Model[]) {
+    let metaContent = '// This file is auto-generated. Do not edit directly.\n\n';
+
+    metaContent += 'export const modelToTable: Record<string, string> = {\n';
+    for (const model of models) {
+      const props = this.getAllPropertyVariations(model.name);
+      const fullTableName = `[${model.schemaName}].[${model.tableName}]`;
+      for (const prop of props) {
+        metaContent += `  ${prop}: "${fullTableName}",\n`;
+      }
+    }
+    metaContent += '};\n\n';
+
+    metaContent += 'export const modelDescriptions: Record<string, string | undefined> = {\n';
+    for (const model of models) {
+      const props = this.getAllPropertyVariations(model.name);
+      for (const prop of props) {
+        metaContent += `  ${prop}: ${model.description ? JSON.stringify(model.description) : 'undefined'},\n`;
+      }
+    }
+    metaContent += '};\n\n';
+
+    metaContent += 'export const modelFields: Record<string, Record<string, { ts: string; sql: string; description?: string }>> = {\n';
+    for (const model of models) {
+      const props = this.getAllPropertyVariations(model.name);
+      const fieldsStr = `{ ${model.fields.map(f => `${f.name}: ${this.formatFieldMetadata(f)}`).join(', ')} }`;
+      for (const prop of props) {
+        metaContent += `  ${prop}: ${fieldsStr},\n`;
+      }
+    }
+    metaContent += '};\n\n';
+
+    metaContent += 'export interface RelationDef {\n  modelName: string;\n  relationType: "many" | "one";\n  foreignKey: string;\n  localKey: string;\n}\n\n';
+
+    metaContent += 'export const relationMap: Record<string, Record<string, RelationDef>> = {\n';
+    for (const model of models) {
+      const props = this.getAllPropertyVariations(model.name);
+      let relationsContent = `  {\n`;
+      for (const rel of model.relations) {
+        relationsContent += `    ${rel.name}: { modelName: "${this.toCamelCase(rel.type)}", relationType: "${rel.isArray ? 'many' : 'one'}", foreignKey: "${rel.foreignKey || 'id'}", localKey: "${rel.localKey || 'id'}" },\n`;
+      }
+      relationsContent += `  }`;
+
+      for (const prop of props) {
+        metaContent += `  ${prop}: ${relationsContent},\n`;
+      }
+    }
+    metaContent += '};\n';
+
+    fs.writeFileSync(this.outputPath, metaContent);
+  }
+
+  private getAllPropertyVariations(modelName: string): string[] {
+    const variations = new Set<string>();
+    
+    // 1. Standard camelCase (e.g. User -> user, McpServer -> mcpServer)
+    variations.add(this.toCamelCase(modelName));
+
+    // 2. Handle known acronyms at start (e.g. LLMProvider -> lLMProvider)
+    // This is already handled by toCamelCase if it only lowercases the first letter.
+    // But we might want 'llmProvider' as well.
+    const acronyms = ['LLM', 'AI', 'MCP', 'IT', 'QC', 'HR', 'MR', 'WH', 'SSIS', 'API', 'URL', 'ID', 'JSON'];
+    
+    for (const acronym of acronyms) {
+      if (modelName.startsWith(acronym)) {
+        // e.g. LLMProvider -> llmProvider
+        variations.add(acronym.toLowerCase() + modelName.slice(acronym.length));
+        
+        // e.g. LLMProvider -> lLMProvider (Prisma style)
+        variations.add(acronym[0].toLowerCase() + acronym.slice(1) + modelName.slice(acronym.length));
+      }
+    }
+
+    return Array.from(variations);
+  }
+
+  private formatFieldMetadata(field: Model['fields'][number]): string {
+    const entries = [
+      `ts: ${JSON.stringify(`${field.type}${field.isOptional ? '?' : ''}`)}`,
+      `sql: ${JSON.stringify(field.sqlType)}`,
+    ];
+    if (field.description) {
+      entries.push(`description: ${JSON.stringify(field.description)}`);
+    }
+    return `{ ${entries.join(', ')} }`;
+  }
+
+  private toCamelCase(str: string): string {
+    if (!str) return '';
+    return str[0].toLowerCase() + str.slice(1);
+  }
+}