Use credential or config in ~/.aws/ by default

Author: Xu Ao

db/dbsession.go

@@ -12,14 +12,28 @@ var DynamoDB *dynamodb.DynamoDB
 
 // GetDynamoSession returns a new dynamodb session
 func GetDynamoSession(accessKeyID, secretAccessKey, region string) *dynamodb.DynamoDB {
-	token := ""
-	creds := credentials.NewStaticCredentials(accessKeyID, secretAccessKey, token)
-	_, err := creds.Get()
-	if err != nil {
-		panic(err)
+	session_config := session.Options{}
+	if accessKeyID != "" || secretAccessKey != "" {
+		token := ""
+		creds := credentials.NewStaticCredentials(accessKeyID, secretAccessKey, token)
+		_, err := creds.Get()
+		if err != nil {
+			panic(err)
+		}
+		session_config.Config.Credentials = creds
 	}
 
-	DynamoDB = dynamodb.New(session.New(&aws.Config{Region: &region}))
+	if region != "" {
+		session_config.Config.Region = &region
+	} else {
+		session_config.SharedConfigState = session.SharedConfigEnable
+	}
+
+	session, err := session.NewSessionWithOptions(session_config)
+	if err != nil{
+		panic(err)
+	}
+	DynamoDB = dynamodb.New(session)
 	return DynamoDB
 }
 

main.go

@@ -213,12 +213,8 @@ func main() {
 			},
 		},
 		Action: func(c *cli.Context) error {
-			if region == "" {
-				fmt.Println("Must provide aws config region")
-			} else if accessKeyID == "" {
-				fmt.Println("Must provide aws config access key id")
-			} else if secretAccessKey == "" {
-				fmt.Println("Must provide aws config secret access key")
+			if !((accessKeyID == "" && secretAccessKey == "") || (accessKeyID != "" && secretAccessKey != "")){
+				fmt.Println("Must provide access key id and secret access key at the same time")
 			} else {
 				db.GetDynamoSession(accessKeyID, secretAccessKey, region)
 				runPrompt(tablePrefix)