fog 0.7.2 → 0.8.0

data/lib/fog/aws/models/elb/load_balancer.rb

@@ -0,0 +1,129 @@
+require 'fog/core/model'
+module Fog
+  module AWS
+    class ELB
+
+      class LoadBalancer < Fog::Model
+
+        identity  :id,                    :aliases => 'LoadBalancerName'
+        attribute :created_at,            :aliases => 'CreatedTime'
+        attribute :health_check,          :aliases => 'HealthCheck'
+        attribute :dns_name,              :aliases => 'DNSName'
+        attribute :availability_zones,    :aliases => 'AvailabilityZones'
+        attribute :instances,             :aliases => 'Instances'
+
+        def initialize(attributes={})
+          attributes[:availability_zones] ||= %w(us-east-1a us-east-1b us-east-1c us-east-1d)
+          attributes['ListenerDescriptions'] ||= [{
+            'Listener' => {'LoadBalancerPort' => 80, 'InstancePort' => 80, 'Protocol' => 'http'},
+            'PolicyNames' => []
+          }]
+          attributes['Policies'] ||= {'AppCookieStickinessPolicies' => [], 'LBCookieStickinessPolicies' => []}
+          super
+        end
+
+        def register_instances(instances)
+          requires :id
+          data = connection.register_instances_with_load_balancer(instances, id).body['RegisterInstancesWithLoadBalancerResult']
+          data['Instances'].map!{|h| h['InstanceId']}
+          merge_attributes(data)
+        end
+
+        def deregister_instances(instances)
+          requires :id
+          data = connection.deregister_instances_from_load_balancer(instances, id).body['DeregisterInstancesFromLoadBalancerResult']
+          data['Instances'].map!{|h| h['InstanceId']}
+          merge_attributes(data)
+        end
+
+        def enable_availability_zones(zones)
+          requires :id
+          data = connection.enable_availability_zones_for_load_balancer(zones, id).body['EnableAvailabilityZonesForLoadBalancerResult']
+          merge_attributes(data)
+        end
+
+        def disable_availability_zones(zones)
+          requires :id
+          data = connection.disable_availability_zones_for_load_balancer(zones, id).body['DisableAvailabilityZonesForLoadBalancerResult']
+          merge_attributes(data)
+        end
+
+        def instance_health
+          requires :id
+          @instance_health ||= connection.describe_instance_health(id).body['DescribeInstanceHealthResult']['InstanceStates']
+        end
+
+        def instances_in_service
+          instance_health.select{|hash| hash['State'] == 'InService'}.map{|hash| hash['InstanceId']}
+        end
+
+        def instances_out_of_service
+          instance_health.select{|hash| hash['State'] == 'OutOfService'}.map{|hash| hash['InstanceId']}
+        end
+
+        def configure_health_check(health_check)
+          requires :id
+          data = connection.configure_health_check(id, health_check).body['ConfigureHealthCheckResult']['HealthCheck']
+          merge_attributes(:health_check => data)
+        end
+
+        def listeners
+          Fog::AWS::ELB::Listeners.new({
+            :data => attributes['ListenerDescriptions'],
+            :connection => connection,
+            :load_balancer => self
+          })
+        end
+
+        def policies
+          Fog::AWS::ELB::Policies.new({
+            :data => attributes['Policies'],
+            :connection => connection,
+            :load_balancer => self
+          })
+        end
+
+        def set_listener_policy(port, policy_name)
+          requires :id
+          policy_name = [policy_name].flatten
+          connection.set_load_balancer_policies_of_listener(id, port, policy_name)
+          reload
+        end
+
+        def unset_listener_policy(port)
+          set_listener_policy(port, [])
+        end
+
+        def ready?
+          # ELB requests are synchronous
+          true
+        end
+
+        def save
+          requires :id
+          requires :listeners
+          requires :availability_zones
+
+          connection.create_load_balancer(availability_zones, id, listeners.map{|l| l.to_params})
+
+          # reload instead of merge attributes b/c some attrs (like HealthCheck)
+          # may be set, but only the DNS name is returned in the create_load_balance
+          # API call
+          reload
+        end
+
+        def reload
+          super
+          @instance_health = nil
+          self
+        end
+
+        def destroy
+          requires :id
+          connection.delete_load_balancer(id)
+        end
+
+      end
+    end
+  end
+end

data/lib/fog/aws/models/elb/load_balancers.rb

@@ -0,0 +1,28 @@
+require 'fog/aws/models/elb/load_balancer'
+module Fog
+  module AWS
+    class ELB
+      class LoadBalancers < Fog::Collection
+        model Fog::AWS::ELB::LoadBalancer
+
+        # Creates a new load balancer
+        def initialize(attributes={})
+          super
+        end
+
+        def all
+          data = connection.describe_load_balancers.body['DescribeLoadBalancersResult']['LoadBalancerDescriptions']
+          load(data)
+        end
+
+        def get(identity)
+          data = connection.describe_load_balancers(identity).body['DescribeLoadBalancersResult']['LoadBalancerDescriptions'].first
+          new(data)
+        rescue Fog::AWS::ELB::NotFound
+          nil
+        end
+
+      end
+    end
+  end
+end

data/lib/fog/aws/models/elb/policies.rb

@@ -0,0 +1,40 @@
+require 'fog/aws/models/elb/policy'
+module Fog
+  module AWS
+    class ELB
+      class Policies < Fog::Collection
+
+        model Fog::AWS::ELB::Policy
+
+        attr_accessor :data, :load_balancer
+
+        def all
+          load(munged_data)
+        end
+
+        def get(id)
+          all.detect{|policy| id == policy.id}
+        end
+
+        private
+        # Munge a hash like:
+        # {'LBCookieStickinessPolicies' => [policies...],
+        # 'AppCookieStickinessPolicies' => [policies...]}
+        # to a single array of policies with a cookie_stickiness value
+        def munged_data
+          munged_data = []
+          data['LBCookieStickinessPolicies'].each do |policy|
+            munged_data << policy.merge(:cookie_stickiness => :lb)
+          end
+           data['AppCookieStickinessPolicies'].each do |policy|
+            munged_data << policy.merge(:cookie_stickiness => :app)
+          end
+           munged_data
+        end
+
+      end
+
+    end
+  end
+end
+

data/lib/fog/aws/models/elb/policy.rb

@@ -0,0 +1,52 @@
+require 'fog/core/model'
+module Fog
+  module AWS
+    class ELB
+
+      class Policy < Fog::Model
+        identity :id, :aliases => 'PolicyName'
+
+        attribute :cookie,     :aliases => 'CookieName'
+        attribute :expiration, :aliases => 'CookieExpirationPeriod'
+
+        attr_accessor :cookie_stickiness # Either :app or :lb
+
+        def save
+          requires :id, :load_balancer, :cookie_stickiness
+          connection_method = nil
+          args = [load_balancer.id, id]
+          case cookie_stickiness
+          when :app
+            requires :cookie
+            method = :create_app_cookie_stickiness_policy
+            args << cookie
+          when :lb
+            method = :create_lb_cookie_stickiness_policy
+            args << expiration if expiration
+          else
+            raise ArgumentError.new('cookie_stickiness must be :app or :lb')
+          end
+
+          connection.send(method, *args)
+          reload
+        end
+
+        def destroy
+          requires :id, :load_balancer
+          connection.delete_load_balancer_policy(load_balancer.id, id)
+          reload
+        end
+
+        def reload
+          load_balancer.reload
+        end
+
+        def load_balancer
+          collection.load_balancer
+        end
+
+      end
+
+    end
+  end
+end

data/lib/fog/aws/{rds/models → models/rds}/parameter_groups.rb

@@ -1,5 +1,5 @@
 require 'fog/core/collection'
-require 'fog/aws/rds/models/parameter_group'
+require 'fog/aws/models/rds/parameter_group'
 
 module Fog
   module AWS

data/lib/fog/aws/{rds/models → models/rds}/parameters.rb

@@ -1,5 +1,5 @@
 require 'fog/core/collection'
-require 'fog/aws/rds/models/parameter'
+require 'fog/aws/models/rds/parameter'
 
 module Fog
   module AWS

data/lib/fog/aws/{rds/models → models/rds}/security_groups.rb

@@ -1,5 +1,5 @@
 require 'fog/core/collection'
-require 'fog/aws/rds/models/security_group'
+require 'fog/aws/models/rds/security_group'
 
 module Fog
   module AWS

data/lib/fog/aws/{rds/models → models/rds}/servers.rb

@@ -1,5 +1,5 @@
 require 'fog/core/collection'
-require 'fog/aws/rds/models/server'
+require 'fog/aws/models/rds/server'
 
 module Fog
   module AWS

data/lib/fog/aws/{rds/models → models/rds}/snapshots.rb

@@ -1,5 +1,5 @@
 require 'fog/core/collection'
-require 'fog/aws/rds/models/snapshot'
+require 'fog/aws/models/rds/snapshot'
 
 module Fog
   module AWS

data/lib/fog/aws/parsers/cloud_formation/create_stack.rb

@@ -8,7 +8,7 @@ module Fog
           def end_element(name)
             case name
             when 'RequestId', 'StackId'
-              @response[name] = @value
+              @response[name] = value
             end
           end
 

data/lib/fog/aws/parsers/cloud_formation/describe_stack_events.rb

@@ -13,14 +13,14 @@ module Fog
           def end_element(name)
             case name
             when 'EventId', 'LogicalResourceId', 'PhysicalResourceId', 'ResourceProperties', 'ResourceStatus', 'ResourceStatusReason', 'ResourceType', 'StackId', 'StackName'
-              @event[name] = @value
+              @event[name] = value
             when 'member'
               @response['StackEvents'] << @event
               @event = {}
             when 'RequestId'
-              @response[name] = @value
+              @response[name] = value
             when 'Timestamp'
-              @event[name] = Time.parse(@value)
+              @event[name] = Time.parse(value)
             end
           end
 

data/lib/fog/aws/parsers/cloud_formation/describe_stack_resources.rb

@@ -13,14 +13,14 @@ module Fog
           def end_element(name)
             case name
             when 'StackId', 'StackName', 'LogicalResourceId', 'PhysicalResourceId', 'ResourceType', 'ResourceStatus'
-              @resource[name] = @value
+              @resource[name] = value
             when 'member'
               @response['StackResources'] << @resource
               @resource = {}
             when 'RequestId'
-              @response[name] = @value
+              @response[name] = value
             when 'Timestamp'
-              @resource[name] = Time.parse(@value)
+              @resource[name] = Time.parse(value)
             end
           end
 

data/lib/fog/aws/parsers/cloud_formation/describe_stacks.rb

@@ -26,7 +26,7 @@ module Fog
             if @in_outputs
               case name
               when 'OutputKey', 'OutputValue'
-                @output[name] = @value
+                @output[name] = value
               when 'member'
                 @stack['Outputs'] << @output
                 @output = {}
@@ -36,7 +36,7 @@ module Fog
             elsif @in_parameters
               case name
               when 'ParameterKey', 'ParameterValue'
-                @parameter[name] = @value
+                @parameter[name] = value
               when 'member'
                 @stack['Parameters'] << @parameter
                 @parameter = {}
@@ -49,18 +49,18 @@ module Fog
                 @response['Stacks'] << @stack
                 @stack = { 'Outputs' => [], 'Parameters' => [] }
               when 'RequestId'
-                @response[name] = @value
+                @response[name] = value
               when 'CreationTime'
-                @stack[name] = Time.parse(@value)
+                @stack[name] = Time.parse(value)
               when 'DisableRollback'
-                case @value
+                case value
                 when 'false'
                   @stack[name] = false
                 when 'true'
                   @stack[name] = true
                 end
               when 'StackName', 'StackId', 'StackStatus'
-                @stack[name] = @value
+                @stack[name] = value
               end
             end
           end

data/lib/fog/aws/parsers/cloud_formation/get_template.rb

@@ -8,7 +8,7 @@ module Fog
           def end_element(name)
             case name
             when 'RequestId', 'TemplateBody'
-              @response[name] = @value
+              @response[name] = value
             end
           end
 

data/lib/fog/aws/parsers/cloud_formation/validate_template.rb

@@ -21,20 +21,20 @@ module Fog
           def end_element(name)
             case name
             when 'DefaultValue', 'ParameterKey'
-              @parameter[name] = @value
+              @parameter[name] = value
             when 'Description'
               if @in_parameters
-                @parameter[name] = @value
+                @parameter[name] = value
               else
-                @response[name] = @value
+                @response[name] = value
               end
             when 'RequestId'
-              @response[name] = @value
+              @response[name] = value
             when 'member'
               @response['Parameters'] << @parameter
               @parameter = {}
             when 'NoEcho'
-              case @value
+              case value
               when 'false'
                 @parameter[name] = false
               when 'true'

data/lib/fog/aws/parsers/elb/configure_health_check.rb

@@ -0,0 +1,37 @@
+module Fog
+  module Parsers
+    module AWS
+      module ELB
+
+        class ConfigureHealthCheck < Fog::Parsers::Base
+
+          def reset
+            @health_check = {}
+            @response = { 'ConfigureHealthCheckResult' => {}, 'ResponseMetadata' => {} }
+          end
+
+          def start_element(name, attrs = [])
+            super
+          end
+
+          def end_element(name)
+            case name
+            when 'Target'
+              @health_check[name] = value
+            when 'Interval', 'Timeout', 'UnhealthyThreshold', 'HealthyThreshold'
+              @health_check[name] = value.to_i
+
+            when 'HealthCheck'
+              @response['ConfigureHealthCheckResult'][name] = @health_check
+
+            when 'RequestId'
+              @response['ResponseMetadata'][name] = value
+            end
+          end
+
+        end
+
+      end
+    end
+  end
+end