Compare commits
8 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 04bf2c61c5 | |||
| e05adcab2b | |||
| 7b62aa005e | |||
| 39efb82678 | |||
| 93d80ca4d2 | |||
| 7e764d000d | |||
| dc12a8e4b0 | |||
| f69a8e374b |
68
cmd/catalog_dashboard.go
Normal file
68
cmd/catalog_dashboard.go
Normal file
@ -0,0 +1,68 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"dbbackup/internal/tui"
|
||||
|
||||
tea "github.com/charmbracelet/bubbletea"
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
var catalogDashboardCmd = &cobra.Command{
|
||||
Use: "dashboard",
|
||||
Short: "Interactive catalog browser (TUI)",
|
||||
Long: `Launch an interactive terminal UI for browsing and managing backup catalog.
|
||||
|
||||
The catalog dashboard provides:
|
||||
- Browse all backups in an interactive table
|
||||
- Sort by date, size, database, or type
|
||||
- Filter backups by database or search term
|
||||
- View detailed backup information
|
||||
- Pagination for large catalogs
|
||||
- Real-time statistics
|
||||
|
||||
Navigation:
|
||||
↑/↓ or k/j - Navigate entries
|
||||
←/→ or h/l - Previous/next page
|
||||
Enter - View backup details
|
||||
s - Cycle sort (date → size → database → type)
|
||||
r - Reverse sort order
|
||||
d - Filter by database (cycle through)
|
||||
/ - Search/filter
|
||||
c - Clear filters
|
||||
R - Reload catalog
|
||||
q or ESC - Quit (or return from details)
|
||||
|
||||
Examples:
|
||||
# Launch catalog dashboard
|
||||
dbbackup catalog dashboard
|
||||
|
||||
# Dashboard shows:
|
||||
# - Total backups and size
|
||||
# - Sortable table with all backups
|
||||
# - Pagination controls
|
||||
# - Interactive filtering`,
|
||||
RunE: runCatalogDashboard,
|
||||
}
|
||||
|
||||
func init() {
|
||||
catalogCmd.AddCommand(catalogDashboardCmd)
|
||||
}
|
||||
|
||||
func runCatalogDashboard(cmd *cobra.Command, args []string) error {
|
||||
// Check if we're in a terminal
|
||||
if !tui.IsInteractiveTerminal() {
|
||||
return fmt.Errorf("catalog dashboard requires an interactive terminal")
|
||||
}
|
||||
|
||||
// Create and run the TUI
|
||||
model := tui.NewCatalogDashboardView()
|
||||
p := tea.NewProgram(model, tea.WithAltScreen())
|
||||
|
||||
if _, err := p.Run(); err != nil {
|
||||
return fmt.Errorf("failed to run catalog dashboard: %w", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
@ -125,7 +125,7 @@ func init() {
|
||||
cloudCmd.AddCommand(cloudUploadCmd, cloudDownloadCmd, cloudListCmd, cloudDeleteCmd)
|
||||
|
||||
// Cloud configuration flags
|
||||
for _, cmd := range []*cobra.Command{cloudUploadCmd, cloudDownloadCmd, cloudListCmd, cloudDeleteCmd} {
|
||||
for _, cmd := range []*cobra.Command{cloudUploadCmd, cloudDownloadCmd, cloudListCmd, cloudDeleteCmd, cloudStatusCmd} {
|
||||
cmd.Flags().StringVar(&cloudProvider, "cloud-provider", getEnv("DBBACKUP_CLOUD_PROVIDER", "s3"), "Cloud provider (s3, minio, b2)")
|
||||
cmd.Flags().StringVar(&cloudBucket, "cloud-bucket", getEnv("DBBACKUP_CLOUD_BUCKET", ""), "Bucket name")
|
||||
cmd.Flags().StringVar(&cloudRegion, "cloud-region", getEnv("DBBACKUP_CLOUD_REGION", "us-east-1"), "Region")
|
||||
|
||||
460
cmd/cloud_status.go
Normal file
460
cmd/cloud_status.go
Normal file
@ -0,0 +1,460 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"dbbackup/internal/cloud"
|
||||
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
var cloudStatusCmd = &cobra.Command{
|
||||
Use: "status",
|
||||
Short: "Check cloud storage connectivity and status",
|
||||
Long: `Check cloud storage connectivity, credentials, and bucket access.
|
||||
|
||||
This command verifies:
|
||||
- Cloud provider configuration
|
||||
- Authentication/credentials
|
||||
- Bucket/container existence and access
|
||||
- List capabilities (read permissions)
|
||||
- Upload capabilities (write permissions)
|
||||
- Network connectivity
|
||||
- Response times
|
||||
|
||||
Supports:
|
||||
- AWS S3
|
||||
- Google Cloud Storage (GCS)
|
||||
- Azure Blob Storage
|
||||
- MinIO
|
||||
- Backblaze B2
|
||||
|
||||
Examples:
|
||||
# Check configured cloud storage
|
||||
dbbackup cloud status
|
||||
|
||||
# Check with JSON output
|
||||
dbbackup cloud status --format json
|
||||
|
||||
# Quick check (skip upload test)
|
||||
dbbackup cloud status --quick
|
||||
|
||||
# Verbose diagnostics
|
||||
dbbackup cloud status --verbose`,
|
||||
RunE: runCloudStatus,
|
||||
}
|
||||
|
||||
var (
|
||||
cloudStatusFormat string
|
||||
cloudStatusQuick bool
|
||||
// cloudStatusVerbose uses the global cloudVerbose flag from cloud.go
|
||||
)
|
||||
|
||||
type CloudStatus struct {
|
||||
Provider string `json:"provider"`
|
||||
Bucket string `json:"bucket"`
|
||||
Region string `json:"region,omitempty"`
|
||||
Endpoint string `json:"endpoint,omitempty"`
|
||||
Connected bool `json:"connected"`
|
||||
BucketExists bool `json:"bucket_exists"`
|
||||
CanList bool `json:"can_list"`
|
||||
CanUpload bool `json:"can_upload"`
|
||||
ObjectCount int `json:"object_count,omitempty"`
|
||||
TotalSize int64 `json:"total_size_bytes,omitempty"`
|
||||
LatencyMs int64 `json:"latency_ms,omitempty"`
|
||||
Error string `json:"error,omitempty"`
|
||||
Checks []CloudStatusCheck `json:"checks"`
|
||||
Details map[string]interface{} `json:"details,omitempty"`
|
||||
}
|
||||
|
||||
type CloudStatusCheck struct {
|
||||
Name string `json:"name"`
|
||||
Status string `json:"status"` // "pass", "fail", "skip"
|
||||
Message string `json:"message,omitempty"`
|
||||
Error string `json:"error,omitempty"`
|
||||
}
|
||||
|
||||
func init() {
|
||||
cloudCmd.AddCommand(cloudStatusCmd)
|
||||
|
||||
cloudStatusCmd.Flags().StringVar(&cloudStatusFormat, "format", "table", "Output format (table, json)")
|
||||
cloudStatusCmd.Flags().BoolVar(&cloudStatusQuick, "quick", false, "Quick check (skip upload test)")
|
||||
// Note: verbose flag is added by cloud.go init()
|
||||
}
|
||||
|
||||
func runCloudStatus(cmd *cobra.Command, args []string) error {
|
||||
if !cfg.CloudEnabled {
|
||||
fmt.Println("[WARN] Cloud storage is not enabled")
|
||||
fmt.Println("Enable with: --cloud-enabled")
|
||||
fmt.Println()
|
||||
fmt.Println("Example configuration:")
|
||||
fmt.Println(" cloud_enabled = true")
|
||||
fmt.Println(" cloud_provider = \"s3\" # s3, gcs, azure, minio, b2")
|
||||
fmt.Println(" cloud_bucket = \"my-backups\"")
|
||||
fmt.Println(" cloud_region = \"us-east-1\" # for S3/GCS")
|
||||
fmt.Println(" cloud_access_key = \"...\"")
|
||||
fmt.Println(" cloud_secret_key = \"...\"")
|
||||
return nil
|
||||
}
|
||||
|
||||
status := &CloudStatus{
|
||||
Provider: cfg.CloudProvider,
|
||||
Bucket: cfg.CloudBucket,
|
||||
Region: cfg.CloudRegion,
|
||||
Endpoint: cfg.CloudEndpoint,
|
||||
Checks: []CloudStatusCheck{},
|
||||
Details: make(map[string]interface{}),
|
||||
}
|
||||
|
||||
fmt.Println("[CHECK] Cloud Storage Status")
|
||||
fmt.Println()
|
||||
fmt.Printf("Provider: %s\n", cfg.CloudProvider)
|
||||
fmt.Printf("Bucket: %s\n", cfg.CloudBucket)
|
||||
if cfg.CloudRegion != "" {
|
||||
fmt.Printf("Region: %s\n", cfg.CloudRegion)
|
||||
}
|
||||
if cfg.CloudEndpoint != "" {
|
||||
fmt.Printf("Endpoint: %s\n", cfg.CloudEndpoint)
|
||||
}
|
||||
fmt.Println()
|
||||
|
||||
// Check configuration
|
||||
checkConfig(status)
|
||||
|
||||
// Initialize cloud storage
|
||||
ctx := context.Background()
|
||||
|
||||
startTime := time.Now()
|
||||
|
||||
// Create cloud config
|
||||
cloudCfg := &cloud.Config{
|
||||
Provider: cfg.CloudProvider,
|
||||
Bucket: cfg.CloudBucket,
|
||||
Region: cfg.CloudRegion,
|
||||
Endpoint: cfg.CloudEndpoint,
|
||||
AccessKey: cfg.CloudAccessKey,
|
||||
SecretKey: cfg.CloudSecretKey,
|
||||
UseSSL: true,
|
||||
PathStyle: cfg.CloudProvider == "minio",
|
||||
Prefix: cfg.CloudPrefix,
|
||||
Timeout: 300,
|
||||
MaxRetries: 3,
|
||||
}
|
||||
|
||||
backend, err := cloud.NewBackend(cloudCfg)
|
||||
if err != nil {
|
||||
status.Connected = false
|
||||
status.Error = fmt.Sprintf("Failed to initialize cloud storage: %v", err)
|
||||
status.Checks = append(status.Checks, CloudStatusCheck{
|
||||
Name: "Initialize",
|
||||
Status: "fail",
|
||||
Error: err.Error(),
|
||||
})
|
||||
|
||||
printStatus(status)
|
||||
return fmt.Errorf("cloud storage initialization failed: %w", err)
|
||||
}
|
||||
|
||||
initDuration := time.Since(startTime)
|
||||
status.Details["init_time_ms"] = initDuration.Milliseconds()
|
||||
|
||||
if cloudVerbose {
|
||||
fmt.Printf("[DEBUG] Initialization took %s\n", initDuration.Round(time.Millisecond))
|
||||
}
|
||||
|
||||
status.Connected = true
|
||||
status.Checks = append(status.Checks, CloudStatusCheck{
|
||||
Name: "Initialize",
|
||||
Status: "pass",
|
||||
Message: fmt.Sprintf("Connected (%s)", initDuration.Round(time.Millisecond)),
|
||||
})
|
||||
|
||||
// Test bucket existence (via list operation)
|
||||
checkBucketAccess(ctx, backend, status)
|
||||
|
||||
// Test list permissions
|
||||
checkListPermissions(ctx, backend, status)
|
||||
|
||||
// Test upload permissions (unless quick mode)
|
||||
if !cloudStatusQuick {
|
||||
checkUploadPermissions(ctx, backend, status)
|
||||
} else {
|
||||
status.Checks = append(status.Checks, CloudStatusCheck{
|
||||
Name: "Upload",
|
||||
Status: "skip",
|
||||
Message: "Skipped (--quick mode)",
|
||||
})
|
||||
}
|
||||
|
||||
// Calculate overall latency
|
||||
totalLatency := int64(0)
|
||||
for _, check := range status.Checks {
|
||||
if check.Status == "pass" {
|
||||
totalLatency++
|
||||
}
|
||||
}
|
||||
if totalLatency > 0 {
|
||||
status.LatencyMs = initDuration.Milliseconds()
|
||||
}
|
||||
|
||||
// Output results
|
||||
if cloudStatusFormat == "json" {
|
||||
enc := json.NewEncoder(os.Stdout)
|
||||
enc.SetIndent("", " ")
|
||||
return enc.Encode(status)
|
||||
}
|
||||
|
||||
printStatus(status)
|
||||
|
||||
// Return error if any checks failed
|
||||
for _, check := range status.Checks {
|
||||
if check.Status == "fail" {
|
||||
return fmt.Errorf("cloud status check failed")
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func checkConfig(status *CloudStatus) {
|
||||
if status.Provider == "" {
|
||||
status.Checks = append(status.Checks, CloudStatusCheck{
|
||||
Name: "Configuration",
|
||||
Status: "fail",
|
||||
Error: "Cloud provider not configured",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
if status.Bucket == "" {
|
||||
status.Checks = append(status.Checks, CloudStatusCheck{
|
||||
Name: "Configuration",
|
||||
Status: "fail",
|
||||
Error: "Bucket/container name not configured",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
status.Checks = append(status.Checks, CloudStatusCheck{
|
||||
Name: "Configuration",
|
||||
Status: "pass",
|
||||
Message: fmt.Sprintf("%s / %s", status.Provider, status.Bucket),
|
||||
})
|
||||
}
|
||||
|
||||
func checkBucketAccess(ctx context.Context, backend cloud.Backend, status *CloudStatus) {
|
||||
fmt.Print("[TEST] Checking bucket access... ")
|
||||
|
||||
startTime := time.Now()
|
||||
// Try to list - this will fail if bucket doesn't exist or no access
|
||||
_, err := backend.List(ctx, "")
|
||||
duration := time.Since(startTime)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("[FAIL] %v\n", err)
|
||||
status.BucketExists = false
|
||||
status.Checks = append(status.Checks, CloudStatusCheck{
|
||||
Name: "Bucket Access",
|
||||
Status: "fail",
|
||||
Error: err.Error(),
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
fmt.Printf("[OK] (%s)\n", duration.Round(time.Millisecond))
|
||||
status.BucketExists = true
|
||||
status.Checks = append(status.Checks, CloudStatusCheck{
|
||||
Name: "Bucket Access",
|
||||
Status: "pass",
|
||||
Message: fmt.Sprintf("Accessible (%s)", duration.Round(time.Millisecond)),
|
||||
})
|
||||
}
|
||||
|
||||
func checkListPermissions(ctx context.Context, backend cloud.Backend, status *CloudStatus) {
|
||||
fmt.Print("[TEST] Checking list permissions... ")
|
||||
|
||||
startTime := time.Now()
|
||||
objects, err := backend.List(ctx, cfg.CloudPrefix)
|
||||
duration := time.Since(startTime)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("[FAIL] %v\n", err)
|
||||
status.CanList = false
|
||||
status.Checks = append(status.Checks, CloudStatusCheck{
|
||||
Name: "List Objects",
|
||||
Status: "fail",
|
||||
Error: err.Error(),
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
fmt.Printf("[OK] Found %d object(s) (%s)\n", len(objects), duration.Round(time.Millisecond))
|
||||
status.CanList = true
|
||||
status.ObjectCount = len(objects)
|
||||
|
||||
// Calculate total size
|
||||
var totalSize int64
|
||||
for _, obj := range objects {
|
||||
totalSize += obj.Size
|
||||
}
|
||||
status.TotalSize = totalSize
|
||||
|
||||
status.Checks = append(status.Checks, CloudStatusCheck{
|
||||
Name: "List Objects",
|
||||
Status: "pass",
|
||||
Message: fmt.Sprintf("%d objects, %s total (%s)", len(objects), formatCloudBytes(totalSize), duration.Round(time.Millisecond)),
|
||||
})
|
||||
|
||||
if cloudVerbose && len(objects) > 0 {
|
||||
fmt.Println("\n[OBJECTS]")
|
||||
limit := 5
|
||||
for i, obj := range objects {
|
||||
if i >= limit {
|
||||
fmt.Printf(" ... and %d more\n", len(objects)-limit)
|
||||
break
|
||||
}
|
||||
fmt.Printf(" %s (%s, %s)\n", obj.Key, formatCloudBytes(obj.Size), obj.LastModified.Format("2006-01-02 15:04"))
|
||||
}
|
||||
fmt.Println()
|
||||
}
|
||||
}
|
||||
|
||||
func checkUploadPermissions(ctx context.Context, backend cloud.Backend, status *CloudStatus) {
|
||||
fmt.Print("[TEST] Checking upload permissions... ")
|
||||
|
||||
// Create a small test file
|
||||
testKey := cfg.CloudPrefix + "/.dbbackup-test-" + time.Now().Format("20060102150405")
|
||||
testData := []byte("dbbackup cloud status test")
|
||||
|
||||
// Create temp file for upload
|
||||
tmpFile, err := os.CreateTemp("", "dbbackup-test-*")
|
||||
if err != nil {
|
||||
fmt.Printf("[FAIL] Could not create test file: %v\n", err)
|
||||
status.Checks = append(status.Checks, CloudStatusCheck{
|
||||
Name: "Upload Test",
|
||||
Status: "fail",
|
||||
Error: fmt.Sprintf("temp file creation failed: %v", err),
|
||||
})
|
||||
return
|
||||
}
|
||||
defer os.Remove(tmpFile.Name())
|
||||
|
||||
if _, err := tmpFile.Write(testData); err != nil {
|
||||
tmpFile.Close()
|
||||
fmt.Printf("[FAIL] Could not write test file: %v\n", err)
|
||||
status.Checks = append(status.Checks, CloudStatusCheck{
|
||||
Name: "Upload Test",
|
||||
Status: "fail",
|
||||
Error: fmt.Sprintf("test file write failed: %v", err),
|
||||
})
|
||||
return
|
||||
}
|
||||
tmpFile.Close()
|
||||
|
||||
startTime := time.Now()
|
||||
err = backend.Upload(ctx, tmpFile.Name(), testKey, nil)
|
||||
uploadDuration := time.Since(startTime)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("[FAIL] %v\n", err)
|
||||
status.CanUpload = false
|
||||
status.Checks = append(status.Checks, CloudStatusCheck{
|
||||
Name: "Upload Test",
|
||||
Status: "fail",
|
||||
Error: err.Error(),
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
fmt.Printf("[OK] Test file uploaded (%s)\n", uploadDuration.Round(time.Millisecond))
|
||||
|
||||
// Try to delete the test file
|
||||
fmt.Print("[TEST] Checking delete permissions... ")
|
||||
deleteStartTime := time.Now()
|
||||
err = backend.Delete(ctx, testKey)
|
||||
deleteDuration := time.Since(deleteStartTime)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("[WARN] Could not delete test file: %v\n", err)
|
||||
status.Checks = append(status.Checks, CloudStatusCheck{
|
||||
Name: "Upload Test",
|
||||
Status: "pass",
|
||||
Message: fmt.Sprintf("Upload OK (%s), delete failed", uploadDuration.Round(time.Millisecond)),
|
||||
})
|
||||
} else {
|
||||
fmt.Printf("[OK] Test file deleted (%s)\n", deleteDuration.Round(time.Millisecond))
|
||||
status.CanUpload = true
|
||||
status.Checks = append(status.Checks, CloudStatusCheck{
|
||||
Name: "Upload/Delete Test",
|
||||
Status: "pass",
|
||||
Message: fmt.Sprintf("Both successful (upload: %s, delete: %s)",
|
||||
uploadDuration.Round(time.Millisecond),
|
||||
deleteDuration.Round(time.Millisecond)),
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func printStatus(status *CloudStatus) {
|
||||
fmt.Println("\n[RESULTS]")
|
||||
fmt.Println("================================================")
|
||||
|
||||
for _, check := range status.Checks {
|
||||
var statusStr string
|
||||
switch check.Status {
|
||||
case "pass":
|
||||
statusStr = "[OK] "
|
||||
case "fail":
|
||||
statusStr = "[FAIL]"
|
||||
case "skip":
|
||||
statusStr = "[SKIP]"
|
||||
}
|
||||
|
||||
fmt.Printf(" %-20s %s", check.Name+":", statusStr)
|
||||
if check.Message != "" {
|
||||
fmt.Printf(" %s", check.Message)
|
||||
}
|
||||
if check.Error != "" {
|
||||
fmt.Printf(" - %s", check.Error)
|
||||
}
|
||||
fmt.Println()
|
||||
}
|
||||
|
||||
fmt.Println("================================================")
|
||||
|
||||
if status.CanList && status.ObjectCount > 0 {
|
||||
fmt.Printf("\nStorage Usage: %d object(s), %s total\n", status.ObjectCount, formatCloudBytes(status.TotalSize))
|
||||
}
|
||||
|
||||
// Overall status
|
||||
fmt.Println()
|
||||
allPassed := true
|
||||
for _, check := range status.Checks {
|
||||
if check.Status == "fail" {
|
||||
allPassed = false
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if allPassed {
|
||||
fmt.Println("[OK] All checks passed - cloud storage is ready")
|
||||
} else {
|
||||
fmt.Println("[FAIL] Some checks failed - review configuration")
|
||||
}
|
||||
}
|
||||
|
||||
func formatCloudBytes(bytes int64) string {
|
||||
const unit = 1024
|
||||
if bytes < unit {
|
||||
return fmt.Sprintf("%d B", bytes)
|
||||
}
|
||||
div, exp := int64(unit), 0
|
||||
for n := bytes / unit; n >= unit; n /= unit {
|
||||
div *= unit
|
||||
exp++
|
||||
}
|
||||
return fmt.Sprintf("%.1f %cB", float64(bytes)/float64(div), "KMGTPE"[exp])
|
||||
}
|
||||
@ -7,8 +7,29 @@ import (
|
||||
"strings"
|
||||
|
||||
"dbbackup/internal/crypto"
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
var encryptionCmd = &cobra.Command{
|
||||
Use: "encryption",
|
||||
Short: "Encryption key management",
|
||||
Long: `Manage encryption keys for database backups.
|
||||
|
||||
This command group provides encryption key management utilities:
|
||||
- rotate: Generate new encryption keys and rotate existing ones
|
||||
|
||||
Examples:
|
||||
# Generate new encryption key
|
||||
dbbackup encryption rotate
|
||||
|
||||
# Show rotation workflow
|
||||
dbbackup encryption rotate --show-reencrypt`,
|
||||
}
|
||||
|
||||
func init() {
|
||||
rootCmd.AddCommand(encryptionCmd)
|
||||
}
|
||||
|
||||
// loadEncryptionKey loads encryption key from file or environment variable
|
||||
func loadEncryptionKey(keyFile, keyEnvVar string) ([]byte, error) {
|
||||
// Priority 1: Key file
|
||||
|
||||
223
cmd/encryption_rotate.go
Normal file
223
cmd/encryption_rotate.go
Normal file
@ -0,0 +1,223 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"crypto/rand"
|
||||
"encoding/base64"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"time"
|
||||
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
var encryptionRotateCmd = &cobra.Command{
|
||||
Use: "rotate",
|
||||
Short: "Rotate encryption keys",
|
||||
Long: `Generate new encryption keys and provide migration instructions.
|
||||
|
||||
This command helps with encryption key management:
|
||||
- Generates new secure encryption keys
|
||||
- Provides safe key rotation workflow
|
||||
- Creates backup of old keys
|
||||
- Shows re-encryption commands for existing backups
|
||||
|
||||
Key Rotation Workflow:
|
||||
1. Generate new key with this command
|
||||
2. Back up existing backups with old key
|
||||
3. Update configuration with new key
|
||||
4. Re-encrypt old backups (optional)
|
||||
5. Securely delete old key
|
||||
|
||||
Security Best Practices:
|
||||
- Rotate keys every 90-365 days
|
||||
- Never store keys in version control
|
||||
- Use key management systems (AWS KMS, HashiCorp Vault)
|
||||
- Keep old keys until all backups are re-encrypted
|
||||
- Test decryption before deleting old keys
|
||||
|
||||
Examples:
|
||||
# Generate new encryption key
|
||||
dbbackup encryption rotate
|
||||
|
||||
# Generate key with specific strength
|
||||
dbbackup encryption rotate --key-size 256
|
||||
|
||||
# Save key to file
|
||||
dbbackup encryption rotate --output /secure/path/new.key
|
||||
|
||||
# Show re-encryption commands
|
||||
dbbackup encryption rotate --show-reencrypt`,
|
||||
RunE: runEncryptionRotate,
|
||||
}
|
||||
|
||||
var (
|
||||
rotateKeySize int
|
||||
rotateOutput string
|
||||
rotateShowReencrypt bool
|
||||
rotateFormat string
|
||||
)
|
||||
|
||||
func init() {
|
||||
encryptionCmd.AddCommand(encryptionRotateCmd)
|
||||
|
||||
encryptionRotateCmd.Flags().IntVar(&rotateKeySize, "key-size", 256, "Key size in bits (128, 192, 256)")
|
||||
encryptionRotateCmd.Flags().StringVar(&rotateOutput, "output", "", "Save new key to file (default: display only)")
|
||||
encryptionRotateCmd.Flags().BoolVar(&rotateShowReencrypt, "show-reencrypt", true, "Show re-encryption commands")
|
||||
encryptionRotateCmd.Flags().StringVar(&rotateFormat, "format", "base64", "Key format (base64, hex)")
|
||||
}
|
||||
|
||||
func runEncryptionRotate(cmd *cobra.Command, args []string) error {
|
||||
fmt.Println("[KEY ROTATION] Encryption Key Management")
|
||||
fmt.Println("=========================================")
|
||||
fmt.Println()
|
||||
|
||||
// Validate key size
|
||||
if rotateKeySize != 128 && rotateKeySize != 192 && rotateKeySize != 256 {
|
||||
return fmt.Errorf("invalid key size: %d (must be 128, 192, or 256)", rotateKeySize)
|
||||
}
|
||||
|
||||
keyBytes := rotateKeySize / 8
|
||||
|
||||
// Generate new key
|
||||
fmt.Printf("[GENERATE] Creating new %d-bit encryption key...\n", rotateKeySize)
|
||||
|
||||
key := make([]byte, keyBytes)
|
||||
if _, err := rand.Read(key); err != nil {
|
||||
return fmt.Errorf("failed to generate random key: %w", err)
|
||||
}
|
||||
|
||||
// Format key
|
||||
var keyString string
|
||||
switch rotateFormat {
|
||||
case "base64":
|
||||
keyString = base64.StdEncoding.EncodeToString(key)
|
||||
case "hex":
|
||||
keyString = fmt.Sprintf("%x", key)
|
||||
default:
|
||||
return fmt.Errorf("invalid format: %s (use base64 or hex)", rotateFormat)
|
||||
}
|
||||
|
||||
fmt.Println("[OK] New encryption key generated")
|
||||
fmt.Println()
|
||||
|
||||
// Display new key
|
||||
fmt.Println("[NEW KEY]")
|
||||
fmt.Println("=========================================")
|
||||
fmt.Printf("Format: %s\n", rotateFormat)
|
||||
fmt.Printf("Size: %d bits (%d bytes)\n", rotateKeySize, keyBytes)
|
||||
fmt.Printf("Generated: %s\n", time.Now().Format(time.RFC3339))
|
||||
fmt.Println()
|
||||
fmt.Println("Key:")
|
||||
fmt.Printf(" %s\n", keyString)
|
||||
fmt.Println()
|
||||
|
||||
// Save to file if requested
|
||||
if rotateOutput != "" {
|
||||
if err := saveKeyToFile(rotateOutput, keyString); err != nil {
|
||||
return fmt.Errorf("failed to save key: %w", err)
|
||||
}
|
||||
fmt.Printf("[SAVED] Key written to: %s\n", rotateOutput)
|
||||
fmt.Println("[WARN] Secure this file with proper permissions!")
|
||||
fmt.Printf(" chmod 600 %s\n", rotateOutput)
|
||||
fmt.Println()
|
||||
}
|
||||
|
||||
// Show rotation workflow
|
||||
fmt.Println("[KEY ROTATION WORKFLOW]")
|
||||
fmt.Println("=========================================")
|
||||
fmt.Println()
|
||||
fmt.Println("1. [BACKUP] Back up your old key:")
|
||||
fmt.Println(" export OLD_KEY=\"$DBBACKUP_ENCRYPTION_KEY\"")
|
||||
fmt.Println(" echo $OLD_KEY > /secure/backup/old-key.txt")
|
||||
fmt.Println()
|
||||
fmt.Println("2. [UPDATE] Update your configuration:")
|
||||
if rotateOutput != "" {
|
||||
fmt.Printf(" export DBBACKUP_ENCRYPTION_KEY=$(cat %s)\n", rotateOutput)
|
||||
} else {
|
||||
fmt.Printf(" export DBBACKUP_ENCRYPTION_KEY=\"%s\"\n", keyString)
|
||||
}
|
||||
fmt.Println(" # Or update .dbbackup.conf or systemd environment")
|
||||
fmt.Println()
|
||||
fmt.Println("3. [VERIFY] Test new key with a backup:")
|
||||
fmt.Println(" dbbackup backup single testdb --encryption-key-env DBBACKUP_ENCRYPTION_KEY")
|
||||
fmt.Println()
|
||||
fmt.Println("4. [RE-ENCRYPT] Re-encrypt existing backups (optional):")
|
||||
if rotateShowReencrypt {
|
||||
showReencryptCommands()
|
||||
}
|
||||
fmt.Println()
|
||||
fmt.Println("5. [CLEANUP] After all backups re-encrypted:")
|
||||
fmt.Println(" # Securely delete old key")
|
||||
fmt.Println(" shred -u /secure/backup/old-key.txt")
|
||||
fmt.Println(" unset OLD_KEY")
|
||||
fmt.Println()
|
||||
|
||||
// Security warnings
|
||||
fmt.Println("[SECURITY WARNINGS]")
|
||||
fmt.Println("=========================================")
|
||||
fmt.Println()
|
||||
fmt.Println("⚠ DO NOT store keys in:")
|
||||
fmt.Println(" - Version control (git, svn)")
|
||||
fmt.Println(" - Unencrypted files")
|
||||
fmt.Println(" - Email or chat logs")
|
||||
fmt.Println(" - Shell history (use env vars)")
|
||||
fmt.Println()
|
||||
fmt.Println("✓ DO store keys in:")
|
||||
fmt.Println(" - Hardware Security Modules (HSM)")
|
||||
fmt.Println(" - Key Management Systems (AWS KMS, Vault)")
|
||||
fmt.Println(" - Encrypted password managers")
|
||||
fmt.Println(" - Encrypted environment files (0600 permissions)")
|
||||
fmt.Println()
|
||||
fmt.Println("✓ Key Rotation Schedule:")
|
||||
fmt.Println(" - Production: Every 90 days")
|
||||
fmt.Println(" - Development: Every 180 days")
|
||||
fmt.Println(" - After security incident: Immediately")
|
||||
fmt.Println()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func saveKeyToFile(path string, key string) error {
|
||||
// Create directory if needed
|
||||
dir := filepath.Dir(path)
|
||||
if err := os.MkdirAll(dir, 0700); err != nil {
|
||||
return fmt.Errorf("failed to create directory: %w", err)
|
||||
}
|
||||
|
||||
// Write key file with restricted permissions
|
||||
if err := os.WriteFile(path, []byte(key+"\n"), 0600); err != nil {
|
||||
return fmt.Errorf("failed to write file: %w", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func showReencryptCommands() {
|
||||
fmt.Println(" # Option A: Re-encrypt with openssl")
|
||||
fmt.Println(" for backup in /path/to/backups/*.enc; do")
|
||||
fmt.Println(" # Decrypt with old key")
|
||||
fmt.Println(" openssl enc -aes-256-cbc -d \\")
|
||||
fmt.Println(" -in \"$backup\" \\")
|
||||
fmt.Println(" -out \"${backup%.enc}.tmp\" \\")
|
||||
fmt.Println(" -k \"$OLD_KEY\"")
|
||||
fmt.Println()
|
||||
fmt.Println(" # Encrypt with new key")
|
||||
fmt.Println(" openssl enc -aes-256-cbc \\")
|
||||
fmt.Println(" -in \"${backup%.enc}.tmp\" \\")
|
||||
fmt.Println(" -out \"${backup}.new\" \\")
|
||||
fmt.Println(" -k \"$DBBACKUP_ENCRYPTION_KEY\"")
|
||||
fmt.Println()
|
||||
fmt.Println(" # Verify and replace")
|
||||
fmt.Println(" if [ -f \"${backup}.new\" ]; then")
|
||||
fmt.Println(" mv \"${backup}.new\" \"$backup\"")
|
||||
fmt.Println(" rm \"${backup%.enc}.tmp\"")
|
||||
fmt.Println(" fi")
|
||||
fmt.Println(" done")
|
||||
fmt.Println()
|
||||
fmt.Println(" # Option B: Decrypt and re-backup")
|
||||
fmt.Println(" # 1. Restore from old encrypted backups")
|
||||
fmt.Println(" # 2. Create new backups with new key")
|
||||
fmt.Println(" # 3. Verify new backups")
|
||||
fmt.Println(" # 4. Delete old backups")
|
||||
}
|
||||
443
cmd/forecast.go
Normal file
443
cmd/forecast.go
Normal file
@ -0,0 +1,443 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"math"
|
||||
"os"
|
||||
"strings"
|
||||
"text/tabwriter"
|
||||
"time"
|
||||
|
||||
"dbbackup/internal/catalog"
|
||||
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
var forecastCmd = &cobra.Command{
|
||||
Use: "forecast [database]",
|
||||
Short: "Predict future disk space requirements",
|
||||
Long: `Analyze backup growth patterns and predict future disk space needs.
|
||||
|
||||
This command helps with:
|
||||
- Capacity planning (when will we run out of space?)
|
||||
- Budget forecasting (how much storage to provision?)
|
||||
- Growth trend analysis (is growth accelerating?)
|
||||
- Alert thresholds (when to add capacity?)
|
||||
|
||||
Uses historical backup data to calculate:
|
||||
- Average daily growth rate
|
||||
- Growth acceleration/deceleration
|
||||
- Time until space limit reached
|
||||
- Projected size at future dates
|
||||
|
||||
Examples:
|
||||
# Forecast for specific database
|
||||
dbbackup forecast mydb
|
||||
|
||||
# Forecast all databases
|
||||
dbbackup forecast --all
|
||||
|
||||
# Show projection for 90 days
|
||||
dbbackup forecast mydb --days 90
|
||||
|
||||
# Set capacity limit (alert when approaching)
|
||||
dbbackup forecast mydb --limit 100GB
|
||||
|
||||
# JSON output for automation
|
||||
dbbackup forecast mydb --format json`,
|
||||
Args: cobra.MaximumNArgs(1),
|
||||
RunE: runForecast,
|
||||
}
|
||||
|
||||
var (
|
||||
forecastFormat string
|
||||
forecastAll bool
|
||||
forecastDays int
|
||||
forecastLimitSize string
|
||||
)
|
||||
|
||||
type ForecastResult struct {
|
||||
Database string `json:"database"`
|
||||
CurrentSize int64 `json:"current_size_bytes"`
|
||||
TotalBackups int `json:"total_backups"`
|
||||
OldestBackup time.Time `json:"oldest_backup"`
|
||||
NewestBackup time.Time `json:"newest_backup"`
|
||||
ObservationPeriod time.Duration `json:"observation_period_seconds"`
|
||||
DailyGrowthRate float64 `json:"daily_growth_bytes"`
|
||||
DailyGrowthPct float64 `json:"daily_growth_percent"`
|
||||
Projections []ForecastProjection `json:"projections"`
|
||||
TimeToLimit *time.Duration `json:"time_to_limit_seconds,omitempty"`
|
||||
SizeAtLimit *time.Time `json:"date_reaching_limit,omitempty"`
|
||||
Confidence string `json:"confidence"` // "high", "medium", "low"
|
||||
}
|
||||
|
||||
type ForecastProjection struct {
|
||||
Days int `json:"days_from_now"`
|
||||
Date time.Time `json:"date"`
|
||||
PredictedSize int64 `json:"predicted_size_bytes"`
|
||||
Confidence float64 `json:"confidence_percent"`
|
||||
}
|
||||
|
||||
func init() {
|
||||
rootCmd.AddCommand(forecastCmd)
|
||||
|
||||
forecastCmd.Flags().StringVar(&forecastFormat, "format", "table", "Output format (table, json)")
|
||||
forecastCmd.Flags().BoolVar(&forecastAll, "all", false, "Show forecast for all databases")
|
||||
forecastCmd.Flags().IntVar(&forecastDays, "days", 90, "Days to project into future")
|
||||
forecastCmd.Flags().StringVar(&forecastLimitSize, "limit", "", "Capacity limit (e.g., '100GB', '1TB')")
|
||||
}
|
||||
|
||||
func runForecast(cmd *cobra.Command, args []string) error {
|
||||
cat, err := openCatalog()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer cat.Close()
|
||||
|
||||
ctx := context.Background()
|
||||
|
||||
var forecasts []*ForecastResult
|
||||
|
||||
if forecastAll || len(args) == 0 {
|
||||
// Get all databases
|
||||
databases, err := cat.ListDatabases(ctx)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, db := range databases {
|
||||
forecast, err := calculateForecast(ctx, cat, db)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if forecast != nil {
|
||||
forecasts = append(forecasts, forecast)
|
||||
}
|
||||
}
|
||||
} else {
|
||||
database := args[0]
|
||||
forecast, err := calculateForecast(ctx, cat, database)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if forecast != nil {
|
||||
forecasts = append(forecasts, forecast)
|
||||
}
|
||||
}
|
||||
|
||||
if len(forecasts) == 0 {
|
||||
fmt.Println("No forecast data available.")
|
||||
fmt.Println("\nRun 'dbbackup catalog sync <directory>' to import backups.")
|
||||
return nil
|
||||
}
|
||||
|
||||
// Parse limit if provided
|
||||
var limitBytes int64
|
||||
if forecastLimitSize != "" {
|
||||
limitBytes, err = parseSize(forecastLimitSize)
|
||||
if err != nil {
|
||||
return fmt.Errorf("invalid limit size: %w", err)
|
||||
}
|
||||
}
|
||||
|
||||
// Output results
|
||||
if forecastFormat == "json" {
|
||||
enc := json.NewEncoder(os.Stdout)
|
||||
enc.SetIndent("", " ")
|
||||
return enc.Encode(forecasts)
|
||||
}
|
||||
|
||||
// Table output
|
||||
for i, forecast := range forecasts {
|
||||
if i > 0 {
|
||||
fmt.Println()
|
||||
}
|
||||
printForecast(forecast, limitBytes)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func calculateForecast(ctx context.Context, cat *catalog.SQLiteCatalog, database string) (*ForecastResult, error) {
|
||||
// Get all backups for this database
|
||||
query := &catalog.SearchQuery{
|
||||
Database: database,
|
||||
Limit: 1000,
|
||||
OrderBy: "created_at",
|
||||
OrderDesc: false,
|
||||
}
|
||||
|
||||
entries, err := cat.Search(ctx, query)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if len(entries) < 2 {
|
||||
return nil, nil // Need at least 2 backups for growth rate
|
||||
}
|
||||
|
||||
// Calculate metrics
|
||||
var totalSize int64
|
||||
oldest := entries[0].CreatedAt
|
||||
newest := entries[len(entries)-1].CreatedAt
|
||||
|
||||
for _, entry := range entries {
|
||||
totalSize += entry.SizeBytes
|
||||
}
|
||||
|
||||
// Calculate observation period
|
||||
observationPeriod := newest.Sub(oldest)
|
||||
if observationPeriod == 0 {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
// Calculate daily growth rate
|
||||
firstSize := entries[0].SizeBytes
|
||||
lastSize := entries[len(entries)-1].SizeBytes
|
||||
sizeDelta := float64(lastSize - firstSize)
|
||||
|
||||
daysObserved := observationPeriod.Hours() / 24
|
||||
dailyGrowthRate := sizeDelta / daysObserved
|
||||
|
||||
// Calculate daily growth percentage
|
||||
var dailyGrowthPct float64
|
||||
if firstSize > 0 {
|
||||
dailyGrowthPct = (dailyGrowthRate / float64(firstSize)) * 100
|
||||
}
|
||||
|
||||
// Determine confidence based on sample size and consistency
|
||||
confidence := determineConfidence(entries, dailyGrowthRate)
|
||||
|
||||
// Generate projections
|
||||
projections := make([]ForecastProjection, 0)
|
||||
projectionDates := []int{7, 30, 60, 90, 180, 365}
|
||||
|
||||
if forecastDays > 0 {
|
||||
// Use user-specified days
|
||||
projectionDates = []int{forecastDays}
|
||||
if forecastDays > 30 {
|
||||
projectionDates = []int{7, 30, forecastDays}
|
||||
}
|
||||
}
|
||||
|
||||
for _, days := range projectionDates {
|
||||
if days > 365 && forecastDays == 90 {
|
||||
continue // Skip longer projections unless explicitly requested
|
||||
}
|
||||
|
||||
predictedSize := lastSize + int64(dailyGrowthRate*float64(days))
|
||||
if predictedSize < 0 {
|
||||
predictedSize = 0
|
||||
}
|
||||
|
||||
// Confidence decreases with time
|
||||
confidencePct := calculateConfidence(days, confidence)
|
||||
|
||||
projections = append(projections, ForecastProjection{
|
||||
Days: days,
|
||||
Date: newest.Add(time.Duration(days) * 24 * time.Hour),
|
||||
PredictedSize: predictedSize,
|
||||
Confidence: confidencePct,
|
||||
})
|
||||
}
|
||||
|
||||
result := &ForecastResult{
|
||||
Database: database,
|
||||
CurrentSize: lastSize,
|
||||
TotalBackups: len(entries),
|
||||
OldestBackup: oldest,
|
||||
NewestBackup: newest,
|
||||
ObservationPeriod: observationPeriod,
|
||||
DailyGrowthRate: dailyGrowthRate,
|
||||
DailyGrowthPct: dailyGrowthPct,
|
||||
Projections: projections,
|
||||
Confidence: confidence,
|
||||
}
|
||||
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func determineConfidence(entries []*catalog.Entry, avgGrowth float64) string {
|
||||
if len(entries) < 5 {
|
||||
return "low"
|
||||
}
|
||||
if len(entries) < 15 {
|
||||
return "medium"
|
||||
}
|
||||
|
||||
// Calculate variance in growth rates
|
||||
var variance float64
|
||||
for i := 1; i < len(entries); i++ {
|
||||
timeDiff := entries[i].CreatedAt.Sub(entries[i-1].CreatedAt).Hours() / 24
|
||||
if timeDiff == 0 {
|
||||
continue
|
||||
}
|
||||
sizeDiff := float64(entries[i].SizeBytes - entries[i-1].SizeBytes)
|
||||
growthRate := sizeDiff / timeDiff
|
||||
variance += math.Pow(growthRate-avgGrowth, 2)
|
||||
}
|
||||
variance /= float64(len(entries) - 1)
|
||||
stdDev := math.Sqrt(variance)
|
||||
|
||||
// If standard deviation is more than 50% of average growth, confidence is low
|
||||
if stdDev > math.Abs(avgGrowth)*0.5 {
|
||||
return "medium"
|
||||
}
|
||||
|
||||
return "high"
|
||||
}
|
||||
|
||||
func calculateConfidence(daysAhead int, baseConfidence string) float64 {
|
||||
var base float64
|
||||
switch baseConfidence {
|
||||
case "high":
|
||||
base = 95.0
|
||||
case "medium":
|
||||
base = 75.0
|
||||
case "low":
|
||||
base = 50.0
|
||||
}
|
||||
|
||||
// Decay confidence over time (10% per 30 days)
|
||||
decay := float64(daysAhead) / 30.0 * 10.0
|
||||
confidence := base - decay
|
||||
|
||||
if confidence < 30 {
|
||||
confidence = 30
|
||||
}
|
||||
return confidence
|
||||
}
|
||||
|
||||
func printForecast(f *ForecastResult, limitBytes int64) {
|
||||
fmt.Printf("[FORECAST] %s\n", f.Database)
|
||||
fmt.Println(strings.Repeat("=", 60))
|
||||
|
||||
fmt.Printf("\n[CURRENT STATE]\n")
|
||||
fmt.Printf(" Size: %s\n", catalog.FormatSize(f.CurrentSize))
|
||||
fmt.Printf(" Backups: %d backups\n", f.TotalBackups)
|
||||
fmt.Printf(" Observed: %s (%.0f days)\n",
|
||||
formatForecastDuration(f.ObservationPeriod),
|
||||
f.ObservationPeriod.Hours()/24)
|
||||
|
||||
fmt.Printf("\n[GROWTH RATE]\n")
|
||||
if f.DailyGrowthRate > 0 {
|
||||
fmt.Printf(" Daily: +%s/day (%.2f%%/day)\n",
|
||||
catalog.FormatSize(int64(f.DailyGrowthRate)), f.DailyGrowthPct)
|
||||
fmt.Printf(" Weekly: +%s/week\n", catalog.FormatSize(int64(f.DailyGrowthRate*7)))
|
||||
fmt.Printf(" Monthly: +%s/month\n", catalog.FormatSize(int64(f.DailyGrowthRate*30)))
|
||||
fmt.Printf(" Annual: +%s/year\n", catalog.FormatSize(int64(f.DailyGrowthRate*365)))
|
||||
} else if f.DailyGrowthRate < 0 {
|
||||
fmt.Printf(" Daily: %s/day (shrinking)\n", catalog.FormatSize(int64(f.DailyGrowthRate)))
|
||||
} else {
|
||||
fmt.Printf(" Daily: No growth detected\n")
|
||||
}
|
||||
fmt.Printf(" Confidence: %s (%d samples)\n", f.Confidence, f.TotalBackups)
|
||||
|
||||
if len(f.Projections) > 0 {
|
||||
fmt.Printf("\n[PROJECTIONS]\n")
|
||||
w := tabwriter.NewWriter(os.Stdout, 0, 0, 2, ' ', 0)
|
||||
fmt.Fprintf(w, " Days\tDate\tPredicted Size\tConfidence\n")
|
||||
fmt.Fprintf(w, " ----\t----\t--------------\t----------\n")
|
||||
|
||||
for _, proj := range f.Projections {
|
||||
fmt.Fprintf(w, " %d\t%s\t%s\t%.0f%%\n",
|
||||
proj.Days,
|
||||
proj.Date.Format("2006-01-02"),
|
||||
catalog.FormatSize(proj.PredictedSize),
|
||||
proj.Confidence)
|
||||
}
|
||||
w.Flush()
|
||||
}
|
||||
|
||||
// Check against limit
|
||||
if limitBytes > 0 {
|
||||
fmt.Printf("\n[CAPACITY LIMIT]\n")
|
||||
fmt.Printf(" Limit: %s\n", catalog.FormatSize(limitBytes))
|
||||
|
||||
currentPct := float64(f.CurrentSize) / float64(limitBytes) * 100
|
||||
fmt.Printf(" Current: %.1f%% used\n", currentPct)
|
||||
|
||||
if f.CurrentSize >= limitBytes {
|
||||
fmt.Printf(" Status: [WARN] LIMIT EXCEEDED\n")
|
||||
} else if currentPct >= 80 {
|
||||
fmt.Printf(" Status: [WARN] Approaching limit\n")
|
||||
} else {
|
||||
fmt.Printf(" Status: [OK] Within limit\n")
|
||||
}
|
||||
|
||||
// Calculate when we'll hit the limit
|
||||
if f.DailyGrowthRate > 0 {
|
||||
remaining := limitBytes - f.CurrentSize
|
||||
daysToLimit := float64(remaining) / f.DailyGrowthRate
|
||||
|
||||
if daysToLimit > 0 && daysToLimit < 1000 {
|
||||
dateAtLimit := f.NewestBackup.Add(time.Duration(daysToLimit*24) * time.Hour)
|
||||
fmt.Printf(" Estimated: Limit reached in %.0f days (%s)\n",
|
||||
daysToLimit, dateAtLimit.Format("2006-01-02"))
|
||||
|
||||
if daysToLimit < 30 {
|
||||
fmt.Printf(" Alert: [CRITICAL] Less than 30 days remaining!\n")
|
||||
} else if daysToLimit < 90 {
|
||||
fmt.Printf(" Alert: [WARN] Less than 90 days remaining\n")
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fmt.Println()
|
||||
}
|
||||
|
||||
func formatForecastDuration(d time.Duration) string {
|
||||
hours := d.Hours()
|
||||
if hours < 24 {
|
||||
return fmt.Sprintf("%.1f hours", hours)
|
||||
}
|
||||
days := hours / 24
|
||||
if days < 7 {
|
||||
return fmt.Sprintf("%.1f days", days)
|
||||
}
|
||||
weeks := days / 7
|
||||
if weeks < 4 {
|
||||
return fmt.Sprintf("%.1f weeks", weeks)
|
||||
}
|
||||
months := days / 30
|
||||
if months < 12 {
|
||||
return fmt.Sprintf("%.1f months", months)
|
||||
}
|
||||
years := days / 365
|
||||
return fmt.Sprintf("%.1f years", years)
|
||||
}
|
||||
|
||||
func parseSize(s string) (int64, error) {
|
||||
// Simple size parser (supports KB, MB, GB, TB)
|
||||
s = strings.ToUpper(strings.TrimSpace(s))
|
||||
|
||||
var multiplier int64 = 1
|
||||
var numStr string
|
||||
|
||||
if strings.HasSuffix(s, "TB") {
|
||||
multiplier = 1024 * 1024 * 1024 * 1024
|
||||
numStr = strings.TrimSuffix(s, "TB")
|
||||
} else if strings.HasSuffix(s, "GB") {
|
||||
multiplier = 1024 * 1024 * 1024
|
||||
numStr = strings.TrimSuffix(s, "GB")
|
||||
} else if strings.HasSuffix(s, "MB") {
|
||||
multiplier = 1024 * 1024
|
||||
numStr = strings.TrimSuffix(s, "MB")
|
||||
} else if strings.HasSuffix(s, "KB") {
|
||||
multiplier = 1024
|
||||
numStr = strings.TrimSuffix(s, "KB")
|
||||
} else {
|
||||
numStr = s
|
||||
}
|
||||
|
||||
var num float64
|
||||
_, err := fmt.Sscanf(numStr, "%f", &num)
|
||||
if err != nil {
|
||||
return 0, fmt.Errorf("invalid size format: %s", s)
|
||||
}
|
||||
|
||||
return int64(num * float64(multiplier)), nil
|
||||
}
|
||||
154
cmd/notify.go
Normal file
154
cmd/notify.go
Normal file
@ -0,0 +1,154 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"dbbackup/internal/notify"
|
||||
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
var notifyCmd = &cobra.Command{
|
||||
Use: "notify",
|
||||
Short: "Test notification integrations",
|
||||
Long: `Test notification integrations (webhooks, email).
|
||||
|
||||
This command sends test notifications to verify configuration and connectivity.
|
||||
Helps ensure notifications will work before critical events occur.
|
||||
|
||||
Supports:
|
||||
- Generic Webhooks (HTTP POST)
|
||||
- Email (SMTP)
|
||||
|
||||
Examples:
|
||||
# Test all configured notifications
|
||||
dbbackup notify test
|
||||
|
||||
# Test with custom message
|
||||
dbbackup notify test --message "Hello from dbbackup"
|
||||
|
||||
# Test with verbose output
|
||||
dbbackup notify test --verbose`,
|
||||
}
|
||||
|
||||
var testNotifyCmd = &cobra.Command{
|
||||
Use: "test",
|
||||
Short: "Send test notification",
|
||||
Long: `Send a test notification to verify configuration and connectivity.`,
|
||||
RunE: runNotifyTest,
|
||||
}
|
||||
|
||||
var (
|
||||
notifyMessage string
|
||||
notifyVerbose bool
|
||||
)
|
||||
|
||||
func init() {
|
||||
rootCmd.AddCommand(notifyCmd)
|
||||
notifyCmd.AddCommand(testNotifyCmd)
|
||||
|
||||
testNotifyCmd.Flags().StringVar(¬ifyMessage, "message", "", "Custom test message")
|
||||
testNotifyCmd.Flags().BoolVar(¬ifyVerbose, "verbose", false, "Verbose output")
|
||||
}
|
||||
|
||||
func runNotifyTest(cmd *cobra.Command, args []string) error {
|
||||
if !cfg.NotifyEnabled {
|
||||
fmt.Println("[WARN] Notifications are disabled")
|
||||
fmt.Println("Enable with: --notify-enabled")
|
||||
fmt.Println()
|
||||
fmt.Println("Example configuration:")
|
||||
fmt.Println(" notify_enabled = true")
|
||||
fmt.Println(" notify_on_success = true")
|
||||
fmt.Println(" notify_on_failure = true")
|
||||
fmt.Println(" notify_webhook_url = \"https://your-webhook-url\"")
|
||||
fmt.Println(" # or")
|
||||
fmt.Println(" notify_smtp_host = \"smtp.example.com\"")
|
||||
fmt.Println(" notify_smtp_from = \"backups@example.com\"")
|
||||
fmt.Println(" notify_smtp_to = \"admin@example.com\"")
|
||||
return nil
|
||||
}
|
||||
|
||||
// Use custom message or default
|
||||
message := notifyMessage
|
||||
if message == "" {
|
||||
message = fmt.Sprintf("Test notification from dbbackup at %s", time.Now().Format(time.RFC3339))
|
||||
}
|
||||
|
||||
fmt.Println("[TEST] Testing notification configuration...")
|
||||
fmt.Println()
|
||||
|
||||
// Check what's configured
|
||||
hasWebhook := cfg.NotifyWebhookURL != ""
|
||||
hasSMTP := cfg.NotifySMTPHost != ""
|
||||
|
||||
if !hasWebhook && !hasSMTP {
|
||||
fmt.Println("[WARN] No notification endpoints configured")
|
||||
fmt.Println()
|
||||
fmt.Println("Configure at least one:")
|
||||
fmt.Println(" --notify-webhook-url URL # Generic webhook")
|
||||
fmt.Println(" --notify-smtp-host HOST # Email (requires SMTP settings)")
|
||||
return nil
|
||||
}
|
||||
|
||||
// Show what will be tested
|
||||
if hasWebhook {
|
||||
fmt.Printf("[INFO] Webhook configured: %s\n", cfg.NotifyWebhookURL)
|
||||
}
|
||||
if hasSMTP {
|
||||
fmt.Printf("[INFO] SMTP configured: %s:%d\n", cfg.NotifySMTPHost, cfg.NotifySMTPPort)
|
||||
fmt.Printf(" From: %s\n", cfg.NotifySMTPFrom)
|
||||
if len(cfg.NotifySMTPTo) > 0 {
|
||||
fmt.Printf(" To: %v\n", cfg.NotifySMTPTo)
|
||||
}
|
||||
}
|
||||
fmt.Println()
|
||||
|
||||
// Create notification config
|
||||
notifyCfg := notify.Config{
|
||||
SMTPEnabled: hasSMTP,
|
||||
SMTPHost: cfg.NotifySMTPHost,
|
||||
SMTPPort: cfg.NotifySMTPPort,
|
||||
SMTPUser: cfg.NotifySMTPUser,
|
||||
SMTPPassword: cfg.NotifySMTPPassword,
|
||||
SMTPFrom: cfg.NotifySMTPFrom,
|
||||
SMTPTo: cfg.NotifySMTPTo,
|
||||
SMTPTLS: cfg.NotifySMTPTLS,
|
||||
SMTPStartTLS: cfg.NotifySMTPStartTLS,
|
||||
|
||||
WebhookEnabled: hasWebhook,
|
||||
WebhookURL: cfg.NotifyWebhookURL,
|
||||
WebhookMethod: "POST",
|
||||
|
||||
OnSuccess: true,
|
||||
OnFailure: true,
|
||||
}
|
||||
|
||||
// Create manager
|
||||
manager := notify.NewManager(notifyCfg)
|
||||
|
||||
// Create test event
|
||||
event := notify.NewEvent("test", notify.SeverityInfo, message)
|
||||
event.WithDetail("test", "true")
|
||||
event.WithDetail("command", "dbbackup notify test")
|
||||
|
||||
if notifyVerbose {
|
||||
fmt.Printf("[DEBUG] Sending event: %+v\n", event)
|
||||
}
|
||||
|
||||
// Send notification
|
||||
fmt.Println("[SEND] Sending test notification...")
|
||||
|
||||
ctx := context.Background()
|
||||
if err := manager.NotifySync(ctx, event); err != nil {
|
||||
fmt.Printf("[FAIL] Notification failed: %v\n", err)
|
||||
return err
|
||||
}
|
||||
|
||||
fmt.Println("[OK] Notification sent successfully")
|
||||
fmt.Println()
|
||||
fmt.Println("Check your notification endpoint to confirm delivery.")
|
||||
|
||||
return nil
|
||||
}
|
||||
428
cmd/parallel_restore.go
Normal file
428
cmd/parallel_restore.go
Normal file
@ -0,0 +1,428 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"runtime"
|
||||
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
var parallelRestoreCmd = &cobra.Command{
|
||||
Use: "parallel-restore",
|
||||
Short: "Configure and test parallel restore settings",
|
||||
Long: `Configure parallel restore settings for faster database restoration.
|
||||
|
||||
Parallel restore uses multiple threads to restore databases concurrently:
|
||||
- Parallel jobs within single database (--jobs flag)
|
||||
- Parallel database restoration for cluster backups
|
||||
- CPU-aware thread allocation
|
||||
- Memory-aware resource limits
|
||||
|
||||
This significantly reduces restoration time for:
|
||||
- Large databases with many tables
|
||||
- Cluster backups with multiple databases
|
||||
- Systems with multiple CPU cores
|
||||
|
||||
Configuration:
|
||||
- Set parallel jobs count (default: auto-detect CPU cores)
|
||||
- Configure memory limits for large restores
|
||||
- Tune for specific hardware profiles
|
||||
|
||||
Examples:
|
||||
# Show current parallel restore configuration
|
||||
dbbackup parallel-restore status
|
||||
|
||||
# Test parallel restore performance
|
||||
dbbackup parallel-restore benchmark --file backup.dump
|
||||
|
||||
# Show recommended settings for current system
|
||||
dbbackup parallel-restore recommend
|
||||
|
||||
# Simulate parallel restore (dry-run)
|
||||
dbbackup parallel-restore simulate --file backup.dump --jobs 8`,
|
||||
}
|
||||
|
||||
var parallelRestoreStatusCmd = &cobra.Command{
|
||||
Use: "status",
|
||||
Short: "Show parallel restore configuration",
|
||||
Long: `Display current parallel restore configuration and system capabilities.`,
|
||||
RunE: runParallelRestoreStatus,
|
||||
}
|
||||
|
||||
var parallelRestoreBenchmarkCmd = &cobra.Command{
|
||||
Use: "benchmark",
|
||||
Short: "Benchmark parallel restore performance",
|
||||
Long: `Benchmark parallel restore with different thread counts to find optimal settings.`,
|
||||
RunE: runParallelRestoreBenchmark,
|
||||
}
|
||||
|
||||
var parallelRestoreRecommendCmd = &cobra.Command{
|
||||
Use: "recommend",
|
||||
Short: "Get recommended parallel restore settings",
|
||||
Long: `Analyze system resources and recommend optimal parallel restore settings.`,
|
||||
RunE: runParallelRestoreRecommend,
|
||||
}
|
||||
|
||||
var parallelRestoreSimulateCmd = &cobra.Command{
|
||||
Use: "simulate",
|
||||
Short: "Simulate parallel restore execution plan",
|
||||
Long: `Simulate parallel restore without actually restoring data to show execution plan.`,
|
||||
RunE: runParallelRestoreSimulate,
|
||||
}
|
||||
|
||||
var (
|
||||
parallelRestoreFile string
|
||||
parallelRestoreJobs int
|
||||
parallelRestoreFormat string
|
||||
)
|
||||
|
||||
func init() {
|
||||
rootCmd.AddCommand(parallelRestoreCmd)
|
||||
|
||||
parallelRestoreCmd.AddCommand(parallelRestoreStatusCmd)
|
||||
parallelRestoreCmd.AddCommand(parallelRestoreBenchmarkCmd)
|
||||
parallelRestoreCmd.AddCommand(parallelRestoreRecommendCmd)
|
||||
parallelRestoreCmd.AddCommand(parallelRestoreSimulateCmd)
|
||||
|
||||
parallelRestoreStatusCmd.Flags().StringVar(¶llelRestoreFormat, "format", "text", "Output format (text, json)")
|
||||
parallelRestoreBenchmarkCmd.Flags().StringVar(¶llelRestoreFile, "file", "", "Backup file to benchmark (required)")
|
||||
parallelRestoreBenchmarkCmd.MarkFlagRequired("file")
|
||||
parallelRestoreSimulateCmd.Flags().StringVar(¶llelRestoreFile, "file", "", "Backup file to simulate (required)")
|
||||
parallelRestoreSimulateCmd.Flags().IntVar(¶llelRestoreJobs, "jobs", 0, "Number of parallel jobs (0=auto)")
|
||||
parallelRestoreSimulateCmd.MarkFlagRequired("file")
|
||||
}
|
||||
|
||||
func runParallelRestoreStatus(cmd *cobra.Command, args []string) error {
|
||||
numCPU := runtime.NumCPU()
|
||||
recommendedJobs := numCPU
|
||||
if numCPU > 8 {
|
||||
recommendedJobs = numCPU - 2 // Leave headroom
|
||||
}
|
||||
|
||||
status := ParallelRestoreStatus{
|
||||
SystemCPUs: numCPU,
|
||||
RecommendedJobs: recommendedJobs,
|
||||
MaxJobs: numCPU * 2,
|
||||
CurrentJobs: cfg.Jobs,
|
||||
MemoryGB: getAvailableMemoryGB(),
|
||||
ParallelSupported: true,
|
||||
}
|
||||
|
||||
if parallelRestoreFormat == "json" {
|
||||
data, _ := json.MarshalIndent(status, "", " ")
|
||||
fmt.Println(string(data))
|
||||
return nil
|
||||
}
|
||||
|
||||
fmt.Println("[PARALLEL RESTORE] System Capabilities")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println()
|
||||
fmt.Printf("CPU Cores: %d\n", status.SystemCPUs)
|
||||
fmt.Printf("Available Memory: %.1f GB\n", status.MemoryGB)
|
||||
fmt.Println()
|
||||
|
||||
fmt.Println("[CONFIGURATION]")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Printf("Current Jobs: %d\n", status.CurrentJobs)
|
||||
fmt.Printf("Recommended Jobs: %d\n", status.RecommendedJobs)
|
||||
fmt.Printf("Maximum Jobs: %d\n", status.MaxJobs)
|
||||
fmt.Println()
|
||||
|
||||
fmt.Println("[PARALLEL RESTORE MODES]")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println()
|
||||
fmt.Println("1. Single Database Parallel Restore:")
|
||||
fmt.Println(" Uses pg_restore -j flag or parallel mysql restore")
|
||||
fmt.Println(" Restores tables concurrently within one database")
|
||||
fmt.Println(" Example: dbbackup restore single db.dump --jobs 8 --confirm")
|
||||
fmt.Println()
|
||||
fmt.Println("2. Cluster Parallel Restore:")
|
||||
fmt.Println(" Restores multiple databases concurrently")
|
||||
fmt.Println(" Each database can use parallel jobs")
|
||||
fmt.Println(" Example: dbbackup restore cluster backup.tar --jobs 4 --confirm")
|
||||
fmt.Println()
|
||||
|
||||
fmt.Println("[PERFORMANCE TIPS]")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println()
|
||||
fmt.Println("• Start with recommended jobs count")
|
||||
fmt.Println("• More jobs ≠ always faster (context switching overhead)")
|
||||
fmt.Printf("• For this system: --jobs %d is optimal\n", status.RecommendedJobs)
|
||||
fmt.Println("• Monitor system load during restore")
|
||||
fmt.Println("• Use --profile aggressive for maximum speed")
|
||||
fmt.Println("• SSD storage benefits more from parallelization")
|
||||
fmt.Println()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func runParallelRestoreBenchmark(cmd *cobra.Command, args []string) error {
|
||||
if _, err := os.Stat(parallelRestoreFile); err != nil {
|
||||
return fmt.Errorf("backup file not found: %s", parallelRestoreFile)
|
||||
}
|
||||
|
||||
fmt.Println("[PARALLEL RESTORE] Benchmark Mode")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println()
|
||||
fmt.Printf("Backup File: %s\n", parallelRestoreFile)
|
||||
fmt.Println()
|
||||
|
||||
// Detect backup format
|
||||
ext := filepath.Ext(parallelRestoreFile)
|
||||
format := "unknown"
|
||||
if ext == ".dump" || ext == ".pgdump" {
|
||||
format = "PostgreSQL custom format"
|
||||
} else if ext == ".sql" || ext == ".gz" && filepath.Ext(parallelRestoreFile[:len(parallelRestoreFile)-3]) == ".sql" {
|
||||
format = "SQL format"
|
||||
} else if ext == ".tar" || ext == ".tgz" {
|
||||
format = "Cluster backup"
|
||||
}
|
||||
|
||||
fmt.Printf("Detected Format: %s\n", format)
|
||||
fmt.Println()
|
||||
|
||||
fmt.Println("[BENCHMARK STRATEGY]")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println()
|
||||
fmt.Println("Benchmarking would test restore with different job counts:")
|
||||
fmt.Println()
|
||||
|
||||
numCPU := runtime.NumCPU()
|
||||
testConfigs := []int{1, 2, 4}
|
||||
if numCPU >= 8 {
|
||||
testConfigs = append(testConfigs, 8)
|
||||
}
|
||||
if numCPU >= 16 {
|
||||
testConfigs = append(testConfigs, 16)
|
||||
}
|
||||
|
||||
for i, jobs := range testConfigs {
|
||||
estimatedTime := estimateRestoreTime(parallelRestoreFile, jobs)
|
||||
fmt.Printf("%d. Jobs=%d → Estimated: %s\n", i+1, jobs, estimatedTime)
|
||||
}
|
||||
|
||||
fmt.Println()
|
||||
fmt.Println("[NOTE]")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println("Actual benchmarking requires:")
|
||||
fmt.Println(" - Test database or dry-run mode")
|
||||
fmt.Println(" - Multiple restore attempts with different job counts")
|
||||
fmt.Println(" - Measurement of wall clock time")
|
||||
fmt.Println()
|
||||
fmt.Println("For now, use 'dbbackup restore single --dry-run' to test without")
|
||||
fmt.Println("actually restoring data.")
|
||||
fmt.Println()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func runParallelRestoreRecommend(cmd *cobra.Command, args []string) error {
|
||||
numCPU := runtime.NumCPU()
|
||||
memoryGB := getAvailableMemoryGB()
|
||||
|
||||
fmt.Println("[PARALLEL RESTORE] Recommendations")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println()
|
||||
|
||||
fmt.Println("[SYSTEM ANALYSIS]")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Printf("CPU Cores: %d\n", numCPU)
|
||||
fmt.Printf("Available Memory: %.1f GB\n", memoryGB)
|
||||
fmt.Println()
|
||||
|
||||
// Calculate recommendations
|
||||
var recommendedJobs int
|
||||
var profile string
|
||||
|
||||
if memoryGB < 2 {
|
||||
recommendedJobs = 1
|
||||
profile = "conservative"
|
||||
} else if memoryGB < 8 {
|
||||
recommendedJobs = min(numCPU/2, 4)
|
||||
profile = "conservative"
|
||||
} else if memoryGB < 16 {
|
||||
recommendedJobs = min(numCPU-1, 8)
|
||||
profile = "balanced"
|
||||
} else {
|
||||
recommendedJobs = numCPU
|
||||
if numCPU > 8 {
|
||||
recommendedJobs = numCPU - 2
|
||||
}
|
||||
profile = "aggressive"
|
||||
}
|
||||
|
||||
fmt.Println("[RECOMMENDATIONS]")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Printf("Recommended Profile: %s\n", profile)
|
||||
fmt.Printf("Recommended Jobs: %d\n", recommendedJobs)
|
||||
fmt.Println()
|
||||
|
||||
fmt.Println("[COMMAND EXAMPLES]")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println()
|
||||
fmt.Println("Single database restore (recommended):")
|
||||
fmt.Printf(" dbbackup restore single db.dump --jobs %d --profile %s --confirm\n", recommendedJobs, profile)
|
||||
fmt.Println()
|
||||
fmt.Println("Cluster restore (recommended):")
|
||||
fmt.Printf(" dbbackup restore cluster backup.tar --jobs %d --profile %s --confirm\n", recommendedJobs, profile)
|
||||
fmt.Println()
|
||||
|
||||
if memoryGB < 4 {
|
||||
fmt.Println("[⚠ LOW MEMORY WARNING]")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println("Your system has limited memory. Consider:")
|
||||
fmt.Println(" - Using --low-memory flag")
|
||||
fmt.Println(" - Restoring databases one at a time")
|
||||
fmt.Println(" - Reducing --jobs count")
|
||||
fmt.Println(" - Closing other applications")
|
||||
fmt.Println()
|
||||
}
|
||||
|
||||
if numCPU >= 16 {
|
||||
fmt.Println("[💡 HIGH-PERFORMANCE TIPS]")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println("Your system has many cores. Optimize with:")
|
||||
fmt.Println(" - Use --profile aggressive")
|
||||
fmt.Printf(" - Try up to --jobs %d\n", numCPU)
|
||||
fmt.Println(" - Monitor with 'dbbackup restore ... --verbose'")
|
||||
fmt.Println(" - Use SSD storage for temp files")
|
||||
fmt.Println()
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func runParallelRestoreSimulate(cmd *cobra.Command, args []string) error {
|
||||
if _, err := os.Stat(parallelRestoreFile); err != nil {
|
||||
return fmt.Errorf("backup file not found: %s", parallelRestoreFile)
|
||||
}
|
||||
|
||||
jobs := parallelRestoreJobs
|
||||
if jobs == 0 {
|
||||
jobs = runtime.NumCPU()
|
||||
if jobs > 8 {
|
||||
jobs = jobs - 2
|
||||
}
|
||||
}
|
||||
|
||||
fmt.Println("[PARALLEL RESTORE] Simulation")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println()
|
||||
fmt.Printf("Backup File: %s\n", parallelRestoreFile)
|
||||
fmt.Printf("Parallel Jobs: %d\n", jobs)
|
||||
fmt.Println()
|
||||
|
||||
// Detect backup type
|
||||
ext := filepath.Ext(parallelRestoreFile)
|
||||
isCluster := ext == ".tar" || ext == ".tgz"
|
||||
|
||||
if isCluster {
|
||||
fmt.Println("[CLUSTER RESTORE PLAN]")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println()
|
||||
fmt.Println("Phase 1: Extract archive")
|
||||
fmt.Println(" • Decompress backup archive")
|
||||
fmt.Println(" • Extract globals.sql, schemas, and database dumps")
|
||||
fmt.Println()
|
||||
fmt.Println("Phase 2: Restore globals (sequential)")
|
||||
fmt.Println(" • Restore roles and permissions")
|
||||
fmt.Println(" • Restore tablespaces")
|
||||
fmt.Println()
|
||||
fmt.Println("Phase 3: Parallel database restore")
|
||||
fmt.Printf(" • Restore databases with %d parallel jobs\n", jobs)
|
||||
fmt.Println(" • Each database can use internal parallelization")
|
||||
fmt.Println()
|
||||
fmt.Println("Estimated databases: 3-10 (actual count varies)")
|
||||
fmt.Println("Estimated speedup: 3-5x vs sequential")
|
||||
} else {
|
||||
fmt.Println("[SINGLE DATABASE RESTORE PLAN]")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println()
|
||||
fmt.Println("Phase 1: Pre-restore checks")
|
||||
fmt.Println(" • Verify backup file integrity")
|
||||
fmt.Println(" • Check target database connection")
|
||||
fmt.Println(" • Validate sufficient disk space")
|
||||
fmt.Println()
|
||||
fmt.Println("Phase 2: Schema preparation")
|
||||
fmt.Println(" • Create database (if needed)")
|
||||
fmt.Println(" • Drop existing objects (if --clean)")
|
||||
fmt.Println()
|
||||
fmt.Println("Phase 3: Parallel data restore")
|
||||
fmt.Printf(" • Restore tables with %d parallel jobs\n", jobs)
|
||||
fmt.Println(" • Each job processes different tables")
|
||||
fmt.Println(" • Automatic load balancing")
|
||||
fmt.Println()
|
||||
fmt.Println("Phase 4: Post-restore")
|
||||
fmt.Println(" • Rebuild indexes")
|
||||
fmt.Println(" • Restore constraints")
|
||||
fmt.Println(" • Update statistics")
|
||||
fmt.Println()
|
||||
fmt.Printf("Estimated speedup: %dx vs sequential restore\n", estimateSpeedup(jobs))
|
||||
}
|
||||
|
||||
fmt.Println()
|
||||
fmt.Println("[EXECUTION COMMAND]")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println()
|
||||
fmt.Println("To perform this restore:")
|
||||
if isCluster {
|
||||
fmt.Printf(" dbbackup restore cluster %s --jobs %d --confirm\n", parallelRestoreFile, jobs)
|
||||
} else {
|
||||
fmt.Printf(" dbbackup restore single %s --jobs %d --confirm\n", parallelRestoreFile, jobs)
|
||||
}
|
||||
fmt.Println()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
type ParallelRestoreStatus struct {
|
||||
SystemCPUs int `json:"system_cpus"`
|
||||
RecommendedJobs int `json:"recommended_jobs"`
|
||||
MaxJobs int `json:"max_jobs"`
|
||||
CurrentJobs int `json:"current_jobs"`
|
||||
MemoryGB float64 `json:"memory_gb"`
|
||||
ParallelSupported bool `json:"parallel_supported"`
|
||||
}
|
||||
|
||||
func getAvailableMemoryGB() float64 {
|
||||
// Simple estimation - in production would query actual system memory
|
||||
// For now, return a reasonable default
|
||||
return 8.0
|
||||
}
|
||||
|
||||
func estimateRestoreTime(file string, jobs int) string {
|
||||
// Simplified estimation based on file size and jobs
|
||||
info, err := os.Stat(file)
|
||||
if err != nil {
|
||||
return "unknown"
|
||||
}
|
||||
|
||||
sizeGB := float64(info.Size()) / (1024 * 1024 * 1024)
|
||||
baseTime := sizeGB * 120 // ~2 minutes per GB baseline
|
||||
parallelTime := baseTime / float64(jobs) * 0.7 // 70% efficiency
|
||||
|
||||
if parallelTime < 60 {
|
||||
return fmt.Sprintf("%.0fs", parallelTime)
|
||||
}
|
||||
return fmt.Sprintf("%.1fm", parallelTime/60)
|
||||
}
|
||||
|
||||
func estimateSpeedup(jobs int) int {
|
||||
// Amdahl's law: assume 80% parallelizable
|
||||
if jobs <= 1 {
|
||||
return 1
|
||||
}
|
||||
// Simple linear speedup with diminishing returns
|
||||
speedup := 1.0 + float64(jobs-1)*0.7
|
||||
return int(speedup)
|
||||
}
|
||||
|
||||
func min(a, b int) int {
|
||||
if a < b {
|
||||
return a
|
||||
}
|
||||
return b
|
||||
}
|
||||
308
cmd/progress_webhooks.go
Normal file
308
cmd/progress_webhooks.go
Normal file
@ -0,0 +1,308 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"dbbackup/internal/notify"
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
var progressWebhooksCmd = &cobra.Command{
|
||||
Use: "progress-webhooks",
|
||||
Short: "Configure and test progress webhook notifications",
|
||||
Long: `Configure progress webhook notifications during backup/restore operations.
|
||||
|
||||
Progress webhooks send periodic updates while operations are running:
|
||||
- Bytes processed and percentage complete
|
||||
- Tables/objects processed
|
||||
- Estimated time remaining
|
||||
- Current operation phase
|
||||
|
||||
This allows external monitoring systems to track long-running operations
|
||||
in real-time without polling.
|
||||
|
||||
Configuration:
|
||||
- Set notification webhook URL and credentials via environment
|
||||
- Configure update interval (default: 30s)
|
||||
|
||||
Examples:
|
||||
# Show current progress webhook configuration
|
||||
dbbackup progress-webhooks status
|
||||
|
||||
# Show configuration instructions
|
||||
dbbackup progress-webhooks enable --interval 60s
|
||||
|
||||
# Test progress webhooks with simulated backup
|
||||
dbbackup progress-webhooks test
|
||||
|
||||
# Show disable instructions
|
||||
dbbackup progress-webhooks disable`,
|
||||
}
|
||||
|
||||
var progressWebhooksStatusCmd = &cobra.Command{
|
||||
Use: "status",
|
||||
Short: "Show progress webhook configuration",
|
||||
Long: `Display current progress webhook configuration and status.`,
|
||||
RunE: runProgressWebhooksStatus,
|
||||
}
|
||||
|
||||
var progressWebhooksEnableCmd = &cobra.Command{
|
||||
Use: "enable",
|
||||
Short: "Show how to enable progress webhook notifications",
|
||||
Long: `Display instructions for enabling progress webhook notifications.`,
|
||||
RunE: runProgressWebhooksEnable,
|
||||
}
|
||||
|
||||
var progressWebhooksDisableCmd = &cobra.Command{
|
||||
Use: "disable",
|
||||
Short: "Show how to disable progress webhook notifications",
|
||||
Long: `Display instructions for disabling progress webhook notifications.`,
|
||||
RunE: runProgressWebhooksDisable,
|
||||
}
|
||||
|
||||
var progressWebhooksTestCmd = &cobra.Command{
|
||||
Use: "test",
|
||||
Short: "Test progress webhooks with simulated backup",
|
||||
Long: `Send test progress webhook notifications with simulated backup progress.`,
|
||||
RunE: runProgressWebhooksTest,
|
||||
}
|
||||
|
||||
var (
|
||||
progressInterval time.Duration
|
||||
progressFormat string
|
||||
)
|
||||
|
||||
func init() {
|
||||
rootCmd.AddCommand(progressWebhooksCmd)
|
||||
|
||||
progressWebhooksCmd.AddCommand(progressWebhooksStatusCmd)
|
||||
progressWebhooksCmd.AddCommand(progressWebhooksEnableCmd)
|
||||
progressWebhooksCmd.AddCommand(progressWebhooksDisableCmd)
|
||||
progressWebhooksCmd.AddCommand(progressWebhooksTestCmd)
|
||||
|
||||
progressWebhooksEnableCmd.Flags().DurationVar(&progressInterval, "interval", 30*time.Second, "Progress update interval")
|
||||
progressWebhooksStatusCmd.Flags().StringVar(&progressFormat, "format", "text", "Output format (text, json)")
|
||||
progressWebhooksTestCmd.Flags().DurationVar(&progressInterval, "interval", 5*time.Second, "Test progress update interval")
|
||||
}
|
||||
|
||||
func runProgressWebhooksStatus(cmd *cobra.Command, args []string) error {
|
||||
// Get notification configuration from environment
|
||||
webhookURL := os.Getenv("DBBACKUP_WEBHOOK_URL")
|
||||
smtpHost := os.Getenv("DBBACKUP_SMTP_HOST")
|
||||
progressIntervalEnv := os.Getenv("DBBACKUP_PROGRESS_INTERVAL")
|
||||
|
||||
var interval time.Duration
|
||||
if progressIntervalEnv != "" {
|
||||
if d, err := time.ParseDuration(progressIntervalEnv); err == nil {
|
||||
interval = d
|
||||
}
|
||||
}
|
||||
|
||||
status := ProgressWebhookStatus{
|
||||
Enabled: webhookURL != "" || smtpHost != "",
|
||||
Interval: interval,
|
||||
WebhookURL: webhookURL,
|
||||
SMTPEnabled: smtpHost != "",
|
||||
}
|
||||
|
||||
if progressFormat == "json" {
|
||||
data, _ := json.MarshalIndent(status, "", " ")
|
||||
fmt.Println(string(data))
|
||||
return nil
|
||||
}
|
||||
|
||||
fmt.Println("[PROGRESS WEBHOOKS] Configuration Status")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println()
|
||||
|
||||
if status.Enabled {
|
||||
fmt.Println("Status: ✓ ENABLED")
|
||||
} else {
|
||||
fmt.Println("Status: ✗ DISABLED")
|
||||
}
|
||||
|
||||
if status.Interval > 0 {
|
||||
fmt.Printf("Update Interval: %s\n", status.Interval)
|
||||
} else {
|
||||
fmt.Println("Update Interval: Not set (would use 30s default)")
|
||||
}
|
||||
fmt.Println()
|
||||
|
||||
fmt.Println("[NOTIFICATION BACKENDS]")
|
||||
fmt.Println("==========================================")
|
||||
|
||||
if status.WebhookURL != "" {
|
||||
fmt.Println("✓ Webhook: Configured")
|
||||
fmt.Printf(" URL: %s\n", maskURL(status.WebhookURL))
|
||||
} else {
|
||||
fmt.Println("✗ Webhook: Not configured")
|
||||
}
|
||||
|
||||
if status.SMTPEnabled {
|
||||
fmt.Println("✓ Email (SMTP): Configured")
|
||||
} else {
|
||||
fmt.Println("✗ Email (SMTP): Not configured")
|
||||
}
|
||||
fmt.Println()
|
||||
|
||||
if !status.Enabled {
|
||||
fmt.Println("[SETUP INSTRUCTIONS]")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println()
|
||||
fmt.Println("To enable progress webhooks, configure notification backend:")
|
||||
fmt.Println()
|
||||
fmt.Println(" export DBBACKUP_WEBHOOK_URL=https://your-webhook-url")
|
||||
fmt.Println(" export DBBACKUP_PROGRESS_INTERVAL=30s")
|
||||
fmt.Println()
|
||||
fmt.Println("Or add to .dbbackup.conf:")
|
||||
fmt.Println()
|
||||
fmt.Println(" webhook_url: https://your-webhook-url")
|
||||
fmt.Println(" progress_interval: 30s")
|
||||
fmt.Println()
|
||||
fmt.Println("Then test with:")
|
||||
fmt.Println(" dbbackup progress-webhooks test")
|
||||
fmt.Println()
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func runProgressWebhooksEnable(cmd *cobra.Command, args []string) error {
|
||||
webhookURL := os.Getenv("DBBACKUP_WEBHOOK_URL")
|
||||
smtpHost := os.Getenv("DBBACKUP_SMTP_HOST")
|
||||
|
||||
if webhookURL == "" && smtpHost == "" {
|
||||
fmt.Println("[PROGRESS WEBHOOKS] Setup Required")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println()
|
||||
fmt.Println("No notification backend configured.")
|
||||
fmt.Println()
|
||||
fmt.Println("Configure webhook via environment:")
|
||||
fmt.Println(" export DBBACKUP_WEBHOOK_URL=https://your-webhook-url")
|
||||
fmt.Println()
|
||||
fmt.Println("Or configure SMTP:")
|
||||
fmt.Println(" export DBBACKUP_SMTP_HOST=smtp.example.com")
|
||||
fmt.Println(" export DBBACKUP_SMTP_PORT=587")
|
||||
fmt.Println(" export DBBACKUP_SMTP_USER=user@example.com")
|
||||
fmt.Println()
|
||||
return nil
|
||||
}
|
||||
|
||||
fmt.Println("[PROGRESS WEBHOOKS] Configuration")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println()
|
||||
fmt.Println("To enable progress webhooks, add to your environment:")
|
||||
fmt.Println()
|
||||
fmt.Printf(" export DBBACKUP_PROGRESS_INTERVAL=%s\n", progressInterval)
|
||||
fmt.Println()
|
||||
fmt.Println("Or add to .dbbackup.conf:")
|
||||
fmt.Println()
|
||||
fmt.Printf(" progress_interval: %s\n", progressInterval)
|
||||
fmt.Println()
|
||||
fmt.Println("Progress updates will be sent to configured notification backends")
|
||||
fmt.Println("during backup and restore operations.")
|
||||
fmt.Println()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func runProgressWebhooksDisable(cmd *cobra.Command, args []string) error {
|
||||
fmt.Println("[PROGRESS WEBHOOKS] Disable")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println()
|
||||
fmt.Println("To disable progress webhooks:")
|
||||
fmt.Println()
|
||||
fmt.Println(" unset DBBACKUP_PROGRESS_INTERVAL")
|
||||
fmt.Println()
|
||||
fmt.Println("Or remove from .dbbackup.conf:")
|
||||
fmt.Println()
|
||||
fmt.Println(" # progress_interval: 30s")
|
||||
fmt.Println()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func runProgressWebhooksTest(cmd *cobra.Command, args []string) error {
|
||||
webhookURL := os.Getenv("DBBACKUP_WEBHOOK_URL")
|
||||
smtpHost := os.Getenv("DBBACKUP_SMTP_HOST")
|
||||
|
||||
if webhookURL == "" && smtpHost == "" {
|
||||
return fmt.Errorf("no notification backend configured. Set DBBACKUP_WEBHOOK_URL or DBBACKUP_SMTP_HOST")
|
||||
}
|
||||
|
||||
fmt.Println("[PROGRESS WEBHOOKS] Test Mode")
|
||||
fmt.Println("==========================================")
|
||||
fmt.Println()
|
||||
fmt.Println("Simulating backup with progress updates...")
|
||||
fmt.Printf("Update interval: %s\n", progressInterval)
|
||||
fmt.Println()
|
||||
|
||||
// Create notification manager
|
||||
notifyCfg := notify.Config{
|
||||
WebhookEnabled: webhookURL != "",
|
||||
WebhookURL: webhookURL,
|
||||
WebhookMethod: "POST",
|
||||
SMTPEnabled: smtpHost != "",
|
||||
SMTPHost: smtpHost,
|
||||
OnSuccess: true,
|
||||
OnFailure: true,
|
||||
}
|
||||
|
||||
manager := notify.NewManager(notifyCfg)
|
||||
|
||||
// Create progress tracker
|
||||
tracker := notify.NewProgressTracker(manager, "testdb", "Backup")
|
||||
tracker.SetTotals(1024*1024*1024, 10) // 1GB, 10 tables
|
||||
tracker.Start(progressInterval)
|
||||
|
||||
defer tracker.Stop()
|
||||
|
||||
// Simulate backup progress
|
||||
totalBytes := int64(1024 * 1024 * 1024)
|
||||
totalTables := 10
|
||||
steps := 5
|
||||
|
||||
for i := 1; i <= steps; i++ {
|
||||
phase := fmt.Sprintf("Processing table %d/%d", i*2, totalTables)
|
||||
tracker.SetPhase(phase)
|
||||
|
||||
bytesProcessed := totalBytes * int64(i) / int64(steps)
|
||||
tablesProcessed := totalTables * i / steps
|
||||
|
||||
tracker.UpdateBytes(bytesProcessed)
|
||||
tracker.UpdateTables(tablesProcessed)
|
||||
|
||||
progress := tracker.GetProgress()
|
||||
fmt.Printf("[%d/%d] %s - %s\n", i, steps, phase, progress.FormatSummary())
|
||||
|
||||
if i < steps {
|
||||
time.Sleep(progressInterval)
|
||||
}
|
||||
}
|
||||
|
||||
fmt.Println()
|
||||
fmt.Println("✓ Test completed")
|
||||
fmt.Println()
|
||||
fmt.Println("Check your notification backend for progress updates.")
|
||||
fmt.Println("You should have received approximately 5 progress notifications.")
|
||||
fmt.Println()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
type ProgressWebhookStatus struct {
|
||||
Enabled bool `json:"enabled"`
|
||||
Interval time.Duration `json:"interval"`
|
||||
WebhookURL string `json:"webhook_url,omitempty"`
|
||||
SMTPEnabled bool `json:"smtp_enabled"`
|
||||
}
|
||||
|
||||
func maskURL(url string) string {
|
||||
if len(url) < 20 {
|
||||
return url[:5] + "***"
|
||||
}
|
||||
return url[:20] + "***"
|
||||
}
|
||||
@ -108,7 +108,7 @@ func runSchedule(cmd *cobra.Command, args []string) error {
|
||||
func getSystemdTimers() ([]TimerInfo, error) {
|
||||
// Run systemctl list-timers --all --no-pager
|
||||
cmdArgs := []string{"list-timers", "--all", "--no-pager"}
|
||||
|
||||
|
||||
output, err := exec.Command("systemctl", cmdArgs...).CombinedOutput()
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to list timers: %w\nOutput: %s", err, string(output))
|
||||
@ -137,7 +137,7 @@ func parseTimerList(output string) []TimerInfo {
|
||||
|
||||
// Extract timer info
|
||||
timer := TimerInfo{}
|
||||
|
||||
|
||||
// Check if NEXT field is "n/a" (inactive timer)
|
||||
if fields[0] == "n/a" {
|
||||
timer.NextRun = "n/a"
|
||||
@ -227,11 +227,11 @@ func filterTimers(timers []TimerInfo) []TimerInfo {
|
||||
|
||||
// Default: filter for backup-related timers
|
||||
name := strings.ToLower(timer.Unit)
|
||||
if strings.Contains(name, "backup") ||
|
||||
strings.Contains(name, "dbbackup") ||
|
||||
strings.Contains(name, "postgres") ||
|
||||
strings.Contains(name, "mysql") ||
|
||||
strings.Contains(name, "mariadb") {
|
||||
if strings.Contains(name, "backup") ||
|
||||
strings.Contains(name, "dbbackup") ||
|
||||
strings.Contains(name, "postgres") ||
|
||||
strings.Contains(name, "mysql") ||
|
||||
strings.Contains(name, "mariadb") {
|
||||
filtered = append(filtered, timer)
|
||||
}
|
||||
}
|
||||
@ -243,7 +243,7 @@ func outputTimerTable(timers []TimerInfo) {
|
||||
fmt.Println()
|
||||
fmt.Println("Scheduled Backups")
|
||||
fmt.Println("=====================================================")
|
||||
|
||||
|
||||
for _, timer := range timers {
|
||||
name := timer.Unit
|
||||
if strings.HasSuffix(name, ".timer") {
|
||||
@ -252,7 +252,7 @@ func outputTimerTable(timers []TimerInfo) {
|
||||
|
||||
fmt.Printf("\n[TIMER] %s\n", name)
|
||||
fmt.Printf(" Status: %s\n", timer.Active)
|
||||
|
||||
|
||||
if timer.Active == "active" && timer.NextRun != "" && timer.NextRun != "n/a" {
|
||||
fmt.Printf(" Next Run: %s\n", timer.NextRun)
|
||||
if timer.Left != "" {
|
||||
@ -261,7 +261,7 @@ func outputTimerTable(timers []TimerInfo) {
|
||||
} else {
|
||||
fmt.Printf(" Next Run: Not scheduled (timer inactive)\n")
|
||||
}
|
||||
|
||||
|
||||
if timer.LastRun != "" && timer.LastRun != "n/a" {
|
||||
fmt.Printf(" Last Run: %s\n", timer.LastRun)
|
||||
}
|
||||
|
||||
540
cmd/validate.go
Normal file
540
cmd/validate.go
Normal file
@ -0,0 +1,540 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"dbbackup/internal/config"
|
||||
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
var validateCmd = &cobra.Command{
|
||||
Use: "validate",
|
||||
Short: "Validate configuration and environment",
|
||||
Long: `Validate dbbackup configuration file and runtime environment.
|
||||
|
||||
This command performs comprehensive validation:
|
||||
- Configuration file syntax and structure
|
||||
- Database connection parameters
|
||||
- Directory paths and permissions
|
||||
- External tool availability (pg_dump, mysqldump)
|
||||
- Cloud storage credentials (if configured)
|
||||
- Encryption setup (if enabled)
|
||||
- Resource limits and system requirements
|
||||
- Port accessibility
|
||||
|
||||
Helps identify configuration issues before running backups.
|
||||
|
||||
Examples:
|
||||
# Validate default config (.dbbackup.conf)
|
||||
dbbackup validate
|
||||
|
||||
# Validate specific config file
|
||||
dbbackup validate --config /etc/dbbackup/prod.conf
|
||||
|
||||
# Quick validation (skip connectivity tests)
|
||||
dbbackup validate --quick
|
||||
|
||||
# JSON output for automation
|
||||
dbbackup validate --format json`,
|
||||
RunE: runValidate,
|
||||
}
|
||||
|
||||
var (
|
||||
validateFormat string
|
||||
validateQuick bool
|
||||
)
|
||||
|
||||
type ValidationResult struct {
|
||||
Valid bool `json:"valid"`
|
||||
Issues []ValidationIssue `json:"issues"`
|
||||
Warnings []ValidationIssue `json:"warnings"`
|
||||
Checks []ValidationCheck `json:"checks"`
|
||||
Summary string `json:"summary"`
|
||||
}
|
||||
|
||||
type ValidationIssue struct {
|
||||
Category string `json:"category"`
|
||||
Description string `json:"description"`
|
||||
Suggestion string `json:"suggestion,omitempty"`
|
||||
}
|
||||
|
||||
type ValidationCheck struct {
|
||||
Name string `json:"name"`
|
||||
Status string `json:"status"` // "pass", "warn", "fail"
|
||||
Message string `json:"message,omitempty"`
|
||||
}
|
||||
|
||||
func init() {
|
||||
rootCmd.AddCommand(validateCmd)
|
||||
|
||||
validateCmd.Flags().StringVar(&validateFormat, "format", "table", "Output format (table, json)")
|
||||
validateCmd.Flags().BoolVar(&validateQuick, "quick", false, "Quick validation (skip connectivity tests)")
|
||||
}
|
||||
|
||||
func runValidate(cmd *cobra.Command, args []string) error {
|
||||
result := &ValidationResult{
|
||||
Valid: true,
|
||||
Issues: []ValidationIssue{},
|
||||
Warnings: []ValidationIssue{},
|
||||
Checks: []ValidationCheck{},
|
||||
}
|
||||
|
||||
// Validate configuration file
|
||||
validateConfigFile(cfg, result)
|
||||
|
||||
// Validate database settings
|
||||
validateDatabase(cfg, result)
|
||||
|
||||
// Validate paths
|
||||
validatePaths(cfg, result)
|
||||
|
||||
// Validate external tools
|
||||
validateTools(cfg, result)
|
||||
|
||||
// Validate cloud storage (if enabled)
|
||||
if cfg.CloudEnabled {
|
||||
validateCloud(cfg, result)
|
||||
}
|
||||
|
||||
// Validate encryption (if enabled)
|
||||
if cfg.PITREnabled && cfg.WALEncryption {
|
||||
validateEncryption(cfg, result)
|
||||
}
|
||||
|
||||
// Validate resource limits
|
||||
validateResources(cfg, result)
|
||||
|
||||
// Connectivity tests (unless --quick)
|
||||
if !validateQuick {
|
||||
validateConnectivity(cfg, result)
|
||||
}
|
||||
|
||||
// Determine overall validity
|
||||
result.Valid = len(result.Issues) == 0
|
||||
|
||||
// Generate summary
|
||||
if result.Valid {
|
||||
if len(result.Warnings) > 0 {
|
||||
result.Summary = fmt.Sprintf("Configuration valid with %d warning(s)", len(result.Warnings))
|
||||
} else {
|
||||
result.Summary = "Configuration valid - all checks passed"
|
||||
}
|
||||
} else {
|
||||
result.Summary = fmt.Sprintf("Configuration invalid - %d issue(s) found", len(result.Issues))
|
||||
}
|
||||
|
||||
// Output results
|
||||
if validateFormat == "json" {
|
||||
enc := json.NewEncoder(os.Stdout)
|
||||
enc.SetIndent("", " ")
|
||||
return enc.Encode(result)
|
||||
}
|
||||
|
||||
printValidationResult(result)
|
||||
|
||||
if !result.Valid {
|
||||
return fmt.Errorf("validation failed")
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func validateConfigFile(cfg *config.Config, result *ValidationResult) {
|
||||
check := ValidationCheck{Name: "Configuration File"}
|
||||
|
||||
if cfg.ConfigPath == "" {
|
||||
check.Status = "warn"
|
||||
check.Message = "No config file specified (using defaults)"
|
||||
result.Warnings = append(result.Warnings, ValidationIssue{
|
||||
Category: "config",
|
||||
Description: "No configuration file found",
|
||||
Suggestion: "Run 'dbbackup backup' to create .dbbackup.conf",
|
||||
})
|
||||
} else {
|
||||
if _, err := os.Stat(cfg.ConfigPath); err != nil {
|
||||
check.Status = "warn"
|
||||
check.Message = "Config file not found"
|
||||
result.Warnings = append(result.Warnings, ValidationIssue{
|
||||
Category: "config",
|
||||
Description: fmt.Sprintf("Config file not accessible: %s", cfg.ConfigPath),
|
||||
Suggestion: "Check file path and permissions",
|
||||
})
|
||||
} else {
|
||||
check.Status = "pass"
|
||||
check.Message = fmt.Sprintf("Loaded from %s", cfg.ConfigPath)
|
||||
}
|
||||
}
|
||||
|
||||
result.Checks = append(result.Checks, check)
|
||||
}
|
||||
|
||||
func validateDatabase(cfg *config.Config, result *ValidationResult) {
|
||||
// Database type
|
||||
check := ValidationCheck{Name: "Database Type"}
|
||||
if cfg.DatabaseType != "postgres" && cfg.DatabaseType != "mysql" && cfg.DatabaseType != "mariadb" {
|
||||
check.Status = "fail"
|
||||
check.Message = fmt.Sprintf("Invalid: %s", cfg.DatabaseType)
|
||||
result.Issues = append(result.Issues, ValidationIssue{
|
||||
Category: "database",
|
||||
Description: fmt.Sprintf("Invalid database type: %s", cfg.DatabaseType),
|
||||
Suggestion: "Use 'postgres', 'mysql', or 'mariadb'",
|
||||
})
|
||||
} else {
|
||||
check.Status = "pass"
|
||||
check.Message = cfg.DatabaseType
|
||||
}
|
||||
result.Checks = append(result.Checks, check)
|
||||
|
||||
// Host
|
||||
check = ValidationCheck{Name: "Database Host"}
|
||||
if cfg.Host == "" {
|
||||
check.Status = "fail"
|
||||
check.Message = "Not configured"
|
||||
result.Issues = append(result.Issues, ValidationIssue{
|
||||
Category: "database",
|
||||
Description: "Database host not specified",
|
||||
Suggestion: "Set --host flag or host in config file",
|
||||
})
|
||||
} else {
|
||||
check.Status = "pass"
|
||||
check.Message = cfg.Host
|
||||
}
|
||||
result.Checks = append(result.Checks, check)
|
||||
|
||||
// Port
|
||||
check = ValidationCheck{Name: "Database Port"}
|
||||
if cfg.Port <= 0 || cfg.Port > 65535 {
|
||||
check.Status = "fail"
|
||||
check.Message = fmt.Sprintf("Invalid: %d", cfg.Port)
|
||||
result.Issues = append(result.Issues, ValidationIssue{
|
||||
Category: "database",
|
||||
Description: fmt.Sprintf("Invalid port number: %d", cfg.Port),
|
||||
Suggestion: "Use valid port (1-65535)",
|
||||
})
|
||||
} else {
|
||||
check.Status = "pass"
|
||||
check.Message = strconv.Itoa(cfg.Port)
|
||||
}
|
||||
result.Checks = append(result.Checks, check)
|
||||
|
||||
// User
|
||||
check = ValidationCheck{Name: "Database User"}
|
||||
if cfg.User == "" {
|
||||
check.Status = "warn"
|
||||
check.Message = "Not configured (using current user)"
|
||||
result.Warnings = append(result.Warnings, ValidationIssue{
|
||||
Category: "database",
|
||||
Description: "Database user not specified",
|
||||
Suggestion: "Set --user flag or user in config file",
|
||||
})
|
||||
} else {
|
||||
check.Status = "pass"
|
||||
check.Message = cfg.User
|
||||
}
|
||||
result.Checks = append(result.Checks, check)
|
||||
}
|
||||
|
||||
func validatePaths(cfg *config.Config, result *ValidationResult) {
|
||||
// Backup directory
|
||||
check := ValidationCheck{Name: "Backup Directory"}
|
||||
if cfg.BackupDir == "" {
|
||||
check.Status = "fail"
|
||||
check.Message = "Not configured"
|
||||
result.Issues = append(result.Issues, ValidationIssue{
|
||||
Category: "paths",
|
||||
Description: "Backup directory not specified",
|
||||
Suggestion: "Set --backup-dir flag or backup_dir in config",
|
||||
})
|
||||
} else {
|
||||
info, err := os.Stat(cfg.BackupDir)
|
||||
if err != nil {
|
||||
check.Status = "warn"
|
||||
check.Message = "Does not exist (will be created)"
|
||||
result.Warnings = append(result.Warnings, ValidationIssue{
|
||||
Category: "paths",
|
||||
Description: fmt.Sprintf("Backup directory does not exist: %s", cfg.BackupDir),
|
||||
Suggestion: "Directory will be created automatically",
|
||||
})
|
||||
} else if !info.IsDir() {
|
||||
check.Status = "fail"
|
||||
check.Message = "Not a directory"
|
||||
result.Issues = append(result.Issues, ValidationIssue{
|
||||
Category: "paths",
|
||||
Description: fmt.Sprintf("Backup path is not a directory: %s", cfg.BackupDir),
|
||||
Suggestion: "Specify a valid directory path",
|
||||
})
|
||||
} else {
|
||||
// Check write permissions
|
||||
testFile := filepath.Join(cfg.BackupDir, ".dbbackup-test")
|
||||
if err := os.WriteFile(testFile, []byte("test"), 0644); err != nil {
|
||||
check.Status = "fail"
|
||||
check.Message = "Not writable"
|
||||
result.Issues = append(result.Issues, ValidationIssue{
|
||||
Category: "paths",
|
||||
Description: fmt.Sprintf("Cannot write to backup directory: %s", cfg.BackupDir),
|
||||
Suggestion: "Check directory permissions",
|
||||
})
|
||||
} else {
|
||||
os.Remove(testFile)
|
||||
check.Status = "pass"
|
||||
check.Message = cfg.BackupDir
|
||||
}
|
||||
}
|
||||
}
|
||||
result.Checks = append(result.Checks, check)
|
||||
|
||||
// WAL archive directory (if PITR enabled)
|
||||
if cfg.PITREnabled {
|
||||
check = ValidationCheck{Name: "WAL Archive Directory"}
|
||||
if cfg.WALArchiveDir == "" {
|
||||
check.Status = "warn"
|
||||
check.Message = "Not configured"
|
||||
result.Warnings = append(result.Warnings, ValidationIssue{
|
||||
Category: "pitr",
|
||||
Description: "PITR enabled but WAL archive directory not set",
|
||||
Suggestion: "Set --wal-archive-dir for PITR functionality",
|
||||
})
|
||||
} else {
|
||||
check.Status = "pass"
|
||||
check.Message = cfg.WALArchiveDir
|
||||
}
|
||||
result.Checks = append(result.Checks, check)
|
||||
}
|
||||
}
|
||||
|
||||
func validateTools(cfg *config.Config, result *ValidationResult) {
|
||||
// Skip if using native engine
|
||||
if cfg.UseNativeEngine {
|
||||
check := ValidationCheck{
|
||||
Name: "External Tools",
|
||||
Status: "pass",
|
||||
Message: "Using native Go engine (no external tools required)",
|
||||
}
|
||||
result.Checks = append(result.Checks, check)
|
||||
return
|
||||
}
|
||||
|
||||
// Check for database tools
|
||||
var requiredTools []string
|
||||
if cfg.DatabaseType == "postgres" {
|
||||
requiredTools = []string{"pg_dump", "pg_restore", "psql"}
|
||||
} else if cfg.DatabaseType == "mysql" || cfg.DatabaseType == "mariadb" {
|
||||
requiredTools = []string{"mysqldump", "mysql"}
|
||||
}
|
||||
|
||||
for _, tool := range requiredTools {
|
||||
check := ValidationCheck{Name: fmt.Sprintf("Tool: %s", tool)}
|
||||
path, err := exec.LookPath(tool)
|
||||
if err != nil {
|
||||
check.Status = "fail"
|
||||
check.Message = "Not found in PATH"
|
||||
result.Issues = append(result.Issues, ValidationIssue{
|
||||
Category: "tools",
|
||||
Description: fmt.Sprintf("Required tool not found: %s", tool),
|
||||
Suggestion: fmt.Sprintf("Install %s or use --native flag", tool),
|
||||
})
|
||||
} else {
|
||||
check.Status = "pass"
|
||||
check.Message = path
|
||||
}
|
||||
result.Checks = append(result.Checks, check)
|
||||
}
|
||||
}
|
||||
|
||||
func validateCloud(cfg *config.Config, result *ValidationResult) {
|
||||
check := ValidationCheck{Name: "Cloud Storage"}
|
||||
|
||||
if cfg.CloudProvider == "" {
|
||||
check.Status = "fail"
|
||||
check.Message = "Provider not configured"
|
||||
result.Issues = append(result.Issues, ValidationIssue{
|
||||
Category: "cloud",
|
||||
Description: "Cloud enabled but provider not specified",
|
||||
Suggestion: "Set --cloud-provider (s3, gcs, azure, minio, b2)",
|
||||
})
|
||||
} else {
|
||||
check.Status = "pass"
|
||||
check.Message = cfg.CloudProvider
|
||||
}
|
||||
result.Checks = append(result.Checks, check)
|
||||
|
||||
// Bucket
|
||||
check = ValidationCheck{Name: "Cloud Bucket"}
|
||||
if cfg.CloudBucket == "" {
|
||||
check.Status = "fail"
|
||||
check.Message = "Not configured"
|
||||
result.Issues = append(result.Issues, ValidationIssue{
|
||||
Category: "cloud",
|
||||
Description: "Cloud bucket/container not specified",
|
||||
Suggestion: "Set --cloud-bucket",
|
||||
})
|
||||
} else {
|
||||
check.Status = "pass"
|
||||
check.Message = cfg.CloudBucket
|
||||
}
|
||||
result.Checks = append(result.Checks, check)
|
||||
|
||||
// Credentials
|
||||
check = ValidationCheck{Name: "Cloud Credentials"}
|
||||
if cfg.CloudAccessKey == "" || cfg.CloudSecretKey == "" {
|
||||
check.Status = "warn"
|
||||
check.Message = "Credentials not in config (may use env vars)"
|
||||
result.Warnings = append(result.Warnings, ValidationIssue{
|
||||
Category: "cloud",
|
||||
Description: "Cloud credentials not in config file",
|
||||
Suggestion: "Ensure AWS_ACCESS_KEY_ID/AWS_SECRET_ACCESS_KEY or similar env vars are set",
|
||||
})
|
||||
} else {
|
||||
check.Status = "pass"
|
||||
check.Message = "Configured"
|
||||
}
|
||||
result.Checks = append(result.Checks, check)
|
||||
}
|
||||
|
||||
func validateEncryption(cfg *config.Config, result *ValidationResult) {
|
||||
check := ValidationCheck{Name: "Encryption"}
|
||||
|
||||
// Check for openssl
|
||||
if _, err := exec.LookPath("openssl"); err != nil {
|
||||
check.Status = "fail"
|
||||
check.Message = "openssl not found"
|
||||
result.Issues = append(result.Issues, ValidationIssue{
|
||||
Category: "encryption",
|
||||
Description: "Encryption enabled but openssl not available",
|
||||
Suggestion: "Install openssl or disable WAL encryption",
|
||||
})
|
||||
} else {
|
||||
check.Status = "pass"
|
||||
check.Message = "openssl available"
|
||||
}
|
||||
|
||||
result.Checks = append(result.Checks, check)
|
||||
}
|
||||
|
||||
func validateResources(cfg *config.Config, result *ValidationResult) {
|
||||
// CPU cores
|
||||
check := ValidationCheck{Name: "CPU Cores"}
|
||||
if cfg.MaxCores < 1 {
|
||||
check.Status = "fail"
|
||||
check.Message = "Invalid core count"
|
||||
result.Issues = append(result.Issues, ValidationIssue{
|
||||
Category: "resources",
|
||||
Description: "Invalid max cores setting",
|
||||
Suggestion: "Set --max-cores to positive value",
|
||||
})
|
||||
} else {
|
||||
check.Status = "pass"
|
||||
check.Message = fmt.Sprintf("%d cores", cfg.MaxCores)
|
||||
}
|
||||
result.Checks = append(result.Checks, check)
|
||||
|
||||
// Jobs
|
||||
check = ValidationCheck{Name: "Parallel Jobs"}
|
||||
if cfg.Jobs < 1 {
|
||||
check.Status = "fail"
|
||||
check.Message = "Invalid job count"
|
||||
result.Issues = append(result.Issues, ValidationIssue{
|
||||
Category: "resources",
|
||||
Description: "Invalid jobs setting",
|
||||
Suggestion: "Set --jobs to positive value",
|
||||
})
|
||||
} else if cfg.Jobs > cfg.MaxCores*2 {
|
||||
check.Status = "warn"
|
||||
check.Message = fmt.Sprintf("%d jobs (high)", cfg.Jobs)
|
||||
result.Warnings = append(result.Warnings, ValidationIssue{
|
||||
Category: "resources",
|
||||
Description: "Jobs count higher than CPU cores",
|
||||
Suggestion: "Consider reducing --jobs for better performance",
|
||||
})
|
||||
} else {
|
||||
check.Status = "pass"
|
||||
check.Message = fmt.Sprintf("%d jobs", cfg.Jobs)
|
||||
}
|
||||
result.Checks = append(result.Checks, check)
|
||||
}
|
||||
|
||||
func validateConnectivity(cfg *config.Config, result *ValidationResult) {
|
||||
check := ValidationCheck{Name: "Database Connectivity"}
|
||||
|
||||
// Try to connect to database port
|
||||
address := net.JoinHostPort(cfg.Host, strconv.Itoa(cfg.Port))
|
||||
conn, err := net.DialTimeout("tcp", address, 5*1000000000) // 5 seconds
|
||||
if err != nil {
|
||||
check.Status = "fail"
|
||||
check.Message = fmt.Sprintf("Cannot connect to %s", address)
|
||||
result.Issues = append(result.Issues, ValidationIssue{
|
||||
Category: "connectivity",
|
||||
Description: fmt.Sprintf("Cannot connect to database: %v", err),
|
||||
Suggestion: "Check host, port, and network connectivity",
|
||||
})
|
||||
} else {
|
||||
conn.Close()
|
||||
check.Status = "pass"
|
||||
check.Message = fmt.Sprintf("Connected to %s", address)
|
||||
}
|
||||
|
||||
result.Checks = append(result.Checks, check)
|
||||
}
|
||||
|
||||
func printValidationResult(result *ValidationResult) {
|
||||
fmt.Println("\n[VALIDATION REPORT]")
|
||||
fmt.Println(strings.Repeat("=", 60))
|
||||
|
||||
// Print checks
|
||||
fmt.Println("\n[CHECKS]")
|
||||
for _, check := range result.Checks {
|
||||
var status string
|
||||
switch check.Status {
|
||||
case "pass":
|
||||
status = "[PASS]"
|
||||
case "warn":
|
||||
status = "[WARN]"
|
||||
case "fail":
|
||||
status = "[FAIL]"
|
||||
}
|
||||
|
||||
fmt.Printf(" %-25s %s", check.Name+":", status)
|
||||
if check.Message != "" {
|
||||
fmt.Printf(" %s", check.Message)
|
||||
}
|
||||
fmt.Println()
|
||||
}
|
||||
|
||||
// Print issues
|
||||
if len(result.Issues) > 0 {
|
||||
fmt.Println("\n[ISSUES]")
|
||||
for i, issue := range result.Issues {
|
||||
fmt.Printf(" %d. [%s] %s\n", i+1, strings.ToUpper(issue.Category), issue.Description)
|
||||
if issue.Suggestion != "" {
|
||||
fmt.Printf(" → %s\n", issue.Suggestion)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Print warnings
|
||||
if len(result.Warnings) > 0 {
|
||||
fmt.Println("\n[WARNINGS]")
|
||||
for i, warning := range result.Warnings {
|
||||
fmt.Printf(" %d. [%s] %s\n", i+1, strings.ToUpper(warning.Category), warning.Description)
|
||||
if warning.Suggestion != "" {
|
||||
fmt.Printf(" → %s\n", warning.Suggestion)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Print summary
|
||||
fmt.Println("\n" + strings.Repeat("=", 60))
|
||||
if result.Valid {
|
||||
fmt.Printf("[OK] %s\n\n", result.Summary)
|
||||
} else {
|
||||
fmt.Printf("[FAIL] %s\n\n", result.Summary)
|
||||
}
|
||||
}
|
||||
216
internal/notify/progress.go
Normal file
216
internal/notify/progress.go
Normal file
@ -0,0 +1,216 @@
|
||||
package notify
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
// ProgressTracker tracks backup/restore progress and sends periodic updates
|
||||
type ProgressTracker struct {
|
||||
manager *Manager
|
||||
database string
|
||||
operation string
|
||||
startTime time.Time
|
||||
ticker *time.Ticker
|
||||
stopCh chan struct{}
|
||||
mu sync.RWMutex
|
||||
bytesTotal int64
|
||||
bytesProcessed int64
|
||||
tablesTotal int
|
||||
tablesProcessed int
|
||||
currentPhase string
|
||||
enabled bool
|
||||
}
|
||||
|
||||
// NewProgressTracker creates a new progress tracker
|
||||
func NewProgressTracker(manager *Manager, database, operation string) *ProgressTracker {
|
||||
return &ProgressTracker{
|
||||
manager: manager,
|
||||
database: database,
|
||||
operation: operation,
|
||||
startTime: time.Now(),
|
||||
stopCh: make(chan struct{}),
|
||||
enabled: true,
|
||||
}
|
||||
}
|
||||
|
||||
// Start begins sending periodic progress updates
|
||||
func (pt *ProgressTracker) Start(interval time.Duration) {
|
||||
if !pt.enabled || pt.manager == nil || !pt.manager.HasEnabledNotifiers() {
|
||||
return
|
||||
}
|
||||
|
||||
pt.ticker = time.NewTicker(interval)
|
||||
|
||||
go func() {
|
||||
for {
|
||||
select {
|
||||
case <-pt.ticker.C:
|
||||
pt.sendProgressUpdate()
|
||||
case <-pt.stopCh:
|
||||
return
|
||||
}
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
// Stop stops sending progress updates
|
||||
func (pt *ProgressTracker) Stop() {
|
||||
if pt.ticker != nil {
|
||||
pt.ticker.Stop()
|
||||
}
|
||||
close(pt.stopCh)
|
||||
}
|
||||
|
||||
// SetTotals sets the expected totals for tracking
|
||||
func (pt *ProgressTracker) SetTotals(bytes int64, tables int) {
|
||||
pt.mu.Lock()
|
||||
defer pt.mu.Unlock()
|
||||
pt.bytesTotal = bytes
|
||||
pt.tablesTotal = tables
|
||||
}
|
||||
|
||||
// UpdateBytes updates the number of bytes processed
|
||||
func (pt *ProgressTracker) UpdateBytes(bytes int64) {
|
||||
pt.mu.Lock()
|
||||
defer pt.mu.Unlock()
|
||||
pt.bytesProcessed = bytes
|
||||
}
|
||||
|
||||
// UpdateTables updates the number of tables processed
|
||||
func (pt *ProgressTracker) UpdateTables(tables int) {
|
||||
pt.mu.Lock()
|
||||
defer pt.mu.Unlock()
|
||||
pt.tablesProcessed = tables
|
||||
}
|
||||
|
||||
// SetPhase sets the current operation phase
|
||||
func (pt *ProgressTracker) SetPhase(phase string) {
|
||||
pt.mu.Lock()
|
||||
defer pt.mu.Unlock()
|
||||
pt.currentPhase = phase
|
||||
}
|
||||
|
||||
// GetProgress returns current progress information
|
||||
func (pt *ProgressTracker) GetProgress() ProgressInfo {
|
||||
pt.mu.RLock()
|
||||
defer pt.mu.RUnlock()
|
||||
|
||||
elapsed := time.Since(pt.startTime)
|
||||
|
||||
var percentBytes, percentTables float64
|
||||
if pt.bytesTotal > 0 {
|
||||
percentBytes = float64(pt.bytesProcessed) / float64(pt.bytesTotal) * 100
|
||||
}
|
||||
if pt.tablesTotal > 0 {
|
||||
percentTables = float64(pt.tablesProcessed) / float64(pt.tablesTotal) * 100
|
||||
}
|
||||
|
||||
// Estimate remaining time based on bytes processed
|
||||
var estimatedRemaining time.Duration
|
||||
if pt.bytesProcessed > 0 && pt.bytesTotal > 0 {
|
||||
rate := float64(pt.bytesProcessed) / elapsed.Seconds()
|
||||
remaining := pt.bytesTotal - pt.bytesProcessed
|
||||
estimatedRemaining = time.Duration(float64(remaining) / rate * float64(time.Second))
|
||||
}
|
||||
|
||||
return ProgressInfo{
|
||||
Database: pt.database,
|
||||
Operation: pt.operation,
|
||||
Phase: pt.currentPhase,
|
||||
BytesProcessed: pt.bytesProcessed,
|
||||
BytesTotal: pt.bytesTotal,
|
||||
TablesProcessed: pt.tablesProcessed,
|
||||
TablesTotal: pt.tablesTotal,
|
||||
PercentBytes: percentBytes,
|
||||
PercentTables: percentTables,
|
||||
ElapsedTime: elapsed,
|
||||
EstimatedRemaining: estimatedRemaining,
|
||||
StartTime: pt.startTime,
|
||||
}
|
||||
}
|
||||
|
||||
// sendProgressUpdate sends a progress notification
|
||||
func (pt *ProgressTracker) sendProgressUpdate() {
|
||||
progress := pt.GetProgress()
|
||||
|
||||
message := fmt.Sprintf("%s of database '%s' in progress: %s",
|
||||
pt.operation, pt.database, progress.FormatSummary())
|
||||
|
||||
event := NewEvent(EventType(pt.operation+"_progress"), SeverityInfo, message).
|
||||
WithDatabase(pt.database).
|
||||
WithDetail("operation", pt.operation).
|
||||
WithDetail("phase", progress.Phase).
|
||||
WithDetail("bytes_processed", formatBytes(progress.BytesProcessed)).
|
||||
WithDetail("bytes_total", formatBytes(progress.BytesTotal)).
|
||||
WithDetail("percent_bytes", fmt.Sprintf("%.1f%%", progress.PercentBytes)).
|
||||
WithDetail("tables_processed", fmt.Sprintf("%d", progress.TablesProcessed)).
|
||||
WithDetail("tables_total", fmt.Sprintf("%d", progress.TablesTotal)).
|
||||
WithDetail("percent_tables", fmt.Sprintf("%.1f%%", progress.PercentTables)).
|
||||
WithDetail("elapsed_time", progress.ElapsedTime.String()).
|
||||
WithDetail("estimated_remaining", progress.EstimatedRemaining.String())
|
||||
|
||||
// Send asynchronously
|
||||
go pt.manager.NotifySync(context.Background(), event)
|
||||
}
|
||||
|
||||
// ProgressInfo contains snapshot of current progress
|
||||
type ProgressInfo struct {
|
||||
Database string
|
||||
Operation string
|
||||
Phase string
|
||||
BytesProcessed int64
|
||||
BytesTotal int64
|
||||
TablesProcessed int
|
||||
TablesTotal int
|
||||
PercentBytes float64
|
||||
PercentTables float64
|
||||
ElapsedTime time.Duration
|
||||
EstimatedRemaining time.Duration
|
||||
StartTime time.Time
|
||||
}
|
||||
|
||||
// FormatSummary returns a human-readable progress summary
|
||||
func (pi *ProgressInfo) FormatSummary() string {
|
||||
if pi.TablesTotal > 0 {
|
||||
return fmt.Sprintf("%d/%d tables (%.1f%%), %s elapsed",
|
||||
pi.TablesProcessed, pi.TablesTotal, pi.PercentTables,
|
||||
formatDuration(pi.ElapsedTime))
|
||||
}
|
||||
|
||||
if pi.BytesTotal > 0 {
|
||||
return fmt.Sprintf("%s/%s (%.1f%%), %s elapsed, %s remaining",
|
||||
formatBytes(pi.BytesProcessed), formatBytes(pi.BytesTotal),
|
||||
pi.PercentBytes, formatDuration(pi.ElapsedTime),
|
||||
formatDuration(pi.EstimatedRemaining))
|
||||
}
|
||||
|
||||
return fmt.Sprintf("%s elapsed", formatDuration(pi.ElapsedTime))
|
||||
}
|
||||
|
||||
// Helper function to format bytes
|
||||
func formatProgressBytes(bytes int64) string {
|
||||
const unit = 1024
|
||||
if bytes < unit {
|
||||
return fmt.Sprintf("%d B", bytes)
|
||||
}
|
||||
div, exp := int64(unit), 0
|
||||
for n := bytes / unit; n >= unit; n /= unit {
|
||||
div *= unit
|
||||
exp++
|
||||
}
|
||||
return fmt.Sprintf("%.1f %cB", float64(bytes)/float64(div), "KMGTPE"[exp])
|
||||
}
|
||||
|
||||
// Helper function to format duration
|
||||
func formatProgressDuration(d time.Duration) string {
|
||||
if d < time.Minute {
|
||||
return fmt.Sprintf("%.0fs", d.Seconds())
|
||||
}
|
||||
if d < time.Hour {
|
||||
return fmt.Sprintf("%.1fm", d.Minutes())
|
||||
}
|
||||
return fmt.Sprintf("%.1fh", d.Hours())
|
||||
}
|
||||
533
internal/tui/catalog_dashboard.go
Normal file
533
internal/tui/catalog_dashboard.go
Normal file
@ -0,0 +1,533 @@
|
||||
package tui
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"sort"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"dbbackup/internal/catalog"
|
||||
|
||||
tea "github.com/charmbracelet/bubbletea"
|
||||
"github.com/charmbracelet/lipgloss"
|
||||
)
|
||||
|
||||
// CatalogDashboardView displays an interactive catalog browser
|
||||
type CatalogDashboardView struct {
|
||||
catalog catalog.Catalog
|
||||
entries []*catalog.Entry
|
||||
databases []string
|
||||
cursor int
|
||||
page int
|
||||
pageSize int
|
||||
totalPages int
|
||||
filter string
|
||||
filterMode bool
|
||||
selectedDB string
|
||||
loading bool
|
||||
err error
|
||||
sortBy string // "date", "size", "database", "type"
|
||||
sortDesc bool
|
||||
viewMode string // "list", "detail"
|
||||
selectedIdx int
|
||||
width int
|
||||
height int
|
||||
}
|
||||
|
||||
// Style definitions
|
||||
var (
|
||||
catalogTitleStyle = lipgloss.NewStyle().
|
||||
Bold(true).
|
||||
Foreground(lipgloss.Color("15")).
|
||||
Background(lipgloss.Color("62")).
|
||||
Padding(0, 1)
|
||||
|
||||
catalogHeaderStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("6")).
|
||||
Bold(true)
|
||||
|
||||
catalogRowStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("250"))
|
||||
|
||||
catalogSelectedStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("15")).
|
||||
Background(lipgloss.Color("62")).
|
||||
Bold(true)
|
||||
|
||||
catalogFilterStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("3")).
|
||||
Bold(true)
|
||||
|
||||
catalogStatsStyle = lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("244"))
|
||||
)
|
||||
|
||||
type catalogLoadedMsg struct {
|
||||
entries []*catalog.Entry
|
||||
databases []string
|
||||
err error
|
||||
}
|
||||
|
||||
// NewCatalogDashboardView creates a new catalog dashboard
|
||||
func NewCatalogDashboardView() *CatalogDashboardView {
|
||||
return &CatalogDashboardView{
|
||||
pageSize: 20,
|
||||
sortBy: "date",
|
||||
sortDesc: true,
|
||||
viewMode: "list",
|
||||
selectedIdx: -1,
|
||||
}
|
||||
}
|
||||
|
||||
// Init initializes the view
|
||||
func (v *CatalogDashboardView) Init() tea.Cmd {
|
||||
return v.loadCatalog()
|
||||
}
|
||||
|
||||
// Update handles messages
|
||||
func (v *CatalogDashboardView) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||
switch msg := msg.(type) {
|
||||
case tea.WindowSizeMsg:
|
||||
v.width = msg.Width
|
||||
v.height = msg.Height
|
||||
return v, nil
|
||||
|
||||
case catalogLoadedMsg:
|
||||
v.loading = false
|
||||
v.err = msg.err
|
||||
if msg.err == nil {
|
||||
v.entries = msg.entries
|
||||
v.databases = msg.databases
|
||||
v.sortEntries()
|
||||
v.calculatePages()
|
||||
}
|
||||
return v, nil
|
||||
|
||||
case tea.KeyMsg:
|
||||
if v.filterMode {
|
||||
return v.handleFilterKeys(msg)
|
||||
}
|
||||
|
||||
switch msg.String() {
|
||||
case "q", "esc":
|
||||
if v.selectedIdx >= 0 {
|
||||
v.selectedIdx = -1
|
||||
v.viewMode = "list"
|
||||
return v, nil
|
||||
}
|
||||
return v, tea.Quit
|
||||
|
||||
case "up", "k":
|
||||
if v.cursor > 0 {
|
||||
v.cursor--
|
||||
}
|
||||
|
||||
case "down", "j":
|
||||
maxCursor := len(v.getCurrentPageEntries()) - 1
|
||||
if v.cursor < maxCursor {
|
||||
v.cursor++
|
||||
}
|
||||
|
||||
case "left", "h":
|
||||
if v.page > 0 {
|
||||
v.page--
|
||||
v.cursor = 0
|
||||
}
|
||||
|
||||
case "right", "l":
|
||||
if v.page < v.totalPages-1 {
|
||||
v.page++
|
||||
v.cursor = 0
|
||||
}
|
||||
|
||||
case "enter":
|
||||
entries := v.getCurrentPageEntries()
|
||||
if v.cursor >= 0 && v.cursor < len(entries) {
|
||||
v.selectedIdx = v.page*v.pageSize + v.cursor
|
||||
v.viewMode = "detail"
|
||||
}
|
||||
|
||||
case "/":
|
||||
v.filterMode = true
|
||||
return v, nil
|
||||
|
||||
case "s":
|
||||
// Cycle sort modes
|
||||
switch v.sortBy {
|
||||
case "date":
|
||||
v.sortBy = "size"
|
||||
case "size":
|
||||
v.sortBy = "database"
|
||||
case "database":
|
||||
v.sortBy = "type"
|
||||
case "type":
|
||||
v.sortBy = "date"
|
||||
}
|
||||
v.sortEntries()
|
||||
|
||||
case "r":
|
||||
v.sortDesc = !v.sortDesc
|
||||
v.sortEntries()
|
||||
|
||||
case "d":
|
||||
// Filter by database
|
||||
if len(v.databases) > 0 {
|
||||
return v, v.selectDatabase()
|
||||
}
|
||||
|
||||
case "c":
|
||||
// Clear filters
|
||||
v.filter = ""
|
||||
v.selectedDB = ""
|
||||
v.cursor = 0
|
||||
v.page = 0
|
||||
v.calculatePages()
|
||||
|
||||
case "R":
|
||||
// Reload catalog
|
||||
v.loading = true
|
||||
return v, v.loadCatalog()
|
||||
}
|
||||
}
|
||||
|
||||
return v, nil
|
||||
}
|
||||
|
||||
// View renders the view
|
||||
func (v *CatalogDashboardView) View() string {
|
||||
if v.loading {
|
||||
return catalogTitleStyle.Render("Catalog Dashboard") + "\n\n" +
|
||||
"Loading catalog...\n"
|
||||
}
|
||||
|
||||
if v.err != nil {
|
||||
return catalogTitleStyle.Render("Catalog Dashboard") + "\n\n" +
|
||||
errorStyle.Render(fmt.Sprintf("Error: %v", v.err)) + "\n\n" +
|
||||
infoStyle.Render("Press 'q' to quit")
|
||||
}
|
||||
|
||||
if v.viewMode == "detail" && v.selectedIdx >= 0 && v.selectedIdx < len(v.entries) {
|
||||
return v.renderDetail()
|
||||
}
|
||||
|
||||
return v.renderList()
|
||||
}
|
||||
|
||||
// renderList renders the list view
|
||||
func (v *CatalogDashboardView) renderList() string {
|
||||
var b strings.Builder
|
||||
|
||||
// Title
|
||||
b.WriteString(catalogTitleStyle.Render("Catalog Dashboard"))
|
||||
b.WriteString("\n\n")
|
||||
|
||||
// Stats
|
||||
totalSize := int64(0)
|
||||
for _, e := range v.entries {
|
||||
totalSize += e.SizeBytes
|
||||
}
|
||||
stats := fmt.Sprintf("Total: %d backups | Size: %s | Databases: %d",
|
||||
len(v.entries), formatCatalogBytes(totalSize), len(v.databases))
|
||||
b.WriteString(catalogStatsStyle.Render(stats))
|
||||
b.WriteString("\n\n")
|
||||
|
||||
// Filters and sort
|
||||
filters := []string{}
|
||||
if v.filter != "" {
|
||||
filters = append(filters, fmt.Sprintf("Filter: %s", v.filter))
|
||||
}
|
||||
if v.selectedDB != "" {
|
||||
filters = append(filters, fmt.Sprintf("Database: %s", v.selectedDB))
|
||||
}
|
||||
sortInfo := fmt.Sprintf("Sort: %s (%s)", v.sortBy, map[bool]string{true: "desc", false: "asc"}[v.sortDesc])
|
||||
filters = append(filters, sortInfo)
|
||||
|
||||
if len(filters) > 0 {
|
||||
b.WriteString(catalogFilterStyle.Render(strings.Join(filters, " | ")))
|
||||
b.WriteString("\n\n")
|
||||
}
|
||||
|
||||
// Header
|
||||
header := fmt.Sprintf("%-12s %-20s %-15s %-12s %-10s",
|
||||
"Date", "Database", "Type", "Size", "Status")
|
||||
b.WriteString(catalogHeaderStyle.Render(header))
|
||||
b.WriteString("\n")
|
||||
b.WriteString(strings.Repeat("─", 75))
|
||||
b.WriteString("\n")
|
||||
|
||||
// Entries
|
||||
entries := v.getCurrentPageEntries()
|
||||
if len(entries) == 0 {
|
||||
b.WriteString(infoStyle.Render("No backups found"))
|
||||
b.WriteString("\n")
|
||||
} else {
|
||||
for i, entry := range entries {
|
||||
date := entry.CreatedAt.Format("2006-01-02")
|
||||
time := entry.CreatedAt.Format("15:04")
|
||||
database := entry.Database
|
||||
if len(database) > 18 {
|
||||
database = database[:15] + "..."
|
||||
}
|
||||
backupType := entry.BackupType
|
||||
size := formatCatalogBytes(entry.SizeBytes)
|
||||
status := string(entry.Status)
|
||||
|
||||
line := fmt.Sprintf("%-12s %-20s %-15s %-12s %-10s",
|
||||
date+" "+time, database, backupType, size, status)
|
||||
|
||||
if i == v.cursor {
|
||||
b.WriteString(catalogSelectedStyle.Render(line))
|
||||
} else {
|
||||
b.WriteString(catalogRowStyle.Render(line))
|
||||
}
|
||||
b.WriteString("\n")
|
||||
}
|
||||
}
|
||||
|
||||
// Pagination
|
||||
if v.totalPages > 1 {
|
||||
b.WriteString("\n")
|
||||
pagination := fmt.Sprintf("Page %d/%d", v.page+1, v.totalPages)
|
||||
b.WriteString(catalogStatsStyle.Render(pagination))
|
||||
b.WriteString("\n")
|
||||
}
|
||||
|
||||
// Help
|
||||
b.WriteString("\n")
|
||||
help := "↑/↓:Navigate ←/→:Page Enter:Details s:Sort r:Reverse d:Database /:Filter c:Clear R:Reload q:Quit"
|
||||
b.WriteString(infoStyle.Render(help))
|
||||
|
||||
if v.filterMode {
|
||||
b.WriteString("\n\n")
|
||||
b.WriteString(catalogFilterStyle.Render(fmt.Sprintf("Filter: %s_", v.filter)))
|
||||
}
|
||||
|
||||
return b.String()
|
||||
}
|
||||
|
||||
// renderDetail renders the detail view
|
||||
func (v *CatalogDashboardView) renderDetail() string {
|
||||
entry := v.entries[v.selectedIdx]
|
||||
|
||||
var b strings.Builder
|
||||
|
||||
b.WriteString(catalogTitleStyle.Render("Backup Details"))
|
||||
b.WriteString("\n\n")
|
||||
|
||||
// Basic info
|
||||
b.WriteString(catalogHeaderStyle.Render("Basic Information"))
|
||||
b.WriteString("\n")
|
||||
b.WriteString(fmt.Sprintf("Database: %s\n", entry.Database))
|
||||
b.WriteString(fmt.Sprintf("Type: %s\n", entry.BackupType))
|
||||
b.WriteString(fmt.Sprintf("Status: %s\n", entry.Status))
|
||||
b.WriteString(fmt.Sprintf("Timestamp: %s\n", entry.CreatedAt.Format("2006-01-02 15:04:05")))
|
||||
b.WriteString("\n")
|
||||
|
||||
// File info
|
||||
b.WriteString(catalogHeaderStyle.Render("File Information"))
|
||||
b.WriteString("\n")
|
||||
b.WriteString(fmt.Sprintf("Path: %s\n", entry.BackupPath))
|
||||
b.WriteString(fmt.Sprintf("Size: %s (%d bytes)\n", formatCatalogBytes(entry.SizeBytes), entry.SizeBytes))
|
||||
compressed := entry.Compression != ""
|
||||
b.WriteString(fmt.Sprintf("Compressed: %s\n", map[bool]string{true: "Yes (" + entry.Compression + ")", false: "No"}[compressed]))
|
||||
b.WriteString(fmt.Sprintf("Encrypted: %s\n", map[bool]string{true: "Yes", false: "No"}[entry.Encrypted]))
|
||||
b.WriteString("\n")
|
||||
|
||||
// Duration info
|
||||
if entry.Duration > 0 {
|
||||
b.WriteString(catalogHeaderStyle.Render("Performance"))
|
||||
b.WriteString("\n")
|
||||
duration := time.Duration(entry.Duration * float64(time.Second))
|
||||
b.WriteString(fmt.Sprintf("Duration: %s\n", duration))
|
||||
throughput := float64(entry.SizeBytes) / entry.Duration / (1024 * 1024)
|
||||
b.WriteString(fmt.Sprintf("Throughput: %.2f MB/s\n", throughput))
|
||||
b.WriteString("\n")
|
||||
}
|
||||
|
||||
// Additional metadata
|
||||
if len(entry.Metadata) > 0 {
|
||||
b.WriteString(catalogHeaderStyle.Render("Metadata"))
|
||||
b.WriteString("\n")
|
||||
keys := make([]string, 0, len(entry.Metadata))
|
||||
for k := range entry.Metadata {
|
||||
keys = append(keys, k)
|
||||
}
|
||||
sort.Strings(keys)
|
||||
for _, k := range keys {
|
||||
b.WriteString(fmt.Sprintf("%-15s %s\n", k+":", entry.Metadata[k]))
|
||||
}
|
||||
b.WriteString("\n")
|
||||
}
|
||||
|
||||
// Help
|
||||
b.WriteString("\n")
|
||||
b.WriteString(infoStyle.Render("Press ESC or 'q' to return to list"))
|
||||
|
||||
return b.String()
|
||||
}
|
||||
|
||||
// Helper methods
|
||||
func (v *CatalogDashboardView) loadCatalog() tea.Cmd {
|
||||
return func() tea.Msg {
|
||||
// Open catalog
|
||||
home, err := os.UserHomeDir()
|
||||
if err != nil {
|
||||
return catalogLoadedMsg{err: err}
|
||||
}
|
||||
|
||||
catalogPath := filepath.Join(home, ".dbbackup", "catalog.db")
|
||||
cat, err := catalog.NewSQLiteCatalog(catalogPath)
|
||||
if err != nil {
|
||||
return catalogLoadedMsg{err: err}
|
||||
}
|
||||
defer cat.Close()
|
||||
|
||||
// Load entries
|
||||
entries, err := cat.Search(context.Background(), &catalog.SearchQuery{})
|
||||
if err != nil {
|
||||
return catalogLoadedMsg{err: err}
|
||||
}
|
||||
|
||||
// Load databases
|
||||
databases, err := cat.ListDatabases(context.Background())
|
||||
if err != nil {
|
||||
return catalogLoadedMsg{err: err}
|
||||
}
|
||||
|
||||
return catalogLoadedMsg{
|
||||
entries: entries,
|
||||
databases: databases,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (v *CatalogDashboardView) sortEntries() {
|
||||
sort.Slice(v.entries, func(i, j int) bool {
|
||||
var less bool
|
||||
switch v.sortBy {
|
||||
case "date":
|
||||
less = v.entries[i].CreatedAt.Before(v.entries[j].CreatedAt)
|
||||
case "size":
|
||||
less = v.entries[i].SizeBytes < v.entries[j].SizeBytes
|
||||
case "database":
|
||||
less = v.entries[i].Database < v.entries[j].Database
|
||||
case "type":
|
||||
less = v.entries[i].BackupType < v.entries[j].BackupType
|
||||
default:
|
||||
less = v.entries[i].CreatedAt.Before(v.entries[j].CreatedAt)
|
||||
}
|
||||
if v.sortDesc {
|
||||
return !less
|
||||
}
|
||||
return less
|
||||
})
|
||||
v.calculatePages()
|
||||
}
|
||||
|
||||
func (v *CatalogDashboardView) calculatePages() {
|
||||
filtered := v.getFilteredEntries()
|
||||
v.totalPages = (len(filtered) + v.pageSize - 1) / v.pageSize
|
||||
if v.totalPages == 0 {
|
||||
v.totalPages = 1
|
||||
}
|
||||
if v.page >= v.totalPages {
|
||||
v.page = v.totalPages - 1
|
||||
}
|
||||
if v.page < 0 {
|
||||
v.page = 0
|
||||
}
|
||||
}
|
||||
|
||||
func (v *CatalogDashboardView) getFilteredEntries() []*catalog.Entry {
|
||||
filtered := []*catalog.Entry{}
|
||||
for _, e := range v.entries {
|
||||
if v.selectedDB != "" && e.Database != v.selectedDB {
|
||||
continue
|
||||
}
|
||||
if v.filter != "" {
|
||||
match := strings.Contains(strings.ToLower(e.Database), strings.ToLower(v.filter)) ||
|
||||
strings.Contains(strings.ToLower(e.BackupPath), strings.ToLower(v.filter))
|
||||
if !match {
|
||||
continue
|
||||
}
|
||||
}
|
||||
filtered = append(filtered, e)
|
||||
}
|
||||
return filtered
|
||||
}
|
||||
|
||||
func (v *CatalogDashboardView) getCurrentPageEntries() []*catalog.Entry {
|
||||
filtered := v.getFilteredEntries()
|
||||
start := v.page * v.pageSize
|
||||
end := start + v.pageSize
|
||||
if end > len(filtered) {
|
||||
end = len(filtered)
|
||||
}
|
||||
if start >= len(filtered) {
|
||||
return []*catalog.Entry{}
|
||||
}
|
||||
return filtered[start:end]
|
||||
}
|
||||
|
||||
func (v *CatalogDashboardView) handleFilterKeys(msg tea.KeyMsg) (tea.Model, tea.Cmd) {
|
||||
switch msg.String() {
|
||||
case "enter", "esc":
|
||||
v.filterMode = false
|
||||
v.cursor = 0
|
||||
v.page = 0
|
||||
v.calculatePages()
|
||||
return v, nil
|
||||
|
||||
case "backspace":
|
||||
if len(v.filter) > 0 {
|
||||
v.filter = v.filter[:len(v.filter)-1]
|
||||
}
|
||||
|
||||
default:
|
||||
if len(msg.String()) == 1 {
|
||||
v.filter += msg.String()
|
||||
}
|
||||
}
|
||||
|
||||
return v, nil
|
||||
}
|
||||
|
||||
func (v *CatalogDashboardView) selectDatabase() tea.Cmd {
|
||||
// Simple cycling through databases
|
||||
if v.selectedDB == "" {
|
||||
if len(v.databases) > 0 {
|
||||
v.selectedDB = v.databases[0]
|
||||
}
|
||||
} else {
|
||||
for i, db := range v.databases {
|
||||
if db == v.selectedDB {
|
||||
if i+1 < len(v.databases) {
|
||||
v.selectedDB = v.databases[i+1]
|
||||
} else {
|
||||
v.selectedDB = ""
|
||||
}
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
v.cursor = 0
|
||||
v.page = 0
|
||||
v.calculatePages()
|
||||
return nil
|
||||
}
|
||||
|
||||
func formatCatalogBytes(bytes int64) string {
|
||||
const unit = 1024
|
||||
if bytes < unit {
|
||||
return fmt.Sprintf("%d B", bytes)
|
||||
}
|
||||
div, exp := int64(unit), 0
|
||||
for n := bytes / unit; n >= unit; n /= unit {
|
||||
div *= unit
|
||||
exp++
|
||||
}
|
||||
return fmt.Sprintf("%.1f %cB", float64(bytes)/float64(div), "KMGTPE"[exp])
|
||||
}
|
||||
@ -62,7 +62,7 @@ func (c *ChainView) loadChains() tea.Msg {
|
||||
// Open catalog - use default path
|
||||
home, _ := os.UserHomeDir()
|
||||
catalogPath := filepath.Join(home, ".dbbackup", "catalog.db")
|
||||
|
||||
|
||||
cat, err := catalog.NewSQLiteCatalog(catalogPath)
|
||||
if err != nil {
|
||||
return chainLoadedMsg{err: fmt.Errorf("failed to open catalog: %w", err)}
|
||||
@ -230,7 +230,7 @@ func (c *ChainView) View() string {
|
||||
|
||||
if len(chain.Incrementals) > 0 {
|
||||
b.WriteString(fmt.Sprintf(" [CHAIN] %d Incremental(s)\n", len(chain.Incrementals)))
|
||||
|
||||
|
||||
// Show first few
|
||||
limit := 3
|
||||
for i, inc := range chain.Incrementals {
|
||||
|
||||
@ -455,6 +455,7 @@ func (m *MenuModel) handleDiagnoseBackup() (tea.Model, tea.Cmd) {
|
||||
browser := NewArchiveBrowser(m.config, m.logger, m, m.ctx, "diagnose")
|
||||
return browser, browser.Init()
|
||||
}
|
||||
|
||||
// handleSchedule shows backup schedule
|
||||
func (m *MenuModel) handleSchedule() (tea.Model, tea.Cmd) {
|
||||
schedule := NewScheduleView(m.config, m.logger, m)
|
||||
@ -466,6 +467,7 @@ func (m *MenuModel) handleChain() (tea.Model, tea.Cmd) {
|
||||
chain := NewChainView(m.config, m.logger, m)
|
||||
return chain, chain.Init()
|
||||
}
|
||||
|
||||
// handleTools opens the tools submenu
|
||||
func (m *MenuModel) handleTools() (tea.Model, tea.Cmd) {
|
||||
tools := NewToolsMenu(m.config, m.logger, m, m.ctx)
|
||||
|
||||
@ -14,21 +14,21 @@ import (
|
||||
|
||||
// ScheduleView displays systemd timer schedules
|
||||
type ScheduleView struct {
|
||||
config *config.Config
|
||||
logger logger.Logger
|
||||
parent tea.Model
|
||||
timers []TimerInfo
|
||||
loading bool
|
||||
error string
|
||||
quitting bool
|
||||
config *config.Config
|
||||
logger logger.Logger
|
||||
parent tea.Model
|
||||
timers []TimerInfo
|
||||
loading bool
|
||||
error string
|
||||
quitting bool
|
||||
}
|
||||
|
||||
type TimerInfo struct {
|
||||
Name string
|
||||
NextRun string
|
||||
Left string
|
||||
LastRun string
|
||||
Active string
|
||||
Name string
|
||||
NextRun string
|
||||
Left string
|
||||
LastRun string
|
||||
Active string
|
||||
}
|
||||
|
||||
func NewScheduleView(cfg *config.Config, log logger.Logger, parent tea.Model) *ScheduleView {
|
||||
@ -66,16 +66,16 @@ func (s *ScheduleView) loadTimers() tea.Msg {
|
||||
}
|
||||
|
||||
timers := parseTimerList(string(output))
|
||||
|
||||
|
||||
// Filter for backup-related timers
|
||||
var filtered []TimerInfo
|
||||
for _, timer := range timers {
|
||||
name := strings.ToLower(timer.Name)
|
||||
if strings.Contains(name, "backup") ||
|
||||
strings.Contains(name, "dbbackup") ||
|
||||
strings.Contains(name, "postgres") ||
|
||||
strings.Contains(name, "mysql") ||
|
||||
strings.Contains(name, "mariadb") {
|
||||
if strings.Contains(name, "backup") ||
|
||||
strings.Contains(name, "dbbackup") ||
|
||||
strings.Contains(name, "postgres") ||
|
||||
strings.Contains(name, "mysql") ||
|
||||
strings.Contains(name, "mariadb") {
|
||||
filtered = append(filtered, timer)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user