require 'beaker'

module BeakerRSpec
  # BeakerShim Module
  #
  # This module provides the connection between rspec and the Beaker DSL.
  # Additional wrappers are provided around commonly executed sets of Beaker
  # commands.
  module BeakerShim
    include Beaker::DSL

    # Accessor for logger
    # @return Beaker::Logger object
    def logger
      RSpec.configuration.logger
    end

    # Accessor for options hash
    # @return Hash options
    def options
      RSpec.configuration.options
    end

    # Provision the hosts to run tests on.
    # Assumes #setup has already been called.
    #
    def provision
      @network_manager = Beaker::NetworkManager.new(options, @logger)
      RSpec.configuration.hosts = @network_manager.provision
    end

    # Validate that the SUTs are up and correctly configured.  Checks that required
    # packages are installed and if they are missing attempt installation.
    # Assumes #setup and #provision has already been called.
    def validate
      @network_manager.validate
    end

    # Run configuration steps to have hosts ready to test on (such as ensuring that 
    # hosts are correctly time synched, adding keys, etc).
    # Assumes #setup, #provision and #validate have already been called.
    def configure
      @network_manager.configure
    end

    # Setup the testing environment
    # @param [Array<String>] args The argument array of options for configuring Beaker
    # See 'beaker --help' for full list of supported command line options
    def setup(args = [])
      options_parser = Beaker::Options::Parser.new
      options = options_parser.parse_args(args)
      options[:debug] = true
      RSpec.configuration.logger = Beaker::Logger.new(options)
      options[:logger] = logger
      RSpec.configuration.hosts = []
      RSpec.configuration.options = options
    end

    # Accessor for hosts object
    # @return [Array<Beaker::Host>]
    def hosts
      RSpec.configuration.hosts
    end

    # Accessor for default node
    # @return [Beaker::Host]
    def default_node
      RSpec.configuration.default_node ||= find_only_one :default
    end

    # Cleanup the testing framework, shut down test boxen and tidy up
    def cleanup
      @network_manager.cleanup
    end

  end
end