yake 0.2.1 → 0.4.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d97920003245ea42fec1eb6c56d1addc53a743e8a21540f51a86719fb3fe69aa
-  data.tar.gz: 34f51c677d95df8c6773122838815bfd9cf6124177d03192b5f15cc591ff92d1
+  metadata.gz: 6bd208d64f868cf763f2cb19ff4b766a771a3011b6edda1918cf4e3f9436b1c0
+  data.tar.gz: 439c99ae77a312ad882f8d5b58061dd9841e7f163a6e01476a54845ad38efab5
 SHA512:
-  metadata.gz: 900588eb497598196b0081702bf1ac176b0a88167c1c316414887bf56a7f1f5ffe3d746e7f2760ae06f07080242de9cf1fe519f35d9b2bf221aae04842c18b7b
-  data.tar.gz: 9e01f2b55927a2430138556a008353cb22c8405027f97c3dcd3ac566d6942f0ffefb0281df9d290726689738403278a84c682f5fe761d8d21f8998872a8fab5b
+  metadata.gz: 2052948f82b026240655aaf07d6a5a15dc687aa432d97fdcbec0b9c4cb88aca4c64f50d7051336f1a7b99d005b2f9d6234f50a750221897d86a6d033012200e6
+  data.tar.gz: 8800efb2e6428321aadc94c84788f8918290ab620476f51e20d1e0cfd66170bd4a3002ff86c5d4554de5c3aceaa81ecefb14fcb393fe0f5ac368d70233a83429

data/lib/yake/api.rb

@@ -1,32 +1,33 @@
 # frozen_string_literal: true
 
-require "base64"
-require "json"
+require 'base64'
+require 'json'
 
-require "yake"
-require_relative "errors"
+require 'yake'
+require_relative 'errors'
 
 module Yake
   module API
     module DSL
       ##
       # Proxy handler for HTTP requests from Slack
-      def route(event, context = nil)
+      def route(event, context = nil, &block)
         # Extract route method
-        method = event["routeKey"]
+        method = event['routeKey']
         raise Yake::Errors::UndeclaredRoute, method unless respond_to?(method)
 
         # Normalize headers
-        event["headers"]&.transform_keys!(&:downcase)
+        event['headers']&.transform_keys!(&:downcase)
 
         # Decode body if Base64-encoded
-        if event["isBase64Encoded"]
-          body = Base64.strict_decode64(event["body"])
-          event.update("body" => body, "isBase64Encoded" => false)
+        if event['isBase64Encoded']
+          body = Base64.strict_decode64(event['body'])
+          event.update('body' => body, 'isBase64Encoded' => false)
         end
 
         # Execute request
-        send(method, event, context)
+        res = send(method, event, context)
+        block_given? ? yield(res) : res
       end
 
       ##
@@ -37,10 +38,10 @@ module Yake
         Yake.logger&.send(status_code.to_i >= 400 ? :error : :info, log)
 
         # Set headers
-        content_length = (body&.length || 0).to_s
+        content_length = (body&.bytesize || 0).to_s
         to_s_downcase  = -> (key) { key.to_s.downcase }
         headers = {
-          "content-length" => content_length,
+          'content-length' => content_length,
           **(@headers || {}),
           **headers,
         }.transform_keys(&to_s_downcase).compact

data/lib/yake/datadog.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+require 'datadog/lambda'
+require 'yake'
+
+module Yake
+  module Datadog
+    class MockContext < Struct.new(
+      :clock_diff,
+      :deadline_ms,
+      :aws_request_id,
+      :invoked_function_arn,
+      :log_group_name,
+      :log_stream_name,
+      :function_name,
+      :memory_limit_in_mb,
+      :function_version)
+
+      def invoked_function_arn
+        @invoked_function_arn ||= begin
+          region = ENV['AWS_REGION'] || ENV['AWS_DEFAULT_REGION'] || 'us-east-1'
+          "arn:aws:lambda:#{region}:123456789012:function-name"
+        end
+      end
+    end
+
+    module DSL
+      include Yake::DSL
+
+      ##
+      # Datadog handler wrapper
+      def datadog(name, &block)
+        define_method(name) do |event:nil, context:nil|
+          context ||= MockContext.new
+          ::Datadog::Lambda.wrap(event, context) do
+            Yake.wrap(event, context, &block)
+          end
+        end
+      end
+    end
+  end
+end
+
+extend Yake::Datadog::DSL

data/lib/yake/dsl.rb

@@ -1,8 +1,8 @@
 # frozen_string_literal: true
 
-require "json"
+require 'json'
 
-require_relative "logger"
+require_relative 'logger'
 
 module Yake
   module DSL
@@ -14,6 +14,12 @@ module Yake
       end
     end
 
+    ##
+    # Helper to get logger
+    def logger
+      Yake.logger
+    end
+
     ##
     # Turn logging on/off
     def logging(switch, logger = nil)

data/lib/yake/errors.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Yake
   module Errors
     class Error < StandardError; end

data/lib/yake/logger.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
-require "json"
-require "logger"
+require 'json'
+require 'logger'
 
 module Yake
   module Logger
@@ -21,7 +21,7 @@ module Yake
       Format = "%s %s %s\n"
 
       def call(severity, time, progname, msg)
-        Format % [ severity, progname.nil? ? "-" : "RequestId: #{ progname }", msg2str(msg).strip ]
+        Format % [ severity, progname.nil? ? '-' : "RequestId: #{ progname }", msg2str(msg).strip ]
       end
     end
   end

data/lib/yake/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Yake
-  VERSION = "0.2.1"
+  VERSION = '0.4.1'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: yake
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.4.1
 platform: ruby
 authors:
 - Alexander Mancevice
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-05-29 00:00:00.000000000 Z
+date: 2021-10-15 00:00:00.000000000 Z
 dependencies: []
 description:
 email:
@@ -22,6 +22,7 @@ files:
 - README.md
 - lib/yake.rb
 - lib/yake/api.rb
+- lib/yake/datadog.rb
 - lib/yake/dsl.rb
 - lib/yake/errors.rb
 - lib/yake/logger.rb
@@ -45,7 +46,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.6
+rubygems_version: 3.2.22
 signing_key:
 specification_version: 4
 summary: Rake-like DSL for declaring AWS Lambda function handlers