index.js

const path = require('path');
const core = require('@actions/core');
const tmp = require('tmp');
const fs = require('fs');

async function run() {
  try {
    // Get inputs
    const taskDefinitionFile = core.getInput('task-definition', { required: true });
    const containerName = core.getInput('container-name', { required: true });
    const imageURI = core.getInput('image', { required: true });
    const taskFamily = core.getInput('task-family', { required: true });
    const entryPoint = core.getInput('entry-point', { required: false }) || '';
    const logGroup = core.getInput('log-group', { required: false }) || '';
    const memory = core.getInput('memory', { required: false }) || '';
    const cpu = core.getInput('cpu', { required: false }) || '';
    const storageSize = core.getInput('ephemeral-storage-size-in-gib', { required: false }) || '';
    const environmentVariablesFile = core.getInput('environment-file', { required: false }) || '';
    const secretsFile = core.getInput('secrets-file', { required: false }) || '';

    // Parse the task definition
    const taskDefPath = path.isAbsolute(taskDefinitionFile) ?
      taskDefinitionFile :
      path.join(process.env.GITHUB_WORKSPACE, taskDefinitionFile);
    if (!fs.existsSync(taskDefPath)) {
      throw new Error(`Task definition file does not exist: ${taskDefinitionFile}`);
    }
    const taskDefContents = require(taskDefPath);

    taskDefContents.family = taskFamily;

    // Insert the image URI
    if (!Array.isArray(taskDefContents.containerDefinitions)) {
      throw new Error('Invalid task definition format: containerDefinitions section is not present or is not an array');
    }
    const containerDef = taskDefContents.containerDefinitions.find(function(element) {
      return element.name == containerName;
    });
    if (!containerDef) {
      throw new Error('Invalid task definition: Could not find container definition with matching name');
    }
    containerDef.image = imageURI;

    if (entryPoint) {
      containerDef.entryPoint = entryPoint.split(' ');
    }
    if (logGroup) {
      containerDef.logConfiguration.options['awslogs-group'] = logGroup;
    } else {
      containerDef.logConfiguration.options['awslogs-group'] = `/ecs/${taskFamily}`;
    }

    if (environmentVariablesFile) {
      const envFilePath = path.isAbsolute(environmentVariablesFile) ?
        environmentVariablesFile :
        path.join(process.env.GITHUB_WORKSPACE, environmentVariablesFile);
      containerDef.environment = require(envFilePath);
    }
    if (secretsFile) {
      const secretsFilePath = path.isAbsolute(secretsFile) ?
        secretsFile :
        path.join(process.env.GITHUB_WORKSPACE, secretsFile);
      containerDef.secrets = require(secretsFilePath);
    }

    if (memory) {
      taskDefContents.memory = memory;
    }
    if (cpu) {
      taskDefContents.cpu = cpu;
    }
    if (storageSize) {
      taskDefContents.ephemeralStorage = {
        sizeInGiB: parseInt(storageSize, 10)
      };
    }

    // Write out a new task definition file
    var updatedTaskDefFile = tmp.fileSync({
      tmpdir: process.env.RUNNER_TEMP,
      prefix: 'task-definition-',
      postfix: '.json',
      keep: true,
      discardDescriptor: true
    });
    const newTaskDefContents = JSON.stringify(taskDefContents, null, 2);
    fs.writeFileSync(updatedTaskDefFile.name, newTaskDefContents);
    core.setOutput('task-definition', updatedTaskDefFile.name);
  }
  catch (error) {
    core.setFailed(error.message);
  }
}

module.exports = run;

/* istanbul ignore next */
if (require.main === module) {
    run();
}