require 'optparse'

module VagrantPlugins
  module Invade
    module Command
      class Root < Vagrant.plugin('2', :command)
        def self.synopsis
          'manages invade: validate, build, etc.'
        end

        def initialize(argv, env)
          super

          @main_args, @sub_command, @sub_args = split_main_and_subcommand(argv)
          @subcommands = Vagrant::Registry.new

          # INIT COMMAND ("vagrant invade init")
            @subcommands.register(:init) do
            require_relative 'init'
            Init
          end

          # CHECK COMMAND ("vagrant invade check")
            @subcommands.register(:check) do
            require_relative 'check'
            Check
          end

          # VALIDATE COMMAND ("vagrant invade validate")
          @subcommands.register(:validate) do
            require_relative 'validate'
            Validate
          end

          # BUILD COMMAND ("vagrant invade build")
            @subcommands.register(:build) do
            require_relative 'build'
            Build
          end
        end

        def execute
          if @main_args.include?('-h') || @main_args.include?('--help')
            # Print the help for all the sub-commands.
            return help
          end

        # If we reached this far then we must have a subcommand. If not,
          # then we also just print the help and exit.
          command_class = @subcommands.get(@sub_command.to_sym) if @sub_command
          return help if !command_class || !@sub_command
          @logger.debug("Invoking command class: #{command_class} #{@sub_args.inspect}")

          # Initialize and execute the command class
          command_class.new(@sub_args, @env).execute
        end

        # Prints the help out for this command
        def help
          opts = OptionParser.new do |o|
            o.banner = 'Usage: vagrant invade <command> [<args>]'
            o.separator ''
            o.separator 'Available subcommands:'

            # Add the available subcommands as separators in order to print them
            # out as well.
            keys = []
            @subcommands.each { |key, value| keys << key.to_s }

            keys.sort.each do |key|
              o.separator "     #{key}"
            end

            o.separator ''
            o.separator 'For help on any individual command run `vagrant invade COMMAND -h`'
          end

          @env.ui.info(opts.help, prefix: false)
        end
      end

    end

  end
end