cody 0.1.0 → 0.7.0

data/lib/cody/completer/script.rb

@@ -0,0 +1,6 @@
+class Cody::Completer::Script
+  def self.generate
+    bash_script = File.expand_path("script.sh", File.dirname(__FILE__))
+    puts "source #{bash_script}"
+  end
+end

data/lib/cody/completer/script.sh

@@ -0,0 +1,10 @@
+_cody() {
+  COMPREPLY=()
+  local word="${COMP_WORDS[COMP_CWORD]}"
+  local words=("${COMP_WORDS[@]}")
+  unset words[0]
+  local completion=$(cody completion ${words[@]})
+  COMPREPLY=( $(compgen -W "$completion" -- "$word") )
+}
+
+complete -F _cody cody

data/lib/cody/core.rb

@@ -0,0 +1,63 @@
+require 'pathname'
+require 'yaml'
+require 'active_support/core_ext/string'
+
+module Cody
+  module Core
+    extend Memoist
+
+    def root
+      path = ENV['CODY_ROOT'] || '.'
+      Pathname.new(path)
+    end
+
+    def env
+      # 2-way binding
+      cb_env = env_from_profile || 'development'
+      cb_env = ENV['CODY_ENV'] if ENV['CODY_ENV'] # highest precedence
+      ActiveSupport::StringInquirer.new(cb_env)
+    end
+    memoize :env
+
+    def env_extra
+      env_extra = ENV['CODY_ENV_EXTRA'] if ENV['CODY_ENV_EXTRA'] # highest precedence
+      return if env_extra&.empty?
+      env_extra
+    end
+    memoize :env_extra
+
+    # Overrides AWS_PROFILE based on the Cody.env if set in configs/settings.yml
+    # 2-way binding.
+    def set_aws_profile!
+      return if ENV['TEST']
+      return unless File.exist?("#{Cody.root}/.cody/settings.yml") # for rake docs
+      return unless settings # Only load if within Cody project and there's a settings.yml
+
+      data = settings || {}
+      if data[:aws_profile]
+        puts "Using AWS_PROFILE=#{data[:aws_profile]} from CODY_ENV=#{Cody.env} in config/settings.yml"
+        ENV['AWS_PROFILE'] = data[:aws_profile]
+      end
+    end
+
+    def settings
+      Setting.new.data
+    end
+    memoize :settings
+
+    def check_codebuild_project!
+      check_path = "#{Cody.root}/.cody"
+      unless File.exist?(check_path)
+        puts "ERROR: No .cody folder found.  Are you sure you are in a project with codebuild setup?".color(:red)
+        puts "Current directory: #{Dir.pwd}"
+        puts "If you want to set up codebuild for this project, please create a settings file via: cody init"
+        exit 1 unless ENV['TEST']
+      end
+    end
+
+  private
+    def env_from_profile
+      Cody::Setting.new.cb_env
+    end
+  end
+end

data/lib/cody/create.rb

@@ -0,0 +1,12 @@
+module Cody
+  class Create < Stack
+    def perform
+      cfn.create_stack(
+        stack_name: @stack_name,
+        template_body: YAML.dump(@template),
+        capabilities: ["CAPABILITY_IAM"]
+      )
+      puts "Creating stack #{@stack_name}. Check CloudFormation console for status."
+    end
+  end
+end

data/lib/cody/default/settings.yml

@@ -0,0 +1,3 @@
+base:
+  stack_naming:
+    append_env: false

data/lib/cody/delete.rb

@@ -0,0 +1,27 @@
+module Cody
+  class Delete
+    include AwsServices
+
+    def initialize(options)
+      @options = options
+      @project_name = options[:project_name] || inferred_project_name
+      @stack_name = options[:stack_name] || inferred_stack_name(@project_name)
+    end
+
+    def run
+      message = "Deleted #{@stack_name} stack."
+      if @options[:noop]
+        puts "NOOP #{message}"
+      else
+        are_you_sure?(@stack_name, :delete)
+
+        if stack_exists?(@stack_name)
+          cfn.delete_stack(stack_name: @stack_name)
+          puts message
+        else
+          puts "#{@stack_name.inspect} stack does not exist".color(:red)
+        end
+      end
+    end
+  end
+end

data/lib/cody/deploy.rb

@@ -0,0 +1,40 @@
+module Cody
+  class Deploy < Stack
+    def run
+      handle_rollback_completed!
+      if stack_exists?(@stack_name)
+        Update.new(@options).run
+      else
+        Create.new(@options).run
+      end
+    end
+
+    def handle_rollback_completed!
+      @stack = find_stack(@stack_name)
+      if @stack && rollback_complete?(@stack)
+        puts "Existing stack in ROLLBACK_COMPLETE state. Deleting stack before continuing."
+        cfn.delete_stack(stack_name: @stack_name)
+        status.wait
+        status.reset
+        @stack = nil # at this point stack has been deleted
+      end
+    end
+
+    def rollback_complete?(stack)
+      stack.stack_status == 'ROLLBACK_COMPLETE'
+    end
+
+    def find_stack(stack_name)
+      return if ENV['TEST']
+      resp = cfn.describe_stacks(stack_name: stack_name)
+      resp.stacks.first
+    rescue Aws::CloudFormation::Errors::ValidationError => e
+      # example: Stack with id demo-web does not exist
+      if e.message =~ /Stack with/ && e.message =~ /does not exist/
+        nil
+      else
+        raise
+      end
+    end
+  end
+end

data/lib/cody/dsl/project.rb

@@ -0,0 +1,119 @@
+module Cody::Dsl
+  module Project
+    include Ssm
+
+    PROPERTIES = %w[
+      artifacts
+      badge_enabled
+      cache
+      description
+      encryption_key
+      environment
+      logs_config
+      name
+      queued_timeout_in_minutes
+      secondary_artifacts
+      secondary_sources
+      service_role
+      source
+      tags
+      timeout_in_minutes
+      triggers
+      vpc_config
+    ]
+    PROPERTIES.each do |prop|
+      define_method(prop) do |v|
+        @properties[prop.to_sym] = v
+      end
+    end
+
+    # Convenience wrapper methods
+    def github_url(url)
+      @properties[:source][:location] = url
+    end
+
+    # So it looks like the auth resource property doesnt really get used.
+    # Instead an account level credential is worked.  Refer to:
+    # https://github.com/tongueroo/cody/blob/master/readme/github_oauth.md
+    #
+    # Keeping this method around in case the CloudFormation method works one day,
+    # or end up figuring out to use it properly.
+    def github_token(token)
+      @properties[:source][:auth][:resource] = token
+    end
+
+    def github_source(options={})
+      source = {
+        type: "GITHUB",
+        location: options[:location],
+        git_clone_depth: 1,
+        git_submodules_config: { fetch_submodules: true },
+        build_spec: options[:buildspec] || ".cody/buildspec.yml", # options[:buildspec] accounts for type already
+        report_build_status: true,
+      }
+
+      if options[:oauth_token]
+        source[:auth] = {
+          type: "OAUTH",
+          resource: options[:oauth_token],
+        }
+      end
+
+      @properties[:source] = source
+    end
+
+    def linux_image(name)
+      linux_environment(image: name)
+    end
+
+    def linux_environment(options={})
+      image = options[:image] || "aws/codebuild/ruby:2.5.3-1.7.0"
+      env = {
+        compute_type: options[:compute_type] || "BUILD_GENERAL1_SMALL",
+        image_pull_credentials_type: "CODEBUILD", # https://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/aws-properties-codebuild-project-environment.html#cfn-codebuild-project-environment-imagepullcredentialstype
+        privileged_mode: true,
+        image: image,
+        type: "LINUX_CONTAINER"
+      }
+      # @mapped_env_vars is in memory
+      env[:environment_variables] = @mapped_env_vars if @mapped_env_vars
+      # options has highest precedence
+      env[:environment_variables] = options[:environment_variables] if options[:environment_variables]
+      @properties[:environment] = env
+    end
+
+    def environment_variables(vars)
+      # Storing @mapped_env_vars as instance variable for later usage in linux_environment
+      @mapped_env_vars = vars.map { |k,v|
+        k, v = k.to_s, v.to_s
+        if v =~ /^ssm:/
+          { type: "PARAMETER_STORE", name: k, value: v.sub('ssm:','') }
+        else
+          { type: "PLAINTEXT", name: k, value: v }
+        end
+      }
+      @properties[:environment] ||= {}
+      @properties[:environment][:environment_variables] = @mapped_env_vars
+    end
+
+    def local_cache(enable=true)
+      cache = if enable
+        {
+          type: "LOCAL",
+          modes: [
+              "LOCAL_DOCKER_LAYER_CACHE",
+              "LOCAL_SOURCE_CACHE",
+              "LOCAL_CUSTOM_CACHE"
+          ]
+        }
+      else
+        {type: "NO_CACHE"}
+      end
+      @properties[:cache] = cache
+    end
+
+    def type
+      @options[:type]
+    end
+  end
+end

data/lib/cody/dsl/project/ssm.rb

@@ -0,0 +1,22 @@
+require "aws-sdk-ssm"
+
+module Cody::Dsl::Project
+  module Ssm
+    # This method grabs the ssm parameter store value at "compile" time vs
+    # CloudFormation run time. In case we need it as part of the DSL compile phase.
+    def ssm(name)
+      resp = ssm_client.get_parameter(name: name)
+      if resp.parameter.type == "SecureString"
+        resp = ssm_client.get_parameter(name: name, with_decryption: true)
+      end
+
+      resp.parameter.value
+    rescue Aws::SSM::Errors::ParameterNotFound
+      puts "WARN: #{name} found on AWS SSM.".color(:yellow)
+    end
+
+    def ssm_client
+      @ssm_client ||= Aws::SSM::Client.new
+    end
+  end
+end

data/lib/cody/dsl/role.rb

@@ -0,0 +1,50 @@
+module Cody::Dsl
+  module Role
+    PROPERTIES = %w[
+      assume_role_policy_document
+      managed_policy_arns
+      max_session_duration
+      path
+      permissions_boundary
+      policies
+      role_name
+    ]
+    PROPERTIES.each do |prop|
+      define_method(prop) do |v|
+        @properties[prop.to_sym] = v
+      end
+    end
+
+    # convenience wrapper methods
+    def iam_policy(*definitions)
+      @iam_statements = definitions.map { |definition| standardize_iam_policy(definition) }
+    end
+
+    # Returns standarized IAM statement
+    def standardize_iam_policy(definition)
+      case definition
+      when String
+        # Expands simple string from: logs => logs:*
+        definition = "#{definition}:*" unless definition.include?(':')
+        {
+          action: [definition],
+          effect: "Allow",
+          resource: "*",
+        }
+      when Hash
+        definition
+      end
+    end
+
+    def managed_iam_policy(*definitions)
+      @managed_policy_arns = definitions.map { |definition| standardize_managed_iam_policy(definition) }
+    end
+
+    # AmazonEC2ReadOnlyAccess => arn:aws:iam::aws:policy/AmazonEC2ReadOnlyAccess
+    def standardize_managed_iam_policy(definition)
+      return definition if definition.include?('iam::aws:policy')
+
+      "arn:aws:iam::aws:policy/#{definition}"
+    end
+  end
+end

data/lib/cody/dsl/schedule.rb

@@ -0,0 +1,30 @@
+module Cody::Dsl
+  module Schedule
+    PROPERTIES = %w[
+      description
+      event_pattern
+      name
+      role_arn
+      schedule_expression
+      state
+      targets
+    ]
+    PROPERTIES.each do |prop|
+      define_method(prop) do |v|
+        @properties[prop.to_sym] = v
+      end
+    end
+
+    def rate(period)
+      @schedule_expression = "rate(#{period})"
+    end
+
+    def cron(expression)
+      @schedule_expression = "cron(#{expression})"
+    end
+
+    def rule_event(props={})
+      @rule_event_props = props
+    end
+  end
+end

data/lib/cody/evaluate.rb

@@ -0,0 +1,47 @@
+module Cody
+  module Evaluate
+    def evaluate(path)
+      source_code = IO.read(path)
+      begin
+        instance_eval(source_code, path)
+      rescue Exception => e
+        if e.class == SystemExit # allow exit to happen normally
+          raise
+        else
+          task_definition_error(e)
+          puts "\nFull error:"
+          raise
+        end
+      end
+    end
+
+  private
+    # Prints out a user friendly task_definition error message
+    def task_definition_error(e)
+      error_info = e.backtrace.first
+      path, line_no, _ = error_info.split(':')
+      line_no = line_no.to_i
+      puts "Error evaluating #{path}:".color(:red)
+      puts e.message
+      puts "Here's the line in #{path} with the error:\n\n"
+
+      contents = IO.read(path)
+      content_lines = contents.split("\n")
+      context = 5 # lines of context
+      top, bottom = [line_no-context-1, 0].max, line_no+context-1
+      spacing = content_lines.size.to_s.size
+      content_lines[top..bottom].each_with_index do |line_content, index|
+        line_number = top+index+1
+        if line_number == line_no
+          printf("%#{spacing}d %s\n".color(:red), line_number, line_content)
+        else
+          printf("%#{spacing}d %s\n", line_number, line_content)
+        end
+      end
+    end
+
+    def lookup_codebuild_file(name)
+      [".cody", @options[:type], name].compact.join("/")
+    end
+  end
+end

data/lib/cody/help.rb

@@ -0,0 +1,9 @@
+module Cody::Help
+  class << self
+    def text(namespaced_command)
+      path = namespaced_command.to_s.gsub(':','/')
+      path = File.expand_path("../help/#{path}.md", __FILE__)
+      IO.read(path) if File.exist?(path)
+    end
+  end
+end