functions_framework 0.2.1 → 0.5.0

data/lib/functions_framework/cloud_events/http_binding.rb

@@ -1,270 +0,0 @@
-# Copyright 2020 Google LLC
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     https://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-require "functions_framework/cloud_events/json_format"
-
-module FunctionsFramework
-  module CloudEvents
-    ##
-    # HTTP binding for CloudEvents.
-    #
-    # This class implements HTTP binding, including unmarshalling of events from
-    # Rack environment data, and marshalling of events to Rack environment data.
-    # It supports binary (i.e. header-based) HTTP content, as well as structured
-    # (body-based) content that can delegate to formatters such as JSON.
-    #
-    # See https://github.com/cloudevents/spec/blob/master/http-protocol-binding.md
-    #
-    class HttpBinding
-      ##
-      # Returns a default binding, with JSON supported.
-      #
-      def self.default
-        @default ||= begin
-          http_binding = new
-          json_format = JsonFormat.new
-          http_binding.register_structured_formatter "json", json_format
-          http_binding.register_batched_formatter "json", json_format
-          http_binding
-        end
-      end
-
-      ##
-      # Create an empty HTTP binding.
-      #
-      def initialize
-        @structured_formatters = {}
-        @batched_formatters = {}
-      end
-
-      ##
-      # Register a formatter for the given type.
-      #
-      # A formatter must respond to the methods `#encode` and `#decode`. See
-      # {FunctionsFramework::CloudEvents::JsonFormat} for an example.
-      #
-      # @param type [String] The subtype format that should be handled by
-      #     this formatter.
-      # @param formatter [Object] The formatter object.
-      # @return [self]
-      #
-      def register_structured_formatter type, formatter
-        formatters = @structured_formatters[type.to_s.strip.downcase] ||= []
-        formatters << formatter unless formatters.include? formatter
-        self
-      end
-
-      ##
-      # Register a batch formatter for the given type.
-      #
-      # A batch formatter must respond to the methods `#encode_batch` and
-      # `#decode_batch`. See {FunctionsFramework::CloudEvents::JsonFormat} for
-      # an example.
-      #
-      # @param type [String] The subtype format that should be handled by
-      #     this formatter.
-      # @param formatter [Object] The formatter object.
-      # @return [self]
-      #
-      def register_batched_formatter type, formatter
-        formatters = @batched_formatters[type.to_s.strip.downcase] ||= []
-        formatters << formatter unless formatters.include? formatter
-        self
-      end
-
-      ##
-      # Decode an event from the given Rack environment hash. Following the
-      # CloudEvents spec, this chooses a handler based on the Content-Type of
-      # the request.
-      #
-      # @param env [Hash] The Rack environment.
-      # @param format_args [keywords] Extra args to pass to the formatter.
-      # @return [FunctionsFramework::CloudEvents::Event] if the request
-      #     includes a single structured or binary event.
-      # @return [Array<FunctionsFramework::CloudEvents::Event>] if the request
-      #     includes a batch of structured events.
-      # @return [nil] if the request was not recognized as a CloudEvent.
-      #
-      def decode_rack_env env, **format_args
-        content_type_header = env["CONTENT_TYPE"]
-        content_type = ContentType.new content_type_header if content_type_header
-        input = env["rack.input"]
-        if input && content_type&.media_type == "application"
-          case content_type.subtype_prefix
-          when "cloudevents"
-            input.set_encoding content_type.charset if content_type.charset
-            return decode_structured_content input.read, content_type.subtype_format, **format_args
-          when "cloudevents-batch"
-            input.set_encoding content_type.charset if content_type.charset
-            return decode_batched_content input.read, content_type.subtype_format, **format_args
-          end
-        end
-        decode_binary_content env, content_type
-      end
-
-      ##
-      # Decode a single event from the given content data. This should be
-      # passed the request body, if the Content-Type is of the form
-      # `application/cloudevents+format`.
-      #
-      # @param input [String] The string content.
-      # @param format [String] The format code (e.g. "json").
-      # @param format_args [keywords] Extra args to pass to the formatter.
-      # @return [FunctionsFramework::CloudEvents::Event]
-      #
-      def decode_structured_content input, format, **format_args
-        handlers = @structured_formatters[format] || []
-        handlers.reverse_each do |handler|
-          event = handler.decode input, **format_args
-          return event if event
-        end
-        raise HttpContentError, "Unknown cloudevents format: #{format.inspect}"
-      end
-
-      ##
-      # Decode a batch of events from the given content data. This should be
-      # passed the request body, if the Content-Type is of the form
-      # `application/cloudevents-batch+format`.
-      #
-      # @param input [String] The string content.
-      # @param format [String] The format code (e.g. "json").
-      # @param format_args [keywords] Extra args to pass to the formatter.
-      # @return [Array<FunctionsFramework::CloudEvents::Event>]
-      #
-      def decode_batched_content input, format, **format_args
-        handlers = @batched_formatters[format] || []
-        handlers.reverse_each do |handler|
-          events = handler.decode_batch input, **format_args
-          return events if events
-        end
-        raise HttpContentError, "Unknown cloudevents batch format: #{format.inspect}"
-      end
-
-      ##
-      # Decode an event from the given Rack environment in binary content mode.
-      #
-      # @param env [Hash] Rack environment hash.
-      # @param content_type [FunctionsFramework::CloudEvents::ContentType]
-      #     the content type from the Rack environment.
-      # @return [FunctionsFramework::CloudEvents::Event] if a CloudEvent
-      #     could be decoded from the Rack environment.
-      # @return [nil] if the Rack environment does not indicate a CloudEvent
-      #
-      def decode_binary_content env, content_type
-        spec_version = env["HTTP_CE_SPECVERSION"]
-        return nil if spec_version.nil?
-        raise SpecVersionError, "Unrecognized specversion: #{spec_version}" unless spec_version == "1.0"
-        input = env["rack.input"]
-        data = if input
-                 input.set_encoding content_type.charset if content_type&.charset
-                 input.read
-               end
-        attributes = { "spec_version" => spec_version, "data" => data }
-        attributes["data_content_type"] = content_type if content_type
-        omit_names = ["specversion", "spec_version", "data", "datacontenttype", "data_content_type"]
-        env.each do |key, value|
-          match = /^HTTP_CE_(\w+)$/.match key
-          next unless match
-          attr_name = match[1].downcase
-          attributes[attr_name] = value unless omit_names.include? attr_name
-        end
-        Event.create spec_version: spec_version, attributes: attributes
-      end
-
-      ##
-      # Encode a single event to content data in the given format.
-      #
-      # The result is a two-element array where the first element is a headers
-      # list (as defined in the Rack specification) and the second is a string
-      # containing the HTTP body content. The headers list will contain only
-      # one header, a `Content-Type` whose value is of the form
-      # `application/cloudevents+format`.
-      #
-      # @param event [FunctionsFramework::CloudEvents::Event] The event.
-      # @param format [String] The format code (e.g. "json")
-      # @param format_args [keywords] Extra args to pass to the formatter.
-      # @return [Array(headers,String)]
-      #
-      def encode_structured_content event, format, **format_args
-        handlers = @structured_formatters[format] || []
-        handlers.reverse_each do |handler|
-          content = handler.encode event, **format_args
-          return [{ "Content-Type" => "application/cloudevents+#{format}" }, content] if content
-        end
-        raise HttpContentError, "Unknown cloudevents format: #{format.inspect}"
-      end
-
-      ##
-      # Encode a batch of events to content data in the given format.
-      #
-      # The result is a two-element array where the first element is a headers
-      # list (as defined in the Rack specification) and the second is a string
-      # containing the HTTP body content. The headers list will contain only
-      # one header, a `Content-Type` whose value is of the form
-      # `application/cloudevents-batch+format`.
-      #
-      # @param events [Array<FunctionsFramework::CloudEvents::Event>] The batch
-      #     of events.
-      # @param format [String] The format code (e.g. "json").
-      # @param format_args [keywords] Extra args to pass to the formatter.
-      # @return [Array(headers,String)]
-      #
-      def encode_batched_content events, format, **format_args
-        handlers = @batched_formatters[format] || []
-        handlers.reverse_each do |handler|
-          content = handler.encode_batch events, **format_args
-          return [{ "Content-Type" => "application/cloudevents-batch+#{format}" }, content] if content
-        end
-        raise HttpContentError, "Unknown cloudevents format: #{format.inspect}"
-      end
-
-      ##
-      # Encode an event to content and headers, in binary content mode.
-      #
-      # The result is a two-element array where the first element is a headers
-      # list (as defined in the Rack specification) and the second is a string
-      # containing the HTTP body content.
-      #
-      # @param event [FunctionsFramework::CloudEvents::Event] The event.
-      # @return [Array(headers,String)]
-      #
-      def encode_binary_content event
-        headers = {}
-        body = nil
-        event.to_h.each do |key, value|
-          if key == "data"
-            body = value
-          elsif key == "datacontenttype"
-            headers["Content-Type"] = value
-          else
-            headers["CE-#{key}"] = value
-          end
-        end
-        if body.is_a? ::String
-          headers["Content-Type"] ||= if body.encoding == ::Encoding.ASCII_8BIT
-                                        "application/octet-stream"
-                                      else
-                                        "text/plain; charset=#{body.encoding.name.downcase}"
-                                      end
-        elsif body.nil?
-          headers.delete "Content-Type"
-        else
-          body = ::JSON.dump body
-          headers["Content-Type"] ||= "application/json; charset=#{body.encoding.name.downcase}"
-        end
-        [headers, body]
-      end
-    end
-  end
-end

data/lib/functions_framework/cloud_events/json_format.rb

@@ -1,122 +0,0 @@
-# Copyright 2020 Google LLC
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     https://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-require "base64"
-require "json"
-
-module FunctionsFramework
-  module CloudEvents
-    ##
-    # An implementation of JSON format and JSON batch format.
-    #
-    # See https://github.com/cloudevents/spec/blob/master/json-format.md
-    #
-    class JsonFormat
-      ##
-      # Decode an event from the given input JSON string.
-      #
-      # @param json [String] A JSON-formatted string
-      # @return [FunctionsFramework::CloudEvents::Event]
-      #
-      def decode json, **_other_kwargs
-        structure = ::JSON.parse json
-        decode_hash_structure structure
-      end
-
-      ##
-      # Encode an event to a JSON string.
-      #
-      # @param event [FunctionsFramework::CloudEvents::Event] An input event.
-      # @param sort [boolean] Whether to sort keys of the JSON output.
-      # @return [String] The JSON representation.
-      #
-      def encode event, sort: false, **_other_kwargs
-        structure = encode_hash_structure event
-        structure = sort_keys structure if sort
-        ::JSON.dump structure
-      end
-
-      ##
-      # Decode a batch of events from the given input string.
-      #
-      # @param json [String] A JSON-formatted string
-      # @return [Array<FunctionsFramework::CloudEvents::Event>]
-      #
-      def decode_batch json, **_other_kwargs
-        structure_array = Array(::JSON.parse(json))
-        structure_array.map do |structure|
-          decode_hash_structure structure
-        end
-      end
-
-      ##
-      # Encode a batch of event to a JSON string.
-      #
-      # @param events [Array<FunctionsFramework::CloudEvents::Event>] An array
-      #     of input events.
-      # @param sort [boolean] Whether to sort keys of the JSON output.
-      # @return [String] The JSON representation.
-      #
-      def encode_batch events, sort: false, **_other_kwargs
-        structure_array = Array(events).map do |event|
-          structure = encode_hash_structure event
-          sort ? sort_keys(structure) : structure
-        end
-        ::JSON.dump structure_array
-      end
-
-      ##
-      # Decode a single event from a hash data structure with keys and types
-      # conforming to the JSON event format.
-      #
-      # @param structure [Hash] An input hash.
-      # @return [FunctionsFramework::CloudEvents::Event]
-      #
-      def decode_hash_structure structure
-        if structure.key? "data_base64"
-          structure = structure.dup
-          structure["data"] = ::Base64.decode64 structure.delete "data_base64"
-        end
-        Event.create spec_version: structure["specversion"], attributes: structure
-      end
-
-      ##
-      # Encode a single event to a hash data structure with keys and types
-      # conforming to the JSON event format.
-      #
-      # @param event [FunctionsFramework::CloudEvents::Event] An input event.
-      # @return [String] The hash structure.
-      #
-      def encode_hash_structure event
-        structure = event.to_h
-        data = structure["data"]
-        if data.is_a?(::String) && data.encoding == ::Encoding::ASCII_8BIT
-          structure.delete "data"
-          structure["data_base64"] = ::Base64.encode64 data
-        end
-        structure
-      end
-
-      private
-
-      def sort_keys hash
-        result = {}
-        hash.keys.sort.each do |key|
-          result[key] = hash[key]
-        end
-        result
-      end
-    end
-  end
-end