Added default values handling for custom CLI flags (#1379)

Co-authored-by: Andriy Knysh <[email protected]>

Author: MacherelR

atmos.yaml

@@ -402,11 +402,10 @@ docs:
         show_outputs: true
         sort_by: "name"
         hide_empty: false
-        indent_level: 2   
+        indent_level: 2
 
 version:
   check:
     enabled: true
     timeout: 1000 # ms
     frequency: 1h
-

cmd/cmd_utils.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"os"
 	"path/filepath"
+	"strconv"
 	"strings"
 	"time"
 
@@ -91,16 +92,28 @@ func processCustomCommands(
 			// Process and add flags to the command
 			for _, flag := range commandConfig.Flags {
 				if flag.Type == "bool" {
+					defaultVal := false
+					if flag.Default != "" {
+						// Accept "true"/"false" as string for bool default
+						parsed, err := strconv.ParseBool(flag.Default)
+						if err == nil {
+							defaultVal = parsed
+						}
+					}
 					if flag.Shorthand != "" {
-						customCommand.PersistentFlags().BoolP(flag.Name, flag.Shorthand, false, flag.Usage)
+						customCommand.PersistentFlags().BoolP(flag.Name, flag.Shorthand, defaultVal, flag.Usage)
 					} else {
-						customCommand.PersistentFlags().Bool(flag.Name, false, flag.Usage)
+						customCommand.PersistentFlags().Bool(flag.Name, defaultVal, flag.Usage)
 					}
 				} else {
+					defaultVal := ""
+					if flag.Default != "" {
+						defaultVal = flag.Default
+					}
 					if flag.Shorthand != "" {
-						customCommand.PersistentFlags().StringP(flag.Name, flag.Shorthand, "", flag.Usage)
+						customCommand.PersistentFlags().StringP(flag.Name, flag.Shorthand, defaultVal, flag.Usage)
 					} else {
-						customCommand.PersistentFlags().String(flag.Name, "", flag.Usage)
+						customCommand.PersistentFlags().String(flag.Name, defaultVal, flag.Usage)
 					}
 				}
 

pkg/schema/schema.go

@@ -622,6 +622,7 @@ type CommandFlag struct {
 	Description string `yaml:"description" json:"description" mapstructure:"description"`
 	Usage       string `yaml:"usage" json:"usage" mapstructure:"usage"`
 	Required    bool   `yaml:"required" json:"required" mapstructure:"required"`
+	Default     string `yaml:"default" json:"default" mapstructure:"default"`
 }
 
 type CommandEnv struct {