lambda-version-manager 0.0.3

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 65c1fc87ac3d5d62b4cbc7fdcd68100c73110ebd
+  data.tar.gz: d6e4ac2ad9dbb56c3d23f1fd2886aeafd103f8fd
+SHA512:
+  metadata.gz: 5f7da2339574a59efb3c099145130db05fab72691120211f93c210a6f1f39123c8d3172458393347933c2131c8f46888a0a68ac2b614bc2af47ddfa21f2ee920
+  data.tar.gz: b8e3b5d321133d329f9a05b8c4ee65ac9fc70705808e5b455e43621021dfa5ee184bbc1b73c0ee4a3d1c82983ed052aa53a95f0d5c082555fa8d11ec0d015115

data/bin/lambda-version-manager

@@ -0,0 +1,83 @@
+#!/usr/bin/env ruby
+#
+require 'thor'
+require 'yaml'
+require 'pp'
+require_relative '../lib/project'
+require_relative '../lib/deployer'
+class GeneratorCLI < ::Thor
+
+  desc 'deploy', 'Deploy the updated lambda versions'
+  option 'project_path', banner: 'PROJECT_PATH', type: :string, desc: 'Path to the lambda version mappings project'
+  option 'environments', banner: 'ENVIRONMENT', type: :array, desc: 'Deploy lambdas in this environment'
+  option 'artifact', banner: 'ARTIFACT', type: :string, desc: 'Deploy lambdas that use this artifact'
+  option 'lambda', banner: 'LAMBDA', type: :string, desc: 'Deploy lambdas with this name'
+  option 'account', banner: 'ACCOUNT', type: :string, desc: 'Account to deploy to', :required => true
+  option 'version_map_file', banner: 'VERSION_MAP_FILE', type: :string, desc: 'A java properties file mapping artifacts to update to version'
+
+  def deploy
+    opts = parse_options(options)
+    if opts['version_map_file']
+      #for each x=y pair we need to call
+      File.readlines(options['version_map_file']).each do |line|
+        array = line.split("=")
+        deployer = Deployer.new(opts['project_path'], opts['account'], array[0].strip)
+        deployer.deploy(opts['environments'])
+      end
+    else
+      deployer = Deployer.new(opts['project_path'], opts['account'], opts['artifact'], opts['lambda'])
+      deployer.deploy(opts['environments'])
+    end
+  end
+
+  desc 'update_project', 'Update the versions'
+  option 'project_path', banner: 'PROJECT_PATH', type: :string, desc: 'Path to the lambda version mappings project'
+  option 'artifact', banner: 'ARTIFACT', type: :string, desc: 'Deploy lambdas that use this artifact'
+  option 'version', banner: 'VERSION', type: :string, desc: 'Version of the new artifact'
+  option 'version_map_file', banner: 'VERSION_MAP_FILE', type: :string, desc: 'A java properties file mapping artifacts to update to version'
+  option 'accounts', banner: 'ACCOUNT', type: :array, desc: 'Account to deploy to', :required => true
+
+  def update_project
+    opts = parse_options(options)
+    project = Project.new(opts['project_path'])
+    lambda_env_map = project.get_lambdas
+    account_env_map =  project.account_env_map
+
+    supportd_envs = []
+    opts['accounts'].each do | account|
+      supportd_envs += account_env_map[account]
+    end
+
+    lambda_env_map.each do |env, lamdas|
+      unless supportd_envs.include?(env)
+        lambda_env_map.delete(env)
+      end
+    end
+
+
+    if opts['version_map_file']
+      #for each artifact=version pair we need to call
+      File.readlines(opts['version_map_file']).each do |line|
+        array = line.split("=")
+        updated_files = project.update_by_artifact(lambda_env_map, array[0].strip, array[1].strip)
+        lambda_env_map = updated_files
+      end
+    else
+      lambda_env_map = project.update_by_artifact(lambda_env_map, opts['artifact'], opts['version'])
+    end
+    project.write_new_files(lambda_env_map)
+  end
+
+  no_commands do
+    def parse_options(options)
+      unless options['version_map_file'] || (options['version'] && (options['lambda'] || options['artifact']))
+        raise 'Either a file must be specified or a version and lambda or artifact.'
+      end
+      #add input validation, either a file must be specified or a version and lambda or artifact.
+      opts = options.dup
+      opts
+    end
+  end
+end
+
+GeneratorCLI.start(ARGV)

data/lib/client.rb

@@ -0,0 +1,12 @@
+require 'aws-sdk-lambda'
+class Client
+  attr_accessor :client
+  def initialize(aws_region)
+    @client = Aws::Lambda::Client.new(region: aws_region)
+  end
+
+  def update_function_code(function_name, s3_bucket, s3_key)
+    response = client.update_function_code({function_name: function_name, s3_bucket: s3_bucket, s3_key: s3_key})
+    puts "Updated: #{response.function_name}"
+  end
+end

data/lib/deployer.rb

@@ -0,0 +1,86 @@
+require_relative 'client'
+require_relative 'project'
+class Deployer
+  attr_accessor :project_path
+  attr_accessor :account
+  attr_accessor :config
+  attr_accessor :project
+  attr_accessor :lambda
+  attr_accessor :artifact
+
+  def initialize(project_path, account, artifact, lambda)
+    @project_path  = project_path
+    @account = account
+    @project = Project.new("#{project_path}")
+    @config = set_config
+    @artifact =  artifact
+    @lambda =  lambda
+  end
+
+  def set_config
+    @project.config
+  end
+
+  def get_deployable_lambdas(lambda_env_map)
+
+    if artifact
+      lambda_env_map.each do |env, lambdas|
+        lambdas.each do |lambda_name, properties|
+          lambdas.delete(lambda_name) unless properties['artifact_name'] == artifact
+        end
+      end
+    end
+
+    if lambda
+      lambda_env_map.each do |env, lambdas|
+        lambdas.each do |lambda_name, properties|
+          lambdas.delete(lambda_name) unless lambda_name == lambda
+        end
+      end
+    end
+    lambda_env_map
+  end
+
+  def deploy(environments)
+    account_env_map = project.account_env_map[account]
+    env_region_map = project.env_region_map
+    #Filter by account as the primary owner of envs
+    lambda_env_map = project.get_lambdas
+    lambda_env_map = get_deployable_lambdas(lambda_env_map)
+    account_env_map.each do |env|
+      #IF users has specified environments, skip if the environment is not a user specified one
+      next if !environments.nil? && environments.include?(env)
+      lambda_env_map[env].each do |lambda_name, properties|
+        client = Client.new(env_region_map[env])
+
+        s3_bucket = get_s3_bucket(properties, env)
+        s3_key = construct_s3_key(properties, env)
+
+        puts "ENV: #{env}"
+        puts "Lambda Name: #{lambda_name}"
+        puts "S3 Bucket: #{s3_bucket}"
+        puts "S3 Key:  #{s3_key}"
+        #client.update_function_code(lambda_name,s3_bucket,s3_key)
+      end
+    end
+  end
+
+
+  def get_s3_bucket(properties, env)
+    if properties['s3_bucket']
+      return properties['s3_bucket']
+    else
+      return config['environments'][env]['s3_bucket']
+    end
+
+  end
+
+  def construct_s3_key(properties, env)
+    if properties['s3_key']
+      return properties['s3_key']
+    else
+      return "#{config['environments'][env]['base_path']}/#{properties['artifact_name']}#{properties['artifact_name']}-#{properties['version']}.#{properties['extension']}"
+    end
+  end
+end
+

data/lib/project.rb

@@ -0,0 +1,67 @@
+class Project
+  attr_accessor :config
+  attr_accessor :project_path
+
+  def initialize(project_path)
+    @config = YAML.load_file("#{project_path}/config/config.yaml")
+    @project_path = project_path
+  end
+
+  def environments
+    config['environments']
+  end
+
+  def get_lambdas
+    lambdas = {}
+    environments.keys.each do |env|
+      lambdas[env] = YAML.load_file("#{project_path}/environments/#{env}.yaml")
+    end
+    lambdas
+  end
+
+  def account_env_map
+    hash = {}
+    config['environments'].each do |env, properties|
+      if hash[properties['account']].nil?
+        hash[properties['account']] = []
+        hash[properties['account']] << env
+      else
+        hash[properties['account']] << env
+      end
+    end
+    hash
+  end
+
+  def env_region_map
+    hash = {}
+    config['environments'].each do |env, properties|
+      if hash[env].nil?
+        hash[env] = properties['region']
+      else
+        raise("An environment cannot map to more than one region ENV: #{env} Region: #{properties['region']}")
+      end
+    end
+    hash
+  end
+
+
+  def write_new_files(lambda_env_map)
+    lambda_env_map.each do |env, contents|
+      File.write("#{project_path}/environments/#{env}.yaml", contents.to_yaml)
+    end
+  end
+
+  def update_by_artifact(lambda_env_map, artifact, version)
+    #iterate through all lambdas and update the ones with version changes
+    lambda_env_map.each do |env, lambda|
+      pp env
+      pp lambda
+      lambda.each do |lambda_name, properties|
+        if properties['artifact_name'] == artifact
+          properties['version'] = version
+        end
+      end
+    end
+    lambda_env_map
+  end
+end

metadata

@@ -0,0 +1,90 @@
+--- !ruby/object:Gem::Specification
+name: lambda-version-manager
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+platform: ruby
+authors:
+- Bryan Call
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2019-02-19 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: aws-sdk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.0.rc2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.0.rc2
+- !ruby/object:Gem::Dependency
+  name: thor
+  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: thor-scmversion
+  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'
+description: Updates aws lambda versions to match a yaml property file
+email: bcall@rapid7.com
+executables:
+- lambda-version-manager
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/lambda-version-manager
+- lib/client.rb
+- lib/deployer.rb
+- lib/project.rb
+homepage: http://rubygems.org/gems/lambda-version-manager
+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: []
+rubyforge_project: 
+rubygems_version: 2.6.12
+signing_key: 
+specification_version: 4
+summary: Lambda version manager
+test_files: []