update.ts

import { Argv } from 'yargs';

import { monitoring, aws } from '../../lib';
import { setVerbose } from '../../lib/logger';

export const command = 'update [options]';
export const desc = 'Update monitoring config';
export const builder = (yargs: Argv<{}>): Argv<{}> => {
  return yargs.options({
    c: {
      alias: 'config',
      describe: 'Config file to update',
      type: 'string',
      default: 'config.yml',
    },
    p: {
      alias: 'profile',
      describe: 'AWS profile used connect to AWS environment',
      type: 'string',
    },
    r: {
      alias: 'region',
      describe: 'Target region',
      type: 'string',
    },
    i: {
      alias: 'include',
      describe: 'List of included names',
      type: 'array',
    },
    e: {
      alias: 'exclude',
      describe: 'List of excluded names',
      type: 'array',
    },
    s: {
      alias: 'service',
      describe: 'List of services',
      type: 'array',
      choices: ['lambda', 'dynamodb', 'ecs', 'apigateway', 'cloudfront', 'rds', 'eks', 'loggroup', 'appsync', 'sqs'],
    },
    d: {
      alias: 'dry',
      default: false,
      type: 'boolean',
    },
    t: {
      alias: 'stage',
      describe: 'Stage of the deployment',
      type: 'string',
      default: 'dev',
    },
    ep: {
      alias: 'endpoints',
      default: [],
      describe:
        'Add endpoints directly or AWS SSM params name to retrieve endpoints from SSM, e.g. ssm:my-endpoint-${stage}, stage is always added to the end',
      type: 'array',
    },
    v: {
      alias: 'verbose',
      describe: 'Set verbose logging',
      type: 'boolean',
      default: false,
    },
    interactive: {
      default: false,
      type: 'boolean',
    },
    sso: {
      default: false,
      describe: 'Use an AWS profile with SSO credentials',
      type: 'boolean',
    },
  });
};

export const handler = async (args: monitoring.Args): Promise<void> => {
  setVerbose(args.verbose);

  const config = new monitoring.ConfigGenerator(args);
  await config.loadFromFile(args.config);
  const combinedArgs = config.combineCLIArgs(args);
  config.updateCLIArgs(combinedArgs);

  await aws.setAWSCredentials(combinedArgs.profile, combinedArgs.region);

  await config.setPagerDutyEndpoint(combinedArgs);

  const awsConfig = await monitoring.getAllFromAWS(combinedArgs);

  const newConfig = new monitoring.ConfigGenerator(combinedArgs);

  newConfig.addAllLocal(awsConfig);

  config.combine(newConfig);

  if (args.dry) {
    config.diff(newConfig);
    return;
  }

  await config.write(args.config);
};