Permalink
Browse files

feat(bin): Added CLI tool

  • Loading branch information...
RWOverdijk committed Oct 20, 2016
1 parent 758fbe4 commit 8a15ee7c33522ef88ac74120e18bf731eb0dae10
Showing with 162 additions and 0 deletions.
  1. +162 −0 bin/wetland.ts
@@ -0,0 +1,162 @@
#!/usr/bin/env node
import * as program from 'commander';
import * as path from 'path';
import {Wetland, Migrator} from '../src/index';
const Table = require('cli-table');
const colors = require('colors/safe');
const checkmark = process.platform === 'win32' ? '\u221A' : '';
const cross = process.platform === 'win32' ? '\u00D7' : '';
program.version(require(__dirname + '/../../package.json').version)
.option('-c, --config <path>', 'defaults to $PWD/(wetland.js|wetland.json)')
.option('-r, --run', 'run the migration(s) on the database')
.option('-d, --dump-sql', 'dump the queries for the migration(s)')
.usage('<command> [options]');
function getMigrator(provided: string): Migrator {
let wetland;
try {
wetland = new Wetland(getConfig(provided));
} catch (error) {
console.error(`\n Caught: ${error}\n`);
process.exit(1);
}
return wetland.getMigrator();
}
function getConfig(provided: string): Object {
let tryPaths = [];
let config = null;
if (!provided) {
tryPaths = [
path.join(process.cwd(), 'wetland.js'),
path.join(process.cwd(), 'wetland.json')
];
} else {
tryPaths.push(provided[0] === '/' ? provided : path.join(process.cwd(), provided));
}
tryPaths.find(tryPath => {
try {
return config = require(tryPath);
} catch (e) {
}
});
if (!config) {
console.error('\n error: config file not found.\n');
process.exit(1);
}
return config;
}
function migrate(options, method) {
let migrator = getMigrator(options.parent.config);
if (options.parent.dumpSql) {
return migrator[method](Migrator.ACTION_GET_SQL).then(queries => {
if (!queries) {
console.log('\n-- No queries found\n');
} else {
console.log('\n-- Queries for next migration');
console.log(queries, '\n');
}
process.exit(0);
}).catch(error => {
console.error(`\n Caught: ${error}\n`);
process.exit(1);
});
}
if (options.parent.run) {
return migrator[method](Migrator.ACTION_RUN).then(migrations => {
if (!migrations) {
console.log('\n Success: No migrations to run.\n');
} else {
console.log(`\n Success: '${parseInt(migrations)}' migrations executed.\n`);
}
process.exit(0);
}).catch(error => {
console.error(`\n Caught: ${error}\n`);
process.exit(1);
});
}
console.error('\n Error: Missing action. Provide one of --run or --dump-sql\n');
process.exit(1);
}
program.command('migrator:create <name>').action((name, options) => {
let migrator = getMigrator(options.parent.config);
migrator.create(name).then(createdPath => {
console.log('\n Success: Migration file created successfully at ' + createdPath + '\n');
process.exit();
}).catch(error => {
console.error(`\n Caught: ${error}\n`);
process.exit(1);
});
});
program.command('migrator:status').action(options => {
let migrator = getMigrator(options.parent.config);
let table = new Table({
head: [
colors.green.bold('Migration'),
colors.green.bold('Run at'),
colors.green.bold('Run ID'),
colors.green.bold('Status')
]
});
Promise.all([migrator.allMigrations(), migrator.appliedMigrations()]).then(result => {
let runMap = {};
result[1].forEach(migration => runMap[migration.name] = migration);
result[0].map(migration => {
let applied = runMap[migration];
table.push([
migration,
applied ? applied.migration_time : 'N/A',
applied ? applied.run : 'N/A',
applied ? `${colors.green(checkmark)} Applied` : `${colors.red(cross)} Not applied`
]);
});
console.log('\n');
console.log(table.toString());
console.log('\n');
process.exit();
}).catch(error => {
console.error(`\n Caught: ${error}\n`);
process.exit(1);
});
});
program.command('migrator:up').action(options => migrate(options, 'up'));
program.command('migrator:down').action(options => migrate(options, 'down'));
program.command('migrator:latest').action(options => migrate(options, 'latest'));
program.command('migrator:revert').action(options => migrate(options, 'revert'));
program.on('--help', function () {
console.log(' Examples:');
console.log('');
console.log(' $ wetland migrator:create create_user');
console.log(' $ wetland migrator:up --run');
console.log(' $ wetland migrator:down --dump-sql');
console.log(' $ wetland migrator:status');
console.log('');
});
program.parse(process.argv);

0 comments on commit 8a15ee7

Please sign in to comment.