korczis/apollo-fat-box

# encoding: utf-8

require 'pmap'

require_relative 'provider_base'

# Apollon Module
module Apollon
  # Apollon Provider Module
  module Provider
    # Apollon Aws Provider
    class Aws < ProviderBase
      # Apollon Aws Machine
      class Machine < ProviderBase::Machine
      end

      REGIONS = %w(
        us-east-1
        us-west-2
        us-west-1
        eu-west-1
        eu-central-1
        ap-southeast-1
        ap-southeast-2
        ap-northeast-1
        sa-east-1
      )

      DEFAULT_OPTS = {
        regions: REGIONS
      }

      def initialize(client, opts = DEFAULT_OPTS)
        @opts = DEFAULT_OPTS.merge(opts)

        super(client, @opts)

        fail 'config not initialized!' if config.nil?
        fail ArgumentError, "No config['key'] specified!" if config['key'].empty?
        fail ArgumentError, "No config['secret'] specified!" if config['secret'].empty?

        @compute = {}
        opts[:regions].each do |region|
          compute_opts = {
            provider: 'AWS',
            aws_access_key_id: config['key'],
            aws_secret_access_key: config['secret'],
            region: region
          }

          @compute[region] = Fog::Compute.new(compute_opts)
        end

        self
      end

      def machines(force_refresh = false)
        if @machines.nil? || force_refresh
          klass = self.class.const_get('Machine')
          res = @compute.pmap do |_k, v|
            v.servers.all.to_a.map { |m| klass.new(self, v, m) }
          end
          @machines = res.reduce([], :+)
        end
        @machines
      end

      def regions
        REGIONS
      end

      def ssh_keys
        res = @compute.pmap do |_k, v|
          v.key_pairs.to_a
        end
        res.flatten!
      end
    end
  end
end