internuity-awsum 0.2 → 0.3

data/Rakefile

@@ -54,7 +54,8 @@ spec = Gem::Specification.new do |s|
   s.files             = FileList["README*",
                                  "LICENSE",
                                  "Rakefile",
-                                 "{lib,test}/**/*"].to_a
+                                 "{lib,test}/**/*",
+                                 "{lib,test}/*"].to_a
   s.require_path      = "lib"
   s.test_files        = FileList["test/**/test_*.rb"].to_a + FileList["test/fixtures/**/*.xml"].to_a
   s.rubyforge_project = "awsum"

data/lib/awsum.rb

@@ -15,7 +15,7 @@ require 's3/s3'
 
 module Awsum
   
-  VERSION = "0.1"
+  VERSION = "0.3"
 
   API_VERSION = '2008-12-01'
   SIGNATURE_VERSION = 2

data/lib/ec2/address.rb

@@ -0,0 +1,118 @@
+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 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
+
+    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/ec2/availability_zone.rb

@@ -0,0 +1,68 @@
+module Awsum
+  class Ec2
+    class AvailabilityZone
+      attr_reader :name, :state, :availability_zone_name
+
+      def initialize(ec2, name, state, availability_zone_name) #:nodoc:
+        @ec2 = ec2
+        @name = name
+        @state = state
+        @availability_zone_name = availability_zone_name
+      end
+    end
+
+    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/ec2/keypair.rb

@@ -0,0 +1,75 @@
+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
+
+    class KeyPairParser < Awsum::Parser #:nodoc:
+      def initialize(ec2)
+        @ec2 = ec2
+        @key_pairs = []
+        @text = nil
+        @stack = []
+      end
+
+      def tag_start(tag, attributes)
+        #Quick hack so we can use the same parser for CreateKeyPair which doesn't use the item tag to wrap the key pair information
+        if tag == 'CreateKeyPairResponse'
+          @stack << 'keySet'
+        end
+
+        case tag
+          when 'keySet'
+            @stack << 'keySet'
+          when 'item', 'CreateKeyPairResponse'
+            case @stack[-1]
+              when 'keySet'
+                @current = {}
+                @text = ''
+            end
+        end
+      end
+
+      def text(text)
+        @text << text unless @text.nil?
+      end
+
+      def tag_end(tag)
+        case tag
+          when 'DescribeKeyPairsResponse'
+            #no-op
+          when 'keySet'
+            @stack.pop
+          when 'item', 'CreateKeyPairResponse'
+            case @stack[-1]
+              when 'keySet'
+                @key_pairs << KeyPair.new(
+                              @ec2,
+                              @current['keyName'], 
+                              @current['keyFingerprint'],
+                              @current['keyMaterial']
+                            )
+                @text = ''
+            end
+          else
+            unless @text.nil?
+              text = @text.strip
+              @current[tag] = (text == '' ? nil : text)
+              @text = ''
+            end
+        end
+      end
+
+      def result
+        @key_pairs
+      end
+    end
+  end
+end

data/lib/ec2/region.rb

@@ -0,0 +1,96 @@
+module Awsum
+  class Ec2
+    class Region
+      attr_reader :name, :end_point
+
+      def initialize(ec2, name, end_point) #:nodoc:
+        @ec2 = ec2
+        @name = name
+        @end_point = end_point
+      end
+
+      # List the AvailabilityZone(s) of this Region
+      def availability_zones
+        @ec2.availability_zones
+      end
+
+      # Operate all Awsum::Ec2 methods against this Region
+      #
+      # ====Example
+      #
+      #   ec2.region('eu-west-1').use do
+      #     #Runs an instance within the eu-west-1 region
+      #     instance = run_instances('i-ABCDEF')
+      #   end
+      def use(&block)
+        old_host = @ec2.host
+        @ec2.host = end_point
+        if block
+          block.arity < 1 ? instance_eval(&block) : block[self]
+        end
+        @ec2.host = old_host
+        self
+      end
+
+    private
+      #--
+      # Will pass all missing methods onto the ec2 object
+      def method_missing(method_name, *args, &block)
+        @ec2.send(method_name, *args, &block)
+      end
+    end
+
+    class RegionParser < Awsum::Parser #:nodoc:
+      def initialize(ec2)
+        @ec2 = ec2
+        @regions = []
+        @text = nil
+        @stack = []
+      end
+
+      def tag_start(tag, attributes)
+        case tag
+          when 'regionInfo'
+            @stack << 'regionInfo'
+          when 'item'
+            case @stack[-1]
+              when 'regionInfo'
+                @current = {}
+            end
+        end
+        @text = ''
+      end
+
+      def text(text)
+        @text << text unless @text.nil?
+      end
+
+      def tag_end(tag)
+        case tag
+          when 'DescribeRegionsResponse', 'requestId'
+            #no-op
+          when 'regionInfo'
+            @stack.pop
+          when 'item'
+            case @stack[-1]
+              when 'regionInfo'
+                @regions << Region.new(
+                                @ec2,
+                                @current['regionName'], 
+                                @current['regionEndpoint']
+                              )
+            end
+          else
+            unless @text.nil? || @current.nil?
+              text = @text.strip
+              @current[tag] = (text == '' ? nil : text)
+            end
+        end
+      end
+
+      def result
+        @regions
+      end
+    end
+  end
+end

data/lib/ec2/security_group.rb

@@ -0,0 +1,175 @@
+module Awsum
+  class Ec2
+    class SecurityGroup
+      attr_reader :name, :description, :owner_id, :ip_permissions, :group_permissions
+
+      def initialize(ec2, name, description, owner_id, ip_permissions, group_permissions)
+        @ec2 = ec2
+        @name = name
+        @description = description
+        @owner_id = owner_id
+        @ip_permissions = ip_permissions
+        @group_permissions = group_permissions
+      end
+
+      # Authorize access for a group
+      def authorize_group(group_name, owner_id)
+        @ec2.authorize_security_group_ingress(@name, :source_security_group_name => group_name, :source_security_group_owner_id => owner_id)
+      end
+
+      # Revoke access for a group
+      def revoke_group(group_name, owner_id)
+        @ec2.revoke_security_group_ingress(@name, :source_security_group_name => group_name, :source_security_group_owner_id => owner_id)
+      end
+
+      # Authorize access for an ip address
+      def authorize_ip(from_port, to_port, protocol = 'tcp', cidr_ip = '0.0.0.0/0')
+        @ec2.authorize_security_group_ingress(@name, :ip_protocol => protocol, :from_port => from_port, :to_port => to_port, :cidr_ip => cidr_ip)
+      end
+
+      # Revoke access from an ip address
+      def revoke_ip(from_port, to_port, protocol = 'tcp', cidr_ip = '0.0.0.0/0')
+        @ec2.revoke_security_group_ingress(@name, :ip_protocol => protocol, :from_port => from_port, :to_port => to_port, :cidr_ip => cidr_ip)
+      end
+
+      # Delete this SecurityGroup
+      def delete
+        @ec2.delete_security_group(@name)
+      end
+
+    private
+      class Permission #:nodoc:
+        attr_reader :protocol, :from_port, :to_port
+
+        def initialize(protocol, from_port, to_port)
+          @protocol = protocol
+          @from_port = from_port.to_i
+          @to_port = to_port.to_i
+        end
+      end
+
+    protected
+      class IpPermission < Permission
+        attr_reader :ip
+
+        def initialize(protocol, from_port, to_port, ip)
+          super(protocol, from_port, to_port)
+          @ip = ip
+        end
+      end
+
+      class GroupPermission < Permission
+        attr_reader :group, :user_id
+
+        def initialize(protocol, from_port, to_port, group, user_id)
+          super(protocol, from_port, to_port)
+          @group = group
+          @user_id = user_id
+        end
+      end
+    end
+
+    class SecurityGroupParser < Awsum::Parser #:nodoc:
+      def initialize(ec2)
+        @ec2 = ec2
+        @security_groups = []
+        @text = nil
+        @stack = []
+        @group_permissions = []
+        @ip_permissions = []
+      end
+
+      def tag_start(tag, attributes)
+        case tag
+          when 'securityGroupInfo'
+            @stack << 'securityGroupInfo'
+          when 'ipPermissions'
+            @stack << 'ipPermissions'
+          when 'groups'
+            @stack << 'groups'
+          when 'ipRanges'
+            @stack << 'ipRanges'
+          when 'item'
+            case @stack[-1]
+              when 'securityGroupInfo'
+                @current = {}
+                @group_permissions = []
+                @ip_permissions = []
+                @text = ''
+              when 'ipPermissions'
+                @permissions = {}
+                @text = ''
+              when 'groups'
+                @group_info = {}
+                @text = ''
+              when 'ipRanges'
+                @ip_info = {}
+                @text = ''
+            end
+        end
+      end
+
+      def text(text)
+        @text << text unless @text.nil?
+      end
+
+      def tag_end(tag)
+        case tag
+          when 'DescribeSecurityGroupsResponse', 'requestId'
+            #no-op
+          when 'securityGroupInfo', 'ipPermissions', 'groups', 'ipRanges'
+            @stack.pop
+          when 'item'
+            case @stack[-1]
+              when 'securityGroupInfo'
+                @security_groups << SecurityGroup.new(
+                              @ec2,
+                              @current['groupName'], 
+                              @current['groupDescription'], 
+                              @current['ownerId'],
+                              @ip_permissions,
+                              @group_permissions
+                            )
+                @text = ''
+              when 'groups'
+                @group_permissions << SecurityGroup::GroupPermission.new(
+                              @permissions['ipProtocol'],
+                              @permissions['fromPort'],
+                              @permissions['toPort'],
+                              @group_info['groupName'],
+                              @group_info['userId']
+                            )
+              when 'ipRanges'
+                @ip_permissions << SecurityGroup::IpPermission.new(
+                              @permissions['ipProtocol'],
+                              @permissions['fromPort'],
+                              @permissions['toPort'],
+                              @ip_info['cidrIp']
+                            )
+            end
+          else
+            unless @text.nil?
+              text = @text.strip
+              text = (text == '' ? nil : text)
+
+              case @stack[-1]
+                when 'securityGroupInfo'
+                  @current[tag] = text
+                when 'ipPermissions'
+                  @permissions[tag] = text
+                when 'groups'
+                  @group_info[tag] = text
+                when 'ipRanges'
+                  @ip_info[tag] = text
+              end
+              @text = ''
+            end
+        end
+      end
+
+      def result
+        @security_groups
+      end
+    end
+  end
+end