bustlelabs/shep

import inquirer from 'inquirer'
import reporter from '../../util/reporter'
import configSet from '../../config-set'
import * as load from '../../util/load'
import merge from 'lodash.merge'

export const command = 'set <vars...>'
export const desc = 'Set environment variables for alias on AWS'
export function builder (yargs) {
  return yargs
  .pkgConf('shep', process.cwd())
  .boolean('quiet')
  .alias('quiet', 'q')
  .describe('quiet', 'Don\'t log anything')
  .example('shep config set --env beta FOO=bar', 'Set environment variable FOO with value BAR for alias beta')
}

export async function handler (opts) {
  const logger = opts.quiet ? () => {} : reporter()
  let envVars = {}
  opts.vars.forEach(function (varPair) {
    const [key, value] = varPair.match(/(.*?)=(.*)/).slice(1)
    envVars[key] = value
  })
  opts.vars = envVars
  const inputs = {}

  if (!opts.env) {
    const envs = await load.envs()
    const questions = [
      {
        name: 'env',
        message: 'Environment',
        type: 'list',
        choices: () => envs
      }
    ]

    merge(inputs, await inquirer.prompt(questions))
  }

  logger({ type: 'start', body: 'Set environment variables on functions in AWS' })
  try {
    const versions = await configSet(merge({}, inputs, opts))
    logger({ type: 'done' })
    versions.forEach(({ FunctionName, Identifier }) => logger(`Updated ${FunctionName} to version ${Identifier.Version}`))
  } catch (e) {
    logger({ type: 'fail' })
    throw e
  }
}