'use strict'

let cli = require('heroku-cli-util')

function buildCommand (args) {
  if (args.length === 1) {
    // do not add quotes around arguments if there is only one argument
    // `heroku run "rake test"` should work like `heroku run rake test`
    return args[0]
  }
  let cmd = ''
  for (let arg of args) {
    if (arg.indexOf(' ') !== -1 || arg.indexOf('"') !== -1) {
      arg = '"' + arg.replace(/"/g, '\\"') + '"'
    }
    cmd = cmd + ' ' + arg
  }
  return cmd.trim()
}

function buildEnvFromFlag (flag) {
  let env = {}
  for (let v of flag.split(';')) {
    let m = v.match(/^\s*([\w.-]+)\s*=\s*(.*)?\s*$/)
    if (m) env[m[1]] = m[2]
    else cli.warn(`env flag ${v} appears invalid. Avoid using ';' in values.`)
  }
  return env
}

module.exports = {
  buildCommand,
  buildEnvFromFlag
}