#!/usr/bin/env ruby
require 'optparse'
require 'bluepill'

# Check for root
unless ::Process.euid == 0
  $stderr.puts "You must run bluepill as root."
  exit(3)
end


# Default options
options = {
  :log_file => "/var/log/bluepill.log",
  :base_dir => "/var/bluepill"
}

OptionParser.new do |opts|
  opts.banner = "Usage: bluepill [app] cmd [options]"
  opts.on('-l', "--logfile LOGFILE", "Path to logfile, defaults to #{options[:log_file]}") do |file|
    options[:log_file] = file
  end

  opts.on('-c', "--base-dir DIR", "Directory to store bluepill socket and pid files, defaults to #{options[:base_dir]}") do |base_dir|
    options[:base_dir] = base_dir
  end

  opts.on("-v", "--version") do
    puts "bluepill, version #{Bluepill::VERSION}"
    exit
  end

  help = lambda do
    puts opts
    puts
    puts "Commands:"
    puts "    load CONFIG_FILE\t\tLoads new instance of bluepill using the specified config file"
    puts "    status\t\t\tLists the status of the proceses for the specified app"
    puts "    start TARGET\t\tIssues the start command for the target process or group"
    puts "    stop TARGET\t\t\tIssues the stop command for the target process or group"
    puts "    restart TARGET\t\tIssues the restart command for the target process or group"
    puts "    unmonitor TARGET\t\tStop monitoring target process or group"
    puts "    log [TARGET]\t\tShow the log for the specified process or group, defaults to all for app"
    puts "    quit\t\t\tStop bluepill"
    puts
    puts "See http://github.com/arya/bluepill for README"
    exit
  end

  opts.on_tail('-h','--help', 'Show this message', &help)
  help.call if ARGV.empty?
end.parse!

APPLICATION_COMMANDS = %w(status start stop restart unmonitor quit log)

controller = Bluepill::Controller.new(options.slice(:base_dir, :log_file))

if controller.running_applications.include?(ARGV.first)
  # the first arg is the application name
  options[:application] = ARGV.shift
elsif APPLICATION_COMMANDS.include?(ARGV.first)
  if controller.running_applications.length == 1
    # there is only one, let's just use that
    options[:application] = controller.running_applications.first
  elsif controller.running_applications.length > 1
    # There is more than one, tell them the list and exit
    $stderr.puts "You must specify an application name to run that command. Here's the list of running applications:"
    controller.running_applications.each_with_index do |app, index|
      $stderr.puts "  #{index + 1}. #{app}"
    end
    $stderr.puts "Usage: bluepill [app] cmd [options]"
    exit(1)
  else
    # There are none running AND they aren't trying to start one
    $stderr.puts "Error: There are no running bluepill daemons.\nTo start a bluepill daemon, use: bluepill load <config file>"
    exit(2)
  end
end

options[:command] = ARGV.shift

if options[:command] == "load"
  file = ARGV.shift
  if File.exists?(file)
    # Restart the ruby interpreter for the config file so that anything loaded here
    # does not stay in memory for the daemon
    require 'rbconfig'
    ruby = File.join(Config::CONFIG['bindir'], Config::CONFIG['ruby_install_name'])
    load_path = File.expand_path("#{File.dirname(__FILE__)}/../lib")
    file_path = File.expand_path(file)

    exec(ruby, "-I#{load_path}", '-rbluepill', file_path)
    
  else
    $stderr.puts "Can't find file: #{file}"
  end
else
  target = ARGV.shift
  controller.handle_command(options[:application], options[:command], target)
end