421 lines
13 KiB
Bash
421 lines
13 KiB
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
# Database Management Master Script
|
||
|
|
# Central hub for all database operations and tools
|
||
|
|
|
||
|
|
set -e
|
||
|
|
|
||
|
|
# Colors for output
|
||
|
|
RED='\033[0;31m'
|
||
|
|
GREEN='\033[0;32m'
|
||
|
|
YELLOW='\033[1;33m'
|
||
|
|
BLUE='\033[0;34m'
|
||
|
|
CYAN='\033[0;36m'
|
||
|
|
BOLD='\033[1m'
|
||
|
|
NC='\033[0m' # No Color
|
||
|
|
|
||
|
|
# Script directory
|
||
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
|
|
PROJECT_ROOT="$(dirname "$SCRIPT_DIR")"
|
||
|
|
|
||
|
|
# Change to project root
|
||
|
|
cd "$PROJECT_ROOT"
|
||
|
|
|
||
|
|
# Logging functions
|
||
|
|
log() {
|
||
|
|
echo -e "${GREEN}[INFO]${NC} $1"
|
||
|
|
}
|
||
|
|
|
||
|
|
log_warn() {
|
||
|
|
echo -e "${YELLOW}[WARN]${NC} $1"
|
||
|
|
}
|
||
|
|
|
||
|
|
log_error() {
|
||
|
|
echo -e "${RED}[ERROR]${NC} $1"
|
||
|
|
}
|
||
|
|
|
||
|
|
log_success() {
|
||
|
|
echo -e "${GREEN}[SUCCESS]${NC} $1"
|
||
|
|
}
|
||
|
|
|
||
|
|
print_header() {
|
||
|
|
echo -e "${BLUE}${BOLD}=== $1 ===${NC}"
|
||
|
|
}
|
||
|
|
|
||
|
|
print_subheader() {
|
||
|
|
echo -e "${CYAN}--- $1 ---${NC}"
|
||
|
|
}
|
||
|
|
|
||
|
|
print_usage() {
|
||
|
|
echo -e "${BOLD}Database Management Hub${NC}"
|
||
|
|
echo
|
||
|
|
echo "Usage: $0 <category> <command> [options]"
|
||
|
|
echo
|
||
|
|
echo -e "${BOLD}Categories:${NC}"
|
||
|
|
echo
|
||
|
|
echo -e "${CYAN}setup${NC} Database setup and initialization"
|
||
|
|
echo " setup Full database setup (create + migrate + seed)"
|
||
|
|
echo " create Create the database"
|
||
|
|
echo " migrate Run migrations"
|
||
|
|
echo " seed Seed database with test data"
|
||
|
|
echo " reset Reset database (drop + create + migrate)"
|
||
|
|
echo " status Show migration status"
|
||
|
|
echo " drop Drop the database"
|
||
|
|
echo " postgres Setup PostgreSQL database"
|
||
|
|
echo " sqlite Setup SQLite database"
|
||
|
|
echo
|
||
|
|
echo -e "${CYAN}backup${NC} Backup and restore operations"
|
||
|
|
echo " backup Create database backup"
|
||
|
|
echo " restore Restore database from backup"
|
||
|
|
echo " list List available backups"
|
||
|
|
echo " clean Clean old backups"
|
||
|
|
echo " export Export data to JSON/CSV"
|
||
|
|
echo " import Import data from JSON/CSV"
|
||
|
|
echo " clone Clone database to different name"
|
||
|
|
echo " compare Compare two databases"
|
||
|
|
echo
|
||
|
|
echo -e "${CYAN}monitor${NC} Monitoring and health checks"
|
||
|
|
echo " health Complete health check"
|
||
|
|
echo " status Quick status check"
|
||
|
|
echo " connections Show active connections"
|
||
|
|
echo " performance Show performance metrics"
|
||
|
|
echo " slow-queries Show slow queries"
|
||
|
|
echo " locks Show database locks"
|
||
|
|
echo " disk-usage Show disk usage"
|
||
|
|
echo " memory-usage Show memory usage"
|
||
|
|
echo " backup-status Check backup status"
|
||
|
|
echo " monitor Start continuous monitoring"
|
||
|
|
echo " alerts Check for alerts"
|
||
|
|
echo " vacuum Perform database maintenance"
|
||
|
|
echo " analyze Update database statistics"
|
||
|
|
echo " report Generate comprehensive report"
|
||
|
|
echo
|
||
|
|
echo -e "${CYAN}migrate${NC} Migration management"
|
||
|
|
echo " status Show migration status"
|
||
|
|
echo " pending List pending migrations"
|
||
|
|
echo " applied List applied migrations"
|
||
|
|
echo " run Run pending migrations"
|
||
|
|
echo " rollback Rollback migrations"
|
||
|
|
echo " create Create new migration"
|
||
|
|
echo " generate Generate migration from schema diff"
|
||
|
|
echo " validate Validate migration files"
|
||
|
|
echo " dry-run Show what would be migrated"
|
||
|
|
echo " force Force migration state"
|
||
|
|
echo " repair Repair migration table"
|
||
|
|
echo " baseline Set migration baseline"
|
||
|
|
echo " history Show migration history"
|
||
|
|
echo " schema-dump Dump current schema"
|
||
|
|
echo " data-migrate Migrate data between schemas"
|
||
|
|
echo " template Manage migration templates"
|
||
|
|
echo
|
||
|
|
echo -e "${CYAN}utils${NC} Database utilities and maintenance"
|
||
|
|
echo " size Show database size information"
|
||
|
|
echo " tables List all tables with row counts"
|
||
|
|
echo " indexes Show index information"
|
||
|
|
echo " constraints Show table constraints"
|
||
|
|
echo " users Show database users (PostgreSQL only)"
|
||
|
|
echo " permissions Show user permissions"
|
||
|
|
echo " sessions Show active sessions"
|
||
|
|
echo " locks Show current locks"
|
||
|
|
echo " queries Show running queries"
|
||
|
|
echo " kill-query Kill a specific query"
|
||
|
|
echo " optimize Optimize database (VACUUM, ANALYZE)"
|
||
|
|
echo " reindex Rebuild indexes"
|
||
|
|
echo " check-integrity Check database integrity"
|
||
|
|
echo " repair Repair database issues"
|
||
|
|
echo " cleanup Clean up temporary data"
|
||
|
|
echo " logs Show database logs"
|
||
|
|
echo " config Show database configuration"
|
||
|
|
echo " extensions List database extensions (PostgreSQL)"
|
||
|
|
echo " sequences Show sequence information"
|
||
|
|
echo " triggers Show table triggers"
|
||
|
|
echo " functions Show user-defined functions"
|
||
|
|
echo " views Show database views"
|
||
|
|
echo " schema-info Show comprehensive schema information"
|
||
|
|
echo " duplicate-data Find duplicate records"
|
||
|
|
echo " orphaned-data Find orphaned records"
|
||
|
|
echo " table-stats Show detailed table statistics"
|
||
|
|
echo " connection-test Test database connection"
|
||
|
|
echo " benchmark Run database benchmarks"
|
||
|
|
echo " export-schema Export database schema"
|
||
|
|
echo " import-schema Import database schema"
|
||
|
|
echo " copy-table Copy table data"
|
||
|
|
echo " truncate-table Truncate table data"
|
||
|
|
echo " reset-sequence Reset sequence values"
|
||
|
|
echo
|
||
|
|
echo -e "${BOLD}Common Options:${NC}"
|
||
|
|
echo " --env ENV Environment (dev/prod) [default: dev]"
|
||
|
|
echo " --force Skip confirmations"
|
||
|
|
echo " --quiet Suppress verbose output"
|
||
|
|
echo " --debug Enable debug output"
|
||
|
|
echo " --dry-run Show what would be done without executing"
|
||
|
|
echo " --help Show category-specific help"
|
||
|
|
echo
|
||
|
|
echo -e "${BOLD}Quick Commands:${NC}"
|
||
|
|
echo " $0 status Quick database status"
|
||
|
|
echo " $0 health Complete health check"
|
||
|
|
echo " $0 backup Create backup"
|
||
|
|
echo " $0 migrate Run migrations"
|
||
|
|
echo " $0 optimize Optimize database"
|
||
|
|
echo
|
||
|
|
echo -e "${BOLD}Examples:${NC}"
|
||
|
|
echo " $0 setup create # Create database"
|
||
|
|
echo " $0 setup migrate # Run migrations"
|
||
|
|
echo " $0 backup create # Create backup"
|
||
|
|
echo " $0 backup restore --file backup.sql # Restore from backup"
|
||
|
|
echo " $0 monitor health # Health check"
|
||
|
|
echo " $0 monitor connections # Show connections"
|
||
|
|
echo " $0 migrate create --name add_users # Create migration"
|
||
|
|
echo " $0 migrate run # Run pending migrations"
|
||
|
|
echo " $0 utils size # Show database size"
|
||
|
|
echo " $0 utils optimize # Optimize database"
|
||
|
|
echo
|
||
|
|
echo -e "${BOLD}For detailed help on a specific category:${NC}"
|
||
|
|
echo " $0 setup --help"
|
||
|
|
echo " $0 backup --help"
|
||
|
|
echo " $0 monitor --help"
|
||
|
|
echo " $0 migrate --help"
|
||
|
|
echo " $0 utils --help"
|
||
|
|
}
|
||
|
|
|
||
|
|
# Check if required scripts exist
|
||
|
|
check_scripts() {
|
||
|
|
local missing_scripts=()
|
||
|
|
|
||
|
|
if [ ! -f "$SCRIPT_DIR/db-setup.sh" ]; then
|
||
|
|
missing_scripts+=("db-setup.sh")
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [ ! -f "$SCRIPT_DIR/db-backup.sh" ]; then
|
||
|
|
missing_scripts+=("db-backup.sh")
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [ ! -f "$SCRIPT_DIR/db-monitor.sh" ]; then
|
||
|
|
missing_scripts+=("db-monitor.sh")
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [ ! -f "$SCRIPT_DIR/db-migrate.sh" ]; then
|
||
|
|
missing_scripts+=("db-migrate.sh")
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [ ! -f "$SCRIPT_DIR/db-utils.sh" ]; then
|
||
|
|
missing_scripts+=("db-utils.sh")
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [ ${#missing_scripts[@]} -gt 0 ]; then
|
||
|
|
log_error "Missing required scripts: ${missing_scripts[*]}"
|
||
|
|
echo "Please ensure all database management scripts are present in the scripts directory."
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
# Make scripts executable
|
||
|
|
make_scripts_executable() {
|
||
|
|
chmod +x "$SCRIPT_DIR"/db-*.sh 2>/dev/null || true
|
||
|
|
}
|
||
|
|
|
||
|
|
# Show quick status
|
||
|
|
show_quick_status() {
|
||
|
|
print_header "Quick Database Status"
|
||
|
|
|
||
|
|
# Check if .env exists
|
||
|
|
if [ ! -f ".env" ]; then
|
||
|
|
log_error ".env file not found"
|
||
|
|
echo "Run: $0 setup create"
|
||
|
|
return 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Load environment variables
|
||
|
|
export $(grep -v '^#' .env | xargs) 2>/dev/null || true
|
||
|
|
|
||
|
|
# Show basic info
|
||
|
|
log "Environment: ${ENVIRONMENT:-dev}"
|
||
|
|
log "Database URL: ${DATABASE_URL:-not set}"
|
||
|
|
|
||
|
|
# Test connection
|
||
|
|
if command -v "$SCRIPT_DIR/db-utils.sh" >/dev/null 2>&1; then
|
||
|
|
"$SCRIPT_DIR/db-utils.sh" connection-test --quiet 2>/dev/null || log_warn "Database connection failed"
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Show migration status
|
||
|
|
if command -v "$SCRIPT_DIR/db-migrate.sh" >/dev/null 2>&1; then
|
||
|
|
"$SCRIPT_DIR/db-migrate.sh" status --quiet 2>/dev/null || log_warn "Could not check migration status"
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
# Show comprehensive health check
|
||
|
|
show_health_check() {
|
||
|
|
print_header "Comprehensive Database Health Check"
|
||
|
|
|
||
|
|
if [ -f "$SCRIPT_DIR/db-monitor.sh" ]; then
|
||
|
|
"$SCRIPT_DIR/db-monitor.sh" health "$@"
|
||
|
|
else
|
||
|
|
log_error "db-monitor.sh not found"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
# Create quick backup
|
||
|
|
create_quick_backup() {
|
||
|
|
print_header "Quick Database Backup"
|
||
|
|
|
||
|
|
if [ -f "$SCRIPT_DIR/db-backup.sh" ]; then
|
||
|
|
"$SCRIPT_DIR/db-backup.sh" backup --compress "$@"
|
||
|
|
else
|
||
|
|
log_error "db-backup.sh not found"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
# Run migrations
|
||
|
|
run_migrations() {
|
||
|
|
print_header "Running Database Migrations"
|
||
|
|
|
||
|
|
if [ -f "$SCRIPT_DIR/db-migrate.sh" ]; then
|
||
|
|
"$SCRIPT_DIR/db-migrate.sh" run "$@"
|
||
|
|
else
|
||
|
|
log_error "db-migrate.sh not found"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
# Optimize database
|
||
|
|
optimize_database() {
|
||
|
|
print_header "Database Optimization"
|
||
|
|
|
||
|
|
if [ -f "$SCRIPT_DIR/db-utils.sh" ]; then
|
||
|
|
"$SCRIPT_DIR/db-utils.sh" optimize "$@"
|
||
|
|
else
|
||
|
|
log_error "db-utils.sh not found"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
# Parse command line arguments
|
||
|
|
CATEGORY=""
|
||
|
|
COMMAND=""
|
||
|
|
REMAINING_ARGS=()
|
||
|
|
|
||
|
|
# Handle special single commands
|
||
|
|
if [[ $# -eq 1 ]]; then
|
||
|
|
case $1 in
|
||
|
|
"status")
|
||
|
|
show_quick_status
|
||
|
|
exit 0
|
||
|
|
;;
|
||
|
|
"health")
|
||
|
|
show_health_check
|
||
|
|
exit 0
|
||
|
|
;;
|
||
|
|
"backup")
|
||
|
|
create_quick_backup
|
||
|
|
exit 0
|
||
|
|
;;
|
||
|
|
"migrate")
|
||
|
|
run_migrations
|
||
|
|
exit 0
|
||
|
|
;;
|
||
|
|
"optimize")
|
||
|
|
optimize_database
|
||
|
|
exit 0
|
||
|
|
;;
|
||
|
|
"-h"|"--help")
|
||
|
|
print_usage
|
||
|
|
exit 0
|
||
|
|
;;
|
||
|
|
esac
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Parse arguments
|
||
|
|
while [[ $# -gt 0 ]]; do
|
||
|
|
case $1 in
|
||
|
|
-h|--help)
|
||
|
|
if [ -n "$CATEGORY" ]; then
|
||
|
|
REMAINING_ARGS+=("$1")
|
||
|
|
else
|
||
|
|
print_usage
|
||
|
|
exit 0
|
||
|
|
fi
|
||
|
|
shift
|
||
|
|
;;
|
||
|
|
*)
|
||
|
|
if [ -z "$CATEGORY" ]; then
|
||
|
|
CATEGORY="$1"
|
||
|
|
elif [ -z "$COMMAND" ]; then
|
||
|
|
COMMAND="$1"
|
||
|
|
else
|
||
|
|
REMAINING_ARGS+=("$1")
|
||
|
|
fi
|
||
|
|
shift
|
||
|
|
;;
|
||
|
|
esac
|
||
|
|
done
|
||
|
|
|
||
|
|
# Check if we're in the right directory
|
||
|
|
if [ ! -f "Cargo.toml" ]; then
|
||
|
|
log_error "Please run this script from the project root directory"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Check that all required scripts exist
|
||
|
|
check_scripts
|
||
|
|
|
||
|
|
# Make scripts executable
|
||
|
|
make_scripts_executable
|
||
|
|
|
||
|
|
# Validate category and command
|
||
|
|
if [ -z "$CATEGORY" ]; then
|
||
|
|
print_usage
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Route to appropriate script
|
||
|
|
case "$CATEGORY" in
|
||
|
|
"setup")
|
||
|
|
if [ -z "$COMMAND" ]; then
|
||
|
|
log_error "Command required for setup category"
|
||
|
|
echo "Use: $0 setup --help for available commands"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
exec "$SCRIPT_DIR/db-setup.sh" "$COMMAND" "${REMAINING_ARGS[@]}"
|
||
|
|
;;
|
||
|
|
"backup")
|
||
|
|
if [ -z "$COMMAND" ]; then
|
||
|
|
log_error "Command required for backup category"
|
||
|
|
echo "Use: $0 backup --help for available commands"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
exec "$SCRIPT_DIR/db-backup.sh" "$COMMAND" "${REMAINING_ARGS[@]}"
|
||
|
|
;;
|
||
|
|
"monitor")
|
||
|
|
if [ -z "$COMMAND" ]; then
|
||
|
|
log_error "Command required for monitor category"
|
||
|
|
echo "Use: $0 monitor --help for available commands"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
exec "$SCRIPT_DIR/db-monitor.sh" "$COMMAND" "${REMAINING_ARGS[@]}"
|
||
|
|
;;
|
||
|
|
"migrate")
|
||
|
|
if [ -z "$COMMAND" ]; then
|
||
|
|
log_error "Command required for migrate category"
|
||
|
|
echo "Use: $0 migrate --help for available commands"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
exec "$SCRIPT_DIR/db-migrate.sh" "$COMMAND" "${REMAINING_ARGS[@]}"
|
||
|
|
;;
|
||
|
|
"utils")
|
||
|
|
if [ -z "$COMMAND" ]; then
|
||
|
|
log_error "Command required for utils category"
|
||
|
|
echo "Use: $0 utils --help for available commands"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
exec "$SCRIPT_DIR/db-utils.sh" "$COMMAND" "${REMAINING_ARGS[@]}"
|
||
|
|
;;
|
||
|
|
*)
|
||
|
|
log_error "Unknown category: $CATEGORY"
|
||
|
|
echo
|
||
|
|
print_usage
|
||
|
|
exit 1
|
||
|
|
;;
|
||
|
|
esac
|