lamby-no-rails 1.0.3

data/lib/lamby/ssm_parameter_store.rb

@@ -0,0 +1,154 @@
+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.from(1).split('/')
+        env = parts.pop
+        path = "/#{parts.join('/')}"
+        new(path).get!.params.detect do |p|
+          p.env == env
+        end.try(: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, 'w') { |f| f.write(dotenv_contents) }
+    end
+
+    def get!
+      get_all!
+      get_history! if label.present?
+      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 = "export #{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.try(: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.try(: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,15 @@
+namespace :lamby do
+
+  namespace :ssm do
+
+    desc 'Create a Dotenv file.'
+    task :dotenv do
+      require 'lamby/ssm_parameter_store'
+      path = ENV['LAMBY_SSM_PARAMS_PATH']
+      raise ArgumentError, 'The LAMBY_SSM_PARAMS_PATH env is required.' unless path
+      Lamby::SsmParameterStore.dotenv(path)
+    end
+
+  end
+
+end

data/lib/lamby/templates.rake

@@ -0,0 +1,36 @@
+installers = {
+  'API Gateway' => :api_gateway,
+  'Application Load Balancer' => :application_load_balancer
+}.freeze
+
+namespace :lamby do
+  namespace :install do
+
+    installers.each do |name, task_name|
+      desc "Install Lamby files for #{name}"
+      task task_name do
+        exec "#{base_path} LOCATION=#{template(task_name)}"
+      end
+    end
+
+    def template(task_name)
+      File.expand_path "../lamby/templates/#{task_name}.rb", __dir__
+    end
+
+    def bin_path
+      ENV['BUNDLE_BIN'] || './bin'
+    end
+
+    def base_path
+      if Rails::VERSION::MAJOR >= 5
+        "#{RbConfig.ruby} #{bin_path}/rails app:template"
+      else
+        "#{RbConfig.ruby} #{bin_path}/rake rails:template"
+      end
+    end
+
+  end
+
+  desc "Install Lamby files for #{installers.first.first}"
+  task install: 'install:application_load_balancer'
+end

data/lib/lamby/templates/api_gateway.rb

@@ -0,0 +1,25 @@
+$LAMBY_INSTALLER_NAME = 'api_gateway'
+load 'lamby/templates/shared.rb'
+
+say '==> Running Lamby API Gateway 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/api_gateway/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: :api
+end

data/lib/lamby/templates/api_gateway/build

@@ -0,0 +1,23 @@
+#!/usr/bin/env bash
+set -e
+
+# Clean any previous bundle/builds.
+rm -rf ./.aws-sam/build
+
+# Ensure native extensions built for platform.
+sam build --use-container
+
+# [HOOK] Environments & Configuration
+# https://lamby.custominktech.com/docs/environment_and_configuration
+
+# [HOOK] Asset Hosts & Precompiling
+# https://lamby.custominktech.com/docs/asset_host_and_precompiling
+
+# Clean un-needed artifacts.
+pushd ./.aws-sam/build/RailsFunction/
+rm -rf .aws-sam \
+  .git \
+  log \
+  test \
+  tmp
+popd

data/lib/lamby/templates/api_gateway/deploy

@@ -0,0 +1,22 @@
+#!/usr/bin/env bash
+set -e
+
+export RAILS_ENV=${RAILS_ENV:="production"}
+export AWS_DEFAULT_REGION=${AWS_DEFAULT_REGION:=us-east-1}
+export CLOUDFORMATION_BUCKET=${CLOUDFORMATION_BUCKET:="lamby.cloudformation.$(whoami)"}
+
+./bin/build
+
+sam package \
+  --region ${AWS_DEFAULT_REGION} \
+  --template-file ./.aws-sam/build/template.yaml \
+  --output-template-file ./.aws-sam/build/packaged.yaml \
+  --s3-bucket $CLOUDFORMATION_BUCKET \
+  --s3-prefix "APPNAMEHERE-${RAILS_ENV}"
+
+sam deploy \
+  --template-file ./.aws-sam/build/packaged.yaml \
+  --stack-name "APPNAMEHERE-${RAILS_ENV}-${AWS_DEFAULT_REGION}" \
+  --capabilities "CAPABILITY_IAM" \
+  --parameter-overrides \
+    RailsEnv=${RAILS_ENV}

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

@@ -0,0 +1,84 @@
+AWSTemplateFormatVersion: '2010-09-09'
+Transform: AWS::Serverless-2016-10-31
+Description: APPNAMEHERE Lambda (API)
+
+Parameters:
+
+  RailsEnv:
+    Type: String
+    Default: production
+    AllowedValues:
+      - staging
+      - production
+
+Resources:
+
+  RailsApi:
+    Type: AWS::Serverless::Api
+    Properties:
+      StageName: !Ref RailsEnv
+      EndpointConfiguration: REGIONAL
+      DefinitionBody:
+        swagger: 2.0
+        info: { title: !Ref 'AWS::StackName' }
+        basePath: !Join [ '', [ '/', !Ref RailsEnv ] ]
+        schemes: [ 'https' ]
+        paths:
+          /:
+            x-amazon-apigateway-any-method:
+              x-amazon-apigateway-integration:
+                uri:
+                  Fn::Sub: arn:aws:apigateway:${AWS::Region}:lambda:path/2015-03-31/functions/${RailsFunction.Arn}:live/invocations
+                httpMethod: POST
+                type: aws_proxy
+          /{resource+}:
+            x-amazon-apigateway-any-method:
+              parameters:
+                - name: resource
+                  in: path
+                  required: true
+                  type: string
+              x-amazon-apigateway-integration:
+                uri:
+                  Fn::Sub: arn:aws:apigateway:${AWS::Region}:lambda:path/2015-03-31/functions/${RailsFunction.Arn}:live/invocations
+                httpMethod: POST
+                type: aws_proxy
+        x-amazon-apigateway-binary-media-types:
+          - '*/*'
+
+  RailsFunction:
+    Type: AWS::Serverless::Function
+    Properties:
+      CodeUri: .
+      Handler: app.handler
+      Runtime: ruby2.5
+      MemorySize: 512
+      Timeout: 30
+      Environment:
+        Variables:
+          RAILS_ENV: !Ref RailsEnv
+      FunctionName: !Join [ '', [ 'APPNAMEHERE-', !Ref RailsEnv, '-', !Ref 'AWS::Region' ] ]
+      Events:
+        Root:
+          Type: Api
+          Properties:
+            Path: /
+            Method: ANY
+            RestApiId: !Ref RailsApi
+        RailsAppAll:
+          Type: Api
+          Properties:
+            Path: /{resource+}
+            Method: ANY
+            RestApiId: !Ref RailsApi
+      AutoPublishAlias: live
+
+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/application_load_balancer.rb

@@ -0,0 +1,25 @@
+$LAMBY_INSTALLER_NAME = 'application_load_balancer'
+load 'lamby/templates/shared.rb'
+
+say '==> Running Lamby Application Load Balancer 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/application_load_balancer/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: :alb
+end

data/lib/lamby/templates/application_load_balancer/build

@@ -0,0 +1,23 @@
+#!/usr/bin/env bash
+set -e
+
+# Clean any previous bundle/builds.
+rm -rf ./.aws-sam/build
+
+# Ensure native extensions built for platform.
+sam build --use-container
+
+# [HOOK] Environments & Configuration
+# https://lamby.custominktech.com/docs/environment_and_configuration
+
+# [HOOK] Asset Hosts & Precompiling
+# https://lamby.custominktech.com/docs/asset_host_and_precompiling
+
+# Clean un-needed artifacts.
+pushd ./.aws-sam/build/RailsFunction/
+rm -rf .aws-sam \
+  .git \
+  log \
+  test \
+  tmp
+popd

data/lib/lamby/templates/application_load_balancer/deploy

@@ -0,0 +1,39 @@
+#!/usr/bin/env bash
+set -e
+
+export RAILS_ENV=${RAILS_ENV:="production"}
+export AWS_DEFAULT_REGION=${AWS_DEFAULT_REGION:=us-east-1}
+export CLOUDFORMATION_BUCKET=${CLOUDFORMATION_BUCKET:="lamby.cloudformation.$(whoami)"}
+
+./bin/build
+
+export VPCID=${VPCID:=$(
+  aws ec2 describe-vpcs \
+    --output text \
+    --filters 'Name=state,Values=available' \
+    --query 'Vpcs[0].VpcId'
+)}
+
+export SUBNETS=${SUBNETS:=$(
+  aws ec2 describe-subnets \
+    --output text \
+    --filters "Name=state,Values=available,Name=vpc-id,Values=$VPCID" \
+    --query 'Subnets[*].SubnetId' | \
+    tr -s '[:blank:]' ','
+)}
+
+sam package \
+  --region ${AWS_DEFAULT_REGION} \
+  --template-file ./.aws-sam/build/template.yaml \
+  --output-template-file ./.aws-sam/build/packaged.yaml \
+  --s3-bucket $CLOUDFORMATION_BUCKET \
+  --s3-prefix "APPNAMEHERE-${RAILS_ENV}"
+
+sam deploy \
+  --template-file ./.aws-sam/build/packaged.yaml \
+  --stack-name "APPNAMEHERE-${RAILS_ENV}-${AWS_DEFAULT_REGION}" \
+  --capabilities "CAPABILITY_IAM" \
+  --parameter-overrides \
+    RailsEnv=${RAILS_ENV} \
+    VpcId=${VPCID} \
+    Subnets=${SUBNETS}