awsum 0.5

data/lib/awsum/ec2/address.rb

@@ -0,0 +1,67 @@
+require 'awsum/ec2/parsers/address_parser'
+
+module Awsum
+  class Ec2
+    class Address
+      attr_reader :public_ip, :instance_id
+
+      def initialize(ec2, public_ip, instance_id) #:nodoc:
+        @ec2 = ec2
+        @public_ip = public_ip
+        @instance_id = instance_id
+      end
+
+      # Get the Instance associated with this address.
+      #
+      # Returns nil if no instance is associated.
+      def instance
+        @ec2.instance(@instance_id) if @instance_id
+      end
+
+      # Will associate this address with an instance
+      #
+      # Raises an error if the address is already associated with an instance
+      def associate(instance_id)
+        if @instance_id.nil?
+          @ec2.associate_address instance_id, @public_ip
+        else
+          raise 'Cannot associate with an already associated instance' #FIXME: Write a better Awsum error here'
+        end
+      end
+
+      # Will associate this address with an instance (even if it is already associated with another instance)
+      def associate!(instance_id)
+        @ec2.associate_address instance_id, @public_ip
+      end
+
+      # Will disassociate this address from it's instance
+      #
+      # Raises an error if the address is not associated with an instance
+      def disassociate
+        if @instance_id.nil?
+          raise 'Not associated' #FIXME: Write a better Awsum error here'
+        else
+          result = @ec2.disassociate_address @public_ip
+          @instance_id = nil
+          result
+        end
+      end
+
+      # Will release this address
+      #
+      # Raises an error if the address is associated with an instance
+      def release
+        if @instance_id.nil?
+          @ec2.release_address @public_ip
+        else
+          raise 'Associated with an instance' #FIXME: Write a better Awsum error here'
+        end
+      end
+
+      # Will release this address regardless of whether it is associated with an instance or not.
+      def release!
+        @ec2.release_address! @public_ip
+      end
+    end
+  end
+end

data/lib/awsum/ec2/availability_zone.rb

@@ -0,0 +1,16 @@
+require 'awsum/ec2/parsers/availability_zone_parser'
+
+module Awsum
+  class Ec2
+    class AvailabilityZone
+      attr_reader :name, :state, :region_name
+
+      def initialize(ec2, name, state, region_name) #:nodoc:
+        @ec2 = ec2
+        @name = name
+        @state = state
+        @region_name = region_name
+      end
+    end
+  end
+end

data/lib/awsum/ec2/image.rb

@@ -0,0 +1,62 @@
+require 'awsum/ec2/parsers/image_parser.rb'
+require 'awsum/ec2/parsers/register_image_parser.rb'
+
+module Awsum
+  class Ec2
+    class Image
+      attr_reader :id, :location, :state, :owner, :public, :architecture, :type, :kernel_id, :ramdisk_id, :platform, :product_codes
+
+      def initialize(ec2, id, location, state, owner, public, architecture, type, kernel_id, ram_disk_id, platform, product_codes) #:nodoc:
+        @ec2 = ec2
+        @id = id
+        @location = location
+        @state = state
+        @owner = owner
+        @public = public
+        @architecture = architecture
+        @type = type
+        @kernel_id = kernel_id
+        @ramdisk_id = ram_disk_id
+        @platform = platform
+        @product_codes = product_codes
+      end
+
+      def public?
+        @public
+      end
+
+      # Deregister this Image
+      def deregister
+        @ec2.deregister_image @id
+      end
+
+      # Reregister this image
+      #
+      # Will both deregister and then register the Image again
+      def reregister
+        @ec2.deregister_image @id
+        new_id = @ec2.register_image @location
+        @id = new_id
+        self
+      end
+
+      # launches instances of this image
+      #
+      # ===Options:
+      # * <tt>:min</tt> - The minimum number of instances to launch. Default: 1
+      # * <tt>:max</tt> - The maximum number of instances to launch. Default: 1
+      # * <tt>:key_name</tt> - The name of the key pair with which to launch instances
+      # * <tt>:security_groups</tt> - The names of security groups to associate launched instances with
+      # * <tt>:user_data</tt> - User data made available to instances (Note: Must be 16K or less, will be base64 encoded by Awsum)
+      # * <tt>:instance_type</tt> - The size of the instances to launch, can be one of [m1.small, m1.large, m1.xlarge, c1.medium, c1.xlarge], default is m1.small
+      # * <tt>:availability_zone</tt> - The name of the availability zone to launch this instance in
+      # * <tt>:kernel_id</tt> - The ID of the kernel with which to launch instances
+      # * <tt>:ramdisk_id</tt> - The ID of the RAM disk with which to launch instances
+      # * <tt>:block_device_map</tt> - A 'hash' of mappings. E.g. {'instancestore0' => 'sdb'}
+      def run(options = {})
+        @ec2.run_instances(id, options)
+      end
+      alias_method :launch, :run
+    end
+  end
+end

data/lib/awsum/ec2/instance.rb

@@ -0,0 +1,57 @@
+require 'awsum/ec2/parsers/instance_parser.rb'
+
+module Awsum
+  class Ec2
+    class Instance
+      attr_reader :id, :image_id, :type, :state, :dns_name, :private_dns_name, :key_name, :kernel_id, :launch_time, :availability_zone, :product_codes, :ramdisk_id, :reason, :launch_index
+
+      def initialize(ec2, id, image_id, type, state, dns_name, private_dns_name, key_name, kernel_id, launch_time, availability_zone, product_codes, ramdisk_id, reason, launch_index) #:nodoc:
+        @ec2 = ec2
+        @id = id
+        @image_id = image_id
+        @type = type
+        @state = state
+        @dns_name = dns_name
+        @private_dns_name = private_dns_name
+        @key_name = key_name
+        @kernel_id = kernel_id
+        @launch_time = launch_time
+        @availability_zone = availability_zone
+        @product_codes = product_codes
+        @ramdisk_id = ramdisk_id
+        @reason = reason
+        @launch_index = launch_index
+      end
+
+      # Terminates this instance
+      def terminate
+        @ec2.terminate_instances(id)
+      end
+
+      # Lists all the Volume(s) attached to this Instance
+      def volumes
+        volumes = @ec2.volumes
+        volumes.delete_if {|v| v.instance_id != id}
+      end
+
+      # Will create and attach a Volume to this Instance
+      # You must specify a size or a snapshot_id
+      def create_volume(size = nil, snapshot_id = nil, device = '/dev/sdh')
+        raise ArgumentError.new('You must specify a size if not creating a volume from a snapshot') if size.blank? && snapshot_id.blank?
+        raise ArgumentError.new('You must specify a device to attach the volume to') unless device
+
+        volume = @ec2.create_volume availability_zone, :size => size, :snapshot_id => snapshot_id
+        while volume.status != 'available'
+          volume.reload
+        end
+        attach volume, device
+        volume
+      end
+
+      # Will attach a Volume to this Instance
+      def attach(volume, device = '/dev/sdh')
+        @ec2.attach_volume volume.id, id, device
+      end
+    end
+  end
+end

data/lib/awsum/ec2/keypair.rb

@@ -0,0 +1,16 @@
+require 'awsum/ec2/parsers/keypair_parser'
+
+module Awsum
+  class Ec2
+    class KeyPair
+      attr_reader :name, :fingerprint, :material
+
+      def initialize(ec2, name, fingerprint, material)
+        @ec2 = ec2
+        @name = name
+        @fingerprint = fingerprint
+        @material = material
+      end
+    end
+  end
+end

data/lib/awsum/ec2/parsers/address_parser.rb

@@ -0,0 +1,61 @@
+module Awsum
+  class Ec2
+    class AddressParser < Awsum::Parser #:nodoc:
+      def initialize(ec2)
+        @ec2 = ec2
+        @addresses = []
+        @text = nil
+        @stack = []
+      end
+
+      def tag_start(tag, attributes)
+        #Quick hack so we can use the same parser for AllocateAddress which doesn't use the item tag to wrap the address information
+        if tag == 'AllocateAddressResponse'
+          @stack << 'addressesSet'
+        end
+
+        case tag
+          when 'addressesSet'
+            @stack << 'addressesSet'
+          when 'item', 'AllocateAddressResponse'
+            case @stack[-1]
+              when 'addressesSet'
+                @current = {}
+            end
+        end
+        @text = ''
+      end
+
+      def text(text)
+        @text << text unless @text.nil?
+      end
+
+      def tag_end(tag)
+        case tag
+          when 'DescribeAddressesResponse'
+            #no-op
+          when 'addressesSet'
+            @stack.pop
+          when 'item', 'AllocateAddressResponse'
+            case @stack[-1]
+              when 'addressesSet'
+                @addresses << Address.new(
+                                @ec2,
+                                @current['publicIp'],
+                                @current['instanceId']
+                              )
+            end
+          else
+            unless @text.nil? || @current.nil?
+              text = @text.strip
+              @current[tag] = (text == '' ? nil : text)
+            end
+        end
+      end
+
+      def result
+        @addresses
+      end
+    end
+  end
+end

data/lib/awsum/ec2/parsers/availability_zone_parser.rb

@@ -0,0 +1,57 @@
+module Awsum
+  class Ec2
+    class AvailabilityZoneParser < Awsum::Parser #:nodoc:
+      def initialize(ec2)
+        @ec2 = ec2
+        @availability_zones = []
+        @text = nil
+        @stack = []
+      end
+
+      def tag_start(tag, attributes)
+        case tag
+          when 'availabilityZoneInfo'
+            @stack << 'availabilityZoneInfo'
+          when 'item'
+            case @stack[-1]
+              when 'availabilityZoneInfo'
+                @current = {}
+            end
+        end
+        @text = ''
+      end
+
+      def text(text)
+        @text << text unless @text.nil?
+      end
+
+      def tag_end(tag)
+        case tag
+          when 'DescribeAvailabilityZonesResponse', 'requestId'
+            #no-op
+          when 'availabilityZoneInfo'
+            @stack.pop
+          when 'item'
+            case @stack[-1]
+              when 'availabilityZoneInfo'
+                @availability_zones << AvailabilityZone.new(
+                                @ec2,
+                                @current['zoneName'],
+                                @current['zoneState'],
+                                @current['regionName']
+                              )
+            end
+          else
+            unless @text.nil? || @current.nil?
+              text = @text.strip
+              @current[tag] = (text == '' ? nil : text)
+            end
+        end
+      end
+
+      def result
+        @availability_zones
+      end
+    end
+  end
+end

data/lib/awsum/ec2/parsers/image_parser.rb

@@ -0,0 +1,74 @@
+module Awsum
+  class Ec2
+    class ImageParser < Awsum::Parser #:nodoc:
+      def initialize(ec2)
+        @ec2 = ec2
+        @images = []
+        @text = nil
+        @stack = []
+      end
+
+      def tag_start(tag, attributes)
+        case tag
+          when 'imagesSet'
+            @stack << 'imagesSet'
+          when 'item'
+            case @stack[-1]
+              when 'imagesSet'
+                @current = {}
+                @text = ''
+              when 'productCodes'
+                @product_codes = []
+                @text = ''
+            end
+          when 'productCodes'
+            @stack << 'productCodes'
+        end
+      end
+
+      def text(text)
+        @text << text unless @text.nil?
+      end
+
+      def tag_end(tag)
+        case tag
+          when 'DescribeImagesResponse', 'requestId'
+            #no-op
+          when 'imagesSet', 'productCodes'
+            @stack.pop
+          when 'item'
+            case @stack[-1]
+              when 'imagesSet'
+                @images << Image.new(
+                              @ec2,
+                              @current['imageId'],
+                              @current['imageLocation'],
+                              @current['imageState'],
+                              @current['imageOwnerId'],
+                              @current['isPublic'] == 'true',
+                              @current['architecture'],
+                              @current['imageType'],
+                              @current['kernelId'],
+                              @current['ramdiskId'],
+                              @current['platform'],
+                              @product_codes || []
+                            )
+                @text = ''
+            end
+          when 'productCode'
+            @product_codes << @text.strip
+          else
+            unless @text.nil?
+              text = @text.strip
+              @current[tag] = (text == '' ? nil : text)
+              @text = ''
+            end
+        end
+      end
+
+      def result
+        @images
+      end
+    end
+  end
+end

data/lib/awsum/ec2/parsers/instance_parser.rb

@@ -0,0 +1,90 @@
+module Awsum
+  class Ec2
+    class InstanceParser < Awsum::Parser #:nodoc:
+      def initialize(ec2)
+        @ec2 = ec2
+        @instances = []
+        @text = nil
+        @stack = []
+        @placement = nil
+        @state = nil
+      end
+
+      def tag_start(tag, attributes)
+        case tag
+          when 'reservationSet'
+            @stack << 'reservationSet'
+          when 'instancesSet'
+            @stack << 'instancesSet'
+          when 'productCodes'
+            @stack << 'productCodes'
+          when 'instanceState'
+            @stack << 'instanceState'
+          when 'placement'
+            @stack << 'placement'
+          when 'item'
+            case @stack[-1]
+              when 'reservationSet'
+              when 'instancesSet'
+                @current = {}
+                @state = {}
+              when 'productCodes'
+                @product_codes = []
+            end
+        end
+        @text = ''
+      end
+
+      def text(text)
+        @text << text unless @text.nil?
+      end
+
+      def tag_end(tag)
+        case tag
+          when 'DescribeInstancesResponse', 'requestId', 'reservationId'
+            #no-op
+          when 'reservationSet', 'instancesSet', 'productCodes', 'instanceState', 'placement'
+            @stack.pop
+          when 'item'
+            case @stack[-1]
+              when 'instancesSet'
+                @instances << Instance.new(
+                                @ec2,
+                                @current['instanceId'],
+                                @current['imageId'],
+                                @current['instanceType'],
+                                @state,
+                                @current['dnsName'],
+                                @current['privateDnsName'],
+                                @current['keyName'],
+                                @current['kernalId'],
+                                Time.parse(@current['launchTime']),
+                                @placement,
+                                @product_codes || [],
+                                @current['ramdisk_id'],
+                                @current['reason'],
+                                @current['amiLaunchIndex'].to_i
+                              )
+            end
+          when 'productCode'
+            @product_codes << @text.strip
+          when 'availabilityZone'
+            @placement = @text.strip
+          when 'code'
+            @state[:code] = @text.strip.to_i if @stack[-1] == 'instanceState'
+          when 'name'
+            @state[:name] = @text.strip if @stack[-1] == 'instanceState'
+          else
+            unless @text.nil? || @current.nil?
+              text = @text.strip
+              @current[tag] = (text == '' ? nil : text)
+            end
+        end
+      end
+
+      def result
+        @instances
+      end
+    end
+  end
+end