lamby 1.0.2 → 2.0.0

data/docker-compose.yml

@@ -0,0 +1,14 @@
+version: '3.7'
+services:
+  cicd:
+    build: .
+    environment:
+      - AWS_PROFILE=${AWS_PROFILE-default}
+      - RAILS_ENV=${RAILS_ENV-test}
+      - AWS_DEFAULT_REGION=${AWS_DEFAULT_REGION-us-east-1}
+      - SAM_CLI_TELEMETRY=0
+      - DIDPWD=${PWD}
+    volumes:
+      - ~/.aws:/root/.aws:delegated
+      - .:/var/task:delegated
+      - /var/run/docker.sock:/var/run/docker.sock

data/lamby.gemspec

@@ -24,7 +24,7 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency 'bundler'
   spec.add_development_dependency 'rake'
   spec.add_development_dependency 'minitest'
+  spec.add_development_dependency 'minitest-focus'
   spec.add_development_dependency 'mocha'
   spec.add_development_dependency 'pry'
-  spec.add_development_dependency 'sqlite3'
 end

data/lib/lamby.rb

@@ -6,7 +6,8 @@ require 'lamby/version'
 require 'lamby/sam_helpers'
 require 'lamby/rack'
 require 'lamby/rack_alb'
-require 'lamby/rack_api'
+require 'lamby/rack_rest'
+require 'lamby/rack_http'
 require 'lamby/debug'
 require 'lamby/handler'
 require 'rails/railtie'

data/lib/lamby/handler.rb

@@ -44,14 +44,27 @@ module Lamby
       self
     end
 
+    def base64_encodeable?
+      @headers && (
+        @headers['Content-Transfer-Encoding'] == 'binary' ||
+        @headers['X-Lamby-Base64'] == '1'
+      )
+    end
+
+    def body64
+      Base64.strict_encode64(body)
+    end
+
     private
 
     def rack
       @rack ||= case @options[:rack]
-      when :api
-        Lamby::RackApi.new @event, @context
-      else
+      when :rest, :api
+        Lamby::RackRest.new @event, @context
+      when :alb
         Lamby::RackAlb.new @event, @context
+      else
+        Lamby::RackHttp.new @event, @context
       end
     end
 

data/lib/lamby/logger.rb

@@ -1,19 +1,23 @@
 require 'logger'
 
-ENV['RAILS_LOG_TO_STDOUT'] = '1'
+unless ENV['LAMBY_TEST']
 
-module Lamby
-  module Logger
+  ENV['RAILS_LOG_TO_STDOUT'] = '1'
 
-    def initialize(*args)
-      args[0] = STDOUT
-      super(*args)
-    end
+  module Lamby
+    module Logger
+
+      def initialize(*args)
+        args[0] = STDOUT
+        super(*args)
+      end
 
+    end
   end
-end
 
-Logger.prepend Lamby::Logger
+  Logger.prepend Lamby::Logger
+
+end
 
 # TODO: Railtie initializer
 # Rails.application.config.logger = ActiveSupport::TaggedLogging.new(

data/lib/lamby/rack.rb

@@ -6,6 +6,7 @@ module Lamby
     LAMBDA_EVENT = 'lambda.event'.freeze
     LAMBDA_CONTEXT = 'lambda.context'.freeze
     HTTP_X_REQUESTID = 'HTTP_X_REQUEST_ID'.freeze
+    HTTP_COOKIE = 'HTTP_COOKIE'.freeze
 
     attr_reader :event, :context
 
@@ -22,14 +23,6 @@ module Lamby
       {}
     end
 
-    def api?
-      false
-    end
-
-    def alb?
-      false
-    end
-
     def multi_value?
       false
     end
@@ -70,7 +63,16 @@ module Lamby
     end
 
     def query_string
-      @query_string ||= event['queryStringParameters'].try(:to_query)
+      @query_string ||= if event.key?('rawQueryString')
+        event['rawQueryString']
+      elsif event.key?('multiValueQueryStringParameters')
+        query = event['multiValueQueryStringParameters'] || {}
+        query.map do |key, value|
+          value.map{ |v| "#{key}=#{v}" }.join('&')
+        end.flatten.join('&')
+      else
+        event['queryStringParameters'].try(:to_query)
+      end
     end
 
     def base64_encoded?

data/lib/lamby/rack_alb.rb

@@ -53,20 +53,17 @@ module Lamby
     end
 
     def headers_multi
-      (event['multiValueHeaders'] || {}).transform_values do |v|
-        v.is_a?(Array) ? v.first : v
-      end
-    end
-
-    def query_string
-      @query_string ||= multi_value? ? query_string_multi : super
-    end
-
-    def query_string_multi
-      query = event['multiValueQueryStringParameters'] || {}
-      string = query.map do |key, value|
-        value.map{ |v| "#{key}=#{v}" }.join('&')
-      end.flatten.join('&')
+      Hash[(event['multiValueHeaders'] || {}).map do |k,v|
+        if v.is_a?(Array)
+          if k == 'x-forwarded-for'
+            [k, v.join(', ')]
+          else
+            [k, v.first]
+          end
+        else
+          [k,v]
+        end
+      end]
     end
 
   end

data/lib/lamby/rack_http.rb

@@ -0,0 +1,80 @@
+module Lamby
+  class RackHttp < Lamby::Rack
+
+    def response(handler)
+      if handler.base64_encodeable?
+        { isBase64Encoded: true, body: handler.body64 }
+      else
+        super
+      end
+    end
+
+    private
+
+    def env_base
+      { ::Rack::REQUEST_METHOD => request_method,
+        ::Rack::SCRIPT_NAME => '',
+        ::Rack::PATH_INFO => path_info,
+        ::Rack::QUERY_STRING => query_string,
+        ::Rack::SERVER_NAME => server_name,
+        ::Rack::SERVER_PORT => server_port,
+        ::Rack::SERVER_PROTOCOL => server_protocol,
+        ::Rack::RACK_VERSION => ::Rack::VERSION,
+        ::Rack::RACK_URL_SCHEME => 'https',
+        ::Rack::RACK_INPUT => StringIO.new(body || ''),
+        ::Rack::RACK_ERRORS => $stderr,
+        ::Rack::RACK_MULTITHREAD => false,
+        ::Rack::RACK_MULTIPROCESS => false,
+        ::Rack::RACK_RUNONCE => false,
+        LAMBDA_EVENT => event,
+        LAMBDA_CONTEXT => context
+      }.tap do |env|
+        ct = content_type
+        cl = content_length
+        env['CONTENT_TYPE'] = ct if ct
+        env['CONTENT_LENGTH'] = cl if cl
+      end
+    end
+
+    def env_headers
+      super.tap do |hdrs|
+        if event['cookies'].present?
+          hdrs[HTTP_COOKIE] = event['cookies'].join('; ')
+        end
+      end
+    end
+
+    def request_method
+      event.dig('requestContext', 'http', 'method') || event['httpMethod']
+    end
+
+    # Using custom domain names with v1.0 yields a good `path` parameter sans
+    # stage. However, v2.0 and others do not. So we are just going to remove stage
+    # no matter waht from other places for both.
+    #
+    def path_info
+      stage = event.dig('requestContext', 'stage')
+      spath = event.dig('requestContext', 'http', 'path') || event.dig('requestContext', 'path')
+      spath.sub /\A\/#{stage}/, ''
+    end
+
+    def server_name
+      headers['x-forwarded-host'] ||
+        headers['X-Forwarded-Host'] ||
+        headers['host'] ||
+        headers['Host']
+    end
+
+    def server_port
+      headers['x-forwarded-port'] || headers['X-Forwarded-Port']
+    end
+
+    def server_protocol
+      event.dig('requestContext', 'http', 'protocol') ||
+        event.dig('requestContext', 'protocol') ||
+        'HTTP/1.1'
+    end
+
+
+  end
+end

data/lib/lamby/{rack_api.rb → rack_rest.rb}

@@ -1,8 +1,12 @@
 module Lamby
-  class RackApi < Lamby::Rack
+  class RackRest < Lamby::Rack
 
-    def api?
-      true
+    def response(handler)
+      if handler.base64_encodeable?
+        { isBase64Encoded: true, body: handler.body64 }
+      else
+        super
+      end
     end
 
     private

data/lib/lamby/ssm_parameter_store.rb

@@ -68,7 +68,7 @@ module Lamby
 
     def dotenv_contents
       params.each_with_object('') do |param, contents|
-        line = "export #{param.env}=#{param.value}\n"
+        line = "#{param.env}=#{param.value}\n"
         contents << line
       end
     end

data/lib/lamby/templates.rake

@@ -1,6 +1,7 @@
 installers = {
-  'API Gateway' => :api_gateway,
-  'Application Load Balancer' => :application_load_balancer
+  'API Gateway (HTTP API)' => :http,
+  'API Gateway (REST API)' => :rest,
+  'Application Load Balancer' => :alb
 }.freeze
 
 namespace :lamby do
@@ -32,5 +33,5 @@ namespace :lamby do
   end
 
   desc "Install Lamby files for #{installers.first.first}"
-  task install: 'install:application_load_balancer'
+  task install: 'install:http'
 end

data/lib/lamby/templates/{application_load_balancer.rb → alb.rb}

@@ -1,4 +1,4 @@
-$LAMBY_INSTALLER_NAME = 'application_load_balancer'
+$LAMBY_INSTALLER_NAME = 'alb'
 load 'lamby/templates/shared.rb'
 
 say '==> Running Lamby Application Load Balancer installer...'

data/lib/lamby/templates/{application_load_balancer → alb}/build

@@ -1,4 +1,4 @@
-#!/bin/bash
+#!/usr/bin/env bash
 set -e
 
 # Clean any previous bundle/builds.

data/lib/lamby/templates/{application_load_balancer → alb}/deploy

@@ -1,4 +1,4 @@
-#!/bin/bash
+#!/usr/bin/env bash
 set -e
 
 export RAILS_ENV=${RAILS_ENV:="production"}

data/lib/lamby/templates/{application_load_balancer → alb}/template.yaml

@@ -70,7 +70,7 @@ Resources:
     Properties:
       CodeUri: .
       Handler: app.handler
-      Runtime: ruby2.5
+      Runtime: ruby2.7
       MemorySize: 512
       Timeout: 30
       Environment:

data/lib/lamby/templates/{api_gateway.rb → http.rb}

@@ -1,7 +1,7 @@
-$LAMBY_INSTALLER_NAME = 'api_gateway'
+$LAMBY_INSTALLER_NAME = 'http'
 load 'lamby/templates/shared.rb'
 
-say '==> Running Lamby API Gateway installer...'
+say '==> Running Lamby API Gateway (HTTP API) installer...'
 
 say 'Copying files...'
 copy_file tpl_file('app.rb'), app_file('app.rb')

data/lib/lamby/templates/{api_gateway → http}/app.rb

@@ -6,5 +6,5 @@ require_relative 'config/environment'
 $app = Rack::Builder.new { run Rails.application }.to_app
 
 def handler(event:, context:)
-  Lamby.handler $app, event, context, rack: :api
+  Lamby.handler $app, event, context, rack: :http
 end

data/lib/lamby/templates/{api_gateway → http}/build

@@ -1,4 +1,4 @@
-#!/bin/bash
+#!/usr/bin/env bash
 set -e
 
 # Clean any previous bundle/builds.

data/lib/lamby/templates/{api_gateway → http}/deploy

@@ -1,4 +1,4 @@
-#!/bin/bash
+#!/usr/bin/env bash
 set -e
 
 export RAILS_ENV=${RAILS_ENV:="production"}

data/lib/lamby/templates/http/template.yaml

@@ -0,0 +1,48 @@
+AWSTemplateFormatVersion: '2010-09-09'
+Transform: AWS::Serverless-2016-10-31
+Description: APPNAMEHERE Lambda (HTTP API)
+
+Parameters:
+
+  RailsEnv:
+    Type: String
+    Default: production
+    AllowedValues:
+      - staging
+      - production
+
+Resources:
+
+  RailsFunction:
+    Type: AWS::Serverless::Function
+    Properties:
+      CodeUri: .
+      Handler: app.handler
+      Runtime: ruby2.7
+      MemorySize: 512
+      Timeout: 30
+      Environment:
+        Variables:
+          RAILS_ENV: !Ref RailsEnv
+      FunctionName: !Join [ '', [ 'APPNAMEHERE-', !Ref RailsEnv, '-', !Ref 'AWS::Region' ] ]
+      Events:
+        RailsApiProxy:
+          Type: HttpApi
+          Properties:
+            ApiId: !Ref RailsApi
+      AutoPublishAlias: live
+
+  RailsApi:
+    Type: AWS::Serverless::HttpApi
+    Properties:
+      StageName: !Ref RailsEnv
+
+Outputs:
+
+  RailsApiUrl:
+    Description: API Gateway Endpoint
+    Value: !Sub "https://${RailsApi}.execute-api.${AWS::Region}.amazonaws.com/${RailsEnv}/"
+
+  RailsFunctionArn:
+    Description: Lambda ARN
+    Value: !GetAtt RailsFunction.Arn

data/lib/lamby/templates/rest.rb

@@ -0,0 +1,25 @@
+$LAMBY_INSTALLER_NAME = 'rest'
+load 'lamby/templates/shared.rb'
+
+say '==> Running Lamby API Gateway (REST API) installer...'
+
+say 'Copying files...'
+copy_file tpl_file('app.rb'), app_file('app.rb')
+copy_file tpl_file('template.yaml'), app_file('template.yaml')
+gsub_file app_file('template.yaml'), /APPNAMEHERE/, appname
+
+say 'Adding to .gitignore...'
+FileUtils.touch app_file('.gitignore')
+append_to_file app_file('.gitignore'), <<-GITIGNORE.strip_heredoc
+  # Lamby
+  /.aws-sam
+GITIGNORE
+
+say 'Creating ./bin files for build and deploy...'
+copy_file tpl_file('build'), app_file('bin/build')
+chmod app_file('bin/build'), 0755
+copy_file tpl_file('deploy'), app_file('bin/deploy')
+chmod app_file('bin/deploy'), 0755
+gsub_file app_file('bin/deploy'), /APPNAMEHERE/, appname.downcase
+
+say 'Welcome to AWS Lambda and Rails 🎉', :green

data/lib/lamby/templates/rest/app.rb

@@ -0,0 +1,10 @@
+require_relative 'config/boot'
+require 'lamby'
+require_relative 'config/application'
+require_relative 'config/environment'
+
+$app = Rack::Builder.new { run Rails.application }.to_app
+
+def handler(event:, context:)
+  Lamby.handler $app, event, context, rack: :rest
+end