automan 2.1.2

data/lib/automan/beanstalk/version.rb

@@ -0,0 +1,100 @@
+require 'automan'
+
+module Automan::Beanstalk
+  class Version < Automan::Base
+    add_option :application, :label
+
+    include Automan::Mixins::Utils
+
+    def versions
+      logger.info "listing versions for #{application}"
+
+      opts = {
+        application_name: application
+      }
+
+      response = eb.describe_application_versions opts
+
+      unless response.successful?
+        raise RequestFailedError "describe_application_versions failed: #{response.error}"
+      end
+
+      response.data[:application_versions]
+    end
+
+    def exists?
+      versions.each do |v|
+        if v[:application_name] == application && v[:version_label] == label
+          return true
+        end
+      end
+
+      return false
+    end
+
+    def create(package)
+      logger.info "creating version #{label}"
+
+      bucket, key = parse_s3_path(package)
+      opts = {
+        application_name: application,
+        version_label:    label,
+        source_bundle: {
+          s3_bucket:      bucket,
+          s3_key:         key
+        }
+      }
+
+      response = eb.create_application_version opts
+
+      unless response.successful?
+        raise RequestFailedError, "create_application_version failed: #{response.error}"
+      end
+    end
+
+    def delete_by_label(version_label)
+      logger.info "deleting version #{version_label} for application #{application}"
+
+      opts = {
+        application_name: application,
+        version_label: version_label,
+        delete_source_bundle: true
+      }
+
+      begin
+        response = eb.delete_application_version opts
+        unless response.successful?
+          raise RequestFailedError, "delete_application_version failed #{response.error}"
+        end
+      rescue AWS::ElasticBeanstalk::Errors::SourceBundleDeletionFailure => e
+        logger.warn e.message
+      end
+    end
+
+    def delete
+      log_options
+
+      delete_by_label label
+    end
+
+    def cull_versions(max_versions)
+      log_options
+
+      my_versions = versions
+      if my_versions.size <= max_versions
+        return
+      end
+
+      to_cull = my_versions.size - max_versions
+      logger.info "culling oldest #{to_cull} versions"
+
+      condemned = my_versions.sort_by {|v| v[:date_created] }[0..to_cull-1]
+
+      condemned.each do |i|
+        delete_by_label i[:version_label]
+      end
+    end
+
+  end
+
+end

data/lib/automan/chef/uploader.rb

@@ -0,0 +1,30 @@
+require 'automan'
+require 'zlib'
+require 'archive/tar/minitar'
+include Archive::Tar
+
+module Automan::Chef
+  class Uploader < Automan::Base
+    add_option :repopath, :s3path, :chefver, :tempdir
+
+    def upload
+      log_options
+
+      logger.info "packaging #{repopath} to #{tempdir}"
+      tgz = Zlib::GzipWriter.new(File.open("#{tempdir}/chef-repo.tgz", 'wb'))
+      Minitar.pack(repopath, tgz)
+
+      versions = ["#{chefver}", "latest"]
+      versions.each do |version|
+        options = {
+          localfile: "#{tempdir}/chef-repo.tgz",
+          s3file: "#{s3path}/#{version}/chef-repo.tgz"
+        }
+        s = Automan::S3::Uploader.new options
+        s.upload
+      end
+
+    end
+
+  end
+end

data/lib/automan/cli/baker.rb

@@ -0,0 +1,63 @@
+require 'thor'
+require 'automan'
+
+module Automan::Cli
+  class Baker < Thor
+    def self.exit_on_failure?
+        return true
+    end
+
+    desc "create", "create an image"
+
+    option :instance,
+      aliases: "-i",
+      desc: "instance to image",
+      required: true
+
+    option :name,
+      aliases: "-n",
+      desc: "what to name the image",
+      required: true
+
+    option :prune,
+      aliases: "-p",
+      type: :boolean,
+      default: true,
+      desc: "delete AMIs older than 30 days"
+
+    def create
+
+      s = Automan::Ec2::Image.new(options)
+      s.create
+      s.prune_amis if options[:prune]
+    end
+
+    desc "upload", "upload a chef repo"
+
+    option :repopath,
+      aliases: "-r",
+      desc: "path to chef repo",
+      required: true
+
+    option :s3path,
+      aliases: "-s",
+      desc: "s3 path to chef artifacts",
+      required: true
+
+    option :chefver,
+      aliases: "-c",
+      desc: "chef version",
+      required: true
+
+    option :tempdir,
+      aliases: "-t",
+      desc: "temporary directory",
+      required: true
+
+    def upload
+
+      s = Automan::Chef::Uploader.new(options)
+      s.upload
+    end
+  end
+end

data/lib/automan/cli/base.rb

@@ -0,0 +1,14 @@
+require 'thor'
+
+module Automan::Cli
+  class Base < Thor
+    def self.exit_on_failure?
+        return true
+    end
+
+    desc "version", "Show version"
+    def version
+      say Automan::VERSION
+    end
+  end
+end

data/lib/automan/cli/mover.rb

@@ -0,0 +1,47 @@
+require 'thor'
+require 'automan'
+
+module Automan::Cli
+  class Mover < Thor
+    def self.exit_on_failure?
+        return true
+    end
+
+    desc "upload", "upload a file"
+
+    option :localfile,
+      aliases: "-l",
+      desc: "local file",
+      required: true
+
+    option :s3file,
+      aliases: "-s",
+      desc: "s3 file",
+      required: true
+
+    def upload
+
+      s = Automan::S3::Uploader.new(options)
+      s.upload
+    end
+
+    desc "download", "download a file"
+
+    option :localfile,
+      aliases: "-l",
+      desc: "local file",
+      required: true
+
+    option :s3file,
+      aliases: "-s",
+      desc: "s3 file",
+      required: true
+
+    def download
+
+      s = Automan::S3::Downloader.new(options)
+      s.download
+    end
+
+  end
+end

data/lib/automan/cli/scanner.rb

@@ -0,0 +1,24 @@
+require 'automan'
+
+module Automan::Cli
+  class Scanner < Base
+
+    desc "show-instances", "Show instance ips and passwords for an environment"
+
+    option :environment,
+      required: true,
+      aliases: "-e",
+      default: "dev1",
+      desc: "Show instances from this environment"
+
+    option :private_key_file,
+      required: true,
+      aliases: "-p",
+      desc: "Path to private key to use when decrypting windows password"
+
+    def show_instances
+      i = Automan::Ec2::Instance.new(options)
+      i.show_env
+    end
+  end
+end

data/lib/automan/cli/snapper.rb

@@ -0,0 +1,78 @@
+require 'automan'
+require 'time'
+
+module Automan::Cli
+  class Snapper < Base
+
+    desc "create", "create a snapshot"
+
+    option :environment,
+      aliases: "-e",
+      desc: "environment of database to snapshot"
+
+    option :database,
+      aliases: "-d",
+      desc: "name of the database to snapshot"
+
+    option :name,
+      aliases: "-n",
+      desc: "what to name the snapshot"
+
+    option :prune,
+      aliases: "-p",
+      type: :boolean,
+      default: true,
+      desc: "make this snapshot prunable and delete other prunable snapshots older than 30 days"
+
+    option :wait_for_completion,
+      aliases: "-w",
+      type: :boolean,
+      default: false,
+      desc: "wait until snapshot is finished before exiting script"
+
+    def create
+      if options[:database].nil? && options[:environment].nil?
+        puts "Must specify either database or environment"
+        help "create"
+        exit 1
+      end
+
+      s = Automan::RDS::Snapshot.new(options)
+      s.log_aws_calls = true
+      s.prune_snapshots if options[:prune]
+      s.create
+    end
+
+    desc "delete", "delete a snapshot"
+
+    option :name,
+      required: true,
+      aliases: "-n",
+      desc: "name of snapshot to delete"
+
+    def delete
+      Automan::RDS::Snapshot.new(options).delete
+    end
+
+    desc "latest", "find the most recent snapshot"
+
+    option :database,
+      aliases: "-d",
+      desc: "name of the database to snapshot"
+
+    option :environment,
+      aliases: "-e",
+      desc: "environment of database to snapshot"
+
+    def latest
+      if options[:database].nil? && options[:environment].nil?
+        puts "Must specify either database or environment"
+        help "latest"
+        exit 1
+      end
+
+      Automan::RDS::Snapshot.new(options).latest
+    end
+
+  end
+end

data/lib/automan/cli/stacker.rb

@@ -0,0 +1,106 @@
+require 'automan'
+require 'json'
+
+module Automan::Cli
+  class Stacker < Base
+
+    desc "launch", "launches a cloudformation stack"
+
+    option :name,
+      required: true,
+      aliases: "-n",
+      desc: "name of the stack"
+
+    option :template,
+      required: true,
+      aliases: "-t",
+      desc: "path to template file (s3://bucket/template or local file)"
+
+    option :disable_rollback,
+      type: :boolean,
+      default: false,
+      desc: "set to true to disable rollback of failed stacks"
+
+    option :enable_update,
+      type: :boolean,
+      default: false,
+      desc: "set to true to update existing stack"
+
+    option :manifest,
+      aliases: "-m",
+      desc: "s3 path to manifest file containing version and package location"
+
+    option :parameters,
+      type: :hash,
+      default: {},
+      aliases: "-p",
+      desc: "stack parameters (e.g. -p Environment:dev InstanceType:m1.small)"
+
+    option :wait_for_completion,
+      aliases: "-w",
+      type: :boolean,
+      default: false,
+      desc: "wait until stack launches/updates before exiting script"
+
+
+    def launch
+      Automan::Cloudformation::Launcher.new(options).launch_or_update
+    end
+
+    desc "terminate", "terminate stack"
+
+    option :name,
+      required: true,
+      aliases: "-n",
+      desc: "name of the stack"
+
+    option :wait_for_completion,
+      aliases: "-w",
+      type: :boolean,
+      default: false,
+      desc: "wait until stack terminates before exiting script"
+
+    def terminate
+      Automan::Cloudformation::Terminator.new(options).terminate
+    end
+
+    desc "replace-instances", "terminate and replace running asg instances"
+
+    option :name,
+      required: true,
+      aliases: "-n",
+      desc: "name of the stack"
+
+    def replace_instances
+      Automan::Cloudformation::Replacer.new(options).replace_instances
+    end
+
+    desc "upload-templates", "validate and upload cloudformation templates"
+
+    option :template_files,
+      required: true,
+      aliases: "-t",
+      desc: "single or globbed templates"
+
+    option :s3_path,
+      required: true,
+      aliases: "-p",
+      desc: "s3 path to folder to drop the templates"
+
+    def upload_templates
+      Automan::Cloudformation::Uploader.new(options).upload_templates
+    end
+
+    desc "params", "print output from validate_template"
+
+    option :template,
+      required: true,
+      aliases: "-t",
+      desc: "path to template file (s3://bucket/template or local file)"
+
+    def params
+      h = Automan::Cloudformation::Launcher.new(options).parse_template_parameters
+      say JSON.pretty_generate h
+    end
+  end
+end