Adding support for profiles in credential files.

Author: morfien101

awsSession/awsSession.go

@@ -1,16 +1,47 @@
 package awsSession
 
 import (
+	"fmt"
 	"os"
 
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/credentials"
 	"github.com/aws/aws-sdk-go/aws/session"
 )
 
+var (
+	credsPath = ".aws/credentials"
+)
+
 // New will create and return an AWS Session
 func New() (*session.Session, error) {
 	return session.NewSession()
 }
 
+// NewWithOptions will create and return an AWS Session with supplied options
+func NewWithOptions(profile, credsFile string) (*session.Session, error) {
+
+	filePath := credsFile
+	if filePath == "" {
+		home, ok := os.LookupEnv("HOME")
+		if !ok {
+			return nil, fmt.Errorf("Can't find home environment variable. Looking for credentials file")
+		}
+		filePath = home + "/" + credsPath
+		_, err := os.Stat(filePath)
+		if err != nil {
+			return nil, fmt.Errorf("Failed to lookup credentials file. Error: %s", err)
+		}
+	}
+
+	creds := credentials.NewSharedCredentials(filePath, profile)
+	config := aws.NewConfig().WithCredentials(creds)
+	return session.NewSessionWithOptions(session.Options{
+		Config:            *config,
+		SharedConfigState: session.SharedConfigEnable,
+	})
+}
+
 // SetRegion publish the supplied region if there is one given
 func SetRegion(region string) error {
 	// Set what was passed in

main.go

@@ -7,6 +7,7 @@ import (
 	"fmt"
 	"log"
 
+	"github.com/aws/aws-sdk-go/aws/session"
 	"github.com/morfien101/go-parameterStoreReader/awsSession"
 	parameterstore "github.com/morfien101/go-parameterStoreReader/parameterStore"
 )
@@ -24,6 +25,8 @@ var (
 	flagDecrypt    = flag.Bool("decrypt", false, "Request decrypted keys")
 	flagAccessKey  = flag.String("access-key", "", "Access key for AWS API")
 	flagSecretKey  = flag.String("secret-key", "", "Secret key for AWS API")
+	flagProfile    = flag.String("profile", "", "AWS Profile to use")
+	flagCredsFile  = flag.String("config-file", "", "AWS Config file override, only valid with -profile")
 	flagRegion     = flag.String("region", "", "Region for AWS API")
 	flagHelp       = flag.Bool("h", false, "Help menu")
 	flagVersion    = flag.Bool("v", false, "Show Version")
@@ -56,7 +59,13 @@ func main() {
 		log.Fatal("Failed to set environment variable AWS_REGION for access to AWS")
 	}
 
-	session, err := awsSession.New()
+	var session *session.Session
+	var err error
+	if *flagProfile == "" {
+		session, err = awsSession.New()
+	} else {
+		session, err = awsSession.NewWithOptions(*flagProfile, *flagCredsFile)
+	}
 	if err != nil {
 		log.Fatalf("Failed to create AWS Session. Error: %s", err)
 	}