feat: add current namespace retrieval and all-namespaces backup functionality

Author: rogosprojects

README.md

@@ -15,10 +15,13 @@ A Go application that backs up Kubernetes resources from a specified namespace b
 
 ## Features
 
-- Exports all standard Kubernetes resources from a namespace
+- Exports all standard Kubernetes resources from a namespace or all namespaces
+- Uses the current namespace from kubeconfig when no namespace is specified
 - Organizes backups by resource kind in separate directories
 - Thoroughly cleans manifests by removing server-side and cluster-specific fields
 - Timestamp-based backup directories
+- Colorful and descriptive console output with emojis
+- Resource type filtering for selective backups
 
 ## Installation
 
@@ -58,9 +61,15 @@ docker run --rm -v ~/.kube:/root/.kube -v $(pwd)/backups:/backups kbak:latest --
 # Show version information
 ./kbak --version
 
-# Backup a namespace using the current kubeconfig
+# Backup resources from your current namespace (from kubeconfig)
+./kbak
+
+# Backup a specific namespace
 ./kbak --namespace your-namespace
 
+# Backup all namespaces
+./kbak --all-namespaces
+
 # Specify a custom kubeconfig file
 ./kbak --namespace your-namespace --kubeconfig /path/to/kubeconfig
 
@@ -72,6 +81,9 @@ docker run --rm -v ~/.kube:/root/.kube -v $(pwd)/backups:/backups kbak:latest --
 
 # Backup only pods and deployments
 ./kbak --namespace your-namespace --pod --deployment --all-resources=false
+
+# Backup resources with verbose output
+./kbak --namespace your-namespace --verbose
 ```
 
 ### Resource Type Filtering
@@ -111,8 +123,7 @@ The tool automatically backs up the following resource types:
 
 ## Output Structure
 
-Backups are organized as follows:
-
+### Single Namespace Backup
 ```
 02Jan2006-15:04/
 └── namespace/
@@ -127,6 +138,24 @@ Backups are organized as follows:
     │   └── ...
     └── ...
 ```
+
+### All Namespaces Backup
+```
+02Jan2006-15:04/
+└── all-namespaces/
+    ├── namespace1/
+    │   ├── Pod/
+    │   │   ├── my-pod.yaml
+    │   │   └── ...
+    │   ├── Deployment/
+    │   │   └── ...
+    │   └── ...
+    ├── namespace2/
+    │   ├── Pod/
+    │   │   └── ...
+    │   └── ...
+    └── ...
+```
 ## License
 
 MIT License

cmd/kbak/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"context"
 	"flag"
 	"fmt"
 	"os"
@@ -10,6 +11,8 @@ import (
 	"github.com/rogosprojects/kbak/pkg/backup"
 	"github.com/rogosprojects/kbak/pkg/client"
 	"github.com/rogosprojects/kbak/pkg/utils"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/client-go/tools/clientcmd"
 	"k8s.io/client-go/util/homedir"
 )
 
@@ -48,7 +51,7 @@ func main() {
 	var resFlags resourceFlags
 
 	// Basic flags
-	flag.StringVar(&namespace, "namespace", "", "Namespace to backup (required unless --all-namespaces is used)")
+	flag.StringVar(&namespace, "namespace", "", "Namespace to backup (uses current namespace from kubeconfig if not specified)")
 	flag.StringVar(&outputDir, "output", "backups", "Output directory for backup files")
 	flag.BoolVar(&verbose, "verbose", false, "Show verbose output")
 	flag.BoolVar(&showVersion, "version", false, "Show version information and exit")
@@ -80,19 +83,11 @@ func main() {
 	flag.Parse()
 
 	if showVersion {
-		fmt.Printf("%s %skbak%s version %s %s%s\n",
+		fmt.Printf("%s %skbak%s version %s%s%s\n",
 			utils.K8sEmoji, utils.Bold, utils.Reset, utils.Cyan, Version, utils.Reset)
 		os.Exit(0)
 	}
 
-	// Validate namespace requirements
-	if namespace == "" && !allNamespaces {
-		fmt.Printf("%s %s%sError: either --namespace or --all-namespaces flag is required%s\n",
-			utils.ErrorEmoji, utils.Red, utils.Bold, utils.Reset)
-		flag.Usage()
-		os.Exit(1)
-	}
-
 	if allNamespaces && namespace != "" {
 		fmt.Printf("%s %s%sWarning: --namespace flag is ignored when --all-namespaces is used%s\n",
 			utils.WarningEmoji, utils.Yellow, utils.Bold, utils.Reset)
@@ -107,13 +102,92 @@ func main() {
 		os.Exit(1)
 	}
 
-	// Handle all-namespaces case (not implemented yet, just a placeholder)
+	// If namespace is not specified and not using all-namespaces, get the current namespace from kubeconfig
+	if namespace == "" && !allNamespaces {
+		loadingRules := clientcmd.NewDefaultClientConfigLoadingRules()
+		loadingRules.ExplicitPath = kubeconfig
+		configOverrides := &clientcmd.ConfigOverrides{}
+		kubeConfig := clientcmd.NewNonInteractiveDeferredLoadingClientConfig(loadingRules, configOverrides)
+		currentNamespace, _, err := kubeConfig.Namespace()
+		if err != nil {
+			fmt.Printf("%s %s%sError getting current namespace: %v%s\n",
+				utils.ErrorEmoji, utils.Red, utils.Bold, err, utils.Reset)
+			os.Exit(1)
+		}
+		namespace = currentNamespace
+		if verbose {
+			fmt.Printf(" %sUsing current namespace: %s%s\n",
+				utils.Cyan, namespace, utils.Reset)
+		}
+	}
+
+	// Handle all-namespaces case
 	if allNamespaces {
-		fmt.Printf("%s %s%sError: --all-namespaces is not implemented yet%s\n",
-			utils.ErrorEmoji, utils.Red, utils.Bold, utils.Reset)
-		os.Exit(1)
-		// Here we would get a list of all namespaces and iterate through them
-		// This feature is left for future implementation
+		// Get all namespaces
+		namespaces, err := k8sClient.Clientset.CoreV1().Namespaces().List(context.TODO(), metav1.ListOptions{})
+		if err != nil {
+			fmt.Printf("%s %s%sError listing namespaces: %v%s\n",
+				utils.ErrorEmoji, utils.Red, utils.Bold, err, utils.Reset)
+			os.Exit(1)
+		}
+
+		// Create parent backup directory
+		timestamp := time.Now().Format("02Jan2006-15:04")
+		parentBackupDir := filepath.Join(outputDir, timestamp, "all-namespaces")
+
+		if err := os.MkdirAll(parentBackupDir, 0755); err != nil {
+			fmt.Printf("%s %s%sError creating output directory: %v%s\n",
+				utils.ErrorEmoji, utils.Red, utils.Bold, err, utils.Reset)
+			os.Exit(1)
+		}
+
+		fmt.Printf("%s %s%sStarting backup of all namespaces to '%s'%s\n\n",
+			utils.StartEmoji, utils.Blue, utils.Bold, parentBackupDir, utils.Reset)
+
+		totalResourceCount := 0
+		totalErrorCount := 0
+
+		// Process each namespace
+		for _, ns := range namespaces.Items {
+			nsName := ns.Name
+			nsBackupDir := filepath.Join(parentBackupDir, nsName)
+
+			if err := os.MkdirAll(nsBackupDir, 0755); err != nil {
+				fmt.Printf("%s %s%sError creating directory for namespace %s: %v%s\n",
+					utils.ErrorEmoji, utils.Red, utils.Bold, nsName, err, utils.Reset)
+				totalErrorCount++
+				continue
+			}
+
+			// Prepare resource type filter
+			selectedTypes := buildResourceTypeMap(resFlags)
+
+			fmt.Printf("%sProcessing namespace: %s%s\n",
+				utils.Blue, nsName, utils.Reset)
+
+			// Perform backup for this namespace
+			resourceCount, errorCount := backup.PerformBackup(k8sClient, nsName, nsBackupDir, selectedTypes, verbose)
+
+			totalResourceCount += resourceCount
+			totalErrorCount += errorCount
+		}
+
+		if totalResourceCount > 0 {
+			fmt.Printf("\n%s %s%sBackup completed successfully to %s (%d resources total across all namespaces)%s\n",
+				utils.SuccessEmoji, utils.Green, utils.Bold, parentBackupDir, totalResourceCount, utils.Reset)
+		} else {
+			fmt.Printf("\n%s %s%sNo resources found to backup in any namespace%s\n",
+				utils.WarningEmoji, utils.Yellow, utils.Bold, utils.Reset)
+		}
+
+		// Exit with error code if there were errors
+		if totalErrorCount > 0 {
+			fmt.Printf("%s %s%sCompleted with %d errors%s\n",
+				utils.ErrorEmoji, utils.Red, utils.Bold, totalErrorCount, utils.Reset)
+			os.Exit(1)
+		}
+
+		os.Exit(0)
 	}
 
 	// Create output directory with timestamp
@@ -145,7 +219,7 @@ func main() {
 			utils.SuccessEmoji, utils.Green, utils.Bold, backupDir, resourceCount, utils.Reset)
 	} else {
 		fmt.Printf("\n%s %s%sNo resources found to backup in namespace '%s'%s\n",
-			utils.InfoEmoji, utils.Yellow, utils.Bold, namespace, utils.Reset)
+			utils.WarningEmoji, utils.Yellow, utils.Bold, namespace, utils.Reset)
 	}
 
 	// Exit with error code if there were errors

pkg/backup/backup.go

@@ -173,8 +173,8 @@ func backupResourceType(k8sClient *client.K8sClient, namespace, backupDir string
 		// Ensure the filename is valid for the filesystem
 		safeName := ensureValidFilename(name)
 		if safeName != name && verbose {
-			fmt.Printf("%s%sResource name %q sanitized to %q for filesystem compatibility%s\n",
-				utils.InfoEmoji, utils.BrightBlue, name, safeName, utils.Reset)
+			fmt.Printf("%sResource name %q sanitized to %q for filesystem compatibility%s\n",
+				utils.BrightBlue, name, safeName, utils.Reset)
 		}
 
 		// Remove cluster-specific and runtime fields

pkg/utils/colors.go

@@ -30,8 +30,8 @@ const (
 
 // Emojis for different message types
 const (
-	InfoEmoji    = "ℹ️"
-	SuccessEmoji = "✅"
+	InfoEmoji    = "ℹ "
+	SuccessEmoji = "✔ "
 	WarningEmoji = "⚠️"
 	ErrorEmoji   = "❌"
 	BackupEmoji  = "💾"