ktheory-right_aws 2.0.1

data/lib/ec2/right_ec2_vpc.rb

@@ -0,0 +1,571 @@
+#
+# Copyright (c) 2009 RightScale Inc
+#
+# Permission is hereby granted, free of charge, to any person obtaining
+# a copy of this software and associated documentation files (the
+# "Software"), to deal in the Software without restriction, including
+# without limitation the rights to use, copy, modify, merge, publish,
+# distribute, sublicense, and/or sell copies of the Software, and to
+# permit persons to whom the Software is furnished to do so, subject to
+# the following conditions:
+#
+# The above copyright notice and this permission notice shall be
+# included in all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+#
+
+module RightAws
+
+  class Ec2
+
+  private
+
+    def vpc__split_list_and_filters(*params) # :nodoc:
+      params = params.flatten
+      filters = params.last.is_a?(Hash) ? params.pop : {}
+      # Make values to be arrays.
+      filters.each{|key, values| filters[key] = Array(values) }
+      [params, filters]
+    end
+
+  public
+
+    #-----------------
+    # VPC
+    #-----------------
+
+    # Describe VPCs
+    #
+    #  ec2.describe_vpcs #=>
+    #    [{:vpc_id=>"vpc-890ce2e0",
+    #      :dhcp_options_id=>"default",
+    #      :cidr_block=>"10.0.0.0/23",
+    #      :state=>"available"}]
+    #
+    #  ec2.describe_vpcs("vpc-890ce2e0")
+    #
+    def describe_vpcs(*list_and_filters)
+      list, filters = vpc__split_list_and_filters(list_and_filters)
+      cache_for = (list.empty? && filters.empty?) ? :describe_vpcs : nil
+      request_hash = {}
+      request_hash.merge!(amazonize_list('VpcId', list))
+      request_hash.merge!(amazonize_list(['Filter.?.Key','Filter.?.Value.?'], filters))
+      link = generate_request("DescribeVpcs", request_hash)
+      request_cache_or_info cache_for, link,  QEc2DescribeVpcsParser, @@bench, cache_for
+    rescue Exception
+      on_exception
+    end
+
+    # Create VPC.
+    #
+    #  ec2.create_vpc('10.0.0.0/23') #=>
+    #    {:vpc_id=>"vpc-890ce2e0",
+    #     :dhcp_options_id=>"default",
+    #     :cidr_block=>"10.0.0.0/23",
+    #     :state=>"pending"}
+    #
+    def create_vpc(cidr_block)
+      link = generate_request("CreateVpc",'CidrBlock' => cidr_block )
+      request_info(link, QEc2DescribeVpcsParser.new(:logger => @logger)).first
+    rescue Exception
+      on_exception
+    end
+
+    # Delete VPC.
+    #
+    #  ec2.delete_vpc("vpc-890ce2e0") #=> true
+    #
+    def delete_vpc(vpc_id)
+      link = generate_request("DeleteVpc", 'VpcId' => vpc_id )
+      request_info(link, RightHttp2xxParser.new(:logger => @logger))
+    rescue Exception
+      on_exception
+    end
+
+    #-----------------
+    # Subnets
+    #-----------------
+
+    # Describe Subnet.
+    #
+    #  ec2.describe_subnets #=>
+    #    [{:available_ip_address_count=>"251",
+    #      :vpc_id=>"vpc-890ce2e0",
+    #      :availability_zone=>"us-east-1a",
+    #      :subnet_id=>"subnet-770de31e",
+    #      :cidr_block=>"10.0.1.0/24",
+    #      :state=>"available"}]
+    #
+    def describe_subnets(*list_and_filters)
+      list, filters = vpc__split_list_and_filters(list_and_filters)
+      cache_for = (list.empty? && filters.empty?) ? :describe_subnets : nil
+      request_hash = {}
+      request_hash.merge!(amazonize_list('SubnetId', list))
+      request_hash.merge!(amazonize_list(['Filter.?.Key','Filter.?.Value.?'], filters))
+      link = generate_request("DescribeSubnets", request_hash)
+      request_cache_or_info cache_for, link,  QEc2DescribeSubnetsParser, @@bench, cache_for
+    rescue Exception
+      on_exception
+    end
+
+    # Create Subnet.
+    #
+    #  ec2.create_subnet("vpc-890ce2e0",'10.0.1.0/24') #=>
+    #    {:available_ip_address_count=>"251",
+    #     :vpc_id=>"vpc-890ce2e0",
+    #     :availability_zone=>"us-east-1a",
+    #     :subnet_id=>"subnet-770de31e",
+    #     :cidr_block=>"10.0.1.0/24",
+    #     :state=>"pending"}
+    #
+    def create_subnet(vpc_id, cidr_block, availability_zone = nil)
+      request_hash = { 'VpcId'     => vpc_id,
+                       'CidrBlock' => cidr_block }
+      request_hash['AvailabilityZone'] = availability_zone unless availability_zone.blank?
+      link = generate_request("CreateSubnet", request_hash)
+      request_info(link, QEc2DescribeSubnetsParser.new(:logger => @logger)).first
+    rescue Exception
+      on_exception
+    end
+
+    # Delete Subnet.
+    #
+    #  ec2.delete_subnet("subnet-770de31e") #=> true
+    #
+    def delete_subnet(subnet_id)
+      link = generate_request("DeleteSubnet", 'SubnetId' => subnet_id )
+      request_info(link, RightHttp2xxParser.new(:logger => @logger))
+    rescue Exception
+      on_exception
+    end
+
+    #-----------------
+    # DHCP Options
+    #-----------------
+
+    # Describe DHCP options.
+    # 
+    # ec2.describe_dhcp_options #=>
+    #    [{:dhcp_options_id=>"dopt-cb0de3a2",
+    #    :dhcp_configuration_set=>
+    #     {"netbios-node-type"=>["1"], "domain-name"=>["my.awesomesite.ru"]}}]
+    #
+    def describe_dhcp_options(*list)
+      list = list.flatten
+      cache_for = list.empty? ? :describe_dhcp_options : nil
+      request_hash = amazonize_list('DhcpOptionsId', list)
+      link = generate_request("DescribeDhcpOptions", request_hash)
+      request_cache_or_info cache_for, link,  QEc2DescribeDhcpOptionsParser, @@bench, cache_for
+    rescue Exception
+      on_exception
+    end
+
+    # Create DHCP options.
+    #
+    #  ec2.create_dhcp_options('domain-name' => 'my.awesomesite.ru',
+    #                          'netbios-node-type' => 1) #=>
+    #    {:dhcp_options_id=>"dopt-cb0de3a2",
+    #     :dhcp_configuration_set=>
+    #      {"netbios-node-type"=>["1"], "domain-name"=>["my.awesomesite.ru"]}}
+    #
+    def create_dhcp_options(dhcp_configuration)
+      dhcp_configuration.each{ |key, values| dhcp_configuration[key] = Array(values) }
+      request_hash = amazonize_list(['DhcpConfiguration.?.Key','DhcpConfiguration.?.Value.?'], dhcp_configuration)
+      link = generate_request("CreateDhcpOptions", request_hash)
+      request_info(link, QEc2DescribeDhcpOptionsParser.new(:logger => @logger)).first
+    rescue Exception
+      on_exception
+    end
+
+    # Associate DHCP options
+    #
+    # ec2.associate_dhcp_options("dopt-cb0de3a2", "vpc-890ce2e0" ) #=> true
+    # ec2.describe_vpcs #=>
+    #    [{:vpc_id=>"vpc-890ce2e0",
+    #      :dhcp_options_id=>"dopt-cb0de3a2",
+    #      :cidr_block=>"10.0.0.0/23",
+    #      :state=>"available"}]
+    #
+    def associate_dhcp_options(dhcp_options_id, vpc_id)
+      link = generate_request("AssociateDhcpOptions", 'DhcpOptionsId' => dhcp_options_id,
+                                                      'VpcId'         => vpc_id)
+      request_info(link, RightHttp2xxParser.new(:logger => @logger))
+    rescue Exception
+      on_exception
+    end
+
+    # Delete DHCP Options.
+    #
+    #  ec2.delete_dhcp_options("dopt-cb0de3a2") #=> true
+    #
+    def delete_dhcp_options(dhcp_options_id)
+      link = generate_request("DeleteDhcpOptions", 'DhcpOptionsId' => dhcp_options_id )
+      request_info(link, RightHttp2xxParser.new(:logger => @logger))
+    rescue Exception
+      on_exception
+    end
+
+    #-----------------
+    # Customer Gateways
+    #-----------------
+
+    # Describe customer gateways.
+    #
+    #  ec2.describe_customer_gateways
+    #
+    #    [{:type=>"ipsec.1",
+    #      :ip_address=>"12.1.2.3",
+    #      :bgp_asn=>"65534",
+    #      :state=>"available",
+    #      :customer_gateway_id=>"cgw-d5a643bc"}]
+    #
+    def describe_customer_gateways(*list_and_filters)
+      list, filters = vpc__split_list_and_filters(list_and_filters)
+      cache_for = (list.empty? && filters.empty?) ? :describe_customer_gateways : nil
+      request_hash = {}
+      request_hash.merge!(amazonize_list('CustomerGatewayId', list))
+      request_hash.merge!(amazonize_list(['Filter.?.Key','Filter.?.Value.?'], filters))
+      link = generate_request("DescribeCustomerGateways", request_hash)
+      request_cache_or_info cache_for, link,  QEc2DescribeCustomerGatewaysParser, @@bench, cache_for
+    rescue Exception
+      on_exception
+    end
+
+    # Create customer gateway.
+    # 
+    #  ec2.create_customer_gateway('ipsec.1', '12.1.2.3', 65534) #=>
+    #    {:type=>"ipsec.1",
+    #     :bgp_asn=>"65534",
+    #     :ip_address=>"12.1.2.3",
+    #     :state=>"pending",
+    #     :customer_gateway_id=>"cgw-d5a643bc"}
+    #
+    def create_customer_gateway(type, ip_address, bgp_asn)
+      link = generate_request("CreateCustomerGateway", 'Type'      => type,
+                                                       'IpAddress' => ip_address,
+                                                       'BgpAsn'    => bgp_asn )
+      request_info(link, QEc2DescribeCustomerGatewaysParser.new(:logger => @logger)).first
+    rescue Exception
+      on_exception
+    end
+
+    # Delete customer gateway.
+    #
+    #  ec2.delete_customer_gateway("cgw-d5a643bc") #=> true
+    #
+    def delete_customer_gateway(customer_gateway_id)
+      link = generate_request("DeleteCustomerGateway", 'CustomerGatewayId' => customer_gateway_id )
+      request_info(link, RightHttp2xxParser.new(:logger => @logger))
+    rescue Exception
+      on_exception
+    end
+
+    #-----------------
+    # VPN Gateways
+    #-----------------
+
+    # Describe VPN gateways.
+    #
+    #  ec2.describe_vpn_gateways #=>
+    #    [{:type=>"ipsec.1",
+    #      :availability_zone=>"us-east-1a",
+    #      :attachments=>[{:vpc_id=>"vpc-890ce2e0", :state=>"attached"}],
+    #      :vpn_gateway_id=>"vgw-dfa144b6"}]
+    #
+    def describe_vpn_gateways(*list_and_filters)
+      list, filters = vpc__split_list_and_filters(list_and_filters)
+      cache_for = (list.empty? && filters.empty?) ? :describe_vpn_gateways : nil
+      request_hash = {}
+      request_hash.merge!(amazonize_list('VpnGatewayId', list))
+      request_hash.merge!(amazonize_list(['Filter.?.Key','Filter.?.Value.?'], filters))
+      link = generate_request("DescribeVpnGateways", request_hash)
+      request_cache_or_info cache_for, link,  QEc2DescribeVpnGatewaysParser, @@bench, cache_for
+    rescue Exception
+      on_exception
+    end
+
+    # Create VPN gateway.
+    #
+    #  ec2.create_vpn_gateway('ipsec.1') #=>
+    #    {:type=>"ipsec.1",
+    #     :availability_zone=>"us-east-1a",
+    #     :attachments=>[nil],
+    #     :vpn_gateway_id=>"vgw-dfa144b6"}
+    #
+    def create_vpn_gateway(type, availability_zone=nil)
+      request_hash = { 'Type' => type }
+      request_hash['AvailabilityZone'] = availability_zone unless availability_zone.blank?
+      link = generate_request("CreateVpnGateway", request_hash )
+      request_info(link, QEc2DescribeVpnGatewaysParser.new(:logger => @logger)).first
+    rescue Exception
+      on_exception
+    end
+
+    # Attach VPN gateway.
+    #
+    #  ec2.attach_vpn_gateway('vgw-dfa144b6','vpc-890ce2e0') #=>
+    #     {:vpc_id=>"vpc-890ce2e0", :state=>"attaching"}
+    #
+    def attach_vpn_gateway(vpn_gateway_id, vpc_id)
+      link = generate_request("AttachVpnGateway", 'VpnGatewayId' => vpn_gateway_id,
+                                                  'VpcId'        => vpc_id )
+      request_info(link, QEc2AttachVpnGatewayParser.new(:logger => @logger))
+    rescue Exception
+      on_exception
+    end
+
+    # Detach VPN gateway.
+    #
+    #  ec2.detach_vpn_gateway('vgw-dfa144b6','vpc-890ce2e0') #=> true
+    #
+    def detach_vpn_gateway(vpn_gateway_id, vpc_id)
+      link = generate_request("DetachVpnGateway", 'VpnGatewayId' => vpn_gateway_id,
+                                                  'VpcId'        => vpc_id )
+      request_info(link, RightHttp2xxParser.new(:logger => @logger))
+    rescue Exception
+      on_exception
+    end
+
+    # Delete vpn gateway.
+    #
+    #  ec2.delete_vpn_gateway("vgw-dfa144b6") #=> true
+    #
+    def delete_vpn_gateway(vpn_gateway_id)
+      link = generate_request("DeleteVpnGateway", 'VpnGatewayId' => vpn_gateway_id )
+      request_info(link, RightHttp2xxParser.new(:logger => @logger))
+    rescue Exception
+      on_exception
+    end
+
+    #-----------------
+    # VPN Connections
+    #-----------------
+
+    # Describe VPN connections.
+    #
+    #  ec2.describe_vpn_connections #=>
+    #    [{:type=>"ipsec.1",
+    #      :vpn_connection_id=>"vpn-a9a643c0",
+    #      :customer_gateway_configuration=>
+    #       "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n<vpn_connection id=\"vpn-a9a643c0\">\n...</vpn_connection>\n",
+    #      :state=>"available",
+    #      :vpn_gateway_id=>"vgw-dfa144b6",
+    #      :customer_gateway_id=>"cgw-81a643e8"}]
+    #
+    def describe_vpn_connections(*list_and_filters)
+      list, filters = vpc__split_list_and_filters(list_and_filters)
+      cache_for = (list.empty? && filters.empty?) ? :describe_vpn_connections : nil
+      request_hash = {}
+      request_hash.merge!(amazonize_list('VpnConnectionId', list))
+      request_hash.merge!(amazonize_list(['Filter.?.Key','Filter.?.Value.?'], filters))
+      link = generate_request("DescribeVpnConnections", request_hash)
+      request_cache_or_info cache_for, link,  QEc2DescribeVpnConnectionsParser, @@bench, cache_for
+    rescue Exception
+      on_exception
+    end
+
+    # Create VPN connection.
+    #
+    #  ec2.create_vpn_connection('ipsec.1', 'cgw-81a643e8' ,'vgw-dfa144b6')
+    #    {:customer_gateway_id=>"cgw-81a643e8",
+    #     :vpn_connection_id=>"vpn-a9a643c0",
+    #     :customer_gateway_configuration=>
+    #      "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n<vpn_connection id=\"vpn-a9a643c0\">\n...</vpn_connection>\n",
+    #     :state=>"pending",
+    #     :vpn_gateway_id=>"vgw-dfa144b6"}
+    #
+    def create_vpn_connection(type, customer_gateway_id, vpn_gateway_id)
+      link = generate_request("CreateVpnConnection", 'Type'              => type,
+                                                     'CustomerGatewayId' => customer_gateway_id,
+                                                     'VpnGatewayId'      => vpn_gateway_id )
+      request_info(link, QEc2DescribeVpnConnectionsParser.new(:logger => @logger)).first
+    rescue Exception
+      on_exception
+    end
+
+    # Delete VPN connection.
+    #
+    #  ec2.delete_vpn_connection("vpn-a9a643c0") #=> true
+    #
+    def delete_vpn_connection(vpn_connection_id)
+      link = generate_request("DeleteVpnConnection", 'VpnConnectionId' => vpn_connection_id )
+      request_info(link, RightHttp2xxParser.new(:logger => @logger))
+    rescue Exception
+      on_exception
+    end
+
+    #-----------------
+    # Parsers
+    #-----------------
+
+    class QEc2DescribeVpcsParser < RightAWSParser #:nodoc:
+      def tagstart(name, attributes)
+        case name
+        when 'item', 'vpc' then @item = {}
+        end
+      end
+      def tagend(name)
+        case name
+          when 'vpcId'         then @item[:vpc_id] = @text
+          when 'state'         then @item[:state] = @text
+          when 'dhcpOptionsId' then @item[:dhcp_options_id] = @text
+          when 'cidrBlock'     then @item[:cidr_block] = @text
+          when 'item', 'vpc'   then @result << @item
+        end
+      end
+      def reset
+        @result = []
+      end
+    end
+
+    class QEc2DescribeSubnetsParser < RightAWSParser #:nodoc:
+      def tagstart(name, attributes)
+        case name
+        when 'item', 'subnet' then @item = {}
+        end
+      end
+      def tagend(name)
+        case name
+          when 'subnetId'                then @item[:subnet_id] = @text
+          when 'state'                   then @item[:state] = @text
+          when 'vpcId'                   then @item[:vpc_id] = @text
+          when 'cidrBlock'               then @item[:cidr_block] = @text
+          when 'availabilityZone'        then @item[:availability_zone] = @text
+          when 'availableIpAddressCount' then @item[:available_ip_address_count] = @text
+          when 'item', 'subnet'          then @result << @item
+        end
+      end
+      def reset
+        @result = []
+      end
+    end
+
+    class QEc2DescribeDhcpOptionsParser < RightAWSParser #:nodoc:
+      def tagstart(name, attributes)
+        case full_tag_name
+        when @p1, @p2
+          @item = { :dhcp_configuration_set => {} }
+        end
+      end
+      def tagend(name)
+        case name
+        when 'dhcpOptionsId' then @item[:dhcp_options_id] = @text
+        when 'key'           then @conf_item_key = @text
+        when 'value'         then (@item[:dhcp_configuration_set][@conf_item_key] ||= []) << @text
+        end
+        case full_tag_name
+        when @p1, @p2
+          @result << @item
+        end
+      end
+      def reset
+        @p1 = 'DescribeDhcpOptionsResponse/dhcpOptionsSet/item'
+        @p2 = 'CreateDhcpOptionsResponse/dhcpOptions'
+        @result = []
+      end
+    end
+
+    class QEc2DescribeCustomerGatewaysParser < RightAWSParser #:nodoc:
+      def tagstart(name, attributes)
+        case name
+        when 'item', 'customerGateway'
+          @item = {}
+        end
+      end
+      def tagend(name)
+        case name
+        when 'customerGatewayId' then @item[:customer_gateway_id] = @text
+        when 'state'             then @item[:state] = @text
+        when 'type'              then @item[:type] = @text
+        when 'ipAddress'         then @item[:ip_address] = @text
+        when 'bgpAsn'            then @item[:bgp_asn] = @text
+        when 'item', 'customerGateway' then @result << @item
+        end
+      end
+      def reset
+        @result = []
+      end
+    end
+
+    class QEc2DescribeVpnGatewaysParser < RightAWSParser #:nodoc:
+      def tagstart(name, attributes)
+        case full_tag_name
+        when @p1, @p2
+          @item = { :attachments => [] }
+        when "#{@p1}/attachments/item",
+             "#{@p2}/attachments/item"
+          @attachment = {}
+        end
+      end
+      def tagend(name)
+        case name
+        when 'vpnGatewayId'     then @item[:vpn_gateway_id] = @text
+        when 'availabilityZone' then @item[:availability_zone] = @text
+        when 'type'             then @item[:type] = @text
+        when 'vpcId'            then @attachment[:vpc_id] = @text
+        end
+        case full_tag_name
+        when "#{@p1}/state",
+             "#{@p2}/state"
+          @item[:state] = @text
+        when "#{@p1}/attachments/item/state",
+             "#{@p2}/attachments/item/state"
+          @attachment[:state] = @text
+        when "#{@p1}/attachments/item",
+             "#{@p2}/attachments/item"
+          @item[:attachments] << @attachment unless @attachment.blank?
+        when @p1, @p2
+          @result << @item
+        end
+      end
+      def reset
+        @p1 = 'DescribeVpnGatewaysResponse/vpnGatewaySet/item'
+        @p2 = 'CreateVpnGatewayResponse/vpnGateway'
+        @result = []
+      end
+    end
+
+    class QEc2AttachVpnGatewayParser < RightAWSParser #:nodoc:
+      def tagend(name)
+        case name
+        when 'vpcId' then @result[:vpc_id] = @text
+        when 'state' then @result[:state]  = @text
+        end
+      end
+      def reset
+        @result = {}
+      end
+    end
+
+
+    class QEc2DescribeVpnConnectionsParser < RightAWSParser #:nodoc:
+      def tagstart(name, attributes)
+        case name
+        when 'item', 'vpnConnection' then @item = {}
+        end
+      end
+      def tagend(name)
+        case name
+        when 'vpnConnectionId'   then @item[:vpn_connection_id] = @text
+        when 'state'             then @item[:state] = @text
+        when 'type'              then @item[:type] = @text
+        when 'vpnGatewayId'      then @item[:vpn_gateway_id] = @text
+        when 'customerGatewayId' then @item[:customer_gateway_id] = @text
+        when 'customerGatewayConfiguration' then @item[:customer_gateway_configuration] = @text
+        when 'item','vpnConnection'         then @result << @item
+        end
+      end
+      def reset
+        @result = []
+      end
+    end
+
+  end
+end