fog-aws 0.4.1 → 0.5.0

data/lib/fog/aws/parsers/ecs/run_task.rb

@@ -0,0 +1,24 @@
+module Fog
+  module Parsers
+    module AWS
+      module ECS
+        require 'fog/aws/parsers/ecs/task'
+
+        class RunTask < Fog::Parsers::AWS::ECS::Task
+          def reset
+            super
+            @result = 'RunTaskResult'
+            @response[@result] = {'failures' => [], 'tasks' => []}
+            @contexts = %w(failures tasks containers overrides networkBindings containerOverrides)
+            @context             = []
+            @task                = {}
+            @failure             = {}
+            @container           = {}
+            @net_binding         = {}
+            @container_overrides = []
+          end
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/parsers/ecs/service.rb

@@ -0,0 +1,82 @@
+module Fog
+  module Parsers
+    module AWS
+      module ECS
+        require 'fog/aws/parsers/ecs/base'
+
+        class Service < Fog::Parsers::AWS::ECS::Base
+          def start_element(name, attrs = [])
+            super
+            if @contexts.include?(name)
+              @context.push(name)
+            end
+          end
+
+          def end_element(name)
+            super
+            case name
+            when *@contexts
+              @context.pop
+            when 'member'
+              case @context.last
+              when 'services'
+                @response[@result]['services'] << @service
+                @service = {}
+              when 'loadBalancers'
+                @service['loadBalancers'] ||= []
+                @service['loadBalancers'] << @load_balancer
+                @load_balancer = {}
+              when 'events'
+                @service['events'] ||= []
+                @service['events'] << @event
+                @event = {}
+              when 'deployments'
+                @service['deployments'] ||= []
+                @service['deployments'] << @deployment
+                @deployment = {}
+              end
+            when 'clusterArn', 'roleArn', 'serviceArn', 'serviceName'
+              @service[name] = value
+            when 'taskDefinition', 'status'
+              case @context.last
+              when 'service', 'services'
+                @service[name] = value
+              when 'deployments'
+                @deployment[name] = value
+              end
+            when 'desiredCount', 'pendingCount', 'runningCount'
+              case @context.last
+              when 'service', 'services'
+                @service[name] = value.to_i
+              when 'deployments'
+                @deployment[name] = value.to_i
+              end
+            when 'loadBalancerName', 'containerName'
+              @load_balancer[name] = value
+            when 'containerPort'
+              @load_balancer[name] = value.to_i
+            when 'createdAt'
+              case @context.last
+              when 'events'
+                @event[name] = Time.parse(value)
+              when 'deployments'
+                @deployment[name] = Time.parse(value)
+              end
+            when 'id'
+              case @context.last
+              when 'events'
+                @event[name] = value
+              when 'deployments'
+                @deployment[name] = value
+              end
+            when 'message'
+              @event[name] = value
+            when 'updatedAt'
+              @deployment[name] = Time.parse(value)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/parsers/ecs/start_task.rb

@@ -0,0 +1,24 @@
+module Fog
+  module Parsers
+    module AWS
+      module ECS
+        require 'fog/aws/parsers/ecs/task'
+
+        class StartTask < Fog::Parsers::AWS::ECS::Task
+          def reset
+            super
+            @result = 'StartTaskResult'
+            @response[@result] = {'failures' => [], 'tasks' => []}
+            @contexts = %w(failures tasks containers overrides networkBindings containerOverrides)
+            @context             = []
+            @task                = {}
+            @failure             = {}
+            @container           = {}
+            @net_binding         = {}
+            @container_overrides = []
+          end
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/parsers/ecs/stop_task.rb

@@ -0,0 +1,23 @@
+module Fog
+  module Parsers
+    module AWS
+      module ECS
+        require 'fog/aws/parsers/ecs/task'
+
+        class StopTask < Fog::Parsers::AWS::ECS::Task
+          def reset
+            super
+            @result = 'StopTaskResult'
+            @response[@result] = {'task' => {}}
+            @contexts = %w(task containers overrides networkBindings containerOverrides)
+            @context             = []
+            @task                = {}
+            @container           = {}
+            @net_binding         = {}
+            @container_overrides = []
+          end
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/parsers/ecs/task.rb

@@ -0,0 +1,77 @@
+module Fog
+  module Parsers
+    module AWS
+      module ECS
+        require 'fog/aws/parsers/ecs/base'
+
+        class Task < Fog::Parsers::AWS::ECS::Base
+          def start_element(name, attrs = [])
+            super
+            if @contexts.include?(name)
+              @context.push(name)
+            end
+          end
+
+          def end_element(name)
+            super
+            case name
+            when 'containerOverrides'
+              @task['overrides'] ||= {}
+              @task['overrides'][name] = @container_overrides
+              @context.pop
+            when 'task'
+              @response[@result][name] = @task
+            when *@contexts
+              @context.pop
+            when 'member'
+              case @context.last
+              when 'tasks'
+                @response[@result]['tasks'] << @task
+                @task = {}
+              when 'containers'
+                @task['containers'] ||= []
+                @task['containers'] << @container
+                @container = {}
+              when 'networkBindings'
+                @container['networkBindings'] ||= []
+                @container['networkBindings'] << @net_binding
+                @net_binding = {}
+              when 'failures'
+                @response[@result]['failures'] << @failure
+                @failure = {}
+              end
+            when 'clusterArn', 'desiredStatus', 'startedBy', 'containerInstanceArn', 'taskDefinitionArn'
+              @task[name] = value
+            when 'taskArn', 'lastStatus'
+              case @context.last
+              when 'tasks'
+                @task[name] = value
+              when 'containers'
+                @container[name] = value
+              end
+            when 'containerArn'
+              @container[name] = value
+            when 'exitCode'
+              @container[name] = value.to_i
+            when 'name'
+              case @context.last
+              when 'containers'
+                @container[name] = value
+              when 'containerOverrides'
+                @container_overrides << value
+              end
+            when 'networkBindings'
+              @container[name] = @net_bindings
+            when 'bindIP'
+              @net_binding[name] = value
+            when 'hostPort', 'containerPort'
+              @net_binding[name] = value.to_i
+            when 'arn', 'reason'
+              @failure[name] = value
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/parsers/ecs/task_definition.rb

@@ -0,0 +1,95 @@
+module Fog
+  module Parsers
+    module AWS
+      module ECS
+        require 'fog/aws/parsers/ecs/base'
+
+        class TaskDefinition < Fog::Parsers::AWS::ECS::Base
+          def start_element(name, attrs = [])
+            super
+            if @contexts.include?(name)
+              @context.push(name)
+            end
+          end
+
+          def end_element(name)
+            super
+            case name
+            when 'taskDefinitionArn'
+              @response[@result][@definition][name] = value
+            when 'revision'
+              @response[@result][@definition][name] = value.to_i
+            when *@contexts
+              @context.pop
+            when 'member'
+              case @context.last
+              when 'volumes'
+                @response[@result][@definition]['volumes'] << @volume
+                @volume = {}
+              when 'containerDefinitions'
+                @response[@result][@definition]['containerDefinitions'] << @container
+                @container = {}
+              when 'command'
+                @container['command'] ||= []
+                @container['command'] << value
+              when 'entryPoint'
+                @container['entryPoint'] ||= []
+                @container['entryPoint'] << value
+              when 'links'
+                @container['links'] ||= []
+                @container['links'] << value
+              when 'environment'
+                @container['environment'] ||= []
+                @container['environment'] << @environment
+                @environment = {}
+              when 'mountPoints'
+                @container['mountPoints'] ||= []
+                @container['mountPoints'] << @mountpoint
+                @mountpoint = {}
+              when 'portMappings'
+                @container['portMappings'] ||= []
+                @container['portMappings'] << @portmapping
+                @portmapping = {}
+              end
+            when 'name'
+              case @context.last
+              when 'volumes'
+                @volume[name] = value
+              when 'containerDefinitions'
+                @container[name] = value
+              when 'environment'
+                @environment[name] = value
+              end
+            when 'host'
+              @volume[name] = @host
+              @host = {}
+            when 'sourcePath'
+              @host[name] = value
+            when 'cpu', 'memory'
+              @container[name] = value.to_i
+            when 'essential'
+              @container[name] = value == 'true'
+            when 'image'
+              @container[name] = value
+            when 'value'
+              @environment[name] = value
+            when 'readOnly'
+              case @context.last
+              when 'mountPoints'
+                @mountpoint[name] = value == 'true'
+              when 'volumesFrom'
+                @volume_from[name] = value == 'true'
+              end
+            when 'containerPath', 'sourceVolume'
+              @mountpoint[name] = value
+            when 'containerPort', 'hostPort'
+              @portmapping[name] = value.to_i
+            when 'sourceContainer'
+              @volume_from[name] = value
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/parsers/ecs/update_service.rb

@@ -0,0 +1,31 @@
+module Fog
+  module Parsers
+    module AWS
+      module ECS
+        require 'fog/aws/parsers/ecs/service'
+
+        class UpdateService < Fog::Parsers::AWS::ECS::Service
+          def reset
+            super
+            @result = 'UpdateServiceResult'
+            @response[@result] = {}
+            @contexts = %w(service loadBalancers events deployments)
+            @service       = {}
+            @context       = []
+            @deployment    = {}
+            @load_balancer = {}
+            @event         = {}
+          end
+
+          def end_element(name)
+            super
+            case name
+            when 'service'
+              @response[@result]['service'] = @service
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/requests/compute/create_vpc.rb

@@ -37,7 +37,7 @@ module Fog
       end
 
       class Mock
-        def create_vpc(cidrBlock)
+        def create_vpc(cidrBlock, options = {})
           Excon::Response.new.tap do |response|
             if cidrBlock
               response.status = 200

data/lib/fog/aws/requests/ecs/create_cluster.rb

@@ -0,0 +1,64 @@
+module Fog
+  module AWS
+    class ECS
+      class Real
+        require 'fog/aws/parsers/ecs/create_cluster'
+
+        # Creates a new Amazon ECS cluster
+        # http://docs.aws.amazon.com/AmazonECS/latest/APIReference/API_CreateCluster.html
+        # ==== Parameters
+        # * clusterName <~String> - The name of your cluster.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~Hash>:
+        #     * 'Cluster' <~Hash> - The full description of your new cluster
+        def create_cluster(params={})
+          request({
+            'Action' => 'CreateCluster',
+            :parser  => Fog::Parsers::AWS::ECS::CreateCluster.new
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def create_cluster(params={})
+          response = Excon::Response.new
+          response.status = 200
+
+          params.has_key?('clusterName') || params['clusterName'] = 'default'
+
+          owner_id = Fog::AWS::Mock.owner_id
+          cluster_name = params['clusterName']
+          cluster_path = "cluster/#{cluster_name}"
+          cluster_arn = Fog::AWS::Mock.arn('ecs', owner_id, cluster_path, region)
+          cluster = {}
+
+          search_cluster_result = self.data[:clusters].select { |c| c['clusterName'].eql?(cluster_name) }
+          if search_cluster_result.empty?
+            cluster = {
+              'clusterName'                       => cluster_name,
+              'clusterArn'                        => cluster_arn,
+              'status'                            => 'ACTIVE',
+              'registeredContainerInstancesCount' => 0,
+              'runningTasksCount'                 => 0,
+              'pendingTasksCount'                 => 0
+            }
+            self.data[:clusters] << cluster
+          else
+            cluster = search_cluster_result.first
+          end
+
+          response.body = {
+            'CreateClusterResult' => {
+              'cluster' => cluster
+            },
+            'ResponseMetadata' => {
+              'RequestId' => Fog::AWS::Mock.request_id
+            }
+          }
+          response
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/requests/ecs/create_service.rb

@@ -0,0 +1,118 @@
+module Fog
+  module AWS
+    class ECS
+      class Real
+        require 'fog/aws/parsers/ecs/create_service'
+
+        # Runs and maintains a desired number of tasks from a specified task definition.
+        # http://docs.aws.amazon.com/AmazonECS/latest/APIReference/API_CreateService.html
+        # ==== Parameters
+        # * clientToken <~String> - unique, case-sensitive identifier you provide to ensure the idempotency of the request.
+        # * cluster <~String> - short name or full Amazon Resource Name (ARN) of the cluster that you want to run your service on.
+        # * desiredCount <~Integer> - number of instantiations of the specified task definition that you would like to place and keep running on your cluster.
+        # * loadBalancers <~Array> - list of load balancer objects, containing the load balancer name, the container name (as it appears in a container definition), and the container port to access from the load balancer.
+        # * role <~String> - name or full Amazon Resource Name (ARN) of the IAM role that allows your Amazon ECS container agent to make calls to your load balancer on your behalf.
+        # * serviceName <~String> - name of your service
+        # * taskDefinition <~String> - family and revision (family:revision) or full Amazon Resource Name (ARN) of the task definition that you want to run in your service
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~Hash>:
+        #     * 'Service' <~Hash> - The full description of your new service
+        def create_service(params={})
+          if load_balancers = params.delete('loadBalancers')
+            params.merge!(Fog::AWS.indexed_param('loadBalancers.member', [*load_balancers]))
+          end
+          request({
+            'Action' => 'CreateService',
+            :parser  => Fog::Parsers::AWS::ECS::CreateService.new
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def create_service(params={})
+          response = Excon::Response.new
+          response.status = 200
+
+          e = Fog::AWS::ECS::Error
+          msg = 'ClientException => desiredCount cannot be empty.'
+          raise e, msg unless desired_count = params['desiredCount']
+          msg = 'ClientException => serviceName cannot be empty.'
+          raise e unless service_name = params['serviceName']
+          msg = 'ClientException => taskDefinition cannot be empty.'
+          raise e unless task_definition = params['taskDefinition']
+
+          owner_id = Fog::AWS::Mock.owner_id
+
+          service_path = "service/#{service_name}"
+          service_arn = Fog::AWS::Mock.arn('ecs', owner_id, service_path, region)
+
+          cluster = params['cluster'] || 'default'
+          if !cluster.match(/^arn:aws:ecs:.+:.+:cluster\/(.+)$/)
+            cluster_path = "cluster/#{cluster}"
+            cluster_arn = Fog::AWS::Mock.arn('ecs', owner_id, cluster_path, region)
+          else
+            cluster_arn = cluster
+          end
+
+          if params['role']
+            role = params['role'] if params['role']
+            if !role.match(/^arn:aws:iam:.*:.*:role\/(.+)$/)
+              role_path = "role/#{role}"
+              role_arn = Fog::AWS::Mock.arn('iam', owner_id, role_path, region)
+            else
+              role_arn = role
+            end
+          end
+
+          if !task_definition.match(/^arn:aws:ecs:.+:.+:task-definition\/.+$/)
+            task_def_path = "task-definition\/#{task_definition}"
+            task_def_arn = Fog::AWS::Mock.arn('ecs', owner_id, task_def_path, region)
+          else
+            task_def_arn = task_definition
+          end
+
+          load_balancers = params['loadBalancers'] || []
+
+          service = {
+            'events'         => [],
+            'serviceName'    => service_name,
+            'serviceArn'     => service_arn,
+            'taskDefinition' => task_def_arn,
+            'clusterArn'     => cluster_arn,
+            'status'         => 'ACTIVE',
+            'roleArn'        => role_arn,
+            'loadBalancers'  => [*load_balancers],
+            'deployments'    => [],
+            'desiredCount'   => desired_count,
+            'pendingCount'   => 0,
+            'runningCount'   => 0
+          }
+
+          service['deployments'] << {
+            'updatedAt'      => Time.now.utc,
+            'id'             => "ecs-svc/#{Fog::Mock.random_numbers(19)}",
+            'taskDefinition' => task_def_arn,
+            'status'         => 'PRIMARY',
+            'desiredCount'   => desired_count,
+            'createdAt'      => Time.now.utc,
+            'pendingCount'   => 0,
+            'runningCount'   => 0
+          }
+
+          self.data[:services] << service
+
+          response.body = {
+            'CreateServiceResult' => {
+              'service' => service,
+            },
+            'ResponseMetadata' => {
+              'RequestId' => Fog::AWS::Mock.request_id
+            }
+          }
+          response
+        end
+      end
+    end
+  end
+end