@neupgroup/mapper 1.4.1 → 1.4.2

package/dist/cli/create-migration.js

@@ -65,6 +65,7 @@ else {
 const indexFilePath = path.join(migrationDir, 'index.ts');
 let migrations = [];
 let completed = [];
+let currentVersion = -1;
 if (fs.existsSync(indexFilePath)) {
     const content = fs.readFileSync(indexFilePath, 'utf-8');
     const matchMigrations = content.match(/migrations = \[(.*?)\]/s);
@@ -75,6 +76,10 @@ if (fs.existsSync(indexFilePath)) {
     if (matchCompleted) {
         completed = matchCompleted[1].split(',').map(s => s.trim().replace(/['"]/g, '')).filter(Boolean);
     }
+    const matchVersion = content.match(/currentVersion = (.*?);/);
+    if (matchVersion) {
+        currentVersion = parseInt(matchVersion[1]);
+    }
 }
 const migrationName = fileName.replace('.ts', '');
 if (!migrations.includes(migrationName)) {
@@ -88,6 +93,8 @@ ${migrations.map(m => `    '${m}'`).join(',\n')}
 export const completed = [
 ${completed.map(m => `    '${m}'`).join(',\n')}
 ];
+
+export const currentVersion = ${currentVersion};
 `;
 fs.writeFileSync(indexFilePath, indexContent.trim() + '\n');
 console.log(`Updated migration index: ${indexFilePath}`);

package/dist/cli/migrate.js

@@ -1,6 +1,8 @@
 #!/usr/bin/env node
 import * as fs from 'fs';
 import * as path from 'path';
+const args = process.argv.slice(2);
+const command = args[0] || 'up'; // 'up' or 'down'
 const migrationDir = path.resolve(process.cwd(), 'src/migration');
 const indexFilePath = path.join(migrationDir, 'index.ts');
 if (!fs.existsSync(indexFilePath)) {
@@ -11,31 +13,56 @@ async function run() {
     const content = fs.readFileSync(indexFilePath, 'utf-8');
     const matchMigrations = content.match(/migrations = \[(.*?)\]/s);
     const matchCompleted = content.match(/completed = \[(.*?)\]/s);
+    const matchVersion = content.match(/currentVersion = (.*?);/);
     if (!matchMigrations) {
         console.log('No migrations list found in index.ts');
         return;
     }
     const migrations = matchMigrations[1].split(',').map(s => s.trim().replace(/['"]/g, '')).filter(Boolean);
-    const completed = matchCompleted ? matchCompleted[1].split(',').map(s => s.trim().replace(/['"]/g, '')).filter(Boolean) : [];
-    const pending = migrations.filter(m => !completed.includes(m));
-    if (pending.length === 0) {
-        console.log('No pending migrations.');
-        return;
+    let completed = matchCompleted ? matchCompleted[1].split(',').map(s => s.trim().replace(/['"]/g, '')).filter(Boolean) : [];
+    let currentVersion = matchVersion ? parseInt(matchVersion[1]) : -1;
+    if (command === 'up') {
+        const pending = migrations.filter(m => !completed.includes(m));
+        if (pending.length === 0) {
+            console.log('No pending migrations.');
+            return;
+        }
+        console.log(`Found ${pending.length} pending migrations.`);
+        for (const m of pending) {
+            console.log(`Running migration UP: ${m}...`);
+            const filePath = path.join(migrationDir, `${m}.ts`);
+            const absolutePath = path.resolve(filePath);
+            const mod = await import('file://' + absolutePath);
+            if (mod.up) {
+                await mod.up();
+                completed.push(m);
+                currentVersion = migrations.indexOf(m);
+                console.log(`Completed UP: ${m}`);
+            }
+            else {
+                console.error(`Migration ${m} does not have an up() function.`);
+                break;
+            }
+        }
     }
-    console.log(`Found ${pending.length} pending migrations.`);
-    for (const m of pending) {
-        console.log(`Running migration: ${m}...`);
-        const filePath = path.join(migrationDir, `${m}.ts`);
-        // Use path.relative to current working directory or absolute file:// URL
-        const absolutePath = path.isAbsolute(filePath) ? filePath : path.resolve(filePath);
+    else if (command === 'down') {
+        if (completed.length === 0) {
+            console.log('No migrations to roll back.');
+            return;
+        }
+        const lastMigrationName = completed[completed.length - 1];
+        console.log(`Rolling back migration: ${lastMigrationName}...`);
+        const filePath = path.join(migrationDir, `${lastMigrationName}.ts`);
+        const absolutePath = path.resolve(filePath);
         const mod = await import('file://' + absolutePath);
-        if (mod.up) {
-            await mod.up();
-            completed.push(m);
-            console.log(`Completed: ${m}`);
+        if (mod.down) {
+            await mod.down();
+            completed.pop();
+            currentVersion = completed.length > 0 ? migrations.indexOf(completed[completed.length - 1]) : -1;
+            console.log(`Completed DOWN: ${lastMigrationName}`);
         }
         else {
-            console.error(`Migration ${m} does not have an up() function.`);
+            console.error(`Migration ${lastMigrationName} does not have a down() function.`);
         }
     }
     // Update index.ts
@@ -47,9 +74,11 @@ ${migrations.map(m => `    '${m}'`).join(',\n')}
 export const completed = [
 ${completed.map(m => `    '${m}'`).join(',\n')}
 ];
+
+export const currentVersion = ${currentVersion};
 `;
     fs.writeFileSync(indexFilePath, indexContent.trim() + '\n');
-    console.log('Migration runner finished.');
+    console.log(`Migration runner finished. Current version: ${currentVersion}`);
 }
 run().catch(err => {
     console.error('Migration failed:', err);

package/dist/migrator.d.ts

@@ -21,4 +21,8 @@ export declare class TableMigrator {
     addColumn(name: string): ColumnBuilder;
     getColumns(): any[];
     exec(): Promise<void>;
+    drop(): Promise<void>;
+    dropColumn(columnName: string): Promise<void>;
+    dropUnique(columnName: string): Promise<void>;
+    dropPrimaryKey(columnName: string): Promise<void>;
 }

package/dist/migrator.js

@@ -103,4 +103,55 @@ ${fieldsContent}
         fs.writeFileSync(schemaFilePath, schemaContent.trim() + '\n');
         console.log(`Updated schema: ${schemaFilePath}`);
     }
+    async drop() {
+        const fs = await import('fs');
+        const path = await import('path');
+        const schemasDir = path.resolve(process.cwd(), 'src/schemas');
+        const schemaFilePath = path.join(schemasDir, `${this.name}.ts`);
+        if (fs.existsSync(schemaFilePath)) {
+            fs.unlinkSync(schemaFilePath);
+            console.log(`Deleted schema: ${schemaFilePath}`);
+        }
+    }
+    async dropColumn(columnName) {
+        const fs = await import('fs');
+        const path = await import('path');
+        const schemasDir = path.resolve(process.cwd(), 'src/schemas');
+        const schemaFilePath = path.join(schemasDir, `${this.name}.ts`);
+        if (fs.existsSync(schemaFilePath)) {
+            // Very simple implementation: filter out the line with the column
+            const content = fs.readFileSync(schemaFilePath, 'utf-8');
+            const lines = content.split('\n');
+            const filteredLines = lines.filter(line => !line.includes(`name: '${columnName}'`));
+            fs.writeFileSync(schemaFilePath, filteredLines.join('\n'));
+            console.log(`Dropped column ${columnName} from schema ${this.name}`);
+        }
+    }
+    async dropUnique(columnName) {
+        const fs = await import('fs');
+        const path = await import('path');
+        const schemasDir = path.resolve(process.cwd(), 'src/schemas');
+        const schemaFilePath = path.join(schemasDir, `${this.name}.ts`);
+        if (fs.existsSync(schemaFilePath)) {
+            let content = fs.readFileSync(schemaFilePath, 'utf-8');
+            // Remove ", isUnique: true" from the line containing the column
+            const regex = new RegExp(`({ name: '${columnName}', .*?), isUnique: true(.*})`);
+            content = content.replace(regex, '$1$2');
+            fs.writeFileSync(schemaFilePath, content);
+            console.log(`Dropped unique constraint from ${columnName} in schema ${this.name}`);
+        }
+    }
+    async dropPrimaryKey(columnName) {
+        const fs = await import('fs');
+        const path = await import('path');
+        const schemasDir = path.resolve(process.cwd(), 'src/schemas');
+        const schemaFilePath = path.join(schemasDir, `${this.name}.ts`);
+        if (fs.existsSync(schemaFilePath)) {
+            let content = fs.readFileSync(schemaFilePath, 'utf-8');
+            const regex = new RegExp(`({ name: '${columnName}', .*?), isPrimary: true(.*})`);
+            content = content.replace(regex, '$1$2');
+            fs.writeFileSync(schemaFilePath, content);
+            console.log(`Dropped primary key constraint from ${columnName} in schema ${this.name}`);
+        }
+    }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@neupgroup/mapper",
-  "version": "1.4.1",
+  "version": "1.4.2",
   "description": "Neup.Mapper core library for schema and mapping utilities",
   "type": "module",
   "main": "dist/index.js",
@@ -9,7 +9,9 @@
   "bin": {
     "create-migration": "dist/cli/create-migration.js",
     "create-connection": "dist/cli/create-connection.js",
-    "migrate": "dist/cli/migrate.js"
+    "migrate": "dist/cli/migrate.js",
+    "migrate-up": "dist/cli/migrate.js",
+    "migrate-down": "dist/cli/migrate.js"
   },
   "files": [
     "dist"
@@ -19,7 +21,9 @@
     "build": "tsc -p ./tsconfig.json",
     "create-migration": "npx tsx src/cli/create-migration.ts",
     "create-connection": "npx tsx src/cli/create-connection.ts",
-    "migrate": "npx tsx src/cli/migrate.ts"
+    "migrate": "npx tsx src/cli/migrate.ts",
+    "migrate-up": "npx tsx src/cli/migrate.ts up",
+    "migrate-down": "npx tsx src/cli/migrate.ts down"
   },
   "publishConfig": {
     "access": "public"