################################################################################
#
#      Author: Zachary Patten <zachary@jovelabs.com>
#   Copyright: Copyright (c) 2011-2013 Atalanta Systems Ltd
#     License: Apache License, Version 2.0
#
#   Licensed under the Apache License, Version 2.0 (the "License");
#   you may not use this file except in compliance with the License.
#   You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#   Unless required by applicable law or agreed to in writing, software
#   distributed under the License is distributed on an "AS IS" BASIS,
#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#   See the License for the specific language governing permissions and
#   limitations under the License.
#
################################################################################

require 'cucumber/chef/providers/aws'
require 'cucumber/chef/providers/vagrant'

module Cucumber
  module Chef

    class ProviderError < Error; end

    class Provider
      attr_accessor :stdout, :stderr, :stdin, :logger

      PROXY_METHODS = %w(create destroy up down status lab_exists? labs labs_running labs_shutdown id state username ip port chef_server_webui)

################################################################################

      def initialize(stdout=STDOUT, stderr=STDERR, stdin=STDIN, logger=$logger)
        @stdout, @stderr, @stdin, @logger = stdout, stderr, stdin, logger
        @stdout.sync = true if @stdout.respond_to?(:sync=)

        @provider = case Cucumber::Chef::Config[:provider]
        when :aws then
          Cucumber::Chef::Provider::AWS.new(@stdout, @stderr, @stdin, @logger)
        when :vagrant then
          Cucumber::Chef::Provider::Vagrant.new(@stdout, @stderr, @stdin, @logger)
        end
      end

################################################################################

      def chef_server_webui
        "http://#{ip}:4040/"
      end

      def chef_server_api
        "http://#{ip}:4000/"
      end

      def status
        if lab_exists?
          details = {
            "Provider" => @provider.class,
            "ID" => self.id,
            "State" => self.state,
            "Username" => self.username,
            "IP Address" => self.ip,
            "Port" => self.port,
            "Chef-Server API" => self.chef_server_api,
            "Chef-Server WebUI" => self.chef_server_webui
          }
          max_key_length = details.collect{ |k,v| k.to_s.length }.max
          details.each do |key,value|
            @stdout.puts("%#{max_key_length}s: %s" % [key,value.inspect])
          end
        else
          @stdout.puts("There are no cucumber-chef test labs to display information for!")
        end

      rescue Exception => e
        Cucumber::Chef.logger.fatal { e.message }
        Cucumber::Chef.logger.fatal { e.backtrace.join("\n") }
        raise ProviderError, e.message
      end

################################################################################

      def method_missing(method_name, *method_args)
        if Cucumber::Chef::Provider::PROXY_METHODS.include?(method_name.to_s)
          Cucumber::Chef.logger.debug { "provider: #{method_name} #{method_args.inspect}" }
          @provider.send(method_name.to_sym, *method_args)
        else
          super(method_name, *method_args)
        end
      end

################################################################################

    end

  end
end

################################################################################