jets_bb_fork 0.0.4

data/lib/jets/resource/lambda/function/environment.rb

@@ -0,0 +1,62 @@
+class Jets::Resource::Lambda::Function
+  module Environment
+    def env_properties
+      env_vars = Jets::Dotenv.load!(true)
+      variables = environment.merge(env_vars)
+      check_reserved_variables!(variables)
+      {environment: { variables: variables }}
+    end
+
+    def environment
+      env = Jets.config.environment ? Jets.config.environment.to_h : {}
+      env.deep_merge(jets_env)
+    end
+
+    # These jets env variables are special variables that get included to keeps some state
+    def jets_env
+      env = {}
+      env[:JETS_ENV] = Jets.env.to_s
+      env[:JETS_ENV_EXTRA] = Jets.config.env_extra if Jets.config.env_extra
+      env[:JETS_PROJECT_NAME] = ENV['JETS_PROJECT_NAME'] if ENV['JETS_PROJECT_NAME']
+      env[:JETS_STAGE] = Jets::Resource::ApiGateway::Deployment.stage_name
+      env[:JETS_AWS_ACCOUNT] = Jets.aws.account
+      env
+    end
+
+  private
+    def check_reserved_variables!(variables)
+      found_reserved_vars = variables.keys & reserved_variables
+      return if found_reserved_vars.empty?
+
+      puts "You have configured some environment variables that are reserved by AWS Lambda.".color(:red)
+      puts found_reserved_vars
+      puts "The deployment to AWS Lambda will failed when using reserved variables."
+      puts "Please remove these reserved variables. "
+      puts "More info: https://docs.aws.amazon.com/lambda/latest/dg/current-supported-versions.html#lambda-environment-variables"
+      exit 1
+    end
+
+    # https://docs.aws.amazon.com/lambda/latest/dg/current-supported-versions.html#lambda-environment-variables
+    def reserved_variables
+      %w[
+        _HANDLER
+        AWS_REGION
+        AWS_EXECUTION_ENV
+        AWS_LAMBDA_FUNCTION_NAME
+        AWS_LAMBDA_FUNCTION_MEMORY_SIZE
+        AWS_LAMBDA_FUNCTION_VERSION
+        AWS_LAMBDA_LOG_GROUP_NAME
+        AWS_LAMBDA_LOG_STREAM_NAME
+        AWS_ACCESS_KEY_ID
+        AWS_SECRET_ACCESS_KEY
+        AWS_SESSION_TOKEN
+        TZ
+        LAMBDA_TASK_ROOT
+        LAMBDA_RUNTIME_DIR
+        AWS_LAMBDA_RUNTIME_API
+      ]
+    end
+
+
+  end
+end

data/lib/jets/resource/lambda/gem_layer.rb

@@ -0,0 +1,17 @@
+module Jets::Resource::Lambda
+  class GemLayer < LayerVersion
+    def description
+      "Jets Ruby Gems"
+    end
+
+    def layer_name
+      # Do not include the Jets.extra_env to group the layers in same app together
+      "#{Jets.config.short_env}-#{Jets.config.project_name}-gems"
+    end
+
+    def code_s3_key
+      checksum = Jets::Builders::Md5.checksums["stage/opt"]
+      "jets/code/opt-#{checksum}.zip" # s3_key
+    end
+  end
+end

data/lib/jets/resource/lambda/layer_version.rb

@@ -0,0 +1,44 @@
+# Type: "AWS::Lambda::LayerVersion"
+# Properties:
+#   CompatibleRuntimes:
+#     - String
+#     - ...
+#   Content:
+#     Content
+#   Description: String
+#   LayerName: String
+#   LicenseInfo: String
+module Jets::Resource::Lambda
+  class LayerVersion < Jets::Resource::Base
+    def definition
+      {
+        layer_version_logical_id => {
+          type: "AWS::Lambda::LayerVersion",
+          properties: {
+            content: {
+              s3_bucket: s3_bucket,
+              s3_key: code_s3_key,
+            },
+            description: description,
+            layer_name: layer_name,
+            license_info: "MIT",
+          }
+        }
+      }
+    end
+
+    def s3_bucket
+      "!Ref S3Bucket"
+    end
+
+    def layer_version_logical_id
+      self.class.name.split('::').last
+    end
+
+    def outputs
+      {
+        logical_id => "!Ref #{logical_id}",
+      }
+    end
+  end
+end

data/lib/jets/resource/logs/subscription_filter.rb

@@ -0,0 +1,31 @@
+# CloudFormation Log Subscription docs: https://amzn.to/2SNiSpr
+module Jets::Resource::Logs
+  class SubscriptionFilter < Jets::Resource::Base
+    def initialize(props={})
+      @props = props # associated_properties from dsl.rb
+    end
+
+    def definition
+      {
+        log_logical_id => {
+          type: "AWS::Logs::SubscriptionFilter",
+          properties: merged_properties,
+        }
+      }
+    end
+
+    # Do not name this method properties, that is a computed method of `Jets::Resource::Base`
+    def merged_properties
+      {
+        destination_arn: "!GetAtt {namespace}LambdaFunction.Arn",
+        filter_pattern: "", # matches everything https://amzn.to/2N3b39I
+        # log_group_name: string # will be set by log_event
+        # role_arn: string # only required for kinensis, we dont use this for Lambda
+      }.deep_merge(@props)
+    end
+
+    def log_logical_id
+      "{namespace}_subscription_filter"
+    end
+  end
+end

data/lib/jets/resource/permission.rb

@@ -0,0 +1,45 @@
+class Jets::Resource
+  class Permission < Jets::Resource::Base
+    def initialize(replacements, associated_resource)
+      @replacements = replacements
+      @associated_resource = associated_resource
+    end
+
+    def definition
+      logical_id = permission_logical_id
+
+      definition = {
+        logical_id => {
+          type: "AWS::Lambda::Permission",
+          properties: {
+            function_name: "!GetAtt {namespace}LambdaFunction.Arn",
+            action: "lambda:InvokeFunction",
+            principal: principal
+          }
+        }
+      }
+
+      # From AWS docs: https://amzn.to/2N0QXQL
+      # source_arn is "not supported by all event sources"
+      definition[logical_id][:properties][:source_arn] = source_arn if source_arn
+
+      definition
+    end
+
+    def permission_logical_id
+      logical_id = "{namespace}_permission"
+      md = @associated_resource.logical_id.match(/(\d+)$/)
+      counter = md[1] if md
+      [logical_id, counter].compact.join('').underscore
+    end
+
+    # Auto-detect principal from the associated resources.
+    def principal
+      Replacer.principal_map(@associated_resource.type)
+    end
+
+    def source_arn
+      Replacer.source_arn_map(@associated_resource.type)
+    end
+  end
+end

data/lib/jets/resource/replacer.rb

@@ -0,0 +1,85 @@
+class Jets::Resource
+  class Replacer
+    extend Memoist
+
+    def initialize(replacements={})
+      @replacements = replacements
+    end
+
+    # Replace placeholder hash values with replacements.  This does a deep replacement
+    # to the hash values.  The replacement "key" is the string value within the value.
+    #
+    # Example:
+    #
+    #   attributes = {whatever: "foo REPLACE_KEY bar" }
+    #   replace_placeholders(attributes, REPLACE_KEY: "blah:arn")
+    #   => {whatever: "foo blah:arn bar" }
+    #
+    # Also, we always replace the special {namespace} value in the hash values. Example:
+    #
+    #   attributes = {whatever: "{namespace}LambdaFunction" }
+    #   replace_placeholders(attributes, {})
+    #   => {whatever: "foo PostsControllerIndexLambdaFunction bar" }
+    #
+    def replace_placeholders(attributes)
+      update_values(attributes)
+    end
+
+    def update_values(original)
+      case original
+      when Array
+        original.map { |v| update_values(v) }
+      when Hash
+        initializer = original.map do |k, v|
+          [k, update_values(v)]
+        end
+        Hash[initializer]
+      else
+        replace_value(original)
+      end
+    end
+
+    def replace_value(value)
+      # Dont perform replacement on Integers
+      return value if value.is_a?(Integer)
+      # return value unless value.is_a?(String) or value.is_a?(Symbol)
+
+      value = value.to_s # normalize to String
+      @replacements.each do |k,v|
+        # IE: Replaces {namespace} => SecurityJobCheck
+        value = value.gsub("{#{k}}", v)
+      end
+      value
+    end
+
+    class << self
+      # Examples:
+      #   "AWS::Events::Rule" => "events.amazonaws.com",
+      #   "AWS::Config::ConfigRule" => "config.amazonaws.com",
+      #   "AWS::ApiGateway::Method" => "apigateway.amazonaws.com"
+      def principal_map(type)
+        service = type.split('::')[1].downcase
+        service = special_principal_map(service)
+        "#{service}.amazonaws.com"
+      end
+
+      def special_principal_map(service)
+        # special map
+        # s3_event actually uses sns topic events to trigger a Lambda function
+        map = { "s3" => "sns" }
+        map[service] || service
+      end
+
+      # From AWS docs: https://amzn.to/2N0QXQL
+      # source_arn is "not supported by all event sources"
+      #
+      # When it is not available the resource definition should add it.
+      def source_arn_map(type)
+        map = {
+          "AWS::ApiGateway::Method" => "!Sub arn:aws:execute-api:${AWS::Region}:${AWS::AccountId}:${RestApi}/*/*",
+        }
+        map[type]
+      end
+    end
+  end
+end

data/lib/jets/resource/route53/record_set.rb

@@ -0,0 +1,70 @@
+# CloudFormation Docs AWS::Route53::RecordSet: https://amzn.to/2BtP9s5
+#
+# Example:
+#
+#   DnsRecord:
+#     Type: AWS::Route53::RecordSet
+#     Properties:
+#       HostedZoneName: !Ref 'HostedZoneResource'
+#       Comment: DNS name for my instance.
+#       Name: !Join ['', [!Ref 'Ec2Instance', ., !Ref 'AWS::Region', ., !Ref 'HostedZone', .]]
+#       Type: A
+#       TTL: '900'
+#       ResourceRecords:
+#       - !GetAtt Ec2Instance.PublicIp
+module Jets::Resource::Route53
+  class RecordSet < Jets::Resource::Base
+    def definition
+      {
+        dns_record: {
+          type: "AWS::Route53::RecordSet",
+          properties: {
+            hosted_zone_name: hosted_zone_name,
+            comment: "DNS record managed by Jets",
+            name: name,
+            type: "CNAME",
+            ttl: "60",
+            resource_records: [
+              cname,
+            ],
+          }
+        }
+      }
+    end
+
+    def cname
+      if endpoint_types.include?("REGIONAL")
+        "!GetAtt DomainName.RegionalDomainName"
+      else
+        "!GetAtt DomainName.DistributionDomainName"
+      end
+    end
+
+    def domain_name
+      Jets::Resource::ApiGateway::DomainName.new
+    end
+    memoize :domain_name
+
+    def endpoint_types
+      domain_name.endpoint_types
+    end
+
+    # IE: demo-dev.mydomain.com
+    def name
+      # Weird looking but correct: domain_name is object and domain_name is also method
+      domain_name.domain_name
+    end
+
+    # IE: mydomain.com
+    def hosted_zone_name
+      name = Jets.config.domain.hosted_zone_name
+      name.ends_with?('.') ? name : "#{name}." # add trailing period if missing
+    end
+
+    def outputs
+      {
+        "DnsRecord" => "!Ref DnsRecord",
+      }
+    end
+  end
+end

data/lib/jets/resource/s3/bucket.rb

@@ -0,0 +1,24 @@
+module Jets::Resource::S3
+  class Bucket < Jets::Resource::Base
+    attr_reader :bucket_logical_id
+    def initialize(props={})
+      @props = props # associated_properties from dsl.rb
+      @bucket_logical_id = props.delete(:logical_id) || "{namespace}_s3_bucket"
+    end
+
+    def definition
+      {
+        bucket_logical_id => {
+          type: "AWS::S3::Bucket",
+          properties: @props,
+        }
+      }
+    end
+
+    def outputs
+      {
+        bucket_logical_id => "!Ref #{bucket_logical_id.to_s.camelize}",
+      }
+    end
+  end
+end

data/lib/jets/resource/sns/subscription.rb

@@ -0,0 +1,29 @@
+# CloudFormation SNS Subscription docs: https://amzn.to/2SJtN3C
+module Jets::Resource::Sns
+  class Subscription < Jets::Resource::Base
+    def initialize(props={})
+      @props = props # associated_properties from dsl.rb
+    end
+
+    def definition
+      {
+        subscription_logical_id => {
+          type: "AWS::SNS::Subscription",
+          properties: merged_properties,
+        }
+      }
+    end
+
+    # Do not name this method properties, that is a computed method of `Jets::Resource::Base`
+    def merged_properties
+      {
+        endpoint: "!GetAtt {namespace}LambdaFunction.Arn",
+        protocol: "lambda",
+      }.deep_merge(@props)
+    end
+
+    def subscription_logical_id
+      "{namespace}_sns_subscription"
+    end
+  end
+end

data/lib/jets/resource/sns/topic.rb

@@ -0,0 +1,35 @@
+# CloudFormation SNS Topic docs: https://amzn.to/2MYbUZc
+module Jets::Resource::Sns
+  class Topic < Jets::Resource::Base
+    def initialize(props={})
+      @props = props # associated_properties from dsl.rb
+    end
+
+    def definition
+      {
+        topic_logical_id => {
+          type: "AWS::SNS::Topic",
+          properties: merged_properties,
+        }
+      }
+    end
+
+    # Do not name this method properties, that is a computed method of `Jets::Resource::Base`
+    def merged_properties
+      display_name = "{namespace} Topic"[0..99] # limit is 100 chars
+      {
+        display_name: display_name,
+        # Not setting subscription this way but instead with a SNS::Subscription resource so the interface
+        # is consistent. Leaving comment in here to remind me and in case decide to change this.
+        # subscription: [
+        #   endpoint: "!GetAtt {namespace}LambdaFunction.Arn",
+        #   protocol: "lambda"
+        # ]
+      }.deep_merge(@props)
+    end
+
+    def topic_logical_id
+      "{namespace}_sns_topic"
+    end
+  end
+end