fog-aws 0.5.0 → 0.6.0

data/lib/fog/aws/requests/lambda/get_function_configuration.rb

@@ -0,0 +1,44 @@
+module Fog
+  module AWS
+    class Lambda
+      class Real
+        require 'fog/aws/parsers/lambda/base'
+
+        # Returns the configuration information of the Lambda function.
+        # http://docs.aws.amazon.com/lambda/latest/dg/API_GetFunction.html
+        # ==== Parameters
+        # * FunctionName <~String> - Lambda function name.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~Hash>:
+        #     * 'CodeSize' <~Integer> - size, in bytes, of the function .zip file you uploaded.
+        #     * 'Description' <~String> - user-provided description.
+        #     * 'FunctionArn' <~String> - Amazon Resource Name (ARN) assigned to the function.
+        #     * 'FunctionName' <~String> - name of the function.
+        #     * 'Handler' <~String> - function Lambda calls to begin executing your function.
+        #     * 'LastModified' <~Time> - timestamp of the last time you updated the function.
+        #     * 'Memorysize' <~String> - memory size, in MB, you configured for the function.
+        #     * 'Role' <~String> - ARN of the IAM role that Lambda assumes when it executes your function to access any other AWS resources.
+        #     * 'Runtime' <~String> - runtime environment for the Lambda function.
+        #     * 'Timeout' <~Integer> - function execution time at which Lambda should terminate the function.
+        def get_function_configuration(params={})
+          function_name = params.delete('FunctionName')
+          request({
+            :method  => 'GET',
+            :path    => "/functions/#{function_name}/versions/HEAD/configuration",
+            :parser  => Fog::AWS::Parsers::Lambda::Base.new
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def get_function_configuration(params={})
+          response = self.get_function(params)
+          function_configuration = response.body['Configuration']
+          response.body = function_configuration
+          response
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/requests/lambda/get_policy.rb

@@ -0,0 +1,52 @@
+module Fog
+  module AWS
+    class Lambda
+      class Real
+        require 'fog/aws/parsers/lambda/base'
+
+        # Returns the access policy, containing a list of permissions granted via the AddPermission API, associated with the specified bucket.
+        # http://docs.aws.amazon.com/lambda/latest/dg/API_GetPolicy.html
+        # ==== Parameters
+        # * FunctionName <~String> - Function name whose access policy you want to retrieve.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~Hash>:
+        #     * 'Policy' <~Hash> - The access policy associated with the specified function.
+        def get_policy(params={})
+          function_name = params.delete('FunctionName')
+          request({
+            :method  => 'GET',
+            :path    => "/functions/#{function_name}/versions/HEAD/policy",
+            :parser => Fog::AWS::Parsers::Lambda::Base.new
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def get_policy(params={})
+          response = Excon::Response.new
+
+          function     = self.get_function_configuration(params).body
+          function_arn = function['FunctionArn']
+          statements   = self.data[:permissions][function_arn] || []
+
+          if statements.empty?
+            message = "ResourceNotFoundException => "
+            message << "The resource you requested does not exist."
+            raise Fog::AWS::Lambda::Error, message
+          end
+
+          policy = {
+            'Version'   => '2012-10-17',
+            'Statement' => statements,
+            'Id'        => 'default'
+          }
+
+          response.status = 200
+          response.body = { 'Policy' => policy }
+          response
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/requests/lambda/invoke.rb

@@ -0,0 +1,85 @@
+module Fog
+  module AWS
+    class Lambda
+      class Real
+
+        # Invokes a specified Lambda function.
+        # http://docs.aws.amazon.com/lambda/latest/dg/API_Invoke.html
+        # ==== Parameters
+        # * ClientContext <~Hash> - client-specific information to the Lambda function you are invoking.
+        # * FunctionName <~String> - Lambda function name.
+        # * InvocationType <~String> - function invocation type.
+        # * LogType <~String> - logs format for function calls of "RequestResponse" invocation type.
+        # * Payload <~Integer> - Lambda function input.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~Hash> - JSON representation of the object returned by the Lambda function.
+        def invoke(params={})
+          headers = {}
+          if client_context = params.delete('ClientContext')
+            headers['X-Amz-Client-Context'] =
+              Base64::encode64(Fog::Json.encode(client_context))
+          end
+          if client_type = params.delete('InvocationType')
+            headers['X-Amz-Client-Type'] = client_type
+          end
+          if log_type = params.delete('LogType')
+            headers['X-Amz-Log-Type'] = log_type
+          end
+          payload = Fog::JSON.encode(params.delete('Payload'))
+          function_name = params.delete('FunctionName')
+
+          request({
+            :method  => 'POST',
+            :path    => "/functions/#{function_name}/invocations",
+            :headers => headers,
+            :body    => payload,
+            :expects => [200, 202, 204]
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def invoke(params={})
+          response = Excon::Response.new
+          response.status = 200
+          response.body = ''
+
+          unless function_id = params.delete('FunctionName')
+            message = 'AccessDeniedException => '
+            message << 'Unable to determine service/operation name to be authorized'
+            raise Fog::AWS::Lambda::Error, message
+          end
+
+          client_context = params.delete('ClientContext')
+          client_type    = params.delete('InvocationType')
+          log_type       = params.delete('LogType')
+          payload        = params.delete('Payload')
+
+          if (client_context || client_type || log_type)
+            message = "invoke parameters handling are not yet mocked [light_black](#{caller.first})[/]"
+            Fog::Logger.warning message
+            Fog::Mock.not_implemented
+          end
+
+          if payload
+            message = "payload parameter is ignored since we are not really "
+            message << "invoking a function [light_black](#{caller.first})[/]"
+            Fog::Logger.warning message
+          end
+
+          function = self.get_function_configuration('FunctionName' => function_id).body
+
+          if function.is_a?(Hash) && function.has_key?('FunctionArn')
+            response.body = "\"Imagine #{function['FunctionArn']} was invoked\""
+          else
+            message = "ResourceNotFoundException => Function not found: #{function_id}"
+            raise Fog::AWS::Lambda::Error, message
+          end
+
+          response
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/requests/lambda/list_event_source_mappings.rb

@@ -0,0 +1,69 @@
+module Fog
+  module AWS
+    class Lambda
+      class Real
+        # Returns a list of event source mappings where you can identify a stream as an event source.
+        # http://docs.aws.amazon.com/lambda/latest/dg/API_ListEventSourceMappings.html
+        # ==== Parameters
+        # * EventSourceArn <~String> - Amazon Resource Name (ARN) of the stream.
+        # * FunctionName <~String> - name of the Lambda function.
+        # * Marker <~String> - opaque pagination token returned from a previous ListEventSourceMappings operation.
+        # * MaxItems <~Integer> - maximum number of event sources to return in response.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~Hash>:
+        #     * 'EventSourceMappings' <~Array> - array of EventSourceMappingConfiguration objects.
+        #     * 'NextMarker' <~String> - present if there are more event source mappings.
+        def list_event_source_mappings(params={})
+          event_source_arn = params.delete('EventSourceArn')
+          function_name    = params.delete('FunctionName')
+          marker           = params.delete('Marker')
+          max_items        = params.delete('MaxItems')
+
+          query = {}
+          query.merge!('EventSourceArn' => event_source_arn) if event_source_arn
+          query.merge!('FunctionName'   => function_name)    if function_name
+          query.merge!('Marker'         => marker)           if marker
+          query.merge!('MaxItems'       => max_items)        if max_items
+
+          request({
+            :method => 'GET',
+            :path   => '/event-source-mappings/',
+            :query  => query
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def list_event_source_mappings(params={})
+          response = Excon::Response.new
+          response.status = 200
+
+          function_name = params.delete('FunctionName')
+
+          begin
+            function = self.get_function_configuration('FunctionName' => function_name).body
+            function_arn = function['FunctionArn']
+          rescue Fog::AWS::Lambda::Error => e
+            # interestingly enough, if you try to do a list_event_source_mappings
+            # on a nonexisting function, Lambda API endpoint doesn't return
+            # error, just an empty array.
+          end
+
+          event_source_mappings = []
+          if function_arn
+            event_source_mappings = self.data[:event_source_mappings].values.select do |m|
+              m['FunctionArn'].eql?(function_arn)
+            end
+          end
+
+          response.body = {
+            'EventSourceMappings' => event_source_mappings,
+            'NextMarker'          => nil
+          }
+          response
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/requests/lambda/list_functions.rb

@@ -0,0 +1,39 @@
+module Fog
+  module AWS
+    class Lambda
+      class Real
+        require 'fog/aws/parsers/lambda/base'
+
+        # Returns a list of your Lambda functions.
+        # http://docs.aws.amazon.com/lambda/latest/dg/API_ListFunctions.html
+        # ==== Parameters
+        # * Marker <~String> - opaque pagination token returned from a previous ListFunctions operation. If present, indicates where to continue the listing.
+        # * MaxItems <~Integer> - Specifies the maximum number of AWS Lambda functions to return in response.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~Hash>:
+        #     * 'Functions' <~Array> - list of Lambda functions.
+        #     * 'NextMarker' <~String> - present if there are more functions.
+        def list_functions(params={})
+          request({
+            :method => 'GET',
+            :path   => '/functions/',
+            :parser => Fog::AWS::Parsers::Lambda::Base.new
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def list_functions(params={})
+          response = Excon::Response.new
+          response.status = 200
+          response.body = {
+            'Functions'  => self.data[:functions].values,
+            'NextMarker' => nil
+          }
+          response
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/requests/lambda/remove_permission.rb

@@ -0,0 +1,56 @@
+module Fog
+  module AWS
+    class Lambda
+      class Real
+
+        # Remove individual permissions from an access policy associated with a Lambda function by providing a Statement ID.
+        # http://docs.aws.amazon.com/lambda/latest/dg/API_RemovePermission.html
+        # ==== Parameters
+        # * FunctionName <~String> - Lambda function whose access policy you want to remove a permission from.
+        # * StatementId <~String> - Statement ID of the permission to remove.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~String>:
+        def remove_permission(params={})
+          function_name = params.delete('FunctionName')
+          statement_id  = params.delete('StatementId')
+          request({
+            :method  => 'DELETE',
+            :path    => "/functions/#{function_name}/versions/HEAD/policy/#{statement_id}",
+            :expects => 204
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def remove_permission(params={})
+          function_name = params.delete('FunctionName')
+          opts = { 'FunctionName' => function_name }
+          function     = self.get_function_configuration(opts).body
+          function_arn = function['FunctionArn']
+
+          statement_id = params.delete('StatementId')
+          message      = 'Statement ID cannot be blank'
+          raise Fog::AWS::Lambda::Error, message unless statement_id
+
+          permissions_qty = self.data[:permissions][function_arn].size
+
+          self.data[:permissions][function_arn].delete_if do |s|
+            s['Sid'].eql?(statement_id)
+          end
+
+          if self.data[:permissions][function_arn].size.eql?(permissions_qty)
+            message  = "ResourceNotFoundException => "
+            message << "The resource you requested does not exist."
+            raise Fog::AWS::Lambda::Error, message
+          end
+
+          response        = Excon::Response.new
+          response.status = 204
+          response.body   =  ''
+          response
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/requests/lambda/update_event_source_mapping.rb

@@ -0,0 +1,86 @@
+module Fog
+  module AWS
+    class Lambda
+      class Real
+        # Change the parameters of the existing mapping without losing your position in the stream.
+        # http://docs.aws.amazon.com/lambda/latest/dg/API_UpdateEventSourceMapping.html
+        # ==== Parameters
+        # * UUID <~String> - event source mapping identifier.
+        # * BatchSize <~Integer> - maximum number of stream records that can be sent to your Lambda function for a single invocation.
+        # * Enabled <~Boolean> - specifies whether AWS Lambda should actively poll the stream or not.
+        # * FunctionName <~String> - Lambda function to which you want the stream records sent.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~Hash>:
+        #     * 'BatchSize' <~Integer> - largest number of records that AWS Lambda will retrieve from your event source at the time of invoking your function.
+        #     * 'EventSourceArn' <~String> - Amazon Resource Name (ARN) of the stream that is the source of events.
+        #     * 'FunctionArn' <~String> - Lambda function to invoke when AWS Lambda detects an event on the stream.
+        #     * 'LastModified' <~Time> - UTC time string indicating the last time the event mapping was updated.
+        #     * 'LastProcessingResult' <~String> - result of the last AWS Lambda invocation of your Lambda function.
+        #     * 'State' <~String> - state of the event source mapping.
+        #     * 'StateTransitionReason' <~String> - reason the event source mapping is in its current state.
+        #     * 'UUID' <~String> - AWS Lambda assigned opaque identifier for the mapping.
+        def update_event_source_mapping(params={})
+          function_name = params.delete('FunctionName')
+          mapping_id    = params.delete('UUID')
+
+          batch_size = params.delete('BatchSize')
+          enabled    = params.delete('Enabled')
+
+          update = {}
+          update.merge!('BatchSize'    => batch_size)    if batch_size
+          update.merge!('Enabled'      => enabled)       if !enabled.nil?
+          update.merge!('FunctionName' => function_name) if function_name
+
+          request({
+            :method  => 'PUT',
+            :path    => "/event-source-mappings/#{mapping_id}",
+            :expects => 202,
+            :body    => Fog::JSON.encode(update)
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def update_event_source_mapping(params={})
+          mapping_id = params.delete('UUID')
+          mapping = self.data[:event_source_mappings][mapping_id]
+
+          unless mapping
+            message  = 'ResourceNotFoundException => '
+            message << 'The resource you requested does not exist.'
+            raise Fog::AWS::Lambda::Error, message
+          end
+
+          function_name = params.delete('FunctionName')
+          function = {}
+          if function_name
+            function_opts = { 'FunctionName' => function_name }
+            function      = self.get_function_configuration(function_opts).body
+            function_arn  = function['FunctionArn']
+          end
+
+          batch_size = params.delete('BatchSize')
+          enabled    = params.delete('Enabled')
+
+          update = {}
+
+          if function_name && !function.empty? && function_arn
+            update.merge!('FunctionArn' => function_arn)
+          end
+          update.merge!('BatchSize' => batch_size) if batch_size
+          update.merge!('Enabled'   => enabled)    if !enabled.nil?
+
+          mapping.merge!(update)
+          mapping['State'] = 'Disabling' if enabled.eql?(false)
+          mapping['State'] = 'Enabling'  if enabled.eql?(true)
+
+          response = Excon::Response.new
+          response.status = 202
+          response.body = mapping
+          response
+        end
+      end
+    end
+  end
+end

data/lib/fog/aws/requests/lambda/update_function_code.rb

@@ -0,0 +1,75 @@
+module Fog
+  module AWS
+    class Lambda
+      class Real
+        require 'fog/aws/parsers/lambda/base'
+
+        # Updates the code for the specified Lambda function.
+        # http://docs.aws.amazon.com/lambda/latest/dg/API_UpdateFunctionCode.html
+        # ==== Parameters
+        # * FunctionName <~String> - existing Lambda function name whose code you want to replace.
+        # * S3Bucket <~String> - Amazon S3 bucket name where the .zip file containing your deployment package is stored.
+        # * S3Key <~String> - Amazon S3 object (the deployment package) key name you want to upload.
+        # * S3ObjectVersion <~String> - Amazon S3 object (the deployment package) version you want to upload.
+        # * ZipFile <~String> - Based64-encoded .zip file containing your packaged source code.
+        # ==== Returns
+        # * response<~Excon::Response>:
+        #   * body<~Hash>:
+        #     * 'CodeSize' <~Integer> - size, in bytes, of the function .zip file you uploaded.
+        #     * 'Description' <~String> - user-provided description.
+        #     * 'FunctionArn' <~String> - Amazon Resource Name (ARN) assigned to the function.
+        #     * 'FunctionName' <~String> - name of the function.
+        #     * 'Handler' <~String> - function Lambda calls to begin executing your function.
+        #     * 'LastModified' <~Time> - timestamp of the last time you updated the function.
+        #     * 'Memorysize' <~String> - memory size, in MB, you configured for the function.
+        #     * 'Role' <~String> - ARN of the IAM role that Lambda assumes when it executes your function to access any other AWS resources.
+        #     * 'Runtime' <~String> - runtime environment for the Lambda function.
+        #     * 'Timeout' <~Integer> - function execution time at which Lambda should terminate the function.
+        def update_function_code(params={})
+          function_name = params.delete('FunctionName')
+
+          s3_bucket     = params.delete('S3Bucket')
+          s3_key        = params.delete('S3Key')
+          s3_object_ver = params.delete('S3ObjectVersion')
+          zip_file      = params.delete('ZipFile')
+
+          update = {}
+          update.merge!('S3Bucket'        => s3_bucket)     if s3_bucket
+          update.merge!('S3Key'           => s3_key)        if s3_key
+          update.merge!('S3ObjectVersion' => s3_object_ver) if s3_object_ver
+          update.merge!('ZipFile'         => zip_file)      if zip_file
+
+          request({
+            :method  => 'PUT',
+            :path    => "/functions/#{function_name}/versions/HEAD/code",
+            :body    => Fog::JSON.encode(update),
+            :parser  => Fog::AWS::Parsers::Lambda::Base.new
+          }.merge(params))
+        end
+      end
+
+      class Mock
+        def update_function_code(params={})
+          response = self.get_function_configuration(params)
+
+          request_data = []
+          %w(S3Bucket S3Key S3ObjectVersion ZipFile).each do |p|
+            request_data << params.delete(p) if params.has_key?(p)
+          end
+
+          message = 'Please provide a source for function code.'
+          raise Fog::AWS::Lambda::Error, message if request_data.empty?
+          # we ignore any parameters since we are not uploading any code
+
+          function_arn = response.body['FunctionArn']
+
+          response = Excon::Response.new
+          response.status = 200
+          response.body = self.data[:functions][function_arn]
+
+          response
+        end
+      end
+    end
+  end
+end