Fix task and execution role ARN assignment in index.js

aws-actions · Apr 24, 2024 · 4051026 · 4051026
1 parent bcab98a
commit 4051026
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/index.js b/index.js
@@ -40,8 +40,7 @@ async function run() {
       throw new Error('Invalid task definition: Could not find container definition with matching name');
     }
     containerDef.image = imageURI;
-    containerDef.executionRoleArn = taskRoleArn;
-    containerDef.taskRoleArn = executionRoleArn;
+
 
     if (command) {
       containerDef.command = command.split(' ')
@@ -126,6 +125,14 @@ async function run() {
       })
     }
 
+    if (taskRoleArn) {
+      taskDefContents.taskRoleArn = taskRoleArn;
+    }
+
+    if (executionRoleArn) {
+      taskDefContents.executionRoleArn = executionRoleArn;
+    }
+
     // Write out a new task definition file
     var updatedTaskDefFile = tmp.fileSync({
       tmpdir: process.env.RUNNER_TEMP,