Adding initial Elastic Load Balancer support

lib/fog/aws.rb

@@ -1,4 +1,5 @@
 require 'fog/aws/ec2.rb'
+require 'fog/aws/elb.rb'
 require 'fog/aws/s3'
 require 'fog/aws/simpledb'
 

lib/fog/aws/bin.rb

@@ -15,6 +15,8 @@ module AWS
           hash[key] = case key
           when :ec2
             Fog::AWS::EC2.new(credentials)
+          when :elb
+            Fog::AWS::ELB.new(credentials)
           when :simpledb
             Fog::AWS::SimpleDB.new(credentials)
           when :s3

lib/fog/aws/elb.rb

@@ -0,0 +1,105 @@
+module Fog
+  module AWS
+    module ELB
+
+      def self.new(options={})
+
+        unless @required
+          require 'fog/aws/requests/elb/describe_load_balancers'
+          require 'fog/aws/parsers/elb/describe_load_balancers'
+          @required = true
+        end
+
+        unless options[:aws_access_key_id]
+          raise ArgumentError.new('aws_access_key_id is required to access elb')
+        end
+        unless options[:aws_secret_access_key]
+          raise ArgumentError.new('aws_secret_access_key is required to access elb')
+        end
+        Fog::AWS::ELB::Real.new(options)
+      end
+
+      class Real
+
+        # Initialize connection to ELB
+        #
+        # ==== Notes
+        # options parameter must include values for :aws_access_key_id and 
+        # :aws_secret_access_key in order to create a connection
+        #
+        # ==== Examples
+        #   elb = ELB.new(
+        #    :aws_access_key_id => your_aws_access_key_id,
+        #    :aws_secret_access_key => your_aws_secret_access_key
+        #   )
+        #
+        # ==== Parameters
+        # * options<~Hash> - config arguments for connection.  Defaults to {}.
+        #   * region<~String> - optional region to use, in ['eu-west-1', 'us-east-1', 'us-west-1'i, 'ap-southeast-1']
+        #
+        # ==== Returns
+        # * ELB object with connection to AWS.
+        def initialize(options={})
+          @aws_access_key_id      = options[:aws_access_key_id]
+          @aws_secret_access_key  = options[:aws_secret_access_key]
+          @hmac = HMAC::SHA256.new(@aws_secret_access_key)
+          @host = options[:host] || case options[:region]
+          when 'eu-west-1'
+            'elasticloadbalancing.eu-west-1.amazonaws.com'
+          when 'us-east-1'
+            'elasticloadbalancing.us-east-1.amazonaws.com'
+          when 'us-west-1'
+            'elasticloadbalancing.us-west-1.amazonaws.com'
+          when 'ap-southeast-1'
+            'elasticloadbalancing.ap-southeast-1.amazonaws.com'
+          else
+            'elasticloadbalancing.amazonaws.com'
+          end
+          @port       = options[:port]      || 443
+          @scheme     = options[:scheme]    || 'https'
+        end
+
+        private
+
+        def request(params)
+          @connection = Fog::Connection.new("#{@scheme}://#{@host}:#{@port}")
+
+          idempotent  = params.delete(:idempotent)
+          parser      = params.delete(:parser)
+
+          params.merge!({
+            'AWSAccessKeyId' => @aws_access_key_id,
+            'SignatureMethod' => 'HmacSHA256',
+            'SignatureVersion' => '2',
+            'Timestamp' => Time.now.utc.strftime("%Y-%m-%dT%H:%M:%SZ"),
+            'Version' => '2009-11-25'
+          })
+
+          body = ''
+          for key in params.keys.sort
+            unless (value = params[key]).nil?
+              body << "#{key}=#{CGI.escape(value.to_s).gsub(/\+/, '%20')}&"
+            end
+          end
+
+          string_to_sign = "POST\n#{@host}\n/\n" << body.chop
+          hmac = @hmac.update(string_to_sign)
+          body << "Signature=#{CGI.escape(Base64.encode64(hmac.digest).chomp!).gsub(/\+/, '%20')}"
+
+          response = @connection.request({
+            :body       => body,
+            :expects    => 200,
+            :headers    => { 'Content-Type' => 'application/x-www-form-urlencoded' },
+            :idempotent => idempotent,
+            :host       => @host,
+            :method     => 'POST',
+            :parser     => parser
+          })
+
+          response
+        end
+
+      end
+    end
+  end
+end

lib/fog/aws/parsers/elb/describe_load_balancers.rb

@@ -0,0 +1,105 @@
+module Fog
+  module Parsers
+    module AWS
+      module ELB
+
+        class DescribeLoadBalancers < Fog::Parsers::Base
+
+          def reset
+            @load_balancer = { 'ListenerDescriptions' => [], 'Instances' => [], 'AvailabilityZones' => [], 'Policies' => {'AppCookieStickinessPolicies' => [], 'LBCookieStickinessPolicies' => [] }, 'HealthCheck' => {} }
+            @listener = { }
+            @listener_descriptions = []
+            @listener_description = { 'PolicyNames' => [], 'Listener' => {} }
+            @results = { 'LoadBalancerDescriptions' => [] }
+            @response = { 'DescribeLoadBalancersResult' => {}, 'ResponseMetadata' => {} }
+          end
+
+          def start_element(name, attrs = [])
+            case name
+            when 'ListenerDescriptions'
+              @in_listeners = true
+            when 'Instances'
+              @in_instances = true
+            when 'AvailabilityZones'
+              @in_availability_zones = true
+            when 'PolicyNames'
+              @in_policy_names = true
+            when 'Policies'
+              @in_policies = true
+            when 'LBCookieStickinessPolicies'
+              @in_lb_cookies = true
+            when 'AppCookieStickinessPolicies'
+              @in_app_cookies = true
+            end
+            @value = ''
+          end
+
+          def end_element(name)
+            case name
+            when 'member'
+              if @in_policy_names
+                @listener_description['PolicyNames'] << @value
+              elsif @in_availability_zones
+                @load_balancer['AvailabilityZones'] << @value
+              elsif @in_listeners
+                @listener_description['Listener'] = @listener
+                @listener_descriptions << @listener_description
+                @load_balancer['ListenerDescriptions'] = @listener_descriptions
+                @listener_description = { 'PolicyNames' => [], 'Listener' => {} }
+              elsif @in_app_cookies
+                @load_balancer['Policies']['AppCookieStickinessPolicies'] << @value
+              elsif @in_lb_cookies
+                @load_balancer['Policies']['LBCookieStickinessPolicies'] << @value
+              elsif !@in_instances && !@in_policies
+                @results['LoadBalancerDescriptions'] << @load_balancer
+                @load_balancer = { 'ListenerDescriptions' => [], 'Instances' => [], 'AvailabilityZones' => [], 'Policies' => {'AppCookieStickinessPolicies' => [], 'LBCookieStickinessPolicies' => [] }, 'HealthCheck' => {} }
+              end
+
+            when 'LoadBalancerName', 'DNSName'
+              @load_balancer[name] = @value
+            when 'CreatedTime'
+              @load_balancer[name] = Time.parse(@value)
+
+            when 'ListenerDescriptions'
+              @in_listeners = false
+            when 'PolicyNames'
+              @in_policy_names = false
+            when 'Protocol'
+              @listener[name] = @value
+            when 'LoadBalancerPort', 'InstancePort'
+              @listener[name] = @value.to_i
+
+            when 'Instances'
+              @in_instances = false
+            when 'InstanceId'
+              @load_balancer['Instances'] << @value
+
+            when 'AvailabilityZones'
+              @in_availability_zones = false
+
+            when 'Policies'
+              @in_policies = false
+            when 'AppCookieStickinessPolicies'
+              @in_app_cookies = false
+            when 'LBCookieStickinessPolicies'
+              @in_lb_cookies = false
+
+            when 'Interval', 'HealthyThreshold', 'Timeout', 'UnhealthyThreshold'
+              @load_balancer['HealthCheck'][name] = @value.to_i
+            when 'Target'
+              @load_balancer['HealthCheck'][name] = @value
+
+            when 'RequestId'
+              @response['ResponseMetadata'][name] = @value
+
+            when 'DescribeLoadBalancersResponse'
+              @response['DescribeLoadBalancersResult'] = @results
+            end
+          end
+
+        end
+
+      end
+    end
+  end
+end

lib/fog/aws/requests/elb/describe_load_balancers.rb

@@ -0,0 +1,51 @@
+module Fog
+  module AWS
+    module ELB
+      class Real
+
+        # Describe all or specified load balancers
+        #
+        # ==== Parameters
+        # * lb_name<~Array> - List of load balancer names to describe, defaults to all
+        #
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~Hash>:
+        #     * 'ResponseMetadata'<~Hash>:
+        #       * 'RequestId'<~String> - Id of request
+        #     * 'DescribeLoadBalancersResult'<~Hash>:
+        #       * 'LoadBalancerDescriptions'<~Array>
+        #         * 'LoadBalancerName'<~String> - name of load balancer
+        #         * 'DNSName'<~String> - external DNS name of load balancer
+        #         * 'CreatedTime'<~Time> - time load balancer was created
+        #         * 'ListenerDescriptions'<~Array>
+        #           * 'PolicyNames'<~Array> - list of policies enabled
+        #           * 'Listener'<~Hash>:
+        #             * 'InstancePort'<~Integer> - port on instance that requests are sent to
+        #             * 'Protocol'<~String> - transport protocol used for routing in [TCP, HTTP]
+        #             * 'LoadBalancerPort'<~Integer> - port that load balancer listens on for requests
+        #         * 'HealthCheck'<~Hash>:
+        #           * 'HealthyThreshold'<~Integer> - number of consecutive health probe successes required before moving the instance to the Healthy state
+        #           * 'Timeout'<~Integer> - number of seconds after which no response means a failed health probe
+        #           * 'Interval'<~Integer> - interval (in seconds) between health checks of an individual instance
+        #           * 'UnhealthyThreshold'<~Integer> - number of consecutive health probe failures that move the instance to the unhealthy state
+        #           * 'Target'<~String> - string describing protocol type, port and URL to check
+        #         * 'Policies'<~Hash>:
+        #           * 'LBCookieStickinessPolicies'<~Array> - list of Load Balancer Generated Cookie Stickiness policies for the LoadBalancer
+        #           * 'AppCookieStickinessPolicies'<~Array> - list of Application Generated Cookie Stickiness policies for the LoadBalancer
+        #         * 'AvailabilityZones'<~Array> - list of availability zones covered by this load balancer
+        #         * 'Instances'<~Array> - list of instances that the load balancer balances between
+        def describe_load_balancers(lb_name = [])
+          lb_name = [lb_name] if lb_name.instance_of?(String)
+          lb_name.unshift(nil)
+          params = AWS.indexed_param('LoadBalancerNames.member', lb_name)
+          request({
+            'Action'  => 'DescribeLoadBalancers',
+            :parser   => Fog::Parsers::AWS::ELB::DescribeLoadBalancers.new
+          }.merge!(params))
+        end
+
+      end
+    end
+  end
+end