@chaseconey/let-me-in/src/index.js

A CLI tool to easily SSH into an ECS container using SSM Session Manager.

#!/usr/bin/env node

import yargs from "yargs/yargs";
import {
  promptClusters,
  promptServices,
  promptTasks,
  promptContainers,
} from "./prompts.js";
import { checkPrerequisites } from "./prerequisites.js";
import { spawn } from "child_process";
import chalk from "chalk";
import { readFile } from "fs/promises";
import updateNotifier from "update-notifier";

export const getJsonFile = async (path) => {
  const contents = await readFile(new URL(path, import.meta.url));

  return JSON.parse(contents.toString());
};
const packageJson = await getJsonFile("../package.json");
updateNotifier({ pkg: packageJson }).notify();

const argv = yargs(process.argv.slice(2))
  .options({
    r: {
      type: "string",
      default: "us-east-1",
      alias: "region",
      describe: "AWS Region to search",
    },
    c: {
      type: "string",
      alias: "cluster",
      describe: "ECS Cluster Name or ARN, used to short-circuit lookup",
    },
    s: {
      type: "string",
      alias: "service",
      describe: "ECS Service Name or ARN, used to short-circuit lookup",
    },
    p: {
      type: "string",
      alias: "profile",
      describe: "AWS Profile to use",
    },
    print: {
      type: "boolean",
      default: false,
      describe: "Print command instead of executing",
    },
    container: {
      type: "string",
      describe:
        "Container Name to exec into; only required if multiple containers are present in the task",
    },
    command: {
      type: "string",
      describe: "Command to execute in the container",
      default: "/bin/sh",
    },
  })
  .version("version", chalk.yellow("Show version number"), packageJson.version)
  .help()
  .alias("h", "help")
  .showHelpOnFail(true)
  .parse();

// Set region
process.env.AWS_REGION = argv.region;

// Set AWS profile
if (argv.profile) {
  process.env.AWS_PROFILE = argv.profile;
}

// Check prerequisites (AWS CLI and Session Manager plugin)
if (!argv.print) {
  const prerequisitesPassed = await checkPrerequisites();
  if (!prerequisitesPassed) {
    process.exit(1);
  }
}

let cmd = "";
try {
  const cluster = argv.cluster ?? (await promptClusters());

  // Get all services for selected cluster
  const service = argv.service ?? (await promptServices(cluster));

  // Get all tasks for selected service
  const task = await promptTasks(cluster, service);

  // Get all containers for selected task
  const container = argv.container ?? (await promptContainers(task));

  // Get into task with SSM
  // Example: aws ecs execute-command --cluster prod --task $(codecov-prod-task-id) --container ${2:-api} --interactive --command ${1:-"/bin/bash"}
  cmd = `aws ecs execute-command --cluster ${cluster} --task ${task.taskArn} --container ${container} --interactive --command ${argv.command}`;
} catch (e) {
  console.error(chalk.red(e.message));
  process.exit(1);
}

if (argv.print) {
  console.log(cmd);
  process.exit(0);
}

const shell = spawn(cmd, [], { stdio: "inherit", shell: true });

shell.on("close", (code) => {
  console.log("[shell] terminated :", code);
  process.exit(code);
});