fix: Create directory for file and credential_process commands (#71)

cmd/credential_process.go

@@ -51,11 +51,17 @@ func writeConfigFile(roles []string, destination string) error {
 	ini.PrettyEqual = true
 
 	if util.FileExists(destination) {
+		// There's an existing config file, so we'll load it in and update the existing contents
 		configINI, err = ini.Load(destination)
 		if err != nil {
 			return err
 		}
 	} else {
+		// Config file doesn't exist yet. Create it with the same perms as awscli
+		err = util.CreateFile(destination, 0700, 0600)
+		if err != nil {
+			return err
+		}
 		configINI = ini.Empty()
 	}
 

cmd/file.go

@@ -161,11 +161,17 @@ func writeCredentialsFile(credentials *creds.AwsCredentials, profile, filename s
 	ini.PrettyEqual = true
 
 	if util.FileExists(filename) {
+		// There's an existing credentials file, so we'll load it in and update the existing contents
 		credentialsINI, err = ini.Load(filename)
 		if err != nil {
 			return err
 		}
 	} else {
+		// Credentials file doesn't exist yet. Create it with the same perms as awscli
+		err = util.CreateFile(filename, 0700, 0600)
+		if err != nil {
+			return err
+		}
 		credentialsINI = ini.Empty()
 	}
 
@@ -186,5 +192,5 @@ func writeCredentialsFile(credentials *creds.AwsCredentials, profile, filename s
 		return err
 	}
 
-	return nil
+	return err
 }

util/util.go

@@ -18,10 +18,12 @@ package util
 
 import (
 	"fmt"
+	"io/fs"
 	"io/ioutil"
 	"net/http"
 	"os"
 	"os/exec"
+	"path/filepath"
 	"runtime"
 	"strings"
 
@@ -98,6 +100,28 @@ func FileExists(path string) bool {
 	return err == nil
 }
 
+func createEmptyFile(filename string, filemode fs.FileMode) error {
+	d := []byte("")
+	err := ioutil.WriteFile(filename, d, filemode)
+	return err
+}
+
+// CreateFile safely creates an empty file and any missing directory structure.
+// The directories will have directoryPerm filemode. The file will have filePerm filemode.
+func CreateFile(filename string, directoryPerm, filePerm fs.FileMode) error {
+	var err error
+
+	// Make sure the directory exists, using the same perms that awscli uses (0600)
+	dir := filepath.Dir(filename)
+	err = os.MkdirAll(dir, directoryPerm)
+	if err != nil {
+		return err
+	}
+
+	err = createEmptyFile(filename, filePerm)
+	return err
+}
+
 // WriteError writes a status code and plaintext error to the provided http.ResponseWriter.
 // The error is written as plaintext so AWS SDKs will display it inline with an error message.
 func WriteError(w http.ResponseWriter, message string, status int) {