@diagramers/cli 1.0.16 → 1.0.18

package/dist/services/relation-generator.js

@@ -0,0 +1,212 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateRelation = generateRelation;
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+async function generateRelation(table1, table2, relationType = 'one-to-one') {
+    // Validate table names
+    if (!table1 || !table2 || !/^[a-zA-Z][a-zA-Z0-9_-]*$/.test(table1) || !/^[a-zA-Z][a-zA-Z0-9_-]*$/.test(table2)) {
+        throw new Error('Invalid table names. Use only letters, numbers, hyphens, and underscores. Must start with a letter.');
+    }
+    const table1Capitalized = table1.charAt(0).toUpperCase() + table1.slice(1);
+    const table2Capitalized = table2.charAt(0).toUpperCase() + table2.slice(1);
+    const currentDir = process.cwd();
+    // Check for required API project structure
+    const requiredDirs = ['src/entities', 'src/schemas'];
+    const missingDirs = requiredDirs.filter(dir => !fs.existsSync(path.join(currentDir, dir)));
+    if (missingDirs.length > 0) {
+        throw new Error(`This command should be run from a diagramers API project. Missing directories: ${missingDirs.join(', ')}`);
+    }
+    // Check if both tables exist
+    const table1EntityPath = path.join(currentDir, 'src/entities', `${table1}.ts`);
+    const table2EntityPath = path.join(currentDir, 'src/entities', `${table2}.ts`);
+    const table1SchemaPath = path.join(currentDir, 'src/schemas', `${table1}.ts`);
+    const table2SchemaPath = path.join(currentDir, 'src/schemas', `${table2}.ts`);
+    if (!fs.existsSync(table1EntityPath) || !fs.existsSync(table2EntityPath)) {
+        throw new Error(`Both tables must exist before creating relationships. Missing: ${!fs.existsSync(table1EntityPath) ? table1 : ''} ${!fs.existsSync(table2EntityPath) ? table2 : ''}`);
+    }
+    console.log(`🔗 Creating ${relationType} relationship between ${table1} and ${table2}...`);
+    // Update entities
+    console.log('📝 Updating entities...');
+    updateEntity(table1EntityPath, table1Capitalized, table2Capitalized, relationType, 'forward');
+    updateEntity(table2EntityPath, table2Capitalized, table1Capitalized, relationType, 'reverse');
+    // Update schemas
+    console.log('📋 Updating schemas...');
+    updateSchema(table1SchemaPath, table1Capitalized, table2Capitalized, relationType, 'forward');
+    updateSchema(table2SchemaPath, table2Capitalized, table1Capitalized, relationType, 'reverse');
+    console.log('✅ Relationship created successfully!');
+    console.log(`🔗 ${table1} ↔ ${table2} (${relationType})`);
+}
+function updateEntity(entityPath, tableCapitalized, relatedTableCapitalized, relationType, direction) {
+    let entityContent = fs.readFileSync(entityPath, 'utf8');
+    // Add import for the related interface
+    const importStatement = `import { I${relatedTableCapitalized} } from './${direction === 'forward' ? relatedTableCapitalized.toLowerCase() : relatedTableCapitalized.toLowerCase()}';`;
+    if (!entityContent.includes(importStatement)) {
+        // Find the last import statement and add after it
+        const importRegex = /import.*from.*['"];?\s*$/gm;
+        const matches = [...entityContent.matchAll(importRegex)];
+        if (matches.length > 0) {
+            const lastImport = matches[matches.length - 1];
+            const insertIndex = lastImport.index + lastImport[0].length;
+            entityContent = entityContent.slice(0, insertIndex) + '\n' + importStatement + entityContent.slice(insertIndex);
+        }
+    }
+    // Add relationship fields based on type
+    const interfaceRegex = /export interface I\w+ extends mongoose\.Document \{([^}]*)\}/s;
+    const match = entityContent.match(interfaceRegex);
+    if (match) {
+        const interfaceContent = match[1];
+        const newFields = generateEntityFields(tableCapitalized, relatedTableCapitalized, relationType, direction);
+        // Add new fields before the closing brace
+        const updatedInterfaceContent = interfaceContent.trim() + '\n    ' + newFields;
+        entityContent = entityContent.replace(interfaceRegex, `export interface I${tableCapitalized} extends mongoose.Document {$1${updatedInterfaceContent}\n}`);
+    }
+    fs.writeFileSync(entityPath, entityContent);
+}
+function updateSchema(schemaPath, tableCapitalized, relatedTableCapitalized, relationType, direction) {
+    let schemaContent = fs.readFileSync(schemaPath, 'utf8');
+    // Add import for the related schema
+    const importStatement = `import { ${relatedTableCapitalized}Entity } from './${direction === 'forward' ? relatedTableCapitalized.toLowerCase() : relatedTableCapitalized.toLowerCase()}';`;
+    if (!schemaContent.includes(importStatement)) {
+        // Find the last import statement and add after it
+        const importRegex = /import.*from.*['"];?\s*$/gm;
+        const matches = [...schemaContent.matchAll(importRegex)];
+        if (matches.length > 0) {
+            const lastImport = matches[matches.length - 1];
+            const insertIndex = lastImport.index + lastImport[0].length;
+            schemaContent = schemaContent.slice(0, insertIndex) + '\n' + importStatement + schemaContent.slice(insertIndex);
+        }
+    }
+    // Add schema fields based on relationship type
+    const schemaRegex = /export const \w+Schema = new mongoose\.Schema\(([^)]*),/s;
+    const match = schemaContent.match(schemaRegex);
+    if (match) {
+        const schemaFields = match[1];
+        const newFields = generateSchemaFields(tableCapitalized, relatedTableCapitalized, relationType, direction);
+        // Add new fields before the closing brace
+        const updatedSchemaFields = schemaFields.trim() + '\n        ' + newFields;
+        schemaContent = schemaContent.replace(schemaRegex, `export const ${tableCapitalized.toLowerCase()}Schema = new mongoose.Schema(${updatedSchemaFields},`);
+    }
+    // Add virtual fields and population methods
+    const modelRegex = /export const \w+Entity = mongoose\.model<.*>\(.*\);?\s*$/;
+    const virtualFields = generateVirtualFields(tableCapitalized, relatedTableCapitalized, relationType, direction);
+    if (virtualFields) {
+        schemaContent = schemaContent.replace(modelRegex, `${virtualFields}\n\n$&`);
+    }
+    fs.writeFileSync(schemaPath, schemaContent);
+}
+function generateEntityFields(tableCapitalized, relatedTableCapitalized, relationType, direction) {
+    switch (relationType) {
+        case 'one-to-one':
+            if (direction === 'forward') {
+                return `${relatedTableCapitalized.toLowerCase()}Id: ObjectId,\n    ${relatedTableCapitalized.toLowerCase()}?: I${relatedTableCapitalized}`;
+            }
+            else {
+                return `${tableCapitalized.toLowerCase()}Id: ObjectId,\n    ${tableCapitalized.toLowerCase()}?: I${tableCapitalized}`;
+            }
+        case 'one-to-many':
+            if (direction === 'forward') {
+                return `${relatedTableCapitalized.toLowerCase()}Id: ObjectId,\n    ${relatedTableCapitalized.toLowerCase()}?: I${relatedTableCapitalized}`;
+            }
+            else {
+                return `${tableCapitalized.toLowerCase()}s?: I${tableCapitalized}[]`;
+            }
+        case 'many-to-many':
+            if (direction === 'forward') {
+                return `${relatedTableCapitalized.toLowerCase()}Ids: ObjectId[],\n    ${relatedTableCapitalized.toLowerCase()}s?: I${relatedTableCapitalized}[]`;
+            }
+            else {
+                return `${tableCapitalized.toLowerCase()}Ids: ObjectId[],\n    ${tableCapitalized.toLowerCase()}s?: I${tableCapitalized}[]`;
+            }
+        default:
+            return '';
+    }
+}
+function generateSchemaFields(tableCapitalized, relatedTableCapitalized, relationType, direction) {
+    switch (relationType) {
+        case 'one-to-one':
+            if (direction === 'forward') {
+                return `${relatedTableCapitalized.toLowerCase()}Id: {\n            type: mongoose.SchemaTypes.ObjectId,\n            ref: '${relatedTableCapitalized.toLowerCase()}',\n            required: false\n        }`;
+            }
+            else {
+                return `${tableCapitalized.toLowerCase()}Id: {\n            type: mongoose.SchemaTypes.ObjectId,\n            ref: '${tableCapitalized.toLowerCase()}',\n            required: false\n        }`;
+            }
+        case 'one-to-many':
+            if (direction === 'forward') {
+                return `${relatedTableCapitalized.toLowerCase()}Id: {\n            type: mongoose.SchemaTypes.ObjectId,\n            ref: '${relatedTableCapitalized.toLowerCase()}',\n            required: false\n        }`;
+            }
+            else {
+                return ''; // No field needed for reverse one-to-many
+            }
+        case 'many-to-many':
+            if (direction === 'forward') {
+                return `${relatedTableCapitalized.toLowerCase()}Ids: [{\n            type: mongoose.SchemaTypes.ObjectId,\n            ref: '${relatedTableCapitalized.toLowerCase()}'\n        }]`;
+            }
+            else {
+                return `${tableCapitalized.toLowerCase()}Ids: [{\n            type: mongoose.SchemaTypes.ObjectId,\n            ref: '${tableCapitalized.toLowerCase()}'\n        }]`;
+            }
+        default:
+            return '';
+    }
+}
+function generateVirtualFields(tableCapitalized, relatedTableCapitalized, relationType, direction) {
+    switch (relationType) {
+        case 'one-to-one':
+            if (direction === 'forward') {
+                return `// Virtual populate for ${relatedTableCapitalized.toLowerCase()}\n${tableCapitalized.toLowerCase()}Schema.virtual('${relatedTableCapitalized.toLowerCase()}', {\n    ref: '${relatedTableCapitalized.toLowerCase()}',\n    localField: '${relatedTableCapitalized.toLowerCase()}Id',\n    foreignField: '_id',\n    justOne: true\n});`;
+            }
+            else {
+                return `// Virtual populate for ${tableCapitalized.toLowerCase()}\n${tableCapitalized.toLowerCase()}Schema.virtual('${tableCapitalized.toLowerCase()}', {\n    ref: '${tableCapitalized.toLowerCase()}',\n    localField: '${tableCapitalized.toLowerCase()}Id',\n    foreignField: '_id',\n    justOne: true\n});`;
+            }
+        case 'one-to-many':
+            if (direction === 'forward') {
+                return `// Virtual populate for ${relatedTableCapitalized.toLowerCase()}\n${tableCapitalized.toLowerCase()}Schema.virtual('${relatedTableCapitalized.toLowerCase()}', {\n    ref: '${relatedTableCapitalized.toLowerCase()}',\n    localField: '${relatedTableCapitalized.toLowerCase()}Id',\n    foreignField: '_id',\n    justOne: true\n});`;
+            }
+            else {
+                return `// Virtual populate for ${tableCapitalized.toLowerCase()}s\n${tableCapitalized.toLowerCase()}Schema.virtual('${tableCapitalized.toLowerCase()}s', {\n    ref: '${tableCapitalized.toLowerCase()}',\n    localField: '_id',\n    foreignField: '${tableCapitalized.toLowerCase()}Id'\n});`;
+            }
+        case 'many-to-many':
+            if (direction === 'forward') {
+                return `// Virtual populate for ${relatedTableCapitalized.toLowerCase()}s\n${tableCapitalized.toLowerCase()}Schema.virtual('${relatedTableCapitalized.toLowerCase()}s', {\n    ref: '${relatedTableCapitalized.toLowerCase()}',\n    localField: '${relatedTableCapitalized.toLowerCase()}Ids',\n    foreignField: '_id'\n});`;
+            }
+            else {
+                return `// Virtual populate for ${tableCapitalized.toLowerCase()}s\n${tableCapitalized.toLowerCase()}Schema.virtual('${tableCapitalized.toLowerCase()}s', {\n    ref: '${tableCapitalized.toLowerCase()}',\n    localField: '${tableCapitalized.toLowerCase()}Ids',\n    foreignField: '_id'\n});`;
+            }
+        default:
+            return '';
+    }
+}
+//# sourceMappingURL=relation-generator.js.map

package/dist/services/relation-generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"relation-generator.js","sourceRoot":"","sources":["../../src/services/relation-generator.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAKA,4CAyCC;AA9CD,uCAAyB;AACzB,2CAA6B;AAItB,KAAK,UAAU,gBAAgB,CAAC,MAAc,EAAE,MAAc,EAAE,eAA6B,YAAY;IAC9G,uBAAuB;IACvB,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,IAAI,CAAC,0BAA0B,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,0BAA0B,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;QAC/G,MAAM,IAAI,KAAK,CAAC,qGAAqG,CAAC,CAAC;IACzH,CAAC;IAED,MAAM,iBAAiB,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAC3E,MAAM,iBAAiB,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAC3E,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAEjC,2CAA2C;IAC3C,MAAM,YAAY,GAAG,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;IACrD,MAAM,WAAW,GAAG,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;IAC3F,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,kFAAkF,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC9H,CAAC;IAED,6BAA6B;IAC7B,MAAM,gBAAgB,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,cAAc,EAAE,GAAG,MAAM,KAAK,CAAC,CAAC;IAC/E,MAAM,gBAAgB,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,cAAc,EAAE,GAAG,MAAM,KAAK,CAAC,CAAC;IAC/E,MAAM,gBAAgB,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,aAAa,EAAE,GAAG,MAAM,KAAK,CAAC,CAAC;IAC9E,MAAM,gBAAgB,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,aAAa,EAAE,GAAG,MAAM,KAAK,CAAC,CAAC;IAE9E,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,gBAAgB,CAAC,EAAE,CAAC;QACzE,MAAM,IAAI,KAAK,CAAC,kEAAkE,CAAC,EAAE,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IACxL,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,eAAe,YAAY,yBAAyB,MAAM,QAAQ,MAAM,KAAK,CAAC,CAAC;IAE3F,kBAAkB;IAClB,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;IACvC,YAAY,CAAC,gBAAgB,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;IAC9F,YAAY,CAAC,gBAAgB,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;IAE9F,iBAAiB;IACjB,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;IACtC,YAAY,CAAC,gBAAgB,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;IAC9F,YAAY,CAAC,gBAAgB,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;IAE9F,OAAO,CAAC,GAAG,CAAC,sCAAsC,CAAC,CAAC;IACpD,OAAO,CAAC,GAAG,CAAC,MAAM,MAAM,MAAM,MAAM,KAAK,YAAY,GAAG,CAAC,CAAC;AAC5D,CAAC;AAED,SAAS,YAAY,CAAC,UAAkB,EAAE,gBAAwB,EAAE,uBAA+B,EAAE,YAA0B,EAAE,SAAgC;IAC/J,IAAI,aAAa,GAAG,EAAE,CAAC,YAAY,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAExD,uCAAuC;IACvC,MAAM,eAAe,GAAG,aAAa,uBAAuB,cAAc,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,uBAAuB,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,uBAAuB,CAAC,WAAW,EAAE,IAAI,CAAC;IACtL,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;QAC7C,kDAAkD;QAClD,MAAM,WAAW,GAAG,4BAA4B,CAAC;QACjD,MAAM,OAAO,GAAG,CAAC,GAAG,aAAa,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC;QACzD,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC/C,MAAM,WAAW,GAAG,UAAU,CAAC,KAAM,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;YAC7D,aAAa,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC,GAAG,IAAI,GAAG,eAAe,GAAG,aAAa,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QAClH,CAAC;IACH,CAAC;IAED,wCAAwC;IACxC,MAAM,cAAc,GAAG,+DAA+D,CAAC;IACvF,MAAM,KAAK,GAAG,aAAa,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IAElD,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,gBAAgB,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAClC,MAAM,SAAS,GAAG,oBAAoB,CAAC,gBAAgB,EAAE,uBAAuB,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;QAE3G,0CAA0C;QAC1C,MAAM,uBAAuB,GAAG,gBAAgB,CAAC,IAAI,EAAE,GAAG,QAAQ,GAAG,SAAS,CAAC;QAC/E,aAAa,GAAG,aAAa,CAAC,OAAO,CAAC,cAAc,EAAE,qBAAqB,gBAAgB,iCAAiC,uBAAuB,KAAK,CAAC,CAAC;IAC5J,CAAC;IAED,EAAE,CAAC,aAAa,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;AAC9C,CAAC;AAED,SAAS,YAAY,CAAC,UAAkB,EAAE,gBAAwB,EAAE,uBAA+B,EAAE,YAA0B,EAAE,SAAgC;IAC/J,IAAI,aAAa,GAAG,EAAE,CAAC,YAAY,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAExD,oCAAoC;IACpC,MAAM,eAAe,GAAG,YAAY,uBAAuB,oBAAoB,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,uBAAuB,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,uBAAuB,CAAC,WAAW,EAAE,IAAI,CAAC;IAC3L,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;QAC7C,kDAAkD;QAClD,MAAM,WAAW,GAAG,4BAA4B,CAAC;QACjD,MAAM,OAAO,GAAG,CAAC,GAAG,aAAa,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC;QACzD,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC/C,MAAM,WAAW,GAAG,UAAU,CAAC,KAAM,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;YAC7D,aAAa,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC,GAAG,IAAI,GAAG,eAAe,GAAG,aAAa,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QAClH,CAAC;IACH,CAAC;IAED,+CAA+C;IAC/C,MAAM,WAAW,GAAG,0DAA0D,CAAC;IAC/E,MAAM,KAAK,GAAG,aAAa,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;IAE/C,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,YAAY,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,SAAS,GAAG,oBAAoB,CAAC,gBAAgB,EAAE,uBAAuB,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;QAE3G,0CAA0C;QAC1C,MAAM,mBAAmB,GAAG,YAAY,CAAC,IAAI,EAAE,GAAG,YAAY,GAAG,SAAS,CAAC;QAC3E,aAAa,GAAG,aAAa,CAAC,OAAO,CAAC,WAAW,EAAE,gBAAgB,gBAAgB,CAAC,WAAW,EAAE,gCAAgC,mBAAmB,GAAG,CAAC,CAAC;IAC3J,CAAC;IAED,4CAA4C;IAC5C,MAAM,UAAU,GAAG,0DAA0D,CAAC;IAC9E,MAAM,aAAa,GAAG,qBAAqB,CAAC,gBAAgB,EAAE,uBAAuB,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;IAEhH,IAAI,aAAa,EAAE,CAAC;QAClB,aAAa,GAAG,aAAa,CAAC,OAAO,CAAC,UAAU,EAAE,GAAG,aAAa,QAAQ,CAAC,CAAC;IAC9E,CAAC;IAED,EAAE,CAAC,aAAa,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;AAC9C,CAAC;AAED,SAAS,oBAAoB,CAAC,gBAAwB,EAAE,uBAA+B,EAAE,YAA0B,EAAE,SAAgC;IACnJ,QAAQ,YAAY,EAAE,CAAC;QACrB,KAAK,YAAY;YACf,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC5B,OAAO,GAAG,uBAAuB,CAAC,WAAW,EAAE,sBAAsB,uBAAuB,CAAC,WAAW,EAAE,OAAO,uBAAuB,EAAE,CAAC;YAC7I,CAAC;iBAAM,CAAC;gBACN,OAAO,GAAG,gBAAgB,CAAC,WAAW,EAAE,sBAAsB,gBAAgB,CAAC,WAAW,EAAE,OAAO,gBAAgB,EAAE,CAAC;YACxH,CAAC;QAEH,KAAK,aAAa;YAChB,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC5B,OAAO,GAAG,uBAAuB,CAAC,WAAW,EAAE,sBAAsB,uBAAuB,CAAC,WAAW,EAAE,OAAO,uBAAuB,EAAE,CAAC;YAC7I,CAAC;iBAAM,CAAC;gBACN,OAAO,GAAG,gBAAgB,CAAC,WAAW,EAAE,QAAQ,gBAAgB,IAAI,CAAC;YACvE,CAAC;QAEH,KAAK,cAAc;YACjB,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC5B,OAAO,GAAG,uBAAuB,CAAC,WAAW,EAAE,yBAAyB,uBAAuB,CAAC,WAAW,EAAE,QAAQ,uBAAuB,IAAI,CAAC;YACnJ,CAAC;iBAAM,CAAC;gBACN,OAAO,GAAG,gBAAgB,CAAC,WAAW,EAAE,yBAAyB,gBAAgB,CAAC,WAAW,EAAE,QAAQ,gBAAgB,IAAI,CAAC;YAC9H,CAAC;QAEH;YACE,OAAO,EAAE,CAAC;IACd,CAAC;AACH,CAAC;AAED,SAAS,oBAAoB,CAAC,gBAAwB,EAAE,uBAA+B,EAAE,YAA0B,EAAE,SAAgC;IACnJ,QAAQ,YAAY,EAAE,CAAC;QACrB,KAAK,YAAY;YACf,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC5B,OAAO,GAAG,uBAAuB,CAAC,WAAW,EAAE,8EAA8E,uBAAuB,CAAC,WAAW,EAAE,4CAA4C,CAAC;YACjN,CAAC;iBAAM,CAAC;gBACN,OAAO,GAAG,gBAAgB,CAAC,WAAW,EAAE,8EAA8E,gBAAgB,CAAC,WAAW,EAAE,4CAA4C,CAAC;YACnM,CAAC;QAEH,KAAK,aAAa;YAChB,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC5B,OAAO,GAAG,uBAAuB,CAAC,WAAW,EAAE,8EAA8E,uBAAuB,CAAC,WAAW,EAAE,4CAA4C,CAAC;YACjN,CAAC;iBAAM,CAAC;gBACN,OAAO,EAAE,CAAC,CAAC,0CAA0C;YACvD,CAAC;QAEH,KAAK,cAAc;YACjB,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC5B,OAAO,GAAG,uBAAuB,CAAC,WAAW,EAAE,gFAAgF,uBAAuB,CAAC,WAAW,EAAE,eAAe,CAAC;YACtL,CAAC;iBAAM,CAAC;gBACN,OAAO,GAAG,gBAAgB,CAAC,WAAW,EAAE,gFAAgF,gBAAgB,CAAC,WAAW,EAAE,eAAe,CAAC;YACxK,CAAC;QAEH;YACE,OAAO,EAAE,CAAC;IACd,CAAC;AACH,CAAC;AAED,SAAS,qBAAqB,CAAC,gBAAwB,EAAE,uBAA+B,EAAE,YAA0B,EAAE,SAAgC;IACpJ,QAAQ,YAAY,EAAE,CAAC;QACrB,KAAK,YAAY;YACf,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC5B,OAAO,2BAA2B,uBAAuB,CAAC,WAAW,EAAE,KAAK,gBAAgB,CAAC,WAAW,EAAE,mBAAmB,uBAAuB,CAAC,WAAW,EAAE,mBAAmB,uBAAuB,CAAC,WAAW,EAAE,wBAAwB,uBAAuB,CAAC,WAAW,EAAE,wDAAwD,CAAC;YAClV,CAAC;iBAAM,CAAC;gBACN,OAAO,2BAA2B,gBAAgB,CAAC,WAAW,EAAE,KAAK,gBAAgB,CAAC,WAAW,EAAE,mBAAmB,gBAAgB,CAAC,WAAW,EAAE,mBAAmB,gBAAgB,CAAC,WAAW,EAAE,wBAAwB,gBAAgB,CAAC,WAAW,EAAE,wDAAwD,CAAC;YACtT,CAAC;QAEH,KAAK,aAAa;YAChB,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC5B,OAAO,2BAA2B,uBAAuB,CAAC,WAAW,EAAE,KAAK,gBAAgB,CAAC,WAAW,EAAE,mBAAmB,uBAAuB,CAAC,WAAW,EAAE,mBAAmB,uBAAuB,CAAC,WAAW,EAAE,wBAAwB,uBAAuB,CAAC,WAAW,EAAE,wDAAwD,CAAC;YAClV,CAAC;iBAAM,CAAC;gBACN,OAAO,2BAA2B,gBAAgB,CAAC,WAAW,EAAE,MAAM,gBAAgB,CAAC,WAAW,EAAE,mBAAmB,gBAAgB,CAAC,WAAW,EAAE,oBAAoB,gBAAgB,CAAC,WAAW,EAAE,kDAAkD,gBAAgB,CAAC,WAAW,EAAE,UAAU,CAAC;YACpS,CAAC;QAEH,KAAK,cAAc;YACjB,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC5B,OAAO,2BAA2B,uBAAuB,CAAC,WAAW,EAAE,MAAM,gBAAgB,CAAC,WAAW,EAAE,mBAAmB,uBAAuB,CAAC,WAAW,EAAE,oBAAoB,uBAAuB,CAAC,WAAW,EAAE,wBAAwB,uBAAuB,CAAC,WAAW,EAAE,qCAAqC,CAAC;YACjU,CAAC;iBAAM,CAAC;gBACN,OAAO,2BAA2B,gBAAgB,CAAC,WAAW,EAAE,MAAM,gBAAgB,CAAC,WAAW,EAAE,mBAAmB,gBAAgB,CAAC,WAAW,EAAE,oBAAoB,gBAAgB,CAAC,WAAW,EAAE,wBAAwB,gBAAgB,CAAC,WAAW,EAAE,qCAAqC,CAAC;YACrS,CAAC;QAEH;YACE,OAAO,EAAE,CAAC;IACd,CAAC;AACH,CAAC"}

package/dist/services/table-generator.d.ts

@@ -0,0 +1,2 @@
+export declare function generateTable(tableName: string): Promise<void>;
+//# sourceMappingURL=table-generator.d.ts.map

package/dist/services/table-generator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"table-generator.d.ts","sourceRoot":"","sources":["../../src/services/table-generator.ts"],"names":[],"mappings":"AAGA,wBAAsB,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAkCpE"}

package/dist/services/table-generator.js

@@ -0,0 +1,137 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateTable = generateTable;
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+async function generateTable(tableName) {
+    // Validate table name
+    if (!tableName || !/^[a-zA-Z][a-zA-Z0-9_-]*$/.test(tableName)) {
+        throw new Error('Invalid table name. Use only letters, numbers, hyphens, and underscores. Must start with a letter.');
+    }
+    const tableNameCapitalized = tableName.charAt(0).toUpperCase() + tableName.slice(1);
+    const currentDir = process.cwd();
+    // Check for required API project structure
+    const requiredDirs = [
+        'src/entities',
+        'src/schemas'
+    ];
+    const missingDirs = requiredDirs.filter(dir => !fs.existsSync(path.join(currentDir, dir)));
+    if (missingDirs.length > 0) {
+        throw new Error(`This command should be run from a diagramers API project. Missing directories: ${missingDirs.join(', ')}`);
+    }
+    console.log('📝 Creating entity...');
+    const entityContent = generateEntityContent(tableName, tableNameCapitalized);
+    fs.writeFileSync(path.join(currentDir, 'src/entities', `${tableName}.ts`), entityContent);
+    console.log('📋 Creating schema...');
+    const schemaContent = generateSchemaContent(tableName, tableNameCapitalized);
+    fs.writeFileSync(path.join(currentDir, 'src/schemas', `${tableName}.ts`), schemaContent);
+    // Update dbcontext to include the new entity
+    updateDbContext(currentDir, tableName, tableNameCapitalized);
+    console.log('✅ Table generation completed!');
+    console.log(`📊 Entity: src/entities/${tableName}.ts`);
+    console.log(`📋 Schema: src/schemas/${tableName}.ts`);
+    console.log('🔄 dbcontext.ts updated');
+}
+function generateEntityContent(tableName, tableNameCapitalized) {
+    return `import * as mongoose from 'mongoose';
+import { ObjectId } from "bson";
+
+export interface I${tableNameCapitalized} extends mongoose.Document {
+    _id: ObjectId,
+    name: string,
+    description?: string,
+    status: number,
+    createdAt: Date,
+    updatedAt: Date
+}`;
+}
+function generateSchemaContent(tableName, tableNameCapitalized) {
+    return `import * as mongoose from 'mongoose';
+import { I${tableNameCapitalized} } from '../entities/${tableName}';
+
+export const ${tableName}Schema = new mongoose.Schema(
+    {
+        name: {
+            type: mongoose.SchemaTypes.String,
+            required: true,
+        },
+        description: {
+            type: mongoose.SchemaTypes.String,
+            required: false,
+        },
+        status: {
+            type: mongoose.SchemaTypes.Number,
+            default: 1,
+        }
+    },
+    { timestamps: true, suppressReservedKeysWarning: true },
+);
+
+export const ${tableNameCapitalized}Entity = mongoose.model<I${tableNameCapitalized}>('${tableName}', ${tableName}Schema);`;
+}
+function updateDbContext(currentDir, tableName, tableNameCapitalized) {
+    const dbcontextPath = path.join(currentDir, 'src/helpers/dbcontext.ts');
+    if (!fs.existsSync(dbcontextPath)) {
+        console.log('⚠️ dbcontext.ts not found, skipping dbcontext update');
+        return;
+    }
+    let dbcontextContent = fs.readFileSync(dbcontextPath, 'utf8');
+    // Add import
+    const importStatement = `import { ${tableNameCapitalized}Entity } from '../schemas/${tableName}';`;
+    if (!dbcontextContent.includes(importStatement)) {
+        // Find the last import statement and add after it
+        const importRegex = /import.*from.*['"];?\s*$/gm;
+        const matches = [...dbcontextContent.matchAll(importRegex)];
+        if (matches.length > 0) {
+            const lastImport = matches[matches.length - 1];
+            const insertIndex = lastImport.index + lastImport[0].length;
+            dbcontextContent = dbcontextContent.slice(0, insertIndex) + '\n' + importStatement + dbcontextContent.slice(insertIndex);
+        }
+    }
+    // Add to dbcontext object
+    const dbcontextObjectRegex = /export\s+default\s*\{([^}]*)\}/s;
+    const match = dbcontextContent.match(dbcontextObjectRegex);
+    if (match) {
+        const objectContent = match[1];
+        if (!objectContent.includes(`${tableNameCapitalized}Entity`)) {
+            const newObjectContent = objectContent.trim() + `,\n    ${tableNameCapitalized}Entity`;
+            dbcontextContent = dbcontextContent.replace(dbcontextObjectRegex, `export default {$1${newObjectContent}\n}`);
+        }
+    }
+    fs.writeFileSync(dbcontextPath, dbcontextContent);
+    console.log('📊 Updated dbcontext.ts');
+}
+//# sourceMappingURL=table-generator.js.map

package/dist/services/table-generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"table-generator.js","sourceRoot":"","sources":["../../src/services/table-generator.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAGA,sCAkCC;AArCD,uCAAyB;AACzB,2CAA6B;AAEtB,KAAK,UAAU,aAAa,CAAC,SAAiB;IACnD,sBAAsB;IACtB,IAAI,CAAC,SAAS,IAAI,CAAC,0BAA0B,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;QAC9D,MAAM,IAAI,KAAK,CAAC,oGAAoG,CAAC,CAAC;IACxH,CAAC;IAED,MAAM,oBAAoB,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACpF,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAEjC,2CAA2C;IAC3C,MAAM,YAAY,GAAG;QACnB,cAAc;QACd,aAAa;KACd,CAAC;IACF,MAAM,WAAW,GAAG,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;IAC3F,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,kFAAkF,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC9H,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IACrC,MAAM,aAAa,GAAG,qBAAqB,CAAC,SAAS,EAAE,oBAAoB,CAAC,CAAC;IAC7E,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,cAAc,EAAE,GAAG,SAAS,KAAK,CAAC,EAAE,aAAa,CAAC,CAAC;IAE1F,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IACrC,MAAM,aAAa,GAAG,qBAAqB,CAAC,SAAS,EAAE,oBAAoB,CAAC,CAAC;IAC7E,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,aAAa,EAAE,GAAG,SAAS,KAAK,CAAC,EAAE,aAAa,CAAC,CAAC;IAEzF,6CAA6C;IAC7C,eAAe,CAAC,UAAU,EAAE,SAAS,EAAE,oBAAoB,CAAC,CAAC;IAE7D,OAAO,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAC;IAC7C,OAAO,CAAC,GAAG,CAAC,2BAA2B,SAAS,KAAK,CAAC,CAAC;IACvD,OAAO,CAAC,GAAG,CAAC,0BAA0B,SAAS,KAAK,CAAC,CAAC;IACtD,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;AACzC,CAAC;AAED,SAAS,qBAAqB,CAAC,SAAiB,EAAE,oBAA4B;IAC5E,OAAO;;;oBAGW,oBAAoB;;;;;;;EAOtC,CAAC;AACH,CAAC;AAED,SAAS,qBAAqB,CAAC,SAAiB,EAAE,oBAA4B;IAC5E,OAAO;YACG,oBAAoB,wBAAwB,SAAS;;eAElD,SAAS;;;;;;;;;;;;;;;;;;eAkBT,oBAAoB,4BAA4B,oBAAoB,MAAM,SAAS,MAAM,SAAS,UAAU,CAAC;AAC5H,CAAC;AAED,SAAS,eAAe,CAAC,UAAkB,EAAE,SAAiB,EAAE,oBAA4B;IAC1F,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAC;IACxE,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;QAClC,OAAO,CAAC,GAAG,CAAC,sDAAsD,CAAC,CAAC;QACpE,OAAO;IACT,CAAC;IAED,IAAI,gBAAgB,GAAG,EAAE,CAAC,YAAY,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;IAE9D,aAAa;IACb,MAAM,eAAe,GAAG,YAAY,oBAAoB,6BAA6B,SAAS,IAAI,CAAC;IACnG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;QAChD,kDAAkD;QAClD,MAAM,WAAW,GAAG,4BAA4B,CAAC;QACjD,MAAM,OAAO,GAAG,CAAC,GAAG,gBAAgB,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC;QAC5D,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC/C,MAAM,WAAW,GAAG,UAAU,CAAC,KAAM,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;YAC7D,gBAAgB,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC,GAAG,IAAI,GAAG,eAAe,GAAG,gBAAgB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QAC3H,CAAC;IACH,CAAC;IAED,0BAA0B;IAC1B,MAAM,oBAAoB,GAAG,iCAAiC,CAAC;IAC/D,MAAM,KAAK,GAAG,gBAAgB,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;IAC3D,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,aAAa,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAC/B,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,GAAG,oBAAoB,QAAQ,CAAC,EAAE,CAAC;YAC7D,MAAM,gBAAgB,GAAG,aAAa,CAAC,IAAI,EAAE,GAAG,UAAU,oBAAoB,QAAQ,CAAC;YACvF,gBAAgB,GAAG,gBAAgB,CAAC,OAAO,CAAC,oBAAoB,EAAE,qBAAqB,gBAAgB,KAAK,CAAC,CAAC;QAChH,CAAC;IACH,CAAC;IAED,EAAE,CAAC,aAAa,CAAC,aAAa,EAAE,gBAAgB,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;AACzC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diagramers/cli",
-  "version": "1.0.16",
+  "version": "1.0.18",
   "description": "Diagramers CLI - Command-line tools for managing Diagramers projects",
   "main": "dist/index.js",
   "bin": {

package/src/commands/api.ts

@@ -1,5 +1,7 @@
 import { Command } from 'commander';
 import { generateModule } from '../services/api-generator';
+import { generateTable } from '../services/table-generator';
+import { generateRelation } from '../services/relation-generator';
 import { processTemplate } from '../services/template-processor';
 import chalk from 'chalk';
 
@@ -38,5 +40,37 @@ export function apiCommand(program: Command) {
       }
     });
 
+  // Generate table command
+  api
+    .command('generate:table <name>')
+    .description('Generate a new database table with entity and schema only')
+    .action(async (name: string) => {
+      try {
+        console.log(chalk.blue(`🚀 Generating table: ${name}`));
+        await generateTable(name);
+        console.log(chalk.green(`✅ Table '${name}' generated successfully!`));
+      } catch (error: any) {
+        console.error(chalk.red(`❌ Error generating table: ${error.message}`));
+        process.exit(1);
+      }
+    });
+
+  // Generate relation command
+  api
+    .command('generate:relation <table1> <table2> [type]')
+    .description('Generate a relationship between two existing tables')
+    .option('-t, --type <type>', 'Relationship type: one-to-one, one-to-many, many-to-many', 'one-to-one')
+    .action(async (table1: string, table2: string, options: any) => {
+      try {
+        const relationType = options.type || 'one-to-one';
+        console.log(chalk.blue(`🔗 Creating ${relationType} relationship between ${table1} and ${table2}...`));
+        await generateRelation(table1, table2, relationType);
+        console.log(chalk.green(`✅ Relationship created successfully!`));
+      } catch (error: any) {
+        console.error(chalk.red(`❌ Error creating relationship: ${error.message}`));
+        process.exit(1);
+      }
+    });
+
   return api;
 } 

package/src/index.ts

@@ -28,6 +28,8 @@ Examples:
   $ diagramers update
   $ diagramers extend --feature auth
   $ diagramers api generate:module product
+  $ diagramers api generate:table category
+  $ diagramers api generate:relation product category one-to-many
   $ diagramers api process:template my-api-project
 `);
 

package/src/services/relation-generator.ts

@@ -0,0 +1,203 @@
+import * as fs from 'fs';
+import * as path from 'path';
+
+export type RelationType = 'one-to-one' | 'one-to-many' | 'many-to-many';
+
+export async function generateRelation(table1: string, table2: string, relationType: RelationType = 'one-to-one'): Promise<void> {
+  // Validate table names
+  if (!table1 || !table2 || !/^[a-zA-Z][a-zA-Z0-9_-]*$/.test(table1) || !/^[a-zA-Z][a-zA-Z0-9_-]*$/.test(table2)) {
+    throw new Error('Invalid table names. Use only letters, numbers, hyphens, and underscores. Must start with a letter.');
+  }
+
+  const table1Capitalized = table1.charAt(0).toUpperCase() + table1.slice(1);
+  const table2Capitalized = table2.charAt(0).toUpperCase() + table2.slice(1);
+  const currentDir = process.cwd();
+
+  // Check for required API project structure
+  const requiredDirs = ['src/entities', 'src/schemas'];
+  const missingDirs = requiredDirs.filter(dir => !fs.existsSync(path.join(currentDir, dir)));
+  if (missingDirs.length > 0) {
+    throw new Error(`This command should be run from a diagramers API project. Missing directories: ${missingDirs.join(', ')}`);
+  }
+
+  // Check if both tables exist
+  const table1EntityPath = path.join(currentDir, 'src/entities', `${table1}.ts`);
+  const table2EntityPath = path.join(currentDir, 'src/entities', `${table2}.ts`);
+  const table1SchemaPath = path.join(currentDir, 'src/schemas', `${table1}.ts`);
+  const table2SchemaPath = path.join(currentDir, 'src/schemas', `${table2}.ts`);
+
+  if (!fs.existsSync(table1EntityPath) || !fs.existsSync(table2EntityPath)) {
+    throw new Error(`Both tables must exist before creating relationships. Missing: ${!fs.existsSync(table1EntityPath) ? table1 : ''} ${!fs.existsSync(table2EntityPath) ? table2 : ''}`);
+  }
+
+  console.log(`🔗 Creating ${relationType} relationship between ${table1} and ${table2}...`);
+
+  // Update entities
+  console.log('📝 Updating entities...');
+  updateEntity(table1EntityPath, table1Capitalized, table2Capitalized, relationType, 'forward');
+  updateEntity(table2EntityPath, table2Capitalized, table1Capitalized, relationType, 'reverse');
+
+  // Update schemas
+  console.log('📋 Updating schemas...');
+  updateSchema(table1SchemaPath, table1Capitalized, table2Capitalized, relationType, 'forward');
+  updateSchema(table2SchemaPath, table2Capitalized, table1Capitalized, relationType, 'reverse');
+
+  console.log('✅ Relationship created successfully!');
+  console.log(`🔗 ${table1} ↔ ${table2} (${relationType})`);
+}
+
+function updateEntity(entityPath: string, tableCapitalized: string, relatedTableCapitalized: string, relationType: RelationType, direction: 'forward' | 'reverse'): void {
+  let entityContent = fs.readFileSync(entityPath, 'utf8');
+
+  // Add import for the related interface
+  const importStatement = `import { I${relatedTableCapitalized} } from './${direction === 'forward' ? relatedTableCapitalized.toLowerCase() : relatedTableCapitalized.toLowerCase()}';`;
+  if (!entityContent.includes(importStatement)) {
+    // Find the last import statement and add after it
+    const importRegex = /import.*from.*['"];?\s*$/gm;
+    const matches = [...entityContent.matchAll(importRegex)];
+    if (matches.length > 0) {
+      const lastImport = matches[matches.length - 1];
+      const insertIndex = lastImport.index! + lastImport[0].length;
+      entityContent = entityContent.slice(0, insertIndex) + '\n' + importStatement + entityContent.slice(insertIndex);
+    }
+  }
+
+  // Add relationship fields based on type
+  const interfaceRegex = /export interface I\w+ extends mongoose\.Document \{([^}]*)\}/s;
+  const match = entityContent.match(interfaceRegex);
+  
+  if (match) {
+    const interfaceContent = match[1];
+    const newFields = generateEntityFields(tableCapitalized, relatedTableCapitalized, relationType, direction);
+    
+    // Add new fields before the closing brace
+    const updatedInterfaceContent = interfaceContent.trim() + '\n    ' + newFields;
+    entityContent = entityContent.replace(interfaceRegex, `export interface I${tableCapitalized} extends mongoose.Document {$1${updatedInterfaceContent}\n}`);
+  }
+
+  fs.writeFileSync(entityPath, entityContent);
+}
+
+function updateSchema(schemaPath: string, tableCapitalized: string, relatedTableCapitalized: string, relationType: RelationType, direction: 'forward' | 'reverse'): void {
+  let schemaContent = fs.readFileSync(schemaPath, 'utf8');
+
+  // Add import for the related schema
+  const importStatement = `import { ${relatedTableCapitalized}Entity } from './${direction === 'forward' ? relatedTableCapitalized.toLowerCase() : relatedTableCapitalized.toLowerCase()}';`;
+  if (!schemaContent.includes(importStatement)) {
+    // Find the last import statement and add after it
+    const importRegex = /import.*from.*['"];?\s*$/gm;
+    const matches = [...schemaContent.matchAll(importRegex)];
+    if (matches.length > 0) {
+      const lastImport = matches[matches.length - 1];
+      const insertIndex = lastImport.index! + lastImport[0].length;
+      schemaContent = schemaContent.slice(0, insertIndex) + '\n' + importStatement + schemaContent.slice(insertIndex);
+    }
+  }
+
+  // Add schema fields based on relationship type
+  const schemaRegex = /export const \w+Schema = new mongoose\.Schema\(([^)]*),/s;
+  const match = schemaContent.match(schemaRegex);
+  
+  if (match) {
+    const schemaFields = match[1];
+    const newFields = generateSchemaFields(tableCapitalized, relatedTableCapitalized, relationType, direction);
+    
+    // Add new fields before the closing brace
+    const updatedSchemaFields = schemaFields.trim() + '\n        ' + newFields;
+    schemaContent = schemaContent.replace(schemaRegex, `export const ${tableCapitalized.toLowerCase()}Schema = new mongoose.Schema(${updatedSchemaFields},`);
+  }
+
+  // Add virtual fields and population methods
+  const modelRegex = /export const \w+Entity = mongoose\.model<.*>\(.*\);?\s*$/;
+  const virtualFields = generateVirtualFields(tableCapitalized, relatedTableCapitalized, relationType, direction);
+  
+  if (virtualFields) {
+    schemaContent = schemaContent.replace(modelRegex, `${virtualFields}\n\n$&`);
+  }
+
+  fs.writeFileSync(schemaPath, schemaContent);
+}
+
+function generateEntityFields(tableCapitalized: string, relatedTableCapitalized: string, relationType: RelationType, direction: 'forward' | 'reverse'): string {
+  switch (relationType) {
+    case 'one-to-one':
+      if (direction === 'forward') {
+        return `${relatedTableCapitalized.toLowerCase()}Id: ObjectId,\n    ${relatedTableCapitalized.toLowerCase()}?: I${relatedTableCapitalized}`;
+      } else {
+        return `${tableCapitalized.toLowerCase()}Id: ObjectId,\n    ${tableCapitalized.toLowerCase()}?: I${tableCapitalized}`;
+      }
+    
+    case 'one-to-many':
+      if (direction === 'forward') {
+        return `${relatedTableCapitalized.toLowerCase()}Id: ObjectId,\n    ${relatedTableCapitalized.toLowerCase()}?: I${relatedTableCapitalized}`;
+      } else {
+        return `${tableCapitalized.toLowerCase()}s?: I${tableCapitalized}[]`;
+      }
+    
+    case 'many-to-many':
+      if (direction === 'forward') {
+        return `${relatedTableCapitalized.toLowerCase()}Ids: ObjectId[],\n    ${relatedTableCapitalized.toLowerCase()}s?: I${relatedTableCapitalized}[]`;
+      } else {
+        return `${tableCapitalized.toLowerCase()}Ids: ObjectId[],\n    ${tableCapitalized.toLowerCase()}s?: I${tableCapitalized}[]`;
+      }
+    
+    default:
+      return '';
+  }
+}
+
+function generateSchemaFields(tableCapitalized: string, relatedTableCapitalized: string, relationType: RelationType, direction: 'forward' | 'reverse'): string {
+  switch (relationType) {
+    case 'one-to-one':
+      if (direction === 'forward') {
+        return `${relatedTableCapitalized.toLowerCase()}Id: {\n            type: mongoose.SchemaTypes.ObjectId,\n            ref: '${relatedTableCapitalized.toLowerCase()}',\n            required: false\n        }`;
+      } else {
+        return `${tableCapitalized.toLowerCase()}Id: {\n            type: mongoose.SchemaTypes.ObjectId,\n            ref: '${tableCapitalized.toLowerCase()}',\n            required: false\n        }`;
+      }
+    
+    case 'one-to-many':
+      if (direction === 'forward') {
+        return `${relatedTableCapitalized.toLowerCase()}Id: {\n            type: mongoose.SchemaTypes.ObjectId,\n            ref: '${relatedTableCapitalized.toLowerCase()}',\n            required: false\n        }`;
+      } else {
+        return ''; // No field needed for reverse one-to-many
+      }
+    
+    case 'many-to-many':
+      if (direction === 'forward') {
+        return `${relatedTableCapitalized.toLowerCase()}Ids: [{\n            type: mongoose.SchemaTypes.ObjectId,\n            ref: '${relatedTableCapitalized.toLowerCase()}'\n        }]`;
+      } else {
+        return `${tableCapitalized.toLowerCase()}Ids: [{\n            type: mongoose.SchemaTypes.ObjectId,\n            ref: '${tableCapitalized.toLowerCase()}'\n        }]`;
+      }
+    
+    default:
+      return '';
+  }
+}
+
+function generateVirtualFields(tableCapitalized: string, relatedTableCapitalized: string, relationType: RelationType, direction: 'forward' | 'reverse'): string {
+  switch (relationType) {
+    case 'one-to-one':
+      if (direction === 'forward') {
+        return `// Virtual populate for ${relatedTableCapitalized.toLowerCase()}\n${tableCapitalized.toLowerCase()}Schema.virtual('${relatedTableCapitalized.toLowerCase()}', {\n    ref: '${relatedTableCapitalized.toLowerCase()}',\n    localField: '${relatedTableCapitalized.toLowerCase()}Id',\n    foreignField: '_id',\n    justOne: true\n});`;
+      } else {
+        return `// Virtual populate for ${tableCapitalized.toLowerCase()}\n${tableCapitalized.toLowerCase()}Schema.virtual('${tableCapitalized.toLowerCase()}', {\n    ref: '${tableCapitalized.toLowerCase()}',\n    localField: '${tableCapitalized.toLowerCase()}Id',\n    foreignField: '_id',\n    justOne: true\n});`;
+      }
+    
+    case 'one-to-many':
+      if (direction === 'forward') {
+        return `// Virtual populate for ${relatedTableCapitalized.toLowerCase()}\n${tableCapitalized.toLowerCase()}Schema.virtual('${relatedTableCapitalized.toLowerCase()}', {\n    ref: '${relatedTableCapitalized.toLowerCase()}',\n    localField: '${relatedTableCapitalized.toLowerCase()}Id',\n    foreignField: '_id',\n    justOne: true\n});`;
+      } else {
+        return `// Virtual populate for ${tableCapitalized.toLowerCase()}s\n${tableCapitalized.toLowerCase()}Schema.virtual('${tableCapitalized.toLowerCase()}s', {\n    ref: '${tableCapitalized.toLowerCase()}',\n    localField: '_id',\n    foreignField: '${tableCapitalized.toLowerCase()}Id'\n});`;
+      }
+    
+    case 'many-to-many':
+      if (direction === 'forward') {
+        return `// Virtual populate for ${relatedTableCapitalized.toLowerCase()}s\n${tableCapitalized.toLowerCase()}Schema.virtual('${relatedTableCapitalized.toLowerCase()}s', {\n    ref: '${relatedTableCapitalized.toLowerCase()}',\n    localField: '${relatedTableCapitalized.toLowerCase()}Ids',\n    foreignField: '_id'\n});`;
+      } else {
+        return `// Virtual populate for ${tableCapitalized.toLowerCase()}s\n${tableCapitalized.toLowerCase()}Schema.virtual('${tableCapitalized.toLowerCase()}s', {\n    ref: '${tableCapitalized.toLowerCase()}',\n    localField: '${tableCapitalized.toLowerCase()}Ids',\n    foreignField: '_id'\n});`;
+      }
+    
+    default:
+      return '';
+  }
+}