# -*- encoding: utf-8 -*-
#
# Author:: Tyler Ball (<tball@chef.io>)
#
# Copyright (C) 2015, Fletcher Nichol
#
# 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 "aws-sdk"
require "aws-sdk-core/credentials"
require "aws-sdk-core/shared_credentials"
require "aws-sdk-core/instance_profile_credentials"

module Kitchen

  module Driver

    class Aws

      # A class for creating and managing the EC2 client connection
      #
      # @author Tyler Ball <tball@chef.io>
      class Client

        def initialize( # rubocop:disable Metrics/ParameterLists
          region,
          profile_name = nil,
          access_key_id = nil,
          secret_access_key = nil,
          session_token = nil,
          http_proxy = nil,
          retry_limit = nil
        )
          creds = self.class.get_credentials(
            profile_name, access_key_id, secret_access_key, session_token
          )
          ::Aws.config.update(
            :region => region,
            :credentials => creds,
            :http_proxy => http_proxy
          )
          ::Aws.config.update(:retry_limit => retry_limit) unless retry_limit.nil?
        end

        # Try and get the credentials from an ordered list of locations
        # http://docs.aws.amazon.com/sdkforruby/api/index.html#Configuration
        # rubocop:disable Metrics/CyclomaticComplexity, Metrics/PerceivedComplexity
        def self.get_credentials(profile_name, access_key_id, secret_access_key, session_token)
          shared_creds = ::Aws::SharedCredentials.new(:profile_name => profile_name)
          if access_key_id && secret_access_key
            ::Aws::Credentials.new(access_key_id, secret_access_key, session_token)
          elsif ENV["AWS_ACCESS_KEY_ID"] && ENV["AWS_SECRET_ACCESS_KEY"]
            ::Aws::Credentials.new(
              ENV["AWS_ACCESS_KEY_ID"],
              ENV["AWS_SECRET_ACCESS_KEY"],
              ENV["AWS_SESSION_TOKEN"]
            )
          elsif shared_creds.loadable?
            shared_creds
          else
            ::Aws::InstanceProfileCredentials.new(:retries => 1)
          end
        end
        # rubocop:enable Metrics/CyclomaticComplexity, Metrics/PerceivedComplexity

        def create_instance(options)
          resource.create_instances(options)[0]
        end

        def get_instance(id)
          resource.instance(id)
        end

        def get_instance_from_spot_request(request_id)
          resource.instances(
            :filters => [{
              :name => "spot-instance-request-id",
              :values => [request_id]
            }]
          ).to_a[0]
        end

        def client
          @client ||= ::Aws::EC2::Client.new
        end

        def resource
          @resource ||= ::Aws::EC2::Resource.new
        end

      end

    end

  end

end