add logic to add in --age parameter on sops commands

SayHeyD · Aug 9, 2023 · b86e6ce · b86e6ce
1 parent c603420
commit b86e6ce
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/cmd/root.go b/cmd/root.go
@@ -77,14 +77,28 @@ func executeSops(args []string) {
 	if wantedEncryptionKey == nil {
 		log.Printf("Could not find encryption key \"%s\"", appConfig.EncryptionKeyName)
 	} else {
-		err = os.Setenv("SOPS_AGE_KEY", wantedEncryptionKey.PrivateKey)
-		if err != nil {
-			log.Fatalf("could not set env variable: %v", err)
+		for index, arg := range args {
+			if arg == "sops" {
+				argsUntilSops := make([]string, len(args[:index+1]))
+				argsAfterSops := make([]string, len(args[index+1:]))
+
+				copy(argsUntilSops, args[:index+1])
+				copy(argsAfterSops, args[index+1:])
+
+				firstArgHalf := append(argsUntilSops, "--age", wantedEncryptionKey.PublicKey)
+				args = append(firstArgHalf, argsAfterSops...)
+			}
 		}
+
 	}
 
 	if wantedDecryptionKey == nil {
 		log.Printf("Could not find decryption key \"%s\"", appConfig.DecryptionKeyName)
+	} else {
+		err = os.Setenv("SOPS_AGE_KEY", wantedDecryptionKey.PrivateKey)
+		if err != nil {
+			log.Fatalf("could not set env variable: %v", err)
+		}
 	}
 
 	var passThroughOut bytes.Buffer