feat: Add support for task and execution role ARNs in action.yml and …

action.yml

@@ -28,6 +28,12 @@ inputs:
  command:
  description: 'The command used by ECS to start the container image'
  required: false
+ task-role-arn:
+ description: 'The ARN of the IAM role that the ECS container will assume'
+ required: false
+ execution-role-arn:
+ description: 'The ARN of the IAM role that the ECS task execution role will assume'
+ required: false
 outputs:
  task-definition:
  description: 'The path to the rendered task definition file'

index.js

@@ -16,6 +16,9 @@ async function run() {
  const logConfigurationOptions = core.getInput("log-configuration-options", { required: false });
  const dockerLabels = core.getInput('docker-labels', { required: false });
  const command = core.getInput('command', { required: false });
+ const taskRoleArn = core.getInput('task-role-arn', { required: false });
+ const executionRoleArn = core.getInput('execution-role-arn', { required: false });
+
 
  // Parse the task definition
  const taskDefPath = path.isAbsolute(taskDefinitionFile) ?
@@ -38,6 +41,7 @@ async function run() {
  }
  containerDef.image = imageURI;
 
+
  if (command) {
  containerDef.command = command.split(' ')
  }
@@ -121,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,