etude_for_aws 0.7.4

data/lib/etude_for_aws/ec2/ec2.rb

@@ -0,0 +1,218 @@
+module EC2
+  class Ec2
+    attr_reader :config,:gateway,:security_group,:key_pair,:ec2_instances
+
+    def initialize(vpc)
+      @config = Configuration.new
+      @gateway = Ec2ApiGateway.new
+      setup_config(vpc)
+    end
+
+    def create
+
+      create_security_group
+
+      create_key_pair
+
+      create_ec2_instance
+
+    end
+
+    def destroy
+
+      terminate_ec2_instance
+
+      delete_security_group
+
+      delete_key_pair
+
+    end
+
+    def start
+      @ec2_instances.each do |ec2_instance|
+        ec2_instance.start
+      end
+    end
+
+    def stop
+      @ec2_instances.each do |ec2_instance|
+        ec2_instance.stop
+      end
+    end
+
+    def reboot
+      @ec2_instances.each do |ec2_instance|
+        ec2_instance.reboot
+      end
+    end
+
+    private
+    def setup_config(vpc)
+      @config.vpc_id = vpc.get_vpc_id
+      @subnet_infos = vpc.get_subnet_infos
+      @security_group = SecurityGroup.new(self)
+      @key_pair = KeyPair.new(self)
+
+      @ec2_instances = []
+      @config.instance_tags_public.each do |tag|
+        @config.instance_tags = tag
+        values = [@config.instance_tags[0][:value]]
+        instance_ids = @gateway.get_instance_collection(values)
+        instance_ids.each do |instance_id|
+          ec2_instance = Ec2Instance.new(self, instance_id)
+          @ec2_instances << ec2_instance
+        end
+      end
+
+      @config.instance_tags_private.each do |tag|
+        @config.instance_tags = tag
+        values = [@config.instance_tags[0][:value]]
+        instance_ids = @gateway.get_instance_collection(values)
+        instance_ids.each do |instance_id|
+          ec2_instance = Ec2Instance.new(self, instance_id)
+          @ec2_instances << ec2_instance
+        end
+      end
+    end
+
+    def create_security_group
+      @security_group.create
+    end
+
+    def create_key_pair
+      @key_pair.create
+    end
+
+    def create_ec2_instance
+      private_i = 0
+      public_i = 0
+      @subnet_infos.each do |info|
+        @config.subnet_id = info[:subnet_id]
+        @config.az = info[:az]
+        if info[:network] == 'Private'
+          @config.instance_tags = @config.instance_tags_private[private_i]
+          private_i += private_i
+        else
+          @config.instance_tags = @config.instance_tags_public[public_i]
+          public_i += public_i
+        end
+        ec2_instance = Ec2Instance.new(self)
+        ec2_instance.create(@security_group,@key_pair)
+        @ec2_instances << ec2_instance
+      end
+    end
+
+    def terminate_ec2_instance
+      @ec2_instances.each do |ec2_instance|
+        ec2_instance.terminate
+      end
+    end
+
+    def delete_security_group
+      @security_group = nil if @security_group.delete.nil?
+    end
+
+    def delete_key_pair
+      @key_pair.delete
+      @key_pair = nil
+    end
+  end
+
+  class Ec2Stub < Ec2
+    def initialize(vpc)
+      @config = ConfigurationStub.new
+      @gateway = Ec2ApiGatewayStub.new
+      setup_config(vpc)
+    end
+
+    def start
+      instance_id = 'String'
+      @gateway.client.stub_responses(:describe_instances,
+                                     {
+                                         reservations: [
+                                             {
+                                                 instances: [
+                                                     instance_id: instance_id,
+                                                     state: {'code':89}
+                                                 ]
+                                             }
+                                         ]
+                                     })
+      @ec2_instances << Ec2Instance.new(self, instance_id)
+      super
+    end
+
+    def reboot
+      instance_id = 'String'
+      @gateway.client.stub_responses(:describe_instances,
+                                     {
+                                         reservations: [
+                                             {
+                                                 instances: [
+                                                     instance_id: instance_id,
+                                                     state: {'code':16}
+                                                 ]
+                                             }
+                                         ]
+                                     })
+      @ec2_instances << Ec2Instance.new(self, instance_id)
+      super
+    end
+
+    def stop
+      instance_id = 'String'
+      @gateway.client.stub_responses(:describe_instances,
+                                     {
+                                         reservations: [
+                                             {
+                                                 instances: [
+                                                     instance_id: instance_id,
+                                                     state: {'code':16}
+                                                 ]
+                                             }
+                                         ]
+                                     })
+      @ec2_instances << Ec2Instance.new(self, instance_id)
+      super
+    end
+
+    private
+    def create_security_group
+      super
+    end
+
+    def create_key_pair
+      super
+    end
+
+    def create_ec2_instance
+      super
+    end
+
+    def terminate_ec2_instance
+      @gateway.client.stub_responses(:describe_instances,
+                                     {
+                                         reservations: [
+                                             {
+                                                 instances: [
+                                                     instance_id: 'String',
+                                                     state: {'code':16}
+                                                 ]
+                                             }
+                                         ]
+                                     })
+      @ec2_instances.first.instance_variable_set :@instance_id,'String' unless @ec2_instances.first.nil?
+      super
+      @ec2_instances = []
+    end
+
+    def delete_security_group
+      super
+      @security_group = nil
+    end
+
+    def delete_key_pair
+      super
+    end
+  end
+end

data/lib/etude_for_aws/ec2/ec2_api_gateway.rb

@@ -0,0 +1,127 @@
+module EC2
+  class Ec2ApiGateway
+    include CertificationHelper
+
+    attr_reader :client,:resource,:stub
+
+    def initialize
+      aws_certificate
+      @client = Aws::EC2::Client.new
+      @resource = Aws::EC2::Resource.new(client: client)
+      @stub = false
+    end
+
+    def stub?
+      @stub
+    end
+
+    def get_instance_collection(values)
+      instance_ids = []
+      resp = @client.describe_instances(filters: [{name: "tag:Name", values: values}])
+      resp.reservations.each do |reservation|
+        reservation.instances.each do |instance|
+          instance_ids << instance.instance_id
+        end
+      end
+      instance_ids
+    end
+
+    def get_group_id(security_group)
+      group_id = nil
+      @resource.security_groups.each do |sg|
+        if sg.group_name == security_group[:group_name]
+          group_id = sg.group_id
+        end
+      end
+      group_id
+    end
+
+    def select_key_pairs
+      @client.describe_key_pairs
+    end
+
+    def find_instance_by_id(instance_id)
+      @resource.instance(instance_id)
+    end
+
+    def create_security_group(security_group)
+      resp = @resource.create_security_group(security_group)
+      resp.id
+    end
+
+    def authorize_egress(id,authorize_egress)
+      sg = @resource.security_group(id)
+      sg.authorize_egress(authorize_egress)
+    end
+
+    def authorize_ingress(id,authorize_ingress)
+      sg = @resource.security_group(id)
+      sg.authorize_ingress(authorize_ingress)
+    end
+
+    def delete_security_group(security_group_id)
+      @client.delete_security_group({
+                                        group_id: security_group_id,
+                                    })
+    end
+
+    def create_key_pairs(key_pair_name)
+      @client.create_key_pair({
+                                  key_name: key_pair_name
+                              })
+    end
+
+    def create_instances(image_id,min_count,max_count,key_pair_name,security_group,encoded_script,instance_type,config)
+      @resource.create_instances({
+                                     image_id: image_id,
+                                     min_count: min_count,
+                                     max_count: max_count,
+                                     key_name: key_pair_name,
+                                     user_data: encoded_script,
+                                     instance_type: instance_type,
+                                     placement: {
+                                         availability_zone: config.az
+                                     },
+                                     network_interfaces: [
+                                         {
+                                             device_index: 0,
+                                             subnet_id: config.subnet_id,
+                                             associate_public_ip_address: true,
+                                             groups: [security_group.security_group_id],
+                                         },
+                                     ],
+                                 })
+    end
+
+    def delete_key_pairs(key_pair_name)
+      @client.delete_key_pair({
+                                  key_name: key_pair_name
+                              })
+    end
+
+    def wait_for_instance_status_ok(instance_id)
+      @resource.client.wait_until(:instance_status_ok, {instance_ids: [instance_id]}) unless stub?
+    end
+
+    def wait_for_instance_terminated(instance_id)
+      @resource.client.wait_until(:instance_terminated, {instance_ids: [instance_id]}) unless stub?
+    end
+
+    def wait_for_instance_running(instance_id)
+      @resource.client.wait_until(:instance_running, {instance_ids: [instance_id]})  unless stub?
+    end
+
+    def wait_for_instance_stopped(instance_id)
+      @resource.client.wait_until(:instance_stopped, {instance_ids: [instance_id]}) unless stub?
+    end
+
+  end
+
+  class Ec2ApiGatewayStub < Ec2ApiGateway
+    def initialize
+      @client = Aws::EC2::Client.new(stub_responses: true)
+      @resource = Aws::EC2::Resource.new(stub_responses: true,client: client)
+      @stub = true
+    end
+  end
+end

data/lib/etude_for_aws/ec2/ec2_instance.rb

@@ -0,0 +1,107 @@
+module EC2
+  class Ec2Instance
+    attr_reader :instance_id
+
+    def initialize(ec2,instance_id=nil)
+      @config = ec2.config
+      @gateway = ec2.gateway
+      script = ''
+      @encoded_script = Base64.encode64(script)
+      @image_id = @config.image_id
+      @instance_type = @config.instance_type
+      @min_count = @config.min_count
+      @max_count = @config.max_count
+      @instance_tags = @config.instance_tags
+      @instance_id = instance_id
+    end
+
+    def create(security_group,key_pair)
+      instance = @gateway.create_instances(@image_id,
+                                  @min_count,
+                                  @max_count,
+                                  key_pair.key_pair_name,
+                                  security_group,
+                                  @encoded_script,
+                                  @instance_type,
+                                  @config)
+
+      instance.empty? ? @instance_id = nil : @instance_id = instance.first.id
+      @gateway.wait_for_instance_status_ok(@instance_id)
+      instance.create_tags({tags: @instance_tags})
+      instance
+    end
+
+    def terminate
+      i = @gateway.find_instance_by_id(@instance_id)
+
+      unless i.nil?
+        if i.exists?
+          case i.state.code
+            when 48 # terminated
+              puts "#{@instance_id} is already terminated"
+            else
+              i.terminate
+              @gateway.wait_for_instance_terminated(@instance_id)
+          end
+        end
+      end
+    end
+
+    def start
+      i = @gateway.find_instance_by_id(@instance_id)
+
+      unless i.nil?
+        if i.exists?
+          case i.state.code
+            when 0  # pending
+              puts "#{instance_id} is pending, so it will be running in a bit"
+            when 16  # started
+              puts "#{instance_id} is already started"
+            when 48  # terminated
+              puts "#{instance_id} is terminated, so you cannot start it"
+            else
+              puts "#{instance_id} is starting"
+              i.start
+              @gateway.wait_for_instance_running(instance_id)
+          end
+        end
+      end
+    end
+
+    def stop
+      i = @gateway.find_instance_by_id(@instance_id)
+
+      unless i.nil?
+        if i.exists?
+          case i.state.code
+            when 48  # terminated
+              puts "#{instance_id} is terminated, so you cannot stop it"
+            when 64  # stopping
+              puts "#{instance_id} is stopping, so it will be stopped in a bit"
+            when 89  # stopped
+              puts "#{instance_id} is already stopped"
+            else
+              puts "#{instance_id} is stopping"
+              i.stop
+              @gateway.wait_for_instance_stopped(instance_id)
+          end
+        end
+      end
+    end
+
+    def reboot
+      i = @gateway.find_instance_by_id(@instance_id)
+
+      if i.exists?
+        case i.state.code
+          when 48  # terminated
+            puts "#{instance_id} is terminated, so you cannot reboot it"
+          else
+            puts "#{instance_id} is rebooting"
+            i.reboot
+            @gateway.wait_for_instance_status_ok(instance_id)
+        end
+      end
+    end
+  end
+end