breaking!: envfile flags changed and escaped flags support

Author: u-yas

cmd/list.go

@@ -14,9 +14,9 @@ var listCmd = &cobra.Command{
 	Use:   "list",
 	Short: "List all key-value pairs in .env file",
 	Example: heredoc.Doc(`
-$ ennbu list
-$ ennbu list -e .env.development
-$ ennbu list --json
+	$ ennbu list
+	$ ennbu list -e .env.development
+	$ ennbu list --json
 `),
 	RunE: list.Run,
 }

cmd/root.go

@@ -24,6 +24,6 @@ func Execute() {
 }
 
 func init() {
-	rootCmd.PersistentFlags().StringP(flags.FlagEnvFile, "e", ".env", "Path to the .env file")
+	rootCmd.PersistentFlags().StringP(flags.FlagEnvFile, "f", ".env", "Path to the .env file")
 	rootCmd.PersistentFlags().StringP(flags.FlagKey, "k", "", "Key to the .env file")
 }

cmd/set.go

@@ -22,7 +22,7 @@ $ ennbu set -e .env.development -k KEY VALUE
 }
 
 func init() {
-
+	setCmd.Flags().BoolP(set.FlagEscape, "e", false, "value is escaped")
 	_ = setCmd.MarkFlagRequired(flags.FlagKey)
 	rootCmd.AddCommand(setCmd)
 }

flags/global.go

@@ -1,6 +1,6 @@
 package flags
 
 const (
-	FlagEnvFile string = "envFile"
+	FlagEnvFile string = "file"
 	FlagKey     string = "key"
 )

replace/flags.go

@@ -0,0 +1,5 @@
+package replace
+
+const (
+	FlagEscape = "escape"
+)

replace/run.go

@@ -7,12 +7,24 @@ import (
 	"strings"
 
 	"github.com/spf13/cobra"
+	"github.com/u-yas/ennbu/env"
+	"github.com/u-yas/ennbu/flags"
 )
 
 func Run(cmd *cobra.Command, args []string) error {
-	key, _ := cmd.Flags().GetString("key")
-	envFilePath, _ := cmd.Flags().GetString("envFile")
+	key, _ := cmd.Flags().GetString(flags.FlagKey)
+	envFilePath, _ := cmd.Flags().GetString(flags.FlagEnvFile)
+	escapeFlag, _ := cmd.Flags().GetBool(FlagEscape)
+
 	value := args[0]
+	if escapeFlag {
+		value = env.EscapeSpecialChars(value)
+	}
+
+	// Check if the value contains newline characters and wrap it with double quotes if necessary
+	if strings.Contains(value, "\n") {
+		value = fmt.Sprintf("\"%s\"", value)
+	}
 
 	fileName := filepath.Base(envFilePath)
 	envContent, err := os.ReadFile(envFilePath)

set/flags.go

@@ -0,0 +1,5 @@
+package set
+
+const (
+	FlagEscape = "escape"
+)

set/run.go

@@ -6,12 +6,23 @@ import (
 	"strings"
 
 	"github.com/spf13/cobra"
+	"github.com/u-yas/ennbu/env"
+	"github.com/u-yas/ennbu/flags"
 )
 
 func Run(cmd *cobra.Command, args []string) error {
-	key, _ := cmd.Flags().GetString("key")
-	envFilePath, _ := cmd.Flags().GetString("envFile")
+	key, _ := cmd.Flags().GetString(flags.FlagKey)
+	envFilePath, _ := cmd.Flags().GetString(flags.FlagEnvFile)
+	escapeFlag, _ := cmd.Flags().GetBool(FlagEscape)
+
 	value := args[0]
+	if escapeFlag {
+		value = env.EscapeSpecialChars(value)
+	}
+	// Check if the value contains newline characters and wrap it with double quotes if necessary
+	if strings.Contains(value, "\n") {
+		value = fmt.Sprintf("\"%s\"", value)
+	}
 
 	envContent, err := os.ReadFile(envFilePath)
 	if err != nil {