Merge pull request #164 from DopplerHQ/tom_secrets_set

chore: add support for settings secrets w/o equals sign
DopplerHQ · Nov 23, 2020 · e6fb384 · e6fb384
2 parents 94a9eec + 3d8fbf6
commit e6fb384
Showing 1 changed file with 17 additions and 7 deletions.
diff --git a/pkg/cmd/secrets.go b/pkg/cmd/secrets.go
@@ -162,13 +162,23 @@ func setSecrets(cmd *cobra.Command, args []string) {
 
 	secrets := map[string]interface{}{}
 	var keys []string
-	for _, arg := range args {
-		secretArr := strings.Split(arg, "=")
-		keys = append(keys, secretArr[0])
-		if len(secretArr) < 2 {
-			secrets[secretArr[0]] = ""
-		} else {
-			secrets[secretArr[0]] = secretArr[1]
+
+	if len(args) == 2 {
+		// format: 'doppler secrets set KEY value'
+		key := args[0]
+		value := args[1]
+		keys = append(keys, key)
+		secrets[key] = value
+	} else {
+		// format: 'doppler secrets set KEY=value'
+		for _, arg := range args {
+			secretArr := strings.Split(arg, "=")
+			keys = append(keys, secretArr[0])
+			if len(secretArr) < 2 {
+				secrets[secretArr[0]] = ""
+			} else {
+				secrets[secretArr[0]] = secretArr[1]
+			}
 		}
 	}