@fiftyten/db-connect
Version:
CLI tool for database connections and DynamoDB operations via AWS Session Manager
293 lines (283 loc) ā¢ 13.1 kB
JavaScript
;
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const commander_1 = require("commander");
const chalk_1 = __importDefault(require("chalk"));
const database_connector_1 = require("./database-connector");
const dynamodb_connector_1 = require("./dynamodb-connector");
const package_json_1 = require("../package.json");
const program = new commander_1.Command();
program
.name('fiftyten-db')
.description('CLI tool for connecting to Fiftyten databases via AWS Session Manager')
.version(package_json_1.version);
// Tunnel command - creates port forwarding to database
program
.command('tunnel')
.description('Create SSH tunnel to database via Session Manager')
.argument('<environment>', 'Environment (dev/main)')
.option('-p, --port <port>', 'Local port for tunnel', '5433')
.option('-d, --database <app>', 'Application database (platform, copytrading, etc.)', 'platform')
.option('--region <region>', 'AWS region', 'us-west-1')
.action(async (environment, options) => {
try {
const connector = new database_connector_1.DatabaseConnector(options.region);
await connector.createTunnel(environment, options.database, parseInt(options.port));
}
catch (error) {
console.error(chalk_1.default.red('Error creating tunnel:'), error instanceof Error ? error.message : String(error));
process.exit(1);
}
});
// Connect command - direct connection to database
program
.command('connect')
.description('Connect directly to database via Session Manager')
.argument('<environment>', 'Environment (dev/main)')
.option('-d, --database <app>', 'Application database (platform, copytrading, etc.)', 'platform')
.option('--region <region>', 'AWS region', 'us-west-1')
.action(async (environment, options) => {
try {
const connector = new database_connector_1.DatabaseConnector(options.region);
await connector.connectDatabase(environment, options.database);
}
catch (error) {
console.error(chalk_1.default.red('Error connecting to database:'), error instanceof Error ? error.message : String(error));
process.exit(1);
}
});
// SSH command - SSH into bastion host
program
.command('ssh')
.description('SSH into bastion host via Session Manager')
.argument('<environment>', 'Environment (dev/main)')
.option('--region <region>', 'AWS region', 'us-west-1')
.action(async (environment, options) => {
try {
const connector = new database_connector_1.DatabaseConnector(options.region);
await connector.sshBastion(environment);
}
catch (error) {
console.error(chalk_1.default.red('Error connecting to bastion:'), error instanceof Error ? error.message : String(error));
process.exit(1);
}
});
// Info command - show connection information
program
.command('info')
.description('Show connection information for environment')
.argument('<environment>', 'Environment (dev/main)')
.option('--region <region>', 'AWS region', 'us-west-1')
.action(async (environment, options) => {
try {
const connector = new database_connector_1.DatabaseConnector(options.region);
await connector.showInfo(environment);
}
catch (error) {
console.error(chalk_1.default.red('Error fetching info:'), error instanceof Error ? error.message : String(error));
process.exit(1);
}
});
// List command - show available environments
program
.command('list')
.description('List available environments and services')
.option('--region <region>', 'AWS region', 'us-west-1')
.action(async (options) => {
try {
const connector = new database_connector_1.DatabaseConnector(options.region);
await connector.listEnvironments();
}
catch (error) {
console.error(chalk_1.default.red('Error listing environments:'), error instanceof Error ? error.message : String(error));
process.exit(1);
}
});
// Psql command - complete database connection with password
program
.command('psql')
.description('Connect to database with automatic tunnel and password retrieval')
.argument('<environment>', 'Environment (dev/main)')
.option('-p, --port <port>', 'Local port for tunnel', '5433')
.option('-d, --database <app>', 'Application database (platform, copytrading, etc.)', 'platform')
.option('--region <region>', 'AWS region', 'us-west-1')
.action(async (environment, options) => {
try {
const connector = new database_connector_1.DatabaseConnector(options.region);
await connector.connectWithPassword(environment, options.database, parseInt(options.port));
}
catch (error) {
console.error(chalk_1.default.red('Error connecting to database:'), error instanceof Error ? error.message : String(error));
process.exit(1);
}
});
// Databases command - discover available databases
program
.command('databases')
.description('Discover available databases for an environment')
.argument('<environment>', 'Environment (dev/main)')
.option('--region <region>', 'AWS region', 'us-west-1')
.action(async (environment, options) => {
try {
const connector = new database_connector_1.DatabaseConnector(options.region);
await connector.discoverDatabases(environment);
}
catch (error) {
console.error(chalk_1.default.red('Error discovering databases:'), error instanceof Error ? error.message : String(error));
process.exit(1);
}
});
// Password command - get database password for manual configuration
program
.command('password')
.description('Get database password for manual configuration')
.argument('<environment>', 'Environment (dev/main)')
.option('-d, --database <app>', 'Application database (platform, copytrading, etc.)', 'platform')
.option('--region <region>', 'AWS region', 'us-west-1')
.action(async (environment, options) => {
try {
const connector = new database_connector_1.DatabaseConnector(options.region);
const password = await connector.getDatabasePassword(environment, options.database);
console.log(chalk_1.default.green('ā
Database password retrieved:'));
console.log(chalk_1.default.yellow(password));
console.log('');
console.log(chalk_1.default.gray('š” DATABASE_URL for manual configuration:'));
console.log(chalk_1.default.cyan(`DATABASE_URL=postgres://fiftyten:${password}@localhost:5433/${options.database === 'platform' ? 'platform' : options.database}`));
}
catch (error) {
console.error(chalk_1.default.red('Error retrieving password:'), error instanceof Error ? error.message : String(error));
process.exit(1);
}
});
// DynamoDB commands
const dynamoCommand = program
.command('dynamo')
.description('DynamoDB operations (all commands hide sensitive fields for security)');
// List DynamoDB tables
dynamoCommand
.command('list-tables')
.description('List all DynamoDB tables')
.option('--region <region>', 'AWS region', 'us-west-1')
.action(async (options) => {
try {
const connector = new dynamodb_connector_1.DynamoDBConnector(options.region);
await connector.listTables();
}
catch (error) {
console.error(chalk_1.default.red('Error listing tables:'), error instanceof Error ? error.message : String(error));
process.exit(1);
}
});
// Describe a specific table
dynamoCommand
.command('describe')
.description('Describe a DynamoDB table structure and keys')
.argument('<tableName>', 'Name of the table to describe')
.option('--region <region>', 'AWS region', 'us-west-1')
.addHelpText('after', `
Example:
${chalk_1.default.cyan('fiftyten-db dynamo describe fiftyten-exchange-credentials-dev')}
Shows key structure needed for queries:
ā¢ tenant_id (Hash key) + credential_sk (Range key)
ā¢ GSI indexes for efficient lookups
ā¢ Example credential_sk: "USER#john123#PRODUCT#COPY_TRADING#EXCHANGE#gateio"`)
.action(async (tableName, options) => {
try {
const connector = new dynamodb_connector_1.DynamoDBConnector(options.region);
await connector.describeTable(tableName);
}
catch (error) {
console.error(chalk_1.default.red('Error describing table:'), error instanceof Error ? error.message : String(error));
process.exit(1);
}
});
// Scan a table
dynamoCommand
.command('scan')
.description('Scan a DynamoDB table (sensitive fields always hidden)')
.argument('<tableName>', 'Name of the table to scan')
.option('-l, --limit <limit>', 'Maximum number of items to return', '20')
.option('--region <region>', 'AWS region', 'us-west-1')
.addHelpText('after', `
Examples:
${chalk_1.default.gray('# Check recent trading orders')}
${chalk_1.default.cyan('fiftyten-db dynamo scan trading_orders --limit 10')}
${chalk_1.default.gray('# Sample customer credentials to see data format')}
${chalk_1.default.cyan('fiftyten-db dynamo scan fiftyten-exchange-credentials-dev --limit 3')}
${chalk_1.default.gray('# Find all active customers')}
${chalk_1.default.cyan('fiftyten-db dynamo scan fiftyten-exchange-credentials-dev --limit 50')}
ā ļø Scan reads entire table sequentially - expensive! Use query for targeted lookups.`)
.action(async (tableName, options) => {
try {
const connector = new dynamodb_connector_1.DynamoDBConnector(options.region);
await connector.scanTable(tableName, parseInt(options.limit));
}
catch (error) {
console.error(chalk_1.default.red('Error scanning table:'), error instanceof Error ? error.message : String(error));
process.exit(1);
}
});
// Query a table
dynamoCommand
.command('query')
.description('Query a DynamoDB table (sensitive fields always hidden)')
.argument('<tableName>', 'Name of the table to query')
.argument('<keyCondition>', 'Key condition (e.g., "id = value")')
.option('-l, --limit <limit>', 'Maximum number of items to return', '20')
.option('--region <region>', 'AWS region', 'us-west-1')
.addHelpText('after', `
Examples:
${chalk_1.default.gray('# Find all credentials for tenant 5010')}
${chalk_1.default.cyan('fiftyten-db dynamo query fiftyten-exchange-credentials-dev "tenant_id = 5010"')}
${chalk_1.default.gray('# Check trading orders for specific customer')}
${chalk_1.default.cyan('fiftyten-db dynamo query trading_orders "customer_id = john_doe_123"')}
${chalk_1.default.gray('# Find recent positions')}
${chalk_1.default.cyan('fiftyten-db dynamo query trading_positions "customer_id = john_doe_123" --limit 5')}
š” Query is efficient - uses partition key index. For GSI queries, use AWS CLI directly.`)
.action(async (tableName, keyCondition, options) => {
try {
const connector = new dynamodb_connector_1.DynamoDBConnector(options.region);
await connector.queryTable(tableName, keyCondition, parseInt(options.limit));
}
catch (error) {
console.error(chalk_1.default.red('Error querying table:'), error instanceof Error ? error.message : String(error));
process.exit(1);
}
});
// Get a specific item
dynamoCommand
.command('get-item')
.description('Get a specific item from DynamoDB table (sensitive fields always hidden)')
.argument('<tableName>', 'Name of the table')
.argument('<key>', 'Key of the item (format: "keyName:value" or JSON)')
.option('--region <region>', 'AWS region', 'us-west-1')
.addHelpText('after', `
Examples:
${chalk_1.default.gray('# Get specific trading order details')}
${chalk_1.default.cyan('fiftyten-db dynamo get-item trading_orders "id:trd_5f8a2b3c4d5e6f7g8h9i"')}
${chalk_1.default.gray('# Get customer credentials (composite key)')}
${chalk_1.default.cyan('fiftyten-db dynamo get-item fiftyten-exchange-credentials-dev \\\\')}
${chalk_1.default.cyan(' \'{"tenant_id":"5010","credential_sk":"USER#john_doe_123#PRODUCT#COPY_TRADING#EXCHANGE#gateio"}\'')}
${chalk_1.default.gray('# Check specific position status')}
${chalk_1.default.cyan('fiftyten-db dynamo get-item trading_positions "id:pos_abc123def456"')}
š” Use query first to find the exact keys, then get-item for full details.`)
.action(async (tableName, key, options) => {
try {
const connector = new dynamodb_connector_1.DynamoDBConnector(options.region);
await connector.getItem(tableName, key);
}
catch (error) {
console.error(chalk_1.default.red('Error getting item:'), error instanceof Error ? error.message : String(error));
process.exit(1);
}
});
// Parse command line arguments
program.parse();
// Show help if no command provided
if (!process.argv.slice(2).length) {
program.outputHelp();
}
//# sourceMappingURL=index.js.map