fix(s3): pass all config to sub-tasks

kestra-io · Oct 24, 2024 · c996e7e · c996e7e
1 parent 41b7e27
commit c996e7e
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 0 deletions.
diff --git a/src/main/java/io/kestra/plugin/aws/s3/Copy.java b/src/main/java/io/kestra/plugin/aws/s3/Copy.java
@@ -90,6 +90,12 @@ public Output run(RunContext runContext) throws Exception {
                     .endpointOverride(this.endpointOverride)
                     .accessKeyId(this.accessKeyId)
                     .secretKeyId(this.secretKeyId)
+                    .sessionToken(this.sessionToken)
+                    .stsRoleSessionName(this.stsRoleSessionName)
+                    .stsRoleExternalId(this.stsRoleExternalId)
+                    .stsRoleSessionDuration(this.stsRoleSessionDuration)
+                    .stsRoleArn(this.stsRoleArn)
+                    .stsEndpointOverride(this.stsEndpointOverride)
                     .bucket(request.sourceBucket())
                     .key(request.sourceKey())
                     .build()

diff --git a/src/main/java/io/kestra/plugin/aws/s3/Downloads.java b/src/main/java/io/kestra/plugin/aws/s3/Downloads.java
@@ -90,6 +90,7 @@ public Output run(RunContext runContext) throws Exception {
             .endpointOverride(this.endpointOverride)
             .accessKeyId(this.accessKeyId)
             .secretKeyId(this.secretKeyId)
+            .sessionToken(this.sessionToken)
             .requestPayer(this.requestPayer)
             .bucket(this.bucket)
             .prefix(this.prefix)

diff --git a/src/main/java/io/kestra/plugin/aws/s3/Trigger.java b/src/main/java/io/kestra/plugin/aws/s3/Trigger.java
@@ -161,6 +161,7 @@ public Optional<Execution> evaluate(ConditionContext conditionContext, TriggerCo
             .endpointOverride(this.endpointOverride)
             .accessKeyId(this.accessKeyId)
             .secretKeyId(this.secretKeyId)
+            .sessionToken(this.sessionToken)
             .requestPayer(this.requestPayer)
             .bucket(this.bucket)
             .prefix(this.prefix)
@@ -194,6 +195,12 @@ public Optional<Execution> evaluate(ConditionContext conditionContext, TriggerCo
                     .endpointOverride(this.endpointOverride)
                     .accessKeyId(this.accessKeyId)
                     .secretKeyId(this.secretKeyId)
+                    .sessionToken(this.sessionToken)
+                    .stsRoleArn(this.stsRoleArn)
+                    .stsRoleSessionName(this.stsRoleSessionName)
+                    .stsRoleExternalId(this.stsRoleExternalId)
+                    .stsRoleSessionDuration(this.stsRoleSessionDuration)
+                    .stsEndpointOverride(this.stsEndpointOverride)
                     .requestPayer(this.requestPayer)
                     .bucket(this.bucket)
                     .key(object.getKey())