class Appsignal::CLI

def command_option_parser
  {
    "demo" => OptionParser.new do |o|
      o.banner = "Usage: appsignal demo [options]"
      o.on "--environment=<app_env>", "The environment to demo" do |arg|
        options[:environment] = arg
      end
    end,
    "diagnose" => OptionParser.new do |o|
      o.banner = "Usage: appsignal diagnose [options]"
      o.on "--environment=<app_env>", "The environment to diagnose" do |arg|
        options[:environment] = arg
      end
      o.on "--[no-]send-report",
        "Confirm sending the report to AppSignal automatically" do |arg|
        options[:send_report] = arg
      end
      o.on "--[no-]color", "Colorize the output of the diagnose command" do |arg|
        options[:color] = arg
      end
    end,
    "install" => OptionParser.new do |o|
      o.on "--[no-]color", "Colorize the output of the diagnose command" do |arg|
        options[:color] = arg
      end
    end
  }
end

def global_option_parser
  OptionParser.new do |o|
    o.banner = "Usage: appsignal <command> [options]"
    o.on "-v", "--version", "Print version and exit" do |_arg|
      puts "AppSignal #{Appsignal::VERSION}"
      exit(0)
    end
    o.on "-h", "--help", "Show help and exit" do
      puts o
      exit(0)
    end
    o.separator ""
    o.separator "Available commands: #{AVAILABLE_COMMANDS.join(", ")}"
  end
end

def run(argv = ARGV)
  @options = {}
  global = global_option_parser
  commands = command_option_parser
  global.order!(argv)
  command = argv.shift
  if command
    if AVAILABLE_COMMANDS.include?(command)
      commands[command].parse!(argv)
      case command.to_sym
      when :demo
        Appsignal::CLI::Demo.run(options)
      when :diagnose
        Appsignal::CLI::Diagnose.run(options)
      when :install
        Appsignal::CLI::Install.run(argv.shift, options)
      end
    else
      puts "Command '#{command}' does not exist, run appsignal -h to " \
        "see the help"
      exit(1)
    end
  else
    # Print help
    puts global
    exit(0)
  end
end