automan 2.1.2

data/lib/automan/cli/stalker.rb

@@ -0,0 +1,279 @@
+require 'automan'
+
+module Automan::Cli
+  class Stalker < Base
+
+    desc "alias", "point dns to elb"
+
+    option :environment_name,
+      required: true,
+      aliases: "-e",
+      desc: "Beanstalk environment containing target ELB"
+
+    option :hosted_zone_name,
+      required: true,
+      aliases: "-z",
+      desc: "Hosted zone for which apex will be routed to ELB"
+
+    option :hostname,
+      required: true,
+      aliases: "-h",
+      desc: "Alias to create (must be fqdn, e.g. 'dev1.dydev.me')"
+
+    def alias
+      Automan::Beanstalk::Router.new(options).run
+    end
+
+    desc "deploy", "deploy a beanstalk service"
+
+    option :name,
+      required: true,
+      aliases: "-n",
+      desc: "name of the service"
+
+    option :environment,
+      required: true,
+      aliases: "-e",
+      desc: "environment tag (e.g. dev, stage, prd)"
+
+    option :version_label,
+      aliases: "-l",
+      desc: "beanstalk application version label"
+
+    option :package,
+      aliases: "-p",
+      desc: "s3 path to version package (s3://bucket/package.zip)"
+
+    option :manifest,
+      aliases: "-m",
+      desc: "s3 path to manifest file containing version label and package location"
+
+    option :configuration_template,
+      required: true,
+      aliases: "-t",
+      desc: "beanstalk configuration template name"
+
+    option :beanstalk_name,
+      aliases: "-b",
+      desc: "alternate beanstalk environment name"
+
+    option :number_to_keep,
+      aliases: "-n",
+      type: :numeric,
+      desc: "cull old versions and keep newest NUMBER_TO_KEEP"
+
+    def deploy
+      # you must specify either a manifest file or both package and version label
+      if options[:manifest].nil? &&
+        (options[:version_label].nil? || options[:package].nil?)
+        puts "Must specify either manifest or both version_label and package"
+        help "deploy"
+        exit 1
+      end
+
+      Automan::Beanstalk::Deployer.new(options).deploy
+    end
+
+    desc "terminate", "terminate a beanstalk environment"
+
+    option :name,
+      required: true,
+      aliases: "-n",
+      desc: "name of the beanstalk environment"
+
+    def terminate
+      Automan::Beanstalk::Terminator.new(options).terminate
+    end
+
+    desc "create-app", "create beanstalk application"
+
+    option :name,
+      required: true,
+      aliases: "-n",
+      desc: "name of the application"
+
+    def create_app
+      Automan::Beanstalk::Application.new(options).create
+    end
+
+    desc "delete-app", "delete beanstalk application"
+
+    option :name,
+      required: true,
+      aliases: "-n",
+      desc: "name of the application"
+
+    def delete_app
+      Automan::Beanstalk::Application.new(options).delete
+    end
+
+    desc "create-config", "create beanstalk configuration template"
+
+    option :name,
+      required: true,
+      aliases: "-n",
+      desc: "name of the configuration template"
+
+    option :application,
+      required: true,
+      aliases: "-a",
+      desc: "name of the application"
+
+    option :template,
+      required: true,
+      aliases: "-t",
+      desc: "file containing configuration options"
+
+    option :platform,
+      required: "true",
+      aliases: "-p",
+      default: "64bit Windows Server 2012 running IIS 8",
+      desc: "beanstalk solution stack name"
+
+    def create_config
+      Automan::Beanstalk::Configuration.new(options).create
+    end
+
+    desc "update-config", "update beanstalk configuration template"
+
+    option :name,
+      required: true,
+      aliases: "-n",
+      desc: "name of the configuration template"
+
+    option :application,
+      required: true,
+      aliases: "-a",
+      desc: "name of the application"
+
+    option :template,
+      required: true,
+      aliases: "-t",
+      desc: "file containing configuration options"
+
+    option :platform,
+      required: "true",
+      aliases: "-p",
+      default: "64bit Windows Server 2012 running IIS 8",
+      desc: "beanstalk solution stack name"
+
+    def update_config
+      Automan::Beanstalk::Configuration.new(options).update
+    end
+
+    desc "delete-config", "delete beanstalk configuration template"
+
+    option :name,
+      required: true,
+      aliases: "-n",
+      desc: "name of the configuration template"
+
+    option :application,
+      required: true,
+      aliases: "-a",
+      desc: "name of the application"
+
+    def delete_config
+      Automan::Beanstalk::Configuration.new(options).delete
+    end
+
+    desc "upload-config", "validate and upload beanstalk configuration template files to s3"
+
+    option :template_files,
+      required: true,
+      aliases: "-t",
+      desc: "File system glob of configuration templates to upload"
+
+    option :s3_path,
+      required: true,
+      aliases: "-p",
+      desc: "S3 path to destination"
+
+    def upload_config
+      Automan::Beanstalk::Uploader.new(options).upload_config_templates
+    end
+
+    desc "delete-version", "delete an application version"
+
+    option :application,
+      required: true,
+      aliases: "-a",
+      desc: "name of the application"
+
+    option :label,
+      required: true,
+      aliases: "-l",
+      desc: "version label to be deleted"
+
+    def delete_version
+      Automan::Beanstalk::Version.new(options).delete
+    end
+
+    desc "cull-versions", "delete oldest versions, keeping N newest"
+
+    option :application,
+      required: true,
+      aliases: "-a",
+      desc: "name of the application"
+
+    option :number_to_keep,
+      required: true,
+      aliases: "-n",
+      type: :numeric,
+      desc: "keep newest NUMBER_TO_KEEP versions"
+
+    def cull_versions
+      Automan::Beanstalk::Version.new(options).cull_versions(options[:number_to_keep])
+    end
+
+    desc "show-versions", "show all versions for an application"
+
+    option :application,
+      required: true,
+      aliases: "-a",
+      desc: "name of the application"
+
+    def show_versions
+      versions = Automan::Beanstalk::Version.new(options).versions
+      versions.each do |v|
+        tokens = [
+          v[:application_name],
+          v[:version_label],
+          v[:date_created].iso8601
+        ]
+        say tokens.join("\t")
+      end
+    end
+
+    desc "package", "upload a zip file to s3 to be deployed to beanstalk"
+
+    option :destination,
+      required: true,
+      aliases: "-d",
+      desc: "full s3 path of uploaded package"
+
+    option :source,
+      required: true,
+      aliases: "-s",
+      desc: "path to local zip file to be uploaded"
+
+    option :manifest,
+      aliases: "-m",
+      desc: "if set, also upload a manifest file with full s3 path MANIFEST"
+
+    option :version_label,
+      aliases: "-l",
+      desc: "version_label for manifest, required if -m"
+
+    def package
+
+      if !options[:manifest].nil? && options[:version_label].nil?
+        puts "Must specify version_label when uploading manifest file"
+        help "package"
+        exit 1
+      end
+
+      Automan::Beanstalk::Package.new(options).upload_package
+    end
+  end
+end

data/lib/automan/cloudformation/errors.rb

@@ -0,0 +1,40 @@
+module Automan::Cloudformation
+  class MissingParametersError < StandardError
+  end
+
+  class BadTemplateError < StandardError
+  end
+
+  class StackExistsError < StandardError
+  end
+
+  class StackDoesNotExistError < StandardError
+  end
+
+  class StackBrokenError < StandardError
+  end
+
+  class StackDeletionError < StandardError
+  end
+
+  class StackCreationError < StandardError
+  end
+
+  class StackUpdateError < StandardError
+  end
+
+  class WaitTimedOutError < StandardError
+  end
+
+  class MissingAutoScalingGroupError < StandardError
+  end
+
+  class MissingManifestError < StandardError
+  end
+
+  class NoTemplatesError < StandardError
+  end
+
+  class InvalidTemplateError < StandardError
+  end
+end

data/lib/automan/cloudformation/launcher.rb

@@ -0,0 +1,196 @@
+require 'automan'
+require 'json'
+
+# TODO: add timeout
+
+module Automan::Cloudformation
+  class Launcher < Automan::Base
+    add_option :name,
+               :template,
+               :disable_rollback,
+               :enable_iam,
+               :enable_update,
+               :parameters,
+               :manifest,
+               :wait_for_completion
+
+    include Automan::Mixins::Utils
+
+    def initialize(options=nil)
+      @wait_for_completion = false
+      super
+      @wait = Wait.new({
+        delay: 120,
+        attempts: 15, # 15 x 2m == 30m
+        debug: true,
+        rescuer:  WaitRescuer.new(),
+        logger:   @logger
+      })
+    end
+
+    def manifest_exists?
+      s3_object_exists? manifest
+    end
+
+    def read_manifest
+      if !manifest_exists?
+        raise MissingManifestError, "Manifest #{manifest} does not exist"
+      end
+
+      # read manifest from s3 location
+      json = s3_read(manifest)
+
+      # parse it from json
+      data = JSON.parse json
+
+      # merge manifest parameters into cli parameters
+      parameters.merge!(data)
+    end
+
+    def template_handle(template_path)
+      if looks_like_s3_path? template_path
+        bucket, key = parse_s3_path template_path
+        return s3.buckets[bucket].objects[key]
+      else
+        return File.read(template_path)
+      end
+    end
+
+    def parse_template_parameters
+      cfn.validate_template( template_handle(template) )
+    end
+
+    def validate_parameters
+      if parameters.nil?
+        raise MissingParametersError, "there are no parameters!"
+      end
+
+      template_params_hash = parse_template_parameters
+
+      if template_params_hash.has_key? :code
+        mesg = template_params_hash[:message]
+        raise BadTemplateError, "template did not validate: #{mesg}"
+      end
+
+      # make sure any template parameters w/o a default are specified
+      template_params = template_params_hash[:parameters]
+      required_params = template_params.select { |x| !x.has_key? :default_value}
+      required_params.each do |rp|
+        unless parameters.has_key? rp[:parameter_key]
+          raise MissingParametersError, "required parameter #{rp[:parameter_key]} was not specified"
+        end
+      end
+
+      # add iam capabilities if needed
+      if template_params_hash.has_key?(:capabilities) &&
+        template_params_hash[:capabilities].include?('CAPABILITY_IAM')
+        self.enable_iam = true
+      end
+
+    end
+
+    def stack_exists?
+      cfn.stacks[name].exists?
+    end
+
+    def stack_status
+      cfn.stacks[name].status
+    end
+
+    def stack_launch_complete?
+      case stack_status
+      when 'CREATE_COMPLETE'
+        true
+      when 'CREATE_FAILED', /^ROLLBACK_/
+        raise StackCreationError, "Stack #{name} failed to launch"
+      else
+        false
+      end
+    end
+
+    def stack_update_complete?
+      case stack_status
+      when 'UPDATE_COMPLETE'
+        true
+      when 'UPDATE_FAILED', /^UPDATE_ROLLBACK_/
+        raise StackUpdateError, "Stack #{name} failed to update"
+      else
+        false
+      end
+    end
+
+    def launch
+      opts = {
+        parameters: parameters
+      }
+      opts[:capabilities] = ['CAPABILITY_IAM'] if enable_iam
+      opts[:disable_rollback] = disable_rollback
+
+      logger.info "launching stack #{name}"
+      cfn.stacks.create name, template_handle(template), opts
+
+      if wait_for_completion
+        logger.info "waiting for stack #{name} to launch"
+        wait_until { stack_launch_complete? }
+      end
+    end
+
+    def update_stack(name, opts)
+      cfn.stacks[name].update( opts )
+    end
+
+    def update
+      opts = {
+        template: template_handle(template),
+        parameters: parameters
+      }
+      opts[:capabilities] = ['CAPABILITY_IAM'] if enable_iam
+
+      logger.info "updating stack #{name}"
+
+      # if cfn determines that no updates are to be performed,
+      # it raises a ValidationError
+      begin
+        update_stack(name, opts)
+      rescue AWS::CloudFormation::Errors::ValidationError => e
+        if e.message != "No updates are to be performed."
+          raise e
+        else
+          logger.info e.message
+        end
+      end
+
+      if wait_for_completion
+        logger.info "waiting for stack #{name} to update"
+        wait_until { stack_update_complete? }
+      end
+
+    end
+
+    def launch_or_update
+
+      if !manifest.nil?
+        read_manifest
+      end
+
+      log_options
+
+      validate_parameters
+
+      if stack_exists?
+
+        logger.info "stack #{name} exists"
+        if enable_update == true
+          update
+        else
+          raise StackExistsError, "stack #{name} already exists"
+        end
+
+      else
+        launch
+      end
+
+    end
+  end
+
+end