stanchino/sitefull-cloud

require 'aws-sdk'
require 'sitefull-cloud/provider/amazon/networking'

module Sitefull
  module Provider
    module Amazon
      include Networking

      REQUIRED_OPTIONS = %w(role_arn session_name).freeze
      MACHINE_TYPES = %w(t2.nano t2.micro t2.small t2.medium t2.large m4.large m4.xlarge m4.2xlarge m4.4xlarge m4.10xlarge m3.medium m3.large m3.xlarge m3.2xlarge).freeze

      DEFAULT_REGION = 'us-east-1'.freeze
      WAIT = 2.freeze
      INSTANCE_RUNNING_STATE = 'running'.freeze

      def process(options = {})
        options[:region] ||= DEFAULT_REGION
        options
      end

      def connection
        @connection ||= Aws::EC2::Client.new(region: options[:region], credentials: credentials)
      end

      def regions
        @regions ||= connection.describe_regions.regions.map { |r| OpenStruct.new(id: r.region_name, name: r.region_name) }
      end

      def machine_types(_)
        MACHINE_TYPES.map { |mt| OpenStruct.new(id: mt, name: mt) }
      end

      def images(os)
        filters = [{ name: 'name', values: ["#{os}*", "#{os.downcase}*"] }, { name: 'is-public', values: ['true'] }, { name: 'virtualization-type', values: ['hvm'] }]
        connection.describe_images(filters: filters).images.select { |i| i.image_owner_alias.nil? }.map { |i| OpenStruct.new(id: i.image_id, name: i.name) }
      end

      def create_network
        setup_vpc
        setup_internet_gateway
        setup_routing
        subnet.subnet_id
      end

      def create_key(name)
        connection.import_key_pair(key_name: name, public_key_material: "ssh-rsa #{key_data[:public_key]}")
        OpenStruct.new(key_data)
      end

      def create_firewall_rules
        setup_security_group
      end

      # Uses http://docs.aws.amazon.com/sdkforruby/api/Aws/EC2/Client.html#run_instances-instance_method
      def create_instance(_name, machine_type, image, network_id, key)
        instance = connection.run_instances(image_id: image, instance_type: machine_type,
                                            subnet_id: network_id, key_name: key.name,
                                            security_group_ids: [security_group.group_id], min_count: 1, max_count: 1).instances.first
        sleep WAIT unless instance(instance.instance_id).state.name == INSTANCE_RUNNING_STATE
        instance.instance_id
      end

      def instance_data(instance_id)
        OpenStruct.new(id: instance_id, public_ip: instance(instance_id).public_ip_address)
      end

      def valid?
        !connection.nil?
      end

      private

      def instance(instance_id)
        connection.describe_instances(instance_ids: [instance_id]).reservations.first.instances.first
      end
    end
  end
end