def failed; "[ \033[31mFAIL\033[0m ]"; end
def ok;     "[  \033[32mOK\033[0m  ]"; end

def run_as_root
  return true if ENV['USER'] == 'root'
  puts "#{failed} Please run this command as root or with sudo\n\n"
  exit -1
end

def runit_installed
  return true unless `which sv` && $?.to_i != 0
  puts "#{failed} Please install runit: \n\nsudo apt-get install runit\n\n"
  exit -1
end

def remove_upstart_config
  return true unless File.exists?('/etc/init/huginn.conf')
  puts "#{failed} Please stop huginn and remove the huginn upstart init scripts:\n\n"
  puts "sudo stop huginn"
  puts "sudo rm /etc/init/huginn*\n\n"
  exit -1
end

namespace :production do
  task :check do |t|
    remove_upstart_config
    runit_installed
    puts "#{ok} Everything is fine" if t.application.top_level_tasks.include? 'production:check'
  end

  task :stop => :check do
    puts "Stopping huginn ..."
    run_sv('stop')
  end

  task :start => :check do
    puts "Starting huginn ..."
    run_sv('start')
  end

  task :force_stop => :check do
    puts "Force stopping huginn ..."
    run_sv('force-stop')
  end

  task :status => :check do
    run_sv('status')
  end

  task :restart => :check do
    puts "Restarting huginn ..."
    run_sv('restart')
  end

  task :export => :check do
    run_as_root
    Rake::Task['production:stop'].execute
    puts "Exporting new services ..."
    run('rm -rf /etc/service/huginn*')
    run('foreman export runit -a huginn -l /home/huginn/huginn/log /etc/service')
    services = Dir.glob('/etc/service/huginn*')
    while services.length > 0
      services.each do |p|
        supervise = File.join(p, 'supervise')
        next if !Dir.exists?(supervise)
        run("chown -R huginn:huginn #{p}")
        services.delete(p)
      end
      sleep 0.1
    end
  end
end

def run_sv(command)
  Dir.glob('/etc/service/huginn*').each do |p|
    with_retries do
      run("sv #{command} #{File.basename(p)}")
    end
  end
end

def run(cmd, verbose=false)
  output = `#{cmd}`
  if $?.to_i != 0
    raise "'#{cmd}' exited with a non-zero return value: #{output}"
  end
  puts output if verbose && output.strip != ''
  output
end

def with_retries(&block)
  tries ||= 5
  output = block.call
rescue StandardError => e
  retry unless (tries -= 1).zero?
  raise e
else
  puts output
end