Renz
0cf21cd893
- Implemented TUI auto-select for automated testing - Fixed TUI automation: autoSelectMsg handling in Update() - Auto-database selection in DatabaseSelector - Created focused test suite (test_as_postgres.sh) - Created retention policy test (test_retention.sh) - All 10 security tests passing Features validated: ✅ Backup retention policy (30 days, min backups) ✅ Rate limiting (exponential backoff) ✅ Privilege checks (root detection) ✅ Resource limit validation ✅ Path sanitization ✅ Checksum verification (SHA-256) ✅ Audit logging ✅ Secure permissions ✅ Configuration persistence ✅ TUI automation framework Test results: 10/10 passed Backup files created with .dump, .sha256, .info Retention cleanup verified (old files removed)
115 lines
4.9 KiB
Go
Executable File
115 lines
4.9 KiB
Go
Executable File
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"dbbackup/internal/config"
|
|
"dbbackup/internal/logger"
|
|
"dbbackup/internal/security"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
cfg *config.Config
|
|
log logger.Logger
|
|
auditLogger *security.AuditLogger
|
|
rateLimiter *security.RateLimiter
|
|
)
|
|
|
|
// rootCmd represents the base command when called without any subcommands
|
|
var rootCmd = &cobra.Command{
|
|
Use: "dbbackup",
|
|
Short: "Multi-database backup and restore tool",
|
|
Long: `A comprehensive database backup and restore solution supporting both PostgreSQL and MySQL.
|
|
|
|
Features:
|
|
- CPU-aware parallel processing
|
|
- Multiple backup modes (cluster, single database, sample)
|
|
- Interactive UI and CLI modes
|
|
- Archive verification and restore
|
|
- Progress indicators and timing summaries
|
|
- Robust error handling and logging
|
|
|
|
Database Support:
|
|
- PostgreSQL (via pg_dump/pg_restore)
|
|
- MySQL (via mysqldump/mysql)
|
|
|
|
For help with specific commands, use: dbbackup [command] --help`,
|
|
Version: "",
|
|
PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
|
|
if cfg == nil {
|
|
return nil
|
|
}
|
|
|
|
// Load local config if not disabled
|
|
if !cfg.NoLoadConfig {
|
|
if localCfg, err := config.LoadLocalConfig(); err != nil {
|
|
log.Warn("Failed to load local config", "error", err)
|
|
} else if localCfg != nil {
|
|
config.ApplyLocalConfig(cfg, localCfg)
|
|
log.Info("Loaded configuration from .dbbackup.conf")
|
|
}
|
|
}
|
|
|
|
return cfg.SetDatabaseType(cfg.DatabaseType)
|
|
},
|
|
}
|
|
|
|
// Execute adds all child commands to the root command and sets flags appropriately.
|
|
func Execute(ctx context.Context, config *config.Config, logger logger.Logger) error {
|
|
cfg = config
|
|
log = logger
|
|
|
|
// Initialize audit logger
|
|
auditLogger = security.NewAuditLogger(logger, true)
|
|
|
|
// Initialize rate limiter
|
|
rateLimiter = security.NewRateLimiter(config.MaxRetries, logger)
|
|
|
|
// Set version info
|
|
rootCmd.Version = fmt.Sprintf("%s (built: %s, commit: %s)",
|
|
cfg.Version, cfg.BuildTime, cfg.GitCommit)
|
|
|
|
// Add persistent flags
|
|
rootCmd.PersistentFlags().StringVar(&cfg.Host, "host", cfg.Host, "Database host")
|
|
rootCmd.PersistentFlags().IntVar(&cfg.Port, "port", cfg.Port, "Database port")
|
|
rootCmd.PersistentFlags().StringVar(&cfg.User, "user", cfg.User, "Database user")
|
|
rootCmd.PersistentFlags().StringVar(&cfg.Database, "database", cfg.Database, "Database name")
|
|
rootCmd.PersistentFlags().StringVar(&cfg.Password, "password", cfg.Password, "Database password")
|
|
rootCmd.PersistentFlags().StringVarP(&cfg.DatabaseType, "db-type", "d", cfg.DatabaseType, "Database type (postgres|mysql|mariadb)")
|
|
rootCmd.PersistentFlags().StringVar(&cfg.BackupDir, "backup-dir", cfg.BackupDir, "Backup directory")
|
|
rootCmd.PersistentFlags().BoolVar(&cfg.NoColor, "no-color", cfg.NoColor, "Disable colored output")
|
|
rootCmd.PersistentFlags().BoolVar(&cfg.Debug, "debug", cfg.Debug, "Enable debug logging")
|
|
rootCmd.PersistentFlags().IntVar(&cfg.Jobs, "jobs", cfg.Jobs, "Number of parallel jobs")
|
|
rootCmd.PersistentFlags().IntVar(&cfg.DumpJobs, "dump-jobs", cfg.DumpJobs, "Number of parallel dump jobs")
|
|
rootCmd.PersistentFlags().IntVar(&cfg.MaxCores, "max-cores", cfg.MaxCores, "Maximum CPU cores to use")
|
|
rootCmd.PersistentFlags().BoolVar(&cfg.AutoDetectCores, "auto-detect-cores", cfg.AutoDetectCores, "Auto-detect CPU cores")
|
|
rootCmd.PersistentFlags().StringVar(&cfg.CPUWorkloadType, "cpu-workload", cfg.CPUWorkloadType, "CPU workload type (cpu-intensive|io-intensive|balanced)")
|
|
rootCmd.PersistentFlags().StringVar(&cfg.SSLMode, "ssl-mode", cfg.SSLMode, "SSL mode for connections")
|
|
rootCmd.PersistentFlags().BoolVar(&cfg.Insecure, "insecure", cfg.Insecure, "Disable SSL (shortcut for --ssl-mode=disable)")
|
|
rootCmd.PersistentFlags().IntVar(&cfg.CompressionLevel, "compression", cfg.CompressionLevel, "Compression level (0-9)")
|
|
rootCmd.PersistentFlags().BoolVar(&cfg.NoSaveConfig, "no-save-config", false, "Don't save configuration after successful operations")
|
|
rootCmd.PersistentFlags().BoolVar(&cfg.NoLoadConfig, "no-config", false, "Don't load configuration from .dbbackup.conf")
|
|
|
|
// Security flags (MEDIUM priority)
|
|
rootCmd.PersistentFlags().IntVar(&cfg.RetentionDays, "retention-days", cfg.RetentionDays, "Backup retention period in days (0=disabled)")
|
|
rootCmd.PersistentFlags().IntVar(&cfg.MinBackups, "min-backups", cfg.MinBackups, "Minimum number of backups to keep")
|
|
rootCmd.PersistentFlags().IntVar(&cfg.MaxRetries, "max-retries", cfg.MaxRetries, "Maximum connection retry attempts")
|
|
rootCmd.PersistentFlags().BoolVar(&cfg.AllowRoot, "allow-root", cfg.AllowRoot, "Allow running as root/Administrator")
|
|
rootCmd.PersistentFlags().BoolVar(&cfg.CheckResources, "check-resources", cfg.CheckResources, "Check system resource limits")
|
|
|
|
return rootCmd.ExecuteContext(ctx)
|
|
}
|
|
|
|
func init() {
|
|
// Register subcommands
|
|
rootCmd.AddCommand(backupCmd)
|
|
rootCmd.AddCommand(restoreCmd)
|
|
rootCmd.AddCommand(verifyCmd)
|
|
rootCmd.AddCommand(listCmd)
|
|
rootCmd.AddCommand(interactiveCmd)
|
|
rootCmd.AddCommand(statusCmd)
|
|
rootCmd.AddCommand(preflightCmd)
|
|
}
|