require 'pp'

namespace :role do
  desc 'Set security group for a role'
  task :set_sg, [:role, :environment, :security_group] do |_t, args|
    abort('Missing parameters') if args[:role].nil? || args[:environment].nil? || args[:role].empty? || args[:environment].empty? || args[:security_group].nil? || args[:security_group].empty?
    role = DH.get_role_data(args[:role], args[:environment])
    role['sg'].merge!(args[:environment] => args[:security_group])
    DH.save_role_data(args[:role], role)
  end

  desc 'Set ami for role'
  task :set_amis, [:role, :region, :ami_id] do |_t, args|
    abort('Missing parameters') if args[:role].nil? || args[:region].nil? || args[:role].empty? || args[:region].empty? || args[:ami_id].nil? || args[:ami_id].empty?
    role = DH.get_role_data(args[:role])
    role['amis'].merge!(args[:region], args[:ami_id])
    DH.save_role_data(args[:role], role)
  end

  desc "Set runlist for role; Runlist must be a comma separated string of recipes and roles. e.g. 'recipe[recipe],role[otherrole]'"
  task :set_chef_runlist, [:role, :chef_runlist] do |_t, args|
    abort('Missing parameters') if args[:role].nil? || args[:chef_runlist].nil? || args[:role].empty? || args[:chef_runlist].empty?
    role = DH.get_role_data(args[:role])
    rl = args[:chef_runlist].strip.split(',').map(&:strip).to_json
    role['chef_runlist'] = rl
    DH.save_role_data(args[:role], role)
    puts "Added runlist for role #{role}"
  end

  desc 'Remove a role, if there are no instances running'
  task :remove, [:role] do |_t, args|
    abort('Missing parameters') if args[:role].nil? || args[:role].empty?
    res = DH.remove_role(args[:role])
    if res == true
      puts "Removed role #{args[:role]}"
    else
      puts "Cannot remove role: #{res.inspect}"
    end
  end
end

desc 'List all roles (environment is optional but needed to display apps)'
task :role, [:environment] do |_t, args|
  env = args[:environment]
  puts 'Roles:'
  DH.roles(env).each do |role, info|
    puts "* #{role}:"
    puts "  - apps: #{info['apps'].join(', ')}"
    puts "  - sg: #{info['sg'].values.join(', ')}"
    puts "  - amis: #{info['amis'].map { |k, v| "#{v} (#{k})" }.join(', ')}"
    puts "  - runlist: #{info['chef_runlist'].to_a.join(', ')}"
  end
end