lamby_updated 5.2.1

data/lib/lamby/ssm_parameter_store.rb

@@ -0,0 +1,155 @@
+require 'aws-sdk-ssm'
+
+module Lamby
+  class SsmParameterStore
+
+    MAX_RESULTS = 10
+
+    Param = Struct.new :name, :env, :value
+
+    attr_reader :path, :params
+
+    class << self
+
+      def dotenv(path)
+        new(path).get!.to_dotenv
+      end
+
+      def get!(path)
+        parts = path[1..-1].split('/')
+        env = parts.pop
+        path = "/#{parts.join('/')}"
+        param = new(path).get!.params.detect do |p|
+          p.env == env
+        end
+        param&.value
+      end
+
+    end
+
+    def initialize(path, options = {})
+      @path = path
+      @params = []
+      @options = options
+    end
+
+    def to_env(overwrite: true)
+      params.each do |param|
+        overwrite ? ENV[param.env] = param.value : ENV[param.env] ||= param.value
+      end
+    end
+
+    def to_dotenv
+      File.open(dotenv_file, 'a') { |f| f.write(dotenv_contents) }
+    end
+
+    def get!
+      get_all!
+      get_history! unless label.to_s.empty?
+      self
+    end
+
+    def label
+      ENV['LAMBY_SSM_PARAMS_LABEL'] || @options[:label]
+    end
+
+    def client
+      @client ||= begin
+        options = @options[:client_options] || {}
+        Aws::SSM::Client.new(options)
+      end
+    end
+
+
+    private
+
+    def dotenv_file
+      @options[:dotenv_file] || ENV['LAMBY_SSM_PARAMS_FILE'] || Rails.root.join(".env.#{Rails.env}")
+    end
+
+    def dotenv_contents
+      params.each_with_object('') do |param, contents|
+        line = "#{param.env}=#{param.value}\n"
+        contents << line
+      end
+    end
+
+    # Path
+
+    def get_all!
+      return params if @got_all
+      get_parse_all
+      while @all_response.next_token do get_parse_all end
+      @got_all = true
+      params
+    end
+
+    def get_parse_all
+      get_all
+      parse_all
+    end
+
+    def get_all
+      @all_response = client.get_parameters_by_path(get_all_options)
+    end
+
+    def get_all_options
+      { path: path,
+        recursive: true,
+        with_decryption: true,
+        max_results: MAX_RESULTS
+      }.tap { |options|
+        token = @all_response&.next_token
+        options[:next_token] = token if token
+      }
+    end
+
+    def parse_all
+      @all_response.parameters.each do |p|
+        env = p.name.split('/').last
+        params << Param.new(p.name, env, p.value)
+      end
+    end
+
+    # History
+
+    def get_history!
+      return params if @got_history
+      params.each do |param|
+        name = param.name
+        get_parse_history(name)
+        while @hist_response.next_token do get_parse_history(name) end
+      end
+      @got_history = true
+      params
+    end
+
+    def get_parse_history(name)
+      get_history(name)
+      parse_history(name)
+    end
+
+    def get_history(name)
+      @hist_response = client.get_parameter_history(get_history_options(name))
+    end
+
+    def get_history_options(name)
+      { name: name,
+        with_decryption: true,
+        max_results: MAX_RESULTS
+      }.tap { |options|
+        token = @hist_response&.next_token
+        options[:next_token] = token if token
+      }
+    end
+
+    def parse_history(name)
+      @hist_response.parameters.each do |p|
+        next unless p.labels.include? label
+        param = params.detect { |param| param.name == name }
+        param.value = p.value
+      end
+    end
+
+  end
+end

data/lib/lamby/tasks.rake

@@ -0,0 +1,19 @@
+namespace :lamby do
+  task :proxy_server => [:environment] do
+    require 'webrick'
+    port = ENV['LAMBY_PROXY_PORT'] || 3000
+    bind = ENV['LAMBY_PROXY_BIND'] || '0.0.0.0'
+    Rack::Handler::WEBrick.run Lamby::ProxyServer.new, Port: port, BindAddress: bind
+  end
+
+  task :proxy_server_puma => [:environment] do
+    port = ENV['LAMBY_PROXY_PORT'] || 3000
+    host = ENV['LAMBY_PROXY_BIND'] || '0.0.0.0'
+    lamby_proxy = Lamby::ProxyServer.new
+    maybe_later = MaybeLater::Middleware.new(lamby_proxy)
+    server = Puma::Server.new(maybe_later)
+    server.add_tcp_listener host, port
+    puts "Starting Puma server on #{host}:#{port}..."
+    server.run.join
+  end
+end

data/lib/lamby/version.rb

@@ -0,0 +1,3 @@
+module Lamby
+  VERSION = '5.2.1'
+end

data/lib/lamby.rb

@@ -0,0 +1,53 @@
+require 'lamby/logger'
+require 'rack'
+require 'base64'
+require 'lambda-console-ruby'
+require 'lamby/version'
+require 'lamby/config'
+require 'lamby/rack'
+require 'lamby/rack_alb'
+require 'lamby/rack_rest'
+require 'lamby/rack_http'
+require 'lamby/debug'
+require 'lamby/cold_start_metrics'
+require 'lamby/handler'
+
+if defined?(Rails)
+  require 'rails/railtie'
+  require 'lamby/railtie'
+end
+
+module Lamby
+
+  extend self
+
+  def cmd(event:, context:)
+    handler(config.rack_app, event, context)
+  ensure
+    config.handled_proc.call(event, context)
+  end
+
+  def handler(app, event, context, options = {})
+    Handler.call(app, event, context, options)
+  end
+
+  def config
+    Lamby::Config.config
+  end
+
+  autoload :SsmParameterStore, 'lamby/ssm_parameter_store'
+  autoload :ProxyContext, 'lamby/proxy_context'
+  autoload :ProxyServer, 'lamby/proxy_server'
+
+end
+
+# Add signal traps for clean exit
+Signal.trap("TERM") do
+  puts "Received SIGTERM, exiting gracefully..."
+  exit!(0) # exit! ensures no exception is raised
+end
+
+Signal.trap("INT") do
+  puts "Received SIGINT, exiting gracefully..."
+  exit!(0) # exit! ensures no exception is raised
+end

metadata

@@ -0,0 +1,233 @@
+--- !ruby/object:Gem::Specification
+name: lamby_updated
+version: !ruby/object:Gem::Version
+  version: 5.2.1
+platform: ruby
+authors:
+- Ken Collins
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2024-06-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rack
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: lambda-console-ruby
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-ssm
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: minitest-focus
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: mocha
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: timecop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: webrick
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Simple Rails & AWS Lambda Integration using Rack and various utilities.
+email:
+- ken@metaskills.net
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".devcontainer/Dockerfile"
+- ".devcontainer/devcontainer.json"
+- ".github/workflows/test.yml"
+- ".gitignore"
+- CHANGELOG.md
+- CODE_OF_CONDUCT.md
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/bootstrap
+- bin/console
+- bin/setup
+- bin/test
+- bin/update
+- images/social.png
+- images/social2.png
+- lamby.gemspec
+- lib/lamby.rb
+- lib/lamby/cold_start_metrics.rb
+- lib/lamby/config.rb
+- lib/lamby/debug.rb
+- lib/lamby/handler.rb
+- lib/lamby/logger.rb
+- lib/lamby/proxy_context.rb
+- lib/lamby/proxy_server.rb
+- lib/lamby/rack.rb
+- lib/lamby/rack_alb.rb
+- lib/lamby/rack_http.rb
+- lib/lamby/rack_rest.rb
+- lib/lamby/railtie.rb
+- lib/lamby/ssm_parameter_store.rb
+- lib/lamby/tasks.rake
+- lib/lamby/version.rb
+- vendor/.keep
+homepage: https://github.com/rails-lambda/lamby
+licenses:
+- MIT
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.5.5
+signing_key:
+specification_version: 4
+summary: Simple Rails & AWS Lambda Integration using Rack
+test_files: []