class Chorizo

# File lib/chorizo.rb, line 6
def initialize
  @host_names = %w(cloud66 heroku)

  config_file = './config/chorizo.yml'
  @config = YAML.load_file(config_file) if File.exist?(config_file)
end

# File lib/chorizo.rb, line 25
def build_output(env, host)
  configs = load_config
  output = configs[:base]

  # load environment config
  if configs[:envs][env]
    output.merge! configs[:envs][env]
  else
    STDERR.puts "WARNING: #{env} specific configuration not found".red
  end

  # load host config
  if configs[:hosts][host]
    hc = configs[:hosts][host]
    # load embedded env config in host config, if present
    hc_envs, hc_base = hc.partition { |k,v| v.is_a?(Hash) }
    hc_env = hc_envs.to_h[env]
    hc_output = hc_base.to_h
    hc_output.merge! hc_env if hc_env

    output.merge! hc_output
  else
    STDERR.puts "WARNING: #{host} specific configuration not found".red
  end

  output
end

# File lib/chorizo.rb, line 87
def cloud66(env)
  output = build_output(env, 'cloud66')
  output.each do |k,v|
    value = decrypt_value(v)
    puts "#{k.upcase}=#{value}"
  end
end

# File lib/chorizo.rb, line 105
def decrypt_value(value)
  if value =~ /^ENC\[/
    Open3.popen2("eyaml", "decrypt", "-s", value) do |i, o, t|
      o.read.chomp
    end
  else
    value
  end
end

# File lib/chorizo.rb, line 95
def heroku(env, app)
  output = build_output(env, 'heroku')
  cmd_output = output.map do |k,v|
    value = decrypt_value(v)
    escaped_value = "#{value}".shellescape
    "#{k.upcase}=#{escaped_value}"
  end.join(' ')
  system "heroku config:set #{cmd_output} -a #{app}"
end

# File lib/chorizo.rb, line 13
def load_config
  yml = YAML.load_file('./config/application.yml')
  hashes, base = yml.partition { |k,v| v.is_a?(Hash) }
  hashes = hashes.to_h
  base = base.to_h
  hosts = {}
  @host_names.each do |host|
    hosts[host] = hashes.delete(host) if hashes[host]
  end
  { base: base, hosts: hosts, envs: hashes }
end

# File lib/chorizo.rb, line 53
def run(env, opts, target: nil, app: nil)
  if @config
    if target && target != @config[env]['target']
      STDERR.puts 'WARNING: target differs from configuration'.red
    end
    target ||= @config[env]['target']

    if app && app != @config[env]['app']
      STDERR.puts 'WARNING: app differs from configuration'.red
    end
    app ||= @config[env]['app']
  end

  unless target && @host_names.include?(target)
    error = "please specify a valid target [#{@host_names.join(', ')}]"
    STDERR.puts(error.red)
    puts opts
    exit 1
  end

  case target
  when 'cloud66'
    cloud66(env)
  when 'heroku'
    unless app
      STDERR.puts 'please specify an app for heroku'.red
      puts opts
      exit 1
    end

    heroku(env, app)
  end
end