namespace :agents do

desc 'Load agents from YAML with deleting old data'
task load: :environment do
  file_path = "#{Rails.root}/tmp/import/agents.yml"
  if File.exist?(file_path)
    puts 'Deleting old agents...'
    Agent.destroy_all
    puts 'Done. Importing...'
    File.open(file_path, 'r') do |file|
      YAML.load(file).each do |id, data|
        entity = Agent.new id: id
        entity.assign_attributes data
        entity.save!
        print "\r#{id}    "
      end
      puts
    end
    Agent.connection.execute "select setval('agents_id_seq', (select max(id) from agents));"
    puts "Done. We have #{Agent.count} agents now"
  else
    puts "Cannot find file #{file_path}"
  end
end

desc 'Dump agents to YAML'
task dump: :environment do
  file_path = "#{Rails.root}/tmp/export/agents.yml"
  ignored = %w(id)
  File.open file_path, 'w' do |file|
    Agent.order('id asc').each do |entity|
      file.puts "#{entity.id}:"
      entity.attributes.reject { |a, v| ignored.include?(a) || v.nil? }.each do |attribute, value|
        file.puts "  #{attribute}: #{value.inspect}"
      end
    end
  end
end

end