require "acquia_toolbelt"
require "acquia_toolbelt/error"
require "acquia_toolbelt/thor"

module AcquiaToolbelt
  class CLI < AcquiaToolbelt::Thor
    require "acquia_toolbelt/cli/ui"
    require "acquia_toolbelt/cli/api"
    require "acquia_toolbelt/cli/auth"
    require "acquia_toolbelt/cli/database"
    require "acquia_toolbelt/cli/server"
    require "acquia_toolbelt/cli/ssh"
    require "acquia_toolbelt/cli/svn"
    require "acquia_toolbelt/cli/task"
    require "acquia_toolbelt/cli/site"
    require "acquia_toolbelt/cli/domain"
    require "acquia_toolbelt/cli/deploy"
    require "acquia_toolbelt/cli/file"
    require "acquia_toolbelt/cli/environment"

    include Thor::Actions

    def self.start(given_args = ARGV, config = {})
      Thor::Base.shell = AcquiaToolbelt::CLI::UI
      ui = AcquiaToolbelt::CLI::UI.new

      # Use a custom symbol to separate the commands. Useful for rake styled
      # commands.
      if given_args[0].include? ":"
        given_args = given_args.flat_map {|e| e.split(':')}
      end

      super(given_args, {:shell => ui}.merge(config))
    rescue AcquiaToolbelt::Error
      ui.print_exception(e)
      raise
    rescue Interrupt => e
      puts
      ui.print_exception(e)
      ui.say("Quitting...")
      raise
    rescue SystemExit, Errno::EPIPE
      # Don't print a message for safe exits.
      raise
    rescue Exception => e
      ui.print_exception(e)
      raise
    end

    # Define some options that are available to all commands.
    class_option :subscription, :type => :string, :aliases => %w(-s),
      :desc => "Name of a subscription you would like to target."
    class_option :environment, :type => :string, :aliases => %w(-e),
      :desc => "Environment to target for commands."
    class_option :verbose, :type => :boolean, :aliases => %w(-v),
      :desc => "Increase the verbose output from the commands."

    # Authentication.
    desc "auth", ""
    subcommand "auth", AcquiaToolbelt::CLI::Auth

    # Databases.
    desc "databases", ""
    subcommand "databases", AcquiaToolbelt::CLI::Databases

    # Servers.
    desc "servers", ""
    subcommand "servers", AcquiaToolbelt::CLI::Servers

    # SSH.
    desc "ssh", ""
    subcommand "ssh", AcquiaToolbelt::CLI::SSH

    # SVN.
    desc "svn", ""
    subcommand "svn", AcquiaToolbelt::CLI::SVN

    # Tasks.
    desc "tasks", ""
    subcommand "tasks", AcquiaToolbelt::CLI::Tasks

    # Sites.
    desc "sites", ""
    subcommand "sites", AcquiaToolbelt::CLI::Sites

    # Domains.
    desc "domains", ""
    subcommand "domains", AcquiaToolbelt::CLI::Domains

    # Deployments.
    desc "deploy", ""
    subcommand "deploy", AcquiaToolbelt::CLI::Deploy

    # Files.
    desc "files", ""
    subcommand "files", AcquiaToolbelt::CLI::Files

    # Environments.
    desc "environments", ""
    subcommand "environments", AcquiaToolbelt::CLI::Environments
  end
end