Skip to content

Commit 12f31cc

Browse files
committed
Review fixes
1 parent b011f32 commit 12f31cc

4 files changed

Lines changed: 12 additions & 39 deletions

File tree

cmd/kms_priv_key_importer/main.go

Lines changed: 2 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,6 @@ import (
1414
"strings"
1515

1616
"github.com/aws/aws-sdk-go-v2/aws"
17-
awsconfig "github.com/aws/aws-sdk-go-v2/config"
18-
"github.com/aws/aws-sdk-go-v2/credentials"
1917
awskms "github.com/aws/aws-sdk-go-v2/service/kms"
2018
"github.com/aws/aws-sdk-go-v2/service/kms/types"
2119
"github.com/aws/aws-sdk-go-v2/service/secretsmanager"
@@ -149,7 +147,7 @@ func main() {
149147
}
150148

151149
if issuerKMSETHProviderToUse == config.AWSSM {
152-
cfg, err := LoadAWSConfig(ctx)
150+
cfg, err := kms.LoadAWSConfig(ctx)
153151

154152
if err != nil {
155153
log.Error(ctx, "error loading AWSSM config", "err", err)
@@ -233,33 +231,10 @@ func validate(issuerKMSETHProviderToUse string, fPrivateKey *string, ctx context
233231
return nil
234232
}
235233

236-
func LoadAWSConfig(ctx context.Context) (aws.Config, error) {
237-
// Backward-compatible behaviour for AWS SDK configuration
238-
// env variables (DEPRECATED)
239-
// "ISSUER_KMS_AWS_ACCESS_KEY"
240-
// "ISSUER_KMS_AWS_SECRET_KEY"
241-
// "ISSUER_KMS_AWS_REGION"
242-
accessKey := strings.TrimSpace(os.Getenv("ISSUER_KMS_AWS_ACCESS_KEY"))
243-
secretKey := strings.TrimSpace(os.Getenv("ISSUER_KMS_AWS_SECRET_KEY"))
244-
region := strings.TrimSpace(os.Getenv("ISSUER_KMS_AWS_REGION"))
245-
246-
if accessKey != "" && secretKey != "" && region != "" {
247-
return awsconfig.LoadDefaultConfig(
248-
ctx,
249-
awsconfig.WithCredentialsProvider(
250-
credentials.NewStaticCredentialsProvider(accessKey, secretKey, ""),
251-
),
252-
awsconfig.WithRegion(region),
253-
)
254-
}
255-
256-
return awsconfig.LoadDefaultConfig(ctx)
257-
}
258-
259234
//
260235
//nolint:unused
261236
func createEmptyKey(ctx context.Context, privateKeyAlias string) (*string, error) {
262-
cfg, err := LoadAWSConfig(ctx)
237+
cfg, err := kms.LoadAWSConfig(ctx)
263238

264239
if err != nil {
265240
log.Error(ctx, "error loading AWSKMS config", "err", err)

internal/kms/aws.go

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -40,41 +40,39 @@ func AwsSecretsManager(ctx context.Context) (*secretsmanager.Client, error) {
4040
cfg, err := LoadAWSConfig(ctx)
4141

4242
if err != nil {
43-
return nil, fmt.Errorf("unable to load SDK config, %v", err)
43+
return nil, fmt.Errorf("unable to load SDK config, %w", err)
4444
}
4545

46-
var options = make([]func(*secretsmanager.Options), 1)
47-
4846
// LocalStack/OpenStack mode
4947
// https://docs.localstack.cloud/aws/integrations/aws-sdks/go/
5048
// Region is provided from AWS_REGION env variable
5149
url := strings.TrimSpace(os.Getenv("ISSUER_KMS_AWS_URL"))
50+
var options []func(*secretsmanager.Options)
5251
if url != "" {
53-
options[0] = func(o *secretsmanager.Options) {
52+
options = append(options, func(o *secretsmanager.Options) {
5453
o.BaseEndpoint = aws.String(url)
55-
}
54+
})
5655
}
57-
5856
return secretsmanager.NewFromConfig(cfg, options...), nil
5957
}
6058

6159
func AwsKms(ctx context.Context) (*kms.Client, error) {
6260
cfg, err := LoadAWSConfig(ctx)
6361

6462
if err != nil {
65-
return nil, fmt.Errorf("unable to load SDK config, %v", err)
63+
return nil, fmt.Errorf("unable to load SDK config, %w", err)
6664
}
6765

68-
options := make([]func(*kms.Options), 1)
66+
var options []func(*kms.Options)
6967

7068
// LocalStack/OpenStack mode
7169
// https://docs.localstack.cloud/aws/integrations/aws-sdks/go/
7270
// Region is provided from AWS_REGION env variable
7371
url := strings.TrimSpace(os.Getenv("ISSUER_KMS_AWS_URL"))
7472
if url != "" {
75-
options[0] = func(o *kms.Options) {
73+
options = append(options, func(o *kms.Options) {
7674
o.BaseEndpoint = aws.String(url)
77-
}
75+
})
7876
}
7977

8078
return kms.NewFromConfig(cfg, options...), nil

internal/kms/aws_kms_eth_key_provider.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ func NewAwsKMSEthKeyProvider(ctx context.Context, keyType KeyType, issuerETHTran
3636
client, err := AwsKms(ctx)
3737

3838
if err != nil {
39-
return nil, fmt.Errorf("failed to create AWS KMS client: %v", err)
39+
return nil, fmt.Errorf("failed to create AWS KMS client: %w", err)
4040
}
4141

4242
return &awsKmsEthKeyProvider{

internal/kms/aws_secret_storage_provider.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ func NewAwsSecretStorageProvider(ctx context.Context) (*awsSecretStorageProvider
3131
client, err := AwsSecretsManager(ctx)
3232

3333
if err != nil {
34-
return nil, fmt.Errorf("failed to create AWS Secrets Manager client: %v", err)
34+
return nil, fmt.Errorf("failed to create AWS Secrets Manager client: %w", err)
3535
}
3636

3737
return &awsSecretStorageProvider{

0 commit comments

Comments
 (0)