support sts assume role with s3 release sources

internal/component/s3_release_source.go

@@ -16,6 +16,7 @@ import (
 	"github.com/Masterminds/semver/v3"
 	"github.com/aws/aws-sdk-go/aws"
 	"github.com/aws/aws-sdk-go/aws/credentials"
+	"github.com/aws/aws-sdk-go/aws/credentials/stscreds"
 	"github.com/aws/aws-sdk-go/aws/session"
 	"github.com/aws/aws-sdk-go/service/s3"
 	"github.com/aws/aws-sdk-go/service/s3/s3manager"
@@ -72,12 +73,20 @@ func NewS3ReleaseSourceFromConfig(config cargo.ReleaseSourceConfig, logger *log.
 		Region:      aws.String(config.Region),
 		Credentials: credentials.NewStaticCredentials(config.AccessKeyId, config.SecretAccessKey, ""),
 	}
+
+	var assumedRoleAwsConfig aws.Config
+	if config.AwsRoleARN != "" {
+		stsSession := session.Must(session.NewSession(awsConfig))
+		roleCredentials := stscreds.NewCredentials(stsSession, config.AwsRoleARN)
+		assumedRoleAwsConfig.Credentials = roleCredentials
+	}
+
 	if config.Endpoint != "" { // for acceptance testing
 		awsConfig = awsConfig.WithEndpoint(config.Endpoint)
 		awsConfig = awsConfig.WithS3ForcePathStyle(true)
 	}
 
-	sess := session.Must(session.NewSession(awsConfig))
+	sess := session.Must(session.NewSession(awsConfig, &assumedRoleAwsConfig))
 	client := s3.New(sess)
 
 	return NewS3ReleaseSource(

pkg/cargo/files_test.go

@@ -53,6 +53,58 @@ func TestInterpolateAndParseKilnfile(t *testing.T) {
 	})
 }
 
+func TestInterpolateAndParseKilnfileWithRoleARN(t *testing.T) {
+	please := NewWithT(t)
+
+	const validKilnfileWithRoleARN = `---
+release_sources:
+  - type: s3
+    compiled: true
+    bucket: $( variable "bucket" )
+    region: $( variable "region" )
+    access_key_id: $( variable "access_key" )
+    secret_access_key: $( variable "secret_key" )
+    aws_role_arn: $( variable "role_arn" )
+    path_template: $( variable "path_template" )
+`
+
+	variables := map[string]interface{}{
+		"bucket":        "my-bucket",
+		"region":        "middle-earth",
+		"path_template": "not-used",
+
+		"access_key": "id",
+		"secret_key": "key",
+		"role_arn":   "role-arn",
+	}
+
+	kilnfile, err := cargo.InterpolateAndParseKilnfile(
+		strings.NewReader(validKilnfileWithRoleARN), variables,
+	)
+
+	please.Expect(err).NotTo(HaveOccurred())
+
+	please.Expect(kilnfile).To(Equal(cargo.Kilnfile{
+		ReleaseSources: []cargo.ReleaseSourceConfig{
+			{
+				Type:            "s3",
+				Bucket:          "my-bucket",
+				Region:          "middle-earth",
+				AccessKeyId:     "id",
+				AwsRoleARN:      "role-arn",
+				SecretAccessKey: "key",
+				PathTemplate:    "not-used",
+			},
+		},
+	}))
+
+	t.Run("reading fails", func(t *testing.T) {
+		r := iotest.ErrReader(errors.New("lemon"))
+		_, err := cargo.InterpolateAndParseKilnfile(r, make(map[string]any))
+		assert.Error(t, err)
+	})
+}
+
 func TestInterpolateAndParseKilnfile_input_is_not_valid_yaml(t *testing.T) {
 	please := NewWithT(t)
 
@@ -88,7 +140,7 @@ func TestInterpolateAndParseKilnfile_interpolation_variable_not_found(t *testing
 		strings.NewReader(validKilnfile), variables,
 	)
 
-	please.Expect(err).To(HaveOccurred())
+	please.Expect(err).To(MatchError(ContainSubstring(`could not find variable with key 'region'`)))
 }
 
 const validKilnfile = `---

pkg/cargo/kilnfile.go

@@ -109,6 +109,7 @@ type ReleaseSourceConfig struct {
 	Region          string `yaml:"region,omitempty"`
 	AccessKeyId     string `yaml:"access_key_id,omitempty"`
 	SecretAccessKey string `yaml:"secret_access_key,omitempty"`
+	AwsRoleARN      string `yaml:"aws_role_arn,omitempty"`
 	PathTemplate    string `yaml:"path_template,omitempty"`
 	Endpoint        string `yaml:"endpoint,omitempty"`
 	Org             string `yaml:"org,omitempty"`