93 lines
2.8 KiB
Go
93 lines
2.8 KiB
Go
|
package cli
|
||
|
|
|
||
|
|
import (
|
||
|
|
"flag"
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"github.com/aws/aws-sdk-go/aws"
|
||
|
|
"github.com/aws/aws-sdk-go/aws/session"
|
||
|
|
)
|
||
|
|
|
||
|
|
// ParseAndCreateSession returns a Session object, emulating AWS CLI
|
||
|
|
// configuration.
|
||
|
|
//
|
||
|
|
// This is a convenience method to create CLI tools that behave similar to AWS
|
||
|
|
// CLI tools in where they pick up their configuration and credential from.
|
||
|
|
//
|
||
|
|
// It'll add and parse two command line flags: `--profile` and `--endpoint-url`.
|
||
|
|
//
|
||
|
|
// Beyond that, the method will pick up the S3 configuration and credentials
|
||
|
|
// from the same standard places where aws-cli looks for them:
|
||
|
|
//
|
||
|
|
// https://docs.aws.amazon.com/sdk-for-go/v1/developer-guide/configuring-sdk.html
|
||
|
|
//
|
||
|
|
// As a tldr, the easiest way to use this might be to add a new AWS profile:
|
||
|
|
//
|
||
|
|
// # ~/.aws/config
|
||
|
|
// [profile wasabi-test-compliance]
|
||
|
|
// region = eu-central-2
|
||
|
|
//
|
||
|
|
// # ~/.aws/credentials
|
||
|
|
// [wasabi-test-compliance]
|
||
|
|
// aws_access_key_id = test
|
||
|
|
// aws_secret_access_key = test
|
||
|
|
//
|
||
|
|
// And `export AWS_PROFILE=wasabi-test-compliance`, or provide it to the
|
||
|
|
// commands via the `--profile` flag.
|
||
|
|
//
|
||
|
|
// Alternatively, if you don't wish to use AWS profiles, then you can provide
|
||
|
|
// these values using the standard AWS environment variables.
|
||
|
|
//
|
||
|
|
// export AWS_REGION=eu-central-2
|
||
|
|
// export AWS_ACCESS_KEY_ID=test
|
||
|
|
// export AWS_SECRET_ACCESS_KEY=test
|
||
|
|
//
|
||
|
|
// > Tip: If your shell is configured to do so, you can add a leading space `
|
||
|
|
//
|
||
|
|
// export AWS_SECRET_....` to avoid preserving these secrets in your shell
|
||
|
|
// history.
|
||
|
|
//
|
||
|
|
// The endpoint to connect to can be either passed as an (optional) method
|
||
|
|
// parameter, or can be specified at runtime using the `--endpoint-url` flag.
|
||
|
|
//
|
||
|
|
// S3ForcePathStyle can be set to true when connecting to locally running MinIO
|
||
|
|
// instances where each bucket will not have a DNS.
|
||
|
|
func ParseAndCreateSession(endpointURL string, S3ForcePathStyle bool) (*session.Session, error) {
|
||
|
|
logLevel := aws.LogDebugWithHTTPBody
|
||
|
|
cliProfile := flag.String("profile", "AWS_PROFILE",
|
||
|
|
"The profile to use from the S3 config file")
|
||
|
|
cliEndpointURL := flag.String("endpoint-url", "",
|
||
|
|
"The root URL of the S3 compatible API (excluding the bucket)")
|
||
|
|
flag.Parse()
|
||
|
|
|
||
|
|
profile := *cliProfile
|
||
|
|
if profile == "" {
|
||
|
|
profile = os.Getenv("AWS_PROFILE")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Override the passed in value with the CLI always.
|
||
|
|
if *cliEndpointURL != "" {
|
||
|
|
endpointURL = *cliEndpointURL
|
||
|
|
}
|
||
|
|
|
||
|
|
fmt.Printf("Using profile %s, endpoint %s\n", profile, endpointURL)
|
||
|
|
|
||
|
|
sess, err := session.NewSessionWithOptions(session.Options{
|
||
|
|
Profile: profile,
|
||
|
|
// Needed to read region from .aws/profile
|
||
|
|
SharedConfigState: session.SharedConfigEnable,
|
||
|
|
Config: aws.Config{
|
||
|
|
Endpoint: aws.String(endpointURL),
|
||
|
|
S3ForcePathStyle: aws.Bool(S3ForcePathStyle),
|
||
|
|
LogLevel: &logLevel,
|
||
|
|
},
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
fmt.Printf("NewSessionWithOptions error: %s\n", err)
|
||
|
|
return sess, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return sess, err
|
||
|
|
}
|