addy_lambda 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a40fe7dd18f9436441daacde8ea4ea2a6d0ccfd3878a9d2b03e00e7a76c0d8ce
-  data.tar.gz: a8a52aeeafbd076c27b1875cb94d8602eea09272f5f28449e937a8ff581d2d9e
+  metadata.gz: b6e0c3ac7b853a06d75aed86d86c507c8b26eaa012572ec6e075782fc11c554a
+  data.tar.gz: dfd63b81d918faed64600bd88cb5e03773697aa2b6aa2fc09898ebb4923c2b59
 SHA512:
-  metadata.gz: 95519ec57cb521f6e9f8cca6ea1ebaeb9a3a72f55cbc52a67d657a11853ef953096df74512cf6e4bd03b384d1813144467a71ff5c21a4290df451715ad1f2653
-  data.tar.gz: 15db8292c1ea02d3f9868d914dfd2fa381a09dd4bc62a47068fe22c7551f941b6f6676ee47599de36857756c3fa33fca3ab3f9b998b81b7d9bd501e809c02d1c
+  metadata.gz: 1ec0bc266dcbeeae6e680437e9172b773ee71856238a1a285d58395883171f585a052614a38b7e05a1f1094b8253b1b0eae57b48cb983995fcb744adf97db959
+  data.tar.gz: ee662005af9a3efcf3536b3dee8c9db79718d0dc8840fd00756a2cad094c6d1d416cc5f12abdba149b3303d2da978053c75e61940da7fa1aa574c4df83b9046c

data/.gitignore

@@ -7,4 +7,5 @@
 /spec/reports/
 /tmp/
 /vendor
-/bin
+/bin
+addy_lambda-*

data/README.md

@@ -3,6 +3,7 @@
 The purpose of this gem, is to make common functionality for Lambdas be available across all of our Lambda implementaitons.
 
 Things like:
+
 - calling API endpoints
 - Posting to slack
 - Getting a JWT token
@@ -27,9 +28,16 @@ Or install it yourself as:
 
     $ gem install addy_lambda
 
+## Publishing a new version
+
+- Make your changes
+- Bump the version in `lib/addy_lambda/version.rb`
+- Build using `gem build addy_lambda.gemspec`
+
 ## Usage
 
 Run the tests:
+
 ```
 ruby "/home/cabox/workspace/addy_lambda/test/addy_lambda_test.rb"
 ```
@@ -55,7 +63,7 @@ puts(test)
 
 ## Development
 
-After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake test` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `bundle exec ruby test/test_helper.rb` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
 
 To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
 
@@ -63,7 +71,6 @@ To install this gem onto your local machine, run `bundle exec rake install`. To
 
 Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/addy_lambda. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [code of conduct](https://github.com/[USERNAME]/addy_lambda/blob/master/CODE_OF_CONDUCT.md).
 
-
 ## License
 
 The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).

data/deleteme.rb

@@ -2,5 +2,14 @@ require_relative 'lib/addy_lambda'
 
 lambda_client = Aws::Lambda::Client.new
 
-resp = AddyLambda::Common.post_slack_message(lambda_client, '@alon', 'This is the mess', ':tada:')
-puts(resp['message'])
+# resp = AddyLambda::AWS.send_email(lambda_client, ['alon@sabi.me', 'alon@addyinvest.com'], 'alon', 'alon@addyinvest.com',
+#                                      'This is the subject', 'This is the content')
+
+# resp = AddyLambda::AWS.post_slack_message(lambda_client, '@alon', 'This is the mess', ':tada:')
+
+# pdf_annotate / subscription
+resp = AddyLambda::AWS.generate_jwt(lambda_client, AddyLambda::JwtIssuers::ADDY, AddyLambda::JwtAudiences::GENERIC,
+                                    3000, { 'alon' => 'sabi' })
+# resp = AddyLambda::AWS.validate_jwt(lambda_client,
+#                                        'eyJ0eXAiOiJKV1QiLCJhbGciOiJSUzI1NiJ9.eyJhbG9uIjoic2FiaSIsImlzcyI6ImFkZHkiLCJhdWQiOiJnZW5lcmljIiwiaWF0IjoxNjM5NDQxMzk5LCJleHAiOjE2Mzk0NDQzOTl9.jj2OhiEZnJ9s6tamHdc0wGNzlgW3Jwq-S8YquTxe_JWhHuOZKYUhlCM1J1AtaLBqwbnFONCI1geK6cDXIA57lwikh2GcBZGecFfn3hq5IuoSqkhSW-PypTsw2dVCPX-W-qOj6ktbVkTU-d9NvTJJAR4nkJdQMBFtDDgMP_huzxldUjkV9i4ruiebev9blinLuwS_pHIOeqpnTkqRxQ9U6oZscYT8dxHUVoTG03_1dbnnfq3xoBjmBzENz9j2CtZO_fnM96iwqPR4EfxpM7TU7o9NxIzunmN4_Z6Jmjh5yQWpRSp5ItnqQ5N9m6a5bE3NyGDfJh3tVHs7wg3VNqxsLQ')
+puts(resp)

data/lib/addy_lambda/aws.rb

@@ -1,6 +1,20 @@
 # frozen_string_literal: true
 
+require 'aws-sdk-lambda'
+
 module AddyLambda
+  class JwtIssuers
+    ADDY = 0
+  end
+
+  class JwtAudiences
+    GENERIC = 0
+    PDF_ANNOTATE = 1
+    SUBSCRIPTION = 2
+  end
+
+  JWT_AUDIENCES = %w[generic pdf_annotate subscription].freeze
+  JWT_ISSUERS = ['addy'].freeze
   # Common functionality for AWS related activities
   class AWS
     # ssm_client = Aws::SSM::Client.new
@@ -12,5 +26,64 @@ module AddyLambda
       response = ssm_client.get_parameter(request)
       response.parameter.value
     end
+
+    def self.invoke_lambda(lambda_client, function_name, payload, syncronous: true)
+      resp = lambda_client.invoke({
+                                    function_name: function_name,
+                                    invocation_type: syncronous ? 'RequestResponse' : 'Event',
+                                    payload: payload.to_json
+                                  })
+      AddyLambda::Common.valid_json?(resp.payload.string) ? JSON.parse(resp.payload.string) : resp.payload.string
+    end
+
+    # Lambda client can be obtained by running: lambda_client = Aws::Lambda::Client.new
+    def self.post_slack_message(lambda_client, channel, message, emoji = ':tada', unfurl_links: false)
+      input_params = {
+        'user' => 'Signature Request',
+        'channel' => channel,
+        'message' => message,
+        'emoji' => emoji,
+        'unfurl_links' => unfurl_links
+      }
+      response_body = AddyLambda::AWS.invoke_lambda(lambda_client, 'slack-client', input_params)
+      return response_body unless response_body.key?('error')
+
+      logger.error(response_body['error'])
+    end
+
+    def self.send_email(lambda_client, recipient_emails, from_name, from_email, subject, content_html)
+      input_params = {
+        'recipient_list' => recipient_emails,
+        'from_name' => from_name,
+        'from_email' => from_email,
+        'subject' => subject,
+        'content_html' => content_html
+      }
+
+      response_body = AddyLambda::AWS.invoke_lambda(lambda_client, 'mailgun-client', input_params)
+      return response_body unless response_body.key?('error')
+
+      logger.error(response_body['error'])
+    end
+
+    # NOTE: issuer AddyLambda::JwtIssuers::..., Audience AddyLambda::JwtAudiences::...
+    def self.generate_jwt(lambda_client, issuer, audience, expire, payload)
+      lambda_payload = {
+        'issuer' => JWT_ISSUERS[issuer],
+        'audience' => JWT_AUDIENCES[audience],
+        'expire' => expire,
+        'payload' => payload
+      }
+      jwt_response = AddyLambda::AWS.invoke_lambda(lambda_client, 'jwt-generate', lambda_payload)
+      jwt_response.key?('jwt') ? jwt_response['jwt'] : jwt_response
+    end
+
+    def self.validate_jwt(lambda_client, jwt)
+      lambda_payload = {
+        'jwt' => jwt
+      }
+      response = AddyLambda::AWS.invoke_lambda(lambda_client, 'jwt-validate', lambda_payload)
+      response['statusCode'] == 200 ? JSON.parse(response['body']) : response
+    end
   end
 end

data/lib/addy_lambda/common.rb

@@ -1,5 +1,4 @@
 require 'json'
-require 'aws-sdk-lambda'
 
 module AddyLambda
   # Generic methods such as logging, error handling HTTP requests etc.
@@ -51,37 +50,11 @@ module AddyLambda
       ScoutApm::Agent.instance.install
     end
 
-    # Lambda client can be obtained by running: lambda_client = Aws::Lambda::Client.new
-    def self.post_slack_message(lambda_client, channel, message, emoji = ':tada', unfurl_links: false)
-      input_params = {
-        'user' => 'Signature Request',
-        'channel' => channel,
-        'message' => message,
-        'emoji' => emoji,
-        'unfurl_links' => unfurl_links
-      }
-      response_body = invoke_lambda(lambda_client, 'slack-client', input_params)
-      return response_body unless response_body.key?('error')
-
-      logger.error(response_body['error'])
-    end
-
-    def self.invoke_lambda(lambda_client, function_name, payload)
-      resp = lambda_client.invoke({
-                                    function_name: function_name,
-                                    invocation_type: 'RequestResponse',
-                                    payload: payload.to_json
-                                  })
-      valid_json?(resp.payload.string)
-      response_obj = JSON.parse(resp.payload.string)
-      valid_json?(resp.payload.string) ? JSON.parse(response_obj['body']) : resp.payload.string
-    end
-
     def self.get_request(url, params = nil, headers = nil)
       uri = URI(url)
       uri.query = URI.encode_www_form(params) if params
       request = Net::HTTP::Get.new(uri.request_uri)
-      response = http_request(request, uri, headers)
+      http_request(request, uri, headers)
     end
 
     def self.delete_request(url, headers = nil)

data/lib/addy_lambda/version.rb

@@ -1,3 +1,3 @@
 module AddyLambda
-  VERSION = "0.1.0"
+  VERSION = '0.1.1'
 end

data/lib/addy_lambda.rb

@@ -1,8 +1,8 @@
 # frozen_string_literal: true
 
-require 'addy_lambda/version'
-require 'addy_lambda/aws'
-require 'addy_lambda/common'
+require_relative 'addy_lambda/version'
+require_relative 'addy_lambda/aws'
+require_relative 'addy_lambda/common'
 
 # Module for general functionality
 module AddyLambda

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: addy_lambda
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Alon Sabi
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2021-12-08 00:00:00.000000000 Z
+date: 2021-12-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-ssm