internuity-awsum 0.2

data/lib/ec2/image.rb

@@ -0,0 +1,161 @@
+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
+
+    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
+
+    class RegisterImageParser < Awsum::Parser #:nodoc:
+      def initialize(ec2)
+        @ec2 = ec2
+        @image = nil
+        @text = nil
+      end
+
+      def tag_start(tag, attributes)
+        case tag
+          when 'imageId'
+            @text = ''
+        end
+      end
+
+      def text(text)
+        @text << text unless @text.nil?
+      end
+
+      def tag_end(tag)
+        case tag
+          when 'imageId'
+            @image = @text
+            @text = nil
+        end
+      end
+
+      def result
+        @image
+      end
+    end
+  end
+end

data/lib/ec2/instance.rb

@@ -0,0 +1,144 @@
+require 'time'
+
+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
+
+    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

data/lib/ec2/snapshot.rb

@@ -0,0 +1,82 @@
+module Awsum
+  class Ec2
+    class Snapshot
+      attr_reader :id, :volume_id, :status, :start_time, :progress
+
+      def initialize(ec2, id, volume_id, status, start_time, progress) #:nodoc:
+        @ec2 = ec2
+        @id = id
+        @volume_id = volume_id
+        @status = status
+        @start_time = start_time
+        @progress = progress
+      end
+
+      # Delete this Snapshot
+      def delete
+        @ec2.delete_snapshot id
+      end
+    end
+
+    class SnapshotParser < Awsum::Parser #:nodoc:
+      def initialize(ec2)
+        @ec2 = ec2
+        @snapshots = []
+        @text = nil
+        @stack = []
+      end
+
+      def tag_start(tag, attributes)
+        #Quick hack so we can use the same parser for CreateSnapshot which doesn't use the item tag to wrap the snapshot information
+        if tag == 'CreateSnapshotResponse'
+          @stack << 'snapshotSet'
+        end
+
+        case tag
+          when 'snapshotSet'
+            @stack << 'snapshotSet'
+          when 'item', 'CreateSnapshotResponse'
+            case @stack[-1]
+              when 'snapshotSet'
+                @current = {}
+            end
+        end
+        @text = ''
+      end
+
+      def text(text)
+        @text << text unless @text.nil?
+      end
+
+      def tag_end(tag)
+        case tag
+          when 'DescribeSnapshotsResponts'
+            #no-op
+          when 'snapshotSet'
+            @stack.pop
+          when 'item', 'CreateSnapshotResponse'
+            case @stack[-1]
+              when 'snapshotSet'
+                @snapshots << Snapshot.new(
+                                @ec2,
+                                @current['snapshotId'], 
+                                @current['volumeId'], 
+                                @current['status'],
+                                @current['startTime'].blank? ? nil :Time.parse(@current['startTime']),
+                                @current['progress']
+                              )
+            end
+          else
+            unless @text.nil? || @current.nil?
+              text = @text.strip
+              @current[tag] = (text == '' ? nil : text)
+            end
+        end
+      end
+
+      def result
+        @snapshots
+      end
+    end
+  end
+end

data/lib/ec2/volume.rb

@@ -0,0 +1,137 @@
+module Awsum
+  class Ec2
+    class Volume
+      attr_reader :id, :instance_id, :size, :status, :create_time, :snapshot_id, :device, :attach_time, :availability_zone, :attachment_status
+
+      def initialize(ec2, id, instance_id, size, status, create_time, snapshot_id, device, attach_time, availability_zone, attachment_status) #:nodoc:
+        @ec2 = ec2
+        @id = id
+        @instance_id = instance_id
+        @size = size
+        @status = status
+        @create_time = create_time
+        @snapshot_id = snapshot_id
+        @device = device
+        @attach_time = attach_time
+        @availability_zone = availability_zone
+        @attachment_status = attachment_status
+      end
+
+      # Will reload the information about this Volume
+      #
+      # Useful for when a volume has just been created but is not yet available
+      #   while volume.status != 'available'
+      #     volume.reload
+      #   end
+      def reload
+        reloaded_volume = @ec2.volume id
+
+        @id = id
+        @instance_id = reloaded_volume.instance_id
+        @size = reloaded_volume.size
+        @status = reloaded_volume.status
+        @create_time = reloaded_volume.create_time
+        @snapshot_id = reloaded_volume.snapshot_id
+        @device = reloaded_volume.device
+        @attach_time = reloaded_volume.attach_time
+        @availability_zone = reloaded_volume.availability_zone
+        @attachment_status = reloaded_volume.attachment_status
+      end
+
+      # Detach this volume
+      def detach(force = false)
+        @ec2.detach_volume id, :force => force
+      end
+
+      # Delete this volume
+      def delete
+        @ec2.delete_volume id
+      end
+
+      # Creates a Snapshot of this Volume
+      def create_snapshot
+        @ec2.create_snapshot id
+      end
+
+      # Lists the Snapshot(s) of this Volume
+      def snapshots
+        snapshots = @ec2.snapshots
+        snapshots.delete_if {|s| s.volume_id != id}
+      end
+    end
+
+    class VolumeParser < Awsum::Parser #:nodoc:
+      def initialize(ec2)
+        @ec2 = ec2
+        @volumes = []
+        @text = nil
+        @stack = []
+      end
+
+      def tag_start(tag, attributes)
+        #Quick hack so we can use the same parser for CreateVolume which doesn't use the item tag to wrap the volume information
+        if tag == 'CreateVolumeResponse'
+          @stack << 'volumeSet'
+        end
+
+        case tag
+          when 'volumeSet'
+            @stack << 'volumeSet'
+          when 'attachmentSet'
+            @stack << 'attachmentSet'
+          when 'item', 'CreateVolumeResponse'
+            case @stack[-1]
+              when 'volumeSet'
+                @current = {}
+              when 'attachmentSet'
+            end
+        end
+        @text = ''
+      end
+
+      def text(text)
+        @text << text unless @text.nil?
+      end
+
+      def tag_end(tag)
+        case tag
+          when 'DescribeVolumesResponse'
+            #no-op
+          when 'volumeSet', 'attachmentSet'
+            @stack.pop
+          when 'item', 'CreateVolumeResponse'
+            case @stack[-1]
+              when 'volumeSet'
+                @volumes << Volume.new(
+                                @ec2,
+                                @current['volumeId'], 
+                                @current['instanceId'], 
+                                @current['size'].to_i, 
+                                @current['status'],
+                                @current['createTime'].blank? ? nil :Time.parse(@current['createTime']),
+                                @current['snapshotId'], 
+                                @current['device'],
+                                @current['attachTime'].blank? ? nil :Time.parse(@current['attachTime']),
+                                @current['availabilityZone'],
+                                @current['attachment_status']
+                              )
+            end
+          else
+            unless @text.nil? || @current.nil?
+              text = @text.strip
+              #Handle special case for attachmentSet/status
+              if @stack[-1] == 'attachmentSet' && tag == 'status'
+                @current['attachment_status'] = (text == '' ? nil : text)
+              else
+                @current[tag] = (text == '' ? nil : text)
+              end
+            end
+        end
+      end
+
+      def result
+        @volumes
+      end
+    end
+  end
+end