elastic_beans 0.1.0

data/lib/elastic_beans/command/create.rb

@@ -0,0 +1,150 @@
+require "timeout"
+require "ruby-progressbar"
+require "elastic_beans/dns_entry"
+require "elastic_beans/error"
+require "elastic_beans/environment/webserver"
+
+module ElasticBeans
+  module Command
+    class Create
+      USAGE = "create -a APPLICATION [-t TYPE] [-q QUEUE] [-d PRETTY_DNS] ENVIRONMENT"
+      DESC = "Create a new environment in Elastic Beanstalk and attach it to relevant resources"
+      LONG_DESC = <<-LONG_DESC
+Create a new environment in Elastic Beanstalk and attach it to relevant resources
+The environment type must be one of the recognized types: webserver, worker, or scheduled.
+
+Requires the Elastic Beanstalk application name.
+Requires AWS credentials to be set in the environment, i.e. AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY.
+      LONG_DESC
+
+      def initialize(
+        environment_name:,
+        dns: nil,
+        queue: nil,
+        tags:,
+        application:,
+        ui:,
+        elastic_beanstalk:,
+        route53:,
+        s3:
+      )
+        @environment_name = environment_name
+        @environment_type = environment_name
+        @dns = dns
+        if @environment_type == "worker"
+          @queue = queue || "default"
+          @environment_name = "#{@environment_name}-#{@queue}"
+        end
+        @tags = tags
+        @application = application
+        @ui = ui
+        @elastic_beanstalk = elastic_beanstalk
+        @route53 = route53
+        @s3 = s3
+      end
+
+      def run
+        ui.debug { "Creating application version from code at #{Dir.pwd}..." }
+        version = application.deployed_version || ElasticBeans::ApplicationVersion.create(
+          working_directory: Dir.pwd,
+          application: application,
+          elastic_beanstalk: elastic_beanstalk,
+          s3: s3,
+        )
+        environment = ElasticBeans::Environment.new_by_type(
+          environment_type,
+          suffix: environment_name,
+          queue: queue,
+          application: application,
+          elastic_beanstalk: elastic_beanstalk,
+        )
+
+        dns_thread = Thread.new do
+          if environment_type == "webserver" && dns_entry
+            ui.debug { "Updating DNS entry `#{dns_entry.name}'..." }
+            dns_entry.update(dns_value(environment))
+          end
+        end
+
+        ui.info("Creating environment `#{environment.name}'...")
+        progressbar = ProgressBar.create(title: "Creating", total: nil, output: ui.stdout)
+        environment_thread = Thread.new do
+          environment.create(
+            tags: tags,
+            version: version.version_label,
+          )
+        end
+
+        loop do
+          sleep 1
+          progressbar.increment
+          if !environment_thread.alive? && !dns_thread.alive?
+            progressbar.total = progressbar.progress
+            break
+          end
+        end
+
+        environment_thread.join
+        dns_thread.join
+        ui.info("Created!")
+        if environment_type == "webserver"
+          if dns_entry
+            host = dns_entry.name
+          else
+            host = environment.cname
+          end
+          ui.info("#{application.name} reachable at https://#{host}/")
+        end
+      end
+
+      private
+
+      attr_reader(
+        :application,
+        :dns,
+        :environment_name,
+        :environment_type,
+        :queue,
+        :tags,
+        :elastic_beanstalk,
+        :route53,
+        :s3,
+        :ui,
+      )
+
+      def dns_entry
+        if dns.nil?
+          return nil
+        end
+
+        DnsEntry.new(dns, route53: route53)
+      end
+
+      def dns_value(environment)
+        Timeout.timeout(1200) do
+          loop do
+            sleep 1
+            begin
+              value = environment.cname
+              return value if value
+            rescue ElasticBeans::Environment::Webserver::MissingEnvironmentError
+              # keep trying
+            end
+          end
+        end
+      rescue Timeout::Error
+        raise UnhealthyEnvironmentError.new(environment_name: environment.name)
+      end
+
+      class UnhealthyEnvironmentError < ElasticBeans::Error
+        def initialize(environment_name:)
+          @environment_name = environment_name
+        end
+
+        def message
+          "Environment `#{@environment_name}' is still not available"
+        end
+      end
+    end
+  end
+end

data/lib/elastic_beans/command/deploy.rb

@@ -0,0 +1,77 @@
+require "elastic_beans/error/environments_not_ready"
+
+module ElasticBeans
+  module Command
+    class Deploy
+      USAGE = "deploy -a APPLICATION"
+      DESC = "Deploy the HEAD git commit to all environments in Elastic Beanstalk"
+      LONG_DESC = <<-LONG_DESC
+Deploy the HEAD git commit to all environments in Elastic Beanstalk.
+
+Requires the application name.
+Requires AWS credentials to be set in the environment, i.e. AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY.
+      LONG_DESC
+
+      def initialize(application:, elastic_beanstalk:, s3:, ui:)
+        @application = application
+        @elastic_beanstalk = elastic_beanstalk
+        @s3 = s3
+        @ui = ui
+      end
+
+      def run
+        environments = application.environments
+        unready_environments = environments.select { |environment| environment.status != "Ready" }
+        if unready_environments.any?
+          raise EnvironmentsNotReady.new(environments: unready_environments)
+        end
+
+        ui.debug { "Creating application version from code at #{Dir.pwd}..." }
+        progressbar = ProgressBar.create(title: "Deploying", total: nil, output: ui.stdout)
+        version = nil
+        version_thread = Thread.new do
+          version = ElasticBeans::ApplicationVersion.create(
+            working_directory: Dir.pwd,
+            application: application,
+            elastic_beanstalk: elastic_beanstalk,
+            s3: s3,
+          ).version_label
+        end
+
+        loop do
+          sleep 1
+          progressbar.increment
+          if !version_thread.alive?
+            break
+          end
+        end
+        version_thread.join
+
+        progressbar.log("Deploying version #{version} to #{application.name}...")
+        deploy_threads = environments.map { |environment|
+          progressbar.log("Deploying version '#{version}' to environment `#{environment.name}'...")
+          thread = Thread.new do
+            environment.deploy_version(version)
+          end
+          progressbar.increment
+          thread
+        }
+
+        loop do
+          sleep 1
+          progressbar.increment
+          if deploy_threads.none?(&:alive?)
+            progressbar.total = progressbar.progress
+            break
+          end
+        end
+        deploy_threads.each(&:join)
+        ui.info("Deployed!")
+      end
+
+      private
+
+      attr_reader :application, :elastic_beanstalk, :s3, :ui
+    end
+  end
+end

data/lib/elastic_beans/command/exec.rb

@@ -0,0 +1,37 @@
+require "shellwords"
+
+module ElasticBeans
+  module Command
+    class Exec
+      USAGE = "exec -a APPLICATION rake db:migrate"
+      DESC = "Run an arbitrary command in the context of your application"
+      LONG_DESC = <<-LONG_DESC
+Run an arbitrary command in the context of your application.
+The command is run in an "exec" environment, separate from your webserver or worker environments.
+Upload gzipped output from the command to an HTTP endpoint.
+
+Requires the Elastic Beanstalk application name.
+Requires AWS credentials to be set in the environment, i.e. AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY.
+      LONG_DESC
+
+      def initialize(application:, sqs:, ui:)
+        @application = application
+        @sqs = sqs
+        @ui = ui
+      end
+
+      def run(*command_parts)
+        ui.info("Running `#{command_parts.join(" ")}' on #{application.name}...")
+        application.enqueue_command(command(command_parts), sqs: sqs)
+      end
+
+      private
+
+      attr_reader :application, :sqs, :ui
+
+      def command(command_parts)
+        command_parts.map { |word| Shellwords.escape(word) }.join(" ")
+      end
+    end
+  end
+end

data/lib/elastic_beans/command/set_env.rb

@@ -0,0 +1,77 @@
+require "ruby-progressbar"
+require "elastic_beans/error/environments_not_ready"
+
+module ElasticBeans
+  module Command
+    class SetEnv
+      USAGE = "setenv -a APPLICATION KEY=value [KEY=value]..."
+      DESC = "Update environment variables in the Elastic Beanstalk application"
+      LONG_DESC = <<-LONG_DESC
+Update environment variables in the Elastic Beanstalk application.
+Updates all running environments and configuration templates resulting from `configure`.
+
+Requires the application name.
+Requires AWS credentials to be set in the environment, i.e. AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY.
+      LONG_DESC
+
+      def initialize(application:, ui:)
+        @application = application
+        @ui = ui
+      end
+
+      def run(*env_pairs)
+        environments = application.environments
+        unready_environments = environments.select { |environment| environment.status != "Ready" }
+        if unready_environments.any?
+          raise EnvironmentsNotReady.new(environments: unready_environments)
+        end
+
+        env_vars = pairs_as_hash(env_pairs)
+        threads = []
+
+        progressbar = ProgressBar.create(title: "Updating", total: nil, output: ui.stdout)
+
+        progressbar.log("Updating configuration templates in #{application.name}...")
+        threads += application.configuration_templates.map { |config|
+          thread = Thread.new do
+            config.update_environment(env_vars)
+          end
+          progressbar.increment
+          thread
+        }
+
+        threads += environments.map { |environment|
+          progressbar.log("Updating `#{environment.name}'...")
+          thread = Thread.new do
+            environment.update_environment(env_vars)
+          end
+          progressbar.increment
+          thread
+        }
+
+        loop do
+          sleep 1
+          progressbar.increment
+          if threads.none?(&:alive?)
+            progressbar.total = progressbar.progress
+            break
+          end
+        end
+
+        threads.each(&:join)
+      end
+
+      private
+
+      attr_reader :application, :ui
+
+      def pairs_as_hash(pairs)
+        pairs.reduce({}) { |acc, pair|
+          key, value = pair.split('=', 2)
+          acc[key] = value
+          acc
+        }
+      end
+    end
+  end
+end

data/lib/elastic_beans/command/talk.rb

@@ -0,0 +1,74 @@
+module ElasticBeans
+  module Command
+    class Talk
+      BEANSTALK = <<-BEANSTALK
+...........................,:~~:............................
+........................:~~~~~++++=~........................
+.....................,:~~~~~~~++++++~:......................
+...................,~~~~~~~~~~++++++++++=~..................
+...................,~~~~~~~~~~++++++++++++~:................
+...................,~~~~~~~~~~+++++++++++++=:...............
+...................,~~~~~~~~~~+++++++++++++=:...............
+...................,~~~~~~~~~~+++++++++++++=:...............
+...................,~~~~~~~~~~+++++++++++++=:......,::~~:...
+...................,~~~~~~~~~~+++++++++++++=~::~~~~~~~=+++=:
+...................,~~~~~~~~~~++++++=~~~~~~~~~~~~~~~~~=+++=~
+...................,~~~~~~~~~~++++==~~~~~~~~~~~~~~~~~~=+++=~
+...................,~~~~~~~~~~++++==~~~~~~~~~~~~~~~~~~=+++=~
+...................,~~~~~~~~~~++++==~~~~~~~~~~~~~~~~~~=+++=~
+...................,~~~~~~~~~~++++==~~~~~~~~~~~~~~~~~~=+++=~
+...................,~~~~~~~~~~++++==~~~~~~~~~~~~~~~~~~=+++=~
+...................,~~~~~~~~~~++++==~~~~~~~~~~~~~~~~~~=+++=~
+...................,~~~~~~~~~~++++==~~~~~~~~~~~~~~~~~~=+++=~
+...................,~~~~~~~~~~++++==~~~~~~~~~~~~~~~~~~=+++=~
+...................,~~~~~~~~~~++++==~~~~~~~:::::::::::::::::
+...................,~~~~~~~~~~++++==~::::::::,,,,,,.........
+...................,~~~~~~~~~~+++++++++++++=:...............
+...................,~~~~~~~~~~+++++++++++++=:...............
+...................,~~~~~~~~~~+++++++++++++=:...............
+...................,~~~~~~~~~~+++++++++++++=:...............
+...................,~~~~~~~~~~+++++++++++++=:...............
+...................,~~~~~~~~~~+++++++++++++=:...............
+...............:~~~~~~~~~~~~~~+++++++++:....................
+...............:~~~~~~~~~~~~~~+++++++++:....................
+...............:~~~~~~~~~~~~~~+++++++++:....................
+...............:~~~~~~~~~~~~~~+++++++++:....................
+...............:~~~~~~~~~~~~~~+++++++++:....................
+......,:~~=++????????+++~~~~~~+++++++++:....................
+~==II??????????+++++++++~~~~~~+++++++++:....................
+:~~~~=++++++++++++++++++~~~~~~+++++++++:....................
+:~~~~=++++++++++++++++++~~~~~~+++++++++:....................
+:~~~~=++++++++++++++++++~~~~~~+++++++++:....................
+:~~~~=++++++++++++++++++~~~~~~+++++++++:....................
+:~~~~=++++++++++++++++++~~~~~~+++++++++:....................
+:~~~~=++++++++++++++++++~~~~~~+++++++++:....................
+:~~~~=++++++++++++++++++~~~~~~+++++++++:....................
+:~~~~=++++++++++++++++++~~~~~~+++++++++:....................
+:~~~~=+++++++++++++++=~~~~~~~~+++++++++:....................
+:~~~~=++++++=~~~~~~~~~~~~~~~~~+++++++++:....................
+...:~=~~:......:~~~~~~~~~~~~~~+++++++++:....................
+...............:~~~~~~~~~~~~~~+++++++++:....................
+...............:~~~~~~~~~~~~~~+++++++++:....................
+...............:~~~~~~~~~~~~~~+++++++++:....................
+...............,:~~~~~~~~~~~~~+++++++++:....................
+..................:~~~~~~~~~~~+++++++++:....................
+.....................,:~~~~~~~++++++~:......................
+........................:~~~~~++++=~........................
+...........................,:~~:............................
+............................................................
+      BEANSTALK
+
+      def initialize(ui:)
+        @ui = ui
+      end
+
+      def run
+        ui.info BEANSTALK
+      end
+
+      private
+
+      attr_reader :ui
+    end
+  end
+end

data/lib/elastic_beans/command/version.rb

@@ -0,0 +1,17 @@
+module ElasticBeans
+  module Command
+    class Version
+      def initialize(ui:)
+        @ui = ui
+      end
+
+      def run
+        ui.info VERSION
+      end
+
+      private
+
+      attr_reader :ui
+    end
+  end
+end

data/lib/elastic_beans/command.rb

@@ -0,0 +1,12 @@
+require "elastic_beans/command/configure"
+require "elastic_beans/command/create"
+require "elastic_beans/command/deploy"
+require "elastic_beans/command/exec"
+require "elastic_beans/command/set_env"
+require "elastic_beans/command/talk"
+require "elastic_beans/command/version"
+
+module ElasticBeans
+  module Command
+  end
+end