Class: SRack::AWSLambdaHandler

Inherits:
Object
  • Object
show all
Defined in:
lib/srack.rb

Instance Method Summary collapse

Constructor Details

#initialize(rack_app) ⇒ AWSLambdaHandler

Returns a new instance of AWSLambdaHandler.



13
14
15
 
# File 'lib/srack.rb', line 13

def initialize(rack_app)
  @app = rack_app
end

Instance Method Details

#handle(event:, context:) ⇒ Object 



17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
 
# File 'lib/srack.rb', line 17

def handle(event:, context:)
  # Check if the body is base64 encoded. If it is, try to decode it
  body = if event["isBase64Encoded"]
           Base64.decode64 event["body"]
         else
           event["body"]
         end || ""

  # Rack expects the querystring in plain text, not a hash
  headers = event.fetch "headers", {}

  # Environment required by Rack (http://www.rubydoc.info/github/rack/rack/file/SPEC)
  env = {
    "REQUEST_METHOD" => event.fetch("requestContext").fetch("http").fetch("method"),
    "SCRIPT_NAME" => "",
    "PATH_INFO" => event.fetch("rawPath", ""),
    # XXX: fix QUERY_STRING handling
    "QUERY_STRING" => (event["queryStringParameters"] || {}).map { |k, v| "#{k}=#{v}" }.join("&"),
    "SERVER_NAME" => headers.fetch("Host", "localhost"),
    "SERVER_PORT" => headers.fetch("X-Forwarded-Port", 443).to_s,

    "rack.version" => Rack::VERSION,
    "rack.url_scheme" => headers.fetch("CloudFront-Forwarded-Proto") { headers.fetch("X-Forwarded-Proto", "https") },
    "rack.input" => StringIO.new(body),
    "rack.errors" => $stderr
  }

  # Pass request headers to Rack if they are available
  headers.each_pair do |key, value|
    # 'CloudFront-Forwarded-Proto' => 'CLOUDFRONT_FORWARDED_PROTO'
    # Content-Type and Content-Length are handled specially per the Rack SPEC linked above.
    name = key.upcase.tr "-", "_"
    header = case name
    when "CONTENT_TYPE", "CONTENT_LENGTH"
      name
    else
      "HTTP_#{name}"
    end
    env[header] = value.to_s
  end

  begin
    # Response from Rack must have status, headers and body
    status, headers, body = @app.call env

    # body is an array. We combine all the items to a single string
    body_content = ""
    body.each do |item|
      body_content += item.to_s
    end

    # We return the structure required by AWS API Gateway since we integrate with it
    # https://docs.aws.amazon.com/apigateway/latest/developerguide/set-up-lambda-proxy-integrations.html
    response = {
      "statusCode" => status,
      "headers" => headers,
      "body" => body_content
    }
    if event["requestContext"].has_key?("elb")
      # Required if we use Application Load Balancer instead of API Gateway
      response["isBase64Encoded"] = false
    end
  rescue Exception => exception
    # If there is _any_ exception, we return a 500 error with an error message
    response = {
      "statusCode" => 500,
      "body" => exception.message
    }
  end

  # By default, the response serializer will call #to_json for us
  response
end