require 'aws-sdk-ec2'
require 'forwardable'
require 'base64'

module AmiSpec
  class AwsInstance
    extend Forwardable

    def self.start(args)
      new(args).tap do |instance|
        instance.start
      end
    end

    def initialize(options)
      @role = options.fetch(:role)
      @ami = options.fetch(:ami)
      @subnet_id = options.fetch(:subnet_id)
      @key_name = options.fetch(:key_name)
      @instance_type = options.fetch(:aws_instance_type)
      @public_ip = options.fetch(:aws_public_ip)
      @associate_public_ip = options.fetch(:associate_public_ip)
      @region = options.fetch(:aws_region)
      @security_group_ids = options.fetch(:aws_security_groups)
      @tags = ec2ify_tags(options.fetch(:tags))
      @user_data_file = options.fetch(:user_data_file, nil)
      @iam_instance_profile_arn = options.fetch(:iam_instance_profile_arn, nil)
      @logger = options.fetch(:logger)
    end

    def_delegators :@instance, :instance_id, :tags, :private_ip_address, :public_ip_address

    def start
      @logger.info "Creating AWS EC2 instance for #{@ami}"
      client = Aws::EC2::Client.new(client_options)
      placeholder_instance = client.run_instances(instances_options).instances.first

      @instance = Aws::EC2::Instance.new(placeholder_instance.instance_id, client_options)
      @logger.info "Waiting for AWS EC2 instance to start: #{@instance.id}"
      @instance.wait_until_running
      tag_instance
      @logger.info "AWS EC2 instance started: #{@instance.id}"
    end

    def terminate
      @logger.info "Terminating AWS EC2 instance: #{@instance.id}"
      @instance.terminate
      @instance.wait_until_terminated
      @logger.info "AWS EC2 instance terminated: #{@instance.id}"
    end

    private

    def client_options
      !@region.nil? ? {region: @region} : {}
    end

    def ec2ify_tags(tags)
      tags.map { |k,v| {key: k, value: v} }
    end

    def instances_options
      params = {
        image_id: @ami,
        min_count: 1,
        max_count: 1,
        instance_type: @instance_type,
        key_name: @key_name,
        network_interfaces: [{
          device_index: 0,
          associate_public_ip_address: @public_ip || @associate_public_ip,
          subnet_id: @subnet_id,
        }]
      }

      params[:user_data] = Base64.encode64(File.read(@user_data_file)) unless @user_data_file.nil?

      unless @iam_instance_profile_arn.nil?
        params[:iam_instance_profile] = {
            arn: @iam_instance_profile_arn
        }
      end

      unless @security_group_ids.nil?
        params[:network_interfaces][0][:groups] = @security_group_ids
      end

      params
    end

    def tag_instance
      @instance.create_tags(tags: [{ key: 'AmiSpec', value: @role }] + @tags)
    end
  end
end