'use strict'

let co = require('co')
let cli = require('heroku-cli-util')
let helpers = require('../lib/helpers')
let Dyno = require('../lib/dyno')

function * run (context, heroku) {
  context.args.unshift('rake')
  let opts = {
    heroku: heroku,
    app: context.app,
    command: helpers.buildCommand(context.args),
    size: context.flags.size,
    'exit-code': context.flags['exit-code'],
    env: context.flags.env,
    'no-tty': context.flags['no-tty'],
    attach: true
  }

  let dyno = new Dyno(opts)
  try {
    yield dyno.start()
  } catch (err) {
    if (err.exitCode) {
      cli.error(err)
      process.exit(err.exitCode)
    } else throw err
  }
}

module.exports = {
  topic: 'rake',
  hidden: true,
  variableArgs: true,
  needsAuth: true,
  needsApp: true,
  flags: [
    { name: 'size', char: 's', description: 'dyno size', hasValue: true },
    { name: 'exit-code', char: 'x', description: 'passthrough the exit code of the remote command' },
    { name: 'env', char: 'e', description: "environment variables to set (use ';' to split multiple vars)", hasValue: true },
    { name: 'no-tty', description: 'force the command to not run in a tty', hasValue: false }
  ],
  run: cli.command(co.wrap(run))
}