bustlelabs/shep

import inquirer from 'inquirer'
import merge from 'lodash.merge'
import logs from '../logs'
import * as load from '../util/load'

export const command = 'logs [name]'
export const desc = 'Streams logs from the specified version of a function'
export function builder (yargs) {
  return yargs
  .pkgConf('shep', process.cwd())
  .describe('env', 'Specifies which environment to use. If not provided an interactive menu will display the options.')
  .describe('name', 'Name of function to use')
  .describe('time', 'Time in seconds that logs should be streamed')
  .default('time', Infinity)
  .example('shep logs', 'Launch an interactive CLI')
  .example('shep logs --env production foo', 'Shows logs for the `foo` function in the production environment')
}

export async function handler (opts) {
  const fns = await load.funcs()
  const questions = []

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

  questions.push({
    name: 'name',
    message: 'Function',
    type: 'list',
    choices: () => fns
  })

  // Convert seconds to milliseconds
  opts.time = opts.time * 1000
  const inputs = await inquirer.prompt(questions.filter((q) => !opts[q.name]))
  return logs(merge({ logger: console.log }, inputs, opts))
}