Support EventBridge Events

CHANGELOG.md

@@ -4,6 +4,12 @@
 
 See this http://keepachangelog.com link for information on how we want this documented formatted.
 
+## v2.7.0
+
+#### Added
+
+- Support EventBridge events in handler with default proc to log.
+
 ## v2.6.3
 
 #### Added

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    lamby (2.6.3)
+    lamby (2.7.0)
       rack
 
 GEM

lib/lamby.rb

@@ -2,6 +2,7 @@
 require 'rack'
 require 'base64'
 require 'lamby/version'
+require 'lamby/config'
 require 'lamby/sam_helpers'
 require 'lamby/rack'
 require 'lamby/rack_alb'
@@ -23,6 +24,10 @@ def handler(app, event, context, options = {})
     Handler.call(app, event, context, options)
   end
 
+  def config
+    Lamby::Config.config
+  end
+
   autoload :SsmParameterStore, 'lamby/ssm_parameter_store'
 
 end

lib/lamby/config.rb

@@ -0,0 +1,47 @@
+module Lamby
+  module Config
+
+    def configure
+      yield(config)
+      config
+    end
+
+    def reconfigure
+      config.reconfigure { |c| yield(c) if block_given? }
+    end
+
+    def config
+      @config ||= Configuration.new
+    end
+
+    extend self
+
+  end
+
+  class Configuration
+
+    def initialize
+      initialize_defaults
+    end
+
+    def reconfigure
+      instance_variables.each { |var| instance_variable_set var, nil }
+      initialize_defaults
+      yield(self) if block_given?
+      self
+    end
+
+    def initialize_defaults
+      @event_bridge_handler = lambda { |event, context| puts(event) }
+    end
+
+    def event_bridge_handler
+      @event_bridge_handler
+    end
+
+    def event_bridge_handler=(func)
+      @event_bridge_handler = func
+    end
+
+  end
+end

lib/lamby/handler.rb

@@ -47,7 +47,7 @@ def body
     def call
       return self if @called
       @status, @headers, @body = call_app
-      set_cookies
+      set_cookies if rack?
       @called = true
       self
     end
@@ -78,20 +78,34 @@ def rack
     end
 
     def rack_response
-      rack.response(self)
+      rack? ? rack.response(self) : {}
     end
 
     def call_app
       if Debug.on?(@event)
         Debug.call @event, @context, rack.env
-      else
+      elsif rack?
         @app.call rack.env
+      elsif event_bridge?
+        Lamby.config.event_bridge_handler.call @event, @context
+        [200, {}, StringIO.new('')]
+      else
+        [404, {}, StringIO.new('')]
       end
     end
 
     def content_encoding_compressed?(hdrs)
       content_encoding_header = hdrs['Content-Encoding'] || ''
       content_encoding_header.split(', ').any? { |h| ['br', 'gzip'].include?(h) }
     end
+
+    def rack?
+      @event.key?('httpMethod') || @event.dig('requestContext', 'http')
+    end
+
+    def event_bridge?
+      Lamby.config.event_bridge_handler &&
+        @event.key?('source') && @event.key?('detail') && @event.key?('detail-type')
+    end
   end
 end

lib/lamby/railtie.rb

@@ -1,8 +1,6 @@
 module Lamby
   class Railtie < ::Rails::Railtie
 
-    config.lamby = ActiveSupport::OrderedOptions.new
-
     rake_tasks do
       load 'lamby/tasks.rake'
       load 'lamby/templates.rake'

lib/lamby/version.rb

@@ -1,3 +1,3 @@
 module Lamby
-  VERSION = '2.6.3'
+  VERSION = '2.7.0'
 end

test/handler_test.rb

@@ -275,6 +275,36 @@ class HandlerTest < LambySpec
 
   end
 
+  describe 'event_bridge' do
+
+    let(:event) do
+      {
+        "version" => "0",
+        "id"=>"0874bcac-1dac-2393-637f-201025f217b0",
+        "detail-type"=>"orderCreated",
+        "source"=>"com.myorg.stores",
+        "account"=>"123456789012",
+        "time"=>"2021-04-29T13:51:41Z",
+        "region"=>"us-east-1",
+        "resources"=>[],
+        "detail"=>{"id" => "123"}
+      }
+    end
+
+    it 'has a configurable proc' do
+      expect(Lamby.config.event_bridge_handler).must_be_instance_of Proc
+      Lamby.config.event_bridge_handler = lambda { |e,c| "#{e}#{c}" }
+      r = Lamby.config.event_bridge_handler.call(1,2)
+      expect(r).must_equal '12'
+    end
+
+    it 'basic event puts to log' do
+      out = capture(:stdout) { @result = Lamby.handler app, event, context }
+      expect(out).must_match %r{0874bcac-1dac-2393-637f-201025f217b0}
+    end
+
+  end
+
   private
 
   def session_cookie(result)

test/test_helper.rb

@@ -18,6 +18,7 @@ class LambySpec < Minitest::Spec
           TestHelpers::StreamHelpers
 
   before do
+    Lamby.config.reconfigure
     delete_dummy_files
   end