Get it working in lambda

Author: iHiD

Diff for: .gitignore

@@ -1,4 +1,5 @@
+.aws-sam
 /fixtures/
 /tmp/
 
-node_modules
+node_modules

Diff for: Dockerfile

@@ -6,10 +6,12 @@ WORKDIR /var/task
 
 RUN gem install json -v '2.3.1' 
 
-COPY . .
+COPY Gemfile Gemfile.lock ./
 
 RUN bundle config set deployment 'true' && \
     bundle config set without 'development test' && \
     bundle install
 
-CMD [ "source.SnippetExtractor.process_request." ]
+COPY . .
+
+CMD ["bin/run.run"]

Diff for: Gemfile

@@ -8,10 +8,12 @@ end
 gem 'json'
 gem 'mandate'
 gem 'rake'
+gem 'zeitwerk'
 
 group :test do
   gem 'minitest', '~> 5.10', '!= 5.10.2'
   gem 'minitest-stub-const'
   gem 'mocha'
   gem 'simplecov', '~> 0.17.0'
+  gem 'httparty'
 end

Diff for: Gemfile.lock

@@ -2,29 +2,39 @@ GEM
   remote: https://rubygems.org/
   specs:
     docile (1.3.2)
+    httparty (0.18.1)
+      mime-types (~> 3.0)
+      multi_xml (>= 0.5.2)
     json (2.3.1)
     mandate (0.3.0)
+    mime-types (3.3.1)
+      mime-types-data (~> 3.2015)
+    mime-types-data (3.2021.0225)
     minitest (5.14.2)
     minitest-stub-const (0.6)
     mocha (1.11.2)
+    multi_xml (0.6.0)
     rake (13.0.1)
     simplecov (0.17.1)
       docile (~> 1.1)
       json (>= 1.8, < 3)
       simplecov-html (~> 0.10.0)
     simplecov-html (0.10.2)
+    zeitwerk (2.4.2)
 
 PLATFORMS
   ruby
 
 DEPENDENCIES
+  httparty
   json
   mandate
   minitest (~> 5.10, != 5.10.2)
   minitest-stub-const
   mocha
   rake
   simplecov (~> 0.17.0)
+  zeitwerk
 
 BUNDLED WITH
    2.1.4

Diff for: bin/run.rb

@@ -1,3 +1,6 @@
+require "zeitwerk"
 load File.expand_path('../../lib/snippet_extractor.rb', __FILE__)
 
-SnippetExtractor.extract(ARGV[0], ARGV[1], ARGV[2])
+def run(event:, context:)
+  SnippetExtractor.process_request(event: event, context: context)
+end

Diff for: lib/snippet_extractor/process_request.rb

@@ -14,19 +14,18 @@ def call
         headers: {
           'Content-Length': snippet.bytesize,
           'Content-Type': 'application/plain; charset=utf-8',
-          'Content-disposition': 'attachment;snippet.txt'
         },
         isBase64Encoded: false,
         body: snippet
       }
     end
 
     def source_code
-      event['source_code']
+      event['queryStringParameters']['source_code']
     end
 
     def language
-      event['language']
+      event['queryStringParameters']['language']
     end
   end
 end

Diff for: template.yaml

@@ -0,0 +1,41 @@
+AWSTemplateFormatVersion: '2010-09-09'
+Transform: AWS::Serverless-2016-10-31
+Description: >
+  sam-app
+
+  Sample SAM Template for sam-app
+
+# More info about Globals: https://github.com/awslabs/serverless-application-model/blob/master/docs/globals.rst
+Globals:
+  Function:
+    Timeout: 3
+
+Resources:
+  SnippetExtractorFunction:
+    Type: AWS::Serverless::Function # More info about Function Resource: https://github.com/awslabs/serverless-application-model/blob/master/versions/2016-10-31.md#awsserverlessfunction
+    Properties:
+      PackageType: Image
+      Events:
+        SnippetExtractor:
+          Type: Api # More info about API Event Source: https://github.com/awslabs/serverless-application-model/blob/master/versions/2016-10-31.md#api
+          Properties:
+            Path: /extract
+            Method: get
+    Metadata:
+      DockerTag: ruby2.7-v1
+      DockerContext: ./
+      Dockerfile: Dockerfile
+
+Outputs:
+  # ServerlessRestApi is an implicit API created out of Events key under Serverless::Function
+  # Find out more about other implicit resources you can reference within SAM
+  # https://github.com/awslabs/serverless-application-model/blob/master/docs/internals/generated_resources.rst#api
+  SnippetExtractorApi:
+    Description: "API Gateway endpoint URL for Prod stage for Hello World function"
+    Value: !Sub "https://${ServerlessRestApi}.execute-api.${AWS::Region}.amazonaws.com/Prod/extract_snippet/"
+  SnippetExtractorFunction:
+    Description: "Hello World Lambda Function ARN"
+    Value: !GetAtt HelloWorldFunction.Arn
+  SnippetExtractorIamRole:
+    Description: "Implicit IAM Role created for Hello World function"
+    Value: !GetAtt HelloWorldFunctionRole.Arn

Diff for: test/e2e_test.rb

@@ -0,0 +1,92 @@
+require "test_helper"
+
+require_relative '../../hello_world/app'
+
+class HelloWorldTest < Minitest::Test
+  def event
+    {
+      body: 'eyJ0ZXN0IjoiYm9keSJ9',
+      resource: '/{proxy+}',
+      path: '/path/to/resource',
+      httpMethod: 'POST',
+      isBase64Encoded: true,
+      queryStringParameters: {
+        foo: 'bar'
+      },
+      pathParameters: {
+        proxy: '/path/to/resource'
+      },
+      stageVariables: {
+        baz: 'qux'
+      },
+      headers: {
+        'Accept' => 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
+        'Accept-Encoding' => 'gzip, deflate, sdch',
+        'Accept-Language' => 'en-US,en;q=0.8',
+        'Cache-Control' => 'max-age=0',
+        'CloudFront-Forwarded-Proto' => 'https',
+        'CloudFront-Is-Desktop-Viewer' => 'true',
+        'CloudFront-Is-Mobile-Viewer' => 'false',
+        'CloudFront-Is-SmartTV-Viewer' => 'false',
+        'CloudFront-Is-Tablet-Viewer' => 'false',
+        'CloudFront-Viewer-Country' => 'US',
+        'Host' => '1234567890.execute-api.us-east-1.amazonaws.com',
+        'Upgrade-Insecure-Requests' => '1',
+        'User-Agent' => 'Custom User Agent String',
+        'Via' => '1.1 08f323deadbeefa7af34d5feb414ce27.cloudfront.net (CloudFront)',
+        'X-Amz-Cf-Id' => 'cDehVQoZnx43VYQb9j2-nvCh-9z396Uhbp027Y2JvkCPNLmGJHqlaA==',
+        'X-Forwarded-For' => '127.0.0.1, 127.0.0.2',
+        'X-Forwarded-Port' => '443',
+        'X-Forwarded-Proto' => 'https'
+      },
+      requestContext: {
+        accountId: '123456789012',
+        resourceId: '123456',
+        stage: 'prod',
+        requestId: 'c6af9ac6-7b61-11e6-9a41-93e8deadbeef',
+        requestTime: '09/Apr/2015:12:34:56 +0000',
+        requestTimeEpoch: 1428582896000,
+        identity: {
+          cognitoIdentityPoolId: 'null',
+          accountId: 'null',
+          cognitoIdentityId: 'null',
+          caller: 'null',
+          accessKey: 'null',
+          sourceIp: '127.0.0.1',
+          cognitoAuthenticationType: 'null',
+          cognitoAuthenticationProvider: 'null',
+          userArn: 'null',
+          userAgent: 'Custom User Agent String',
+          user: 'null'
+        },
+        path: '/prod/path/to/resource',
+        resourcePath: '/{proxy+}',
+        httpMethod: 'POST',
+        apiId: '1234567890',
+        protocol: 'HTTP/1.1'
+      }
+    }
+  end
+
+  def mock_response
+    Object.new.tap do |mock|
+      mock.expects(:code).returns(200)
+      mock.expects(:body).returns('1.1.1.1')
+    end
+  end
+
+  def expected_result
+    {
+      statusCode: 200,
+      body: {
+        message: 'Hello World!',
+        location: '1.1.1.1'
+      }.to_json
+    }
+  end
+
+  def test_lambda_handler
+    HTTParty.expects(:get).with('http://checkip.amazonaws.com/').returns(mock_response)
+    assert_equal expected, lambda_handler(event: event, context: '')
+  end
+end