asg-rebooter 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: ac27241a28fdcc9c6371c1eb378a67ac6350bc7f232d50fef56228ae622aa50f
+  data.tar.gz: 1eb4afaa574d39eef83a5aac53de1307e25418c6733fda43970bd7f5f71853c1
+SHA512:
+  metadata.gz: dd8c0747be0c121dfbb86a389f7466edeec5db0fbf3eff3729c6918cbdc3bc34d63e655b69a49a11e75be086688378ab03416dd3e9220d457517ec86050be5dc
+  data.tar.gz: bee3f141d30672812527be0a3e43c50e7d1fd4ab29d1df55e13f8fa452f7c3b48235208524975b645b10f6ff102e9460686657485ee583965e3b16385322cce7

data/.gitignore

@@ -0,0 +1,11 @@
+/.bundle/
+/.yardoc
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+
+# rspec failure tracking
+.rspec_status

data/.rspec

@@ -0,0 +1,3 @@
+--format documentation
+--color
+--require spec_helper

data/.travis.yml

@@ -0,0 +1,7 @@
+---
+sudo: false
+language: ruby
+cache: bundler
+rvm:
+  - 2.6.5
+before_install: gem install bundler -v 1.17.3

data/CODE_OF_CONDUCT.md

@@ -0,0 +1,74 @@
+# Contributor Covenant Code of Conduct
+
+## Our Pledge
+
+In the interest of fostering an open and welcoming environment, we as
+contributors and maintainers pledge to making participation in our project and
+our community a harassment-free experience for everyone, regardless of age, body
+size, disability, ethnicity, gender identity and expression, level of experience,
+nationality, personal appearance, race, religion, or sexual identity and
+orientation.
+
+## Our Standards
+
+Examples of behavior that contributes to creating a positive environment
+include:
+
+* Using welcoming and inclusive language
+* Being respectful of differing viewpoints and experiences
+* Gracefully accepting constructive criticism
+* Focusing on what is best for the community
+* Showing empathy towards other community members
+
+Examples of unacceptable behavior by participants include:
+
+* The use of sexualized language or imagery and unwelcome sexual attention or
+advances
+* Trolling, insulting/derogatory comments, and personal or political attacks
+* Public or private harassment
+* Publishing others' private information, such as a physical or electronic
+  address, without explicit permission
+* Other conduct which could reasonably be considered inappropriate in a
+  professional setting
+
+## Our Responsibilities
+
+Project maintainers are responsible for clarifying the standards of acceptable
+behavior and are expected to take appropriate and fair corrective action in
+response to any instances of unacceptable behavior.
+
+Project maintainers have the right and responsibility to remove, edit, or
+reject comments, commits, code, wiki edits, issues, and other contributions
+that are not aligned to this Code of Conduct, or to ban temporarily or
+permanently any contributor for other behaviors that they deem inappropriate,
+threatening, offensive, or harmful.
+
+## Scope
+
+This Code of Conduct applies both within project spaces and in public spaces
+when an individual is representing the project or its community. Examples of
+representing a project or community include using an official project e-mail
+address, posting via an official social media account, or acting as an appointed
+representative at an online or offline event. Representation of a project may be
+further defined and clarified by project maintainers.
+
+## Enforcement
+
+Instances of abusive, harassing, or otherwise unacceptable behavior may be
+reported by contacting the project team at tom@hackerone.com. All
+complaints will be reviewed and investigated and will result in a response that
+is deemed necessary and appropriate to the circumstances. The project team is
+obligated to maintain confidentiality with regard to the reporter of an incident.
+Further details of specific enforcement policies may be posted separately.
+
+Project maintainers who do not follow or enforce the Code of Conduct in good
+faith may face temporary or permanent repercussions as determined by other
+members of the project's leadership.
+
+## Attribution
+
+This Code of Conduct is adapted from the [Contributor Covenant][homepage], version 1.4,
+available at [http://contributor-covenant.org/version/1/4][version]
+
+[homepage]: http://contributor-covenant.org
+[version]: http://contributor-covenant.org/version/1/4/

data/Gemfile

@@ -0,0 +1,6 @@
+source "https://rubygems.org"
+
+git_source(:github) {|repo_name| "https://github.com/#{repo_name}" }
+
+# Specify your gem's dependencies in asg-rebooter.gemspec
+gemspec

data/Gemfile.lock

@@ -0,0 +1,57 @@
+PATH
+  remote: .
+  specs:
+    asg-rebooter (0.1.0)
+      aws-sdk-autoscaling (~> 1.0)
+      aws-sdk-ec2 (~> 1.0)
+      aws-sdk-elasticloadbalancingv2 (~> 1.0)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    aws-eventstream (1.0.3)
+    aws-partitions (1.256.0)
+    aws-sdk-autoscaling (1.30.0)
+      aws-sdk-core (~> 3, >= 3.71.0)
+      aws-sigv4 (~> 1.1)
+    aws-sdk-core (3.86.0)
+      aws-eventstream (~> 1.0, >= 1.0.2)
+      aws-partitions (~> 1, >= 1.239.0)
+      aws-sigv4 (~> 1.1)
+      jmespath (~> 1.0)
+    aws-sdk-ec2 (1.126.0)
+      aws-sdk-core (~> 3, >= 3.71.0)
+      aws-sigv4 (~> 1.1)
+    aws-sdk-elasticloadbalancingv2 (1.39.0)
+      aws-sdk-core (~> 3, >= 3.71.0)
+      aws-sigv4 (~> 1.1)
+    aws-sigv4 (1.1.0)
+      aws-eventstream (~> 1.0, >= 1.0.2)
+    diff-lcs (1.3)
+    jmespath (1.4.0)
+    rake (10.5.0)
+    rspec (3.9.0)
+      rspec-core (~> 3.9.0)
+      rspec-expectations (~> 3.9.0)
+      rspec-mocks (~> 3.9.0)
+    rspec-core (3.9.0)
+      rspec-support (~> 3.9.0)
+    rspec-expectations (3.9.0)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.9.0)
+    rspec-mocks (3.9.0)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.9.0)
+    rspec-support (3.9.0)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  asg-rebooter!
+  bundler (~> 1.17)
+  rake (~> 10.0)
+  rspec (~> 3.0)
+
+BUNDLED WITH
+   1.17.3

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2019 Tom de Vries
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.md

@@ -0,0 +1,43 @@
+# Asg::Rebooter
+
+Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/asg/rebooter`. To experiment with that code, run `bin/console` for an interactive prompt.
+
+TODO: Delete this and the text above, and describe your gem
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'asg-rebooter'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install asg-rebooter
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/asg-rebooter. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [Contributor Covenant](http://contributor-covenant.org) code of conduct.
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
+
+## Code of Conduct
+
+Everyone interacting in the Asg::Rebooter project’s codebases, issue trackers, chat rooms and mailing lists is expected to follow the [code of conduct](https://github.com/[USERNAME]/asg-rebooter/blob/master/CODE_OF_CONDUCT.md).

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/asg-rebooter.gemspec

@@ -0,0 +1,46 @@
+
+lib = File.expand_path("../lib", __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "asg/rebooter/version"
+
+Gem::Specification.new do |spec|
+  spec.name          = "asg-rebooter"
+  spec.version       = Asg::Rebooter::VERSION
+  spec.authors       = ["Tom de Vries"]
+  spec.email         = ["tom@hackerone.com"]
+
+  spec.summary       = %q{Gracefully restart EC2 instances within an Auto Scaling Group.}
+  spec.description   = %q{See summary.}
+  spec.homepage      = "https://github.com/tomdev/asg-rebooter"
+  spec.license       = "MIT"
+
+  # Prevent pushing this gem to RubyGems.org. To allow pushes either set the 'allowed_push_host'
+  # to allow pushing to a single host or delete this section to allow pushing to any host.
+  if spec.respond_to?(:metadata)
+    spec.metadata["allowed_push_host"] = "https://rubygems.org"
+
+    spec.metadata["homepage_uri"] = spec.homepage
+    spec.metadata["source_code_uri"] = spec.homepage
+    spec.metadata["changelog_uri"] = spec.homepage
+  else
+    raise "RubyGems 2.0 or newer is required to protect against " \
+      "public gem pushes."
+  end
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files         = Dir.chdir(File.expand_path('..', __FILE__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "aws-sdk-autoscaling", "~> 1.0"
+  spec.add_dependency "aws-sdk-ec2", "~> 1.0"
+  spec.add_dependency "aws-sdk-elasticloadbalancingv2", "~> 1.0"
+
+  spec.add_development_dependency "bundler", "~> 1.17"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rspec", "~> 3.0"
+end

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "asg/rebooter"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start(__FILE__)

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/exe/asg-rebooter

@@ -0,0 +1,29 @@
+#!/usr/bin/env ruby
+
+require 'asg-rebooter'
+
+argument_present = ARGV.length > 0;
+
+if argument_present
+  asg_name = ARGV[0]
+else
+  puts """
+  asg-rebooter requires
+
+  usage: asg-rebooter <auto_scaling_group_name>
+  """
+
+  exit 1
+end
+
+if ENV['REBOOT']
+  asg = Asg::Rebooter::Wrappers::AutoScalingGroup.new(asg_name).reboot
+else
+  puts """
+  defaulting to dry-run: asg-rebooter would have rebooted AutoScalingGroup with name \"#{asg_name}\"
+
+  To start the reboot procedure use the following command:
+
+      REBOOT=true asg-rebooter #{asg_name}
+  """
+end

data/lib/asg/rebooter.rb

@@ -0,0 +1,23 @@
+require 'aws-sdk-autoscaling'
+require 'aws-sdk-ec2'
+require 'aws-sdk-elasticloadbalancingv2'
+
+require 'asg/rebooter/version'
+require 'asg/rebooter/configuration/auto_scaling_group'
+require 'asg/rebooter/configuration/target_group'
+require 'asg/rebooter/instance_rebooter'
+require 'asg/rebooter/wrappers/auto_scaling_group'
+require 'asg/rebooter/wrappers/instance'
+require 'asg/rebooter/wrappers/target_group'
+
+module Asg
+  module Rebooter
+    class Error < StandardError; end
+
+    class Work
+      def execute
+        puts 'Doing all the hard work for you'
+      end
+    end
+  end
+end

data/lib/asg/rebooter/configuration/auto_scaling_group.rb

@@ -0,0 +1,69 @@
+module Asg
+  module Rebooter
+    module Configuration
+      class AutoScalingGroup
+        ASG_MIN_SIZE_DELTA = 1
+        AZREBALANCE = 'AZRebalance'
+
+        attr_reader :asg, :original_min_size, :original_suspended_processes
+
+        def initialize(asg)
+          @asg = asg
+
+          set_original_values
+        end
+
+        def prepare
+          suspend_az_rebalance
+          decrease_min_size
+        end
+
+        def restore
+          restore_az_rebalance
+          restore_min_size
+        end
+
+        private
+
+        def set_original_values
+          @original_min_size = asg.min_size
+          @original_suspended_processes = asg.suspended_processes
+        end
+
+        def suspend_az_rebalance
+          return unless adjust_az_rebalance?
+
+          asg.suspend_processes([AZREBALANCE])
+        end
+
+        def restore_az_rebalance
+          return unless adjust_az_rebalance?
+
+          asg.resume_processes([AZREBALANCE])
+        end
+
+        def decrease_min_size
+          return unless adjust_min_size?
+
+          new_min_size = @original_min_size - ASG_MIN_SIZE_DELTA
+
+          asg.update_min_size(new_min_size)
+        end
+
+        def restore_min_size
+          return unless adjust_min_size?
+
+          asg.update_min_size(original_min_size)
+        end
+
+        def adjust_min_size?
+          @original_min_size > 0
+        end
+
+        def adjust_az_rebalance?
+          !original_suspended_processes.include?(AZREBALANCE)
+        end
+      end
+    end
+  end
+end

data/lib/asg/rebooter/configuration/target_group.rb

@@ -0,0 +1,34 @@
+module Asg
+  module Rebooter
+    module Configuration
+      class TargetGroup
+        DEREGISTRATION_DELAY_DURING_REBOOT_PROCEDURE = '60'
+
+        DEREGISTRATION_CONFIG = [
+          key: 'deregistration_delay.timeout_seconds',
+          value: DEREGISTRATION_DELAY_DURING_REBOOT_PROCEDURE
+        ]
+
+        attr_reader :target_group, :original_attributes
+
+        def initialize(target_group)
+          @target_group = target_group
+
+          @original_attributes = target_group.attributes
+        end
+
+        def prepare
+          target_group.update_attributes(DEREGISTRATION_CONFIG)
+
+          # Validate?
+        end
+
+        def restore
+          target_group.update_attributes(original_attributes)
+
+          # Validate?
+        end
+      end
+    end
+  end
+end

data/lib/asg/rebooter/instance_rebooter.rb

@@ -0,0 +1,68 @@
+module Asg
+  module Rebooter
+    class InstanceRebooter
+      REBOOT_TIMEOUT = 60
+
+      attr_reader :asg, :asg_config, :target_group_config
+
+      def initialize(asg)
+        @asg = asg
+      end
+
+      def execute
+        puts "[start] Starting reboot sequence for: #{asg.name}"
+
+        begin
+          prepare_configuration
+
+          instances.each do |instance|
+            puts "#{'*' * 4} #{instance.instance_id} #{'*' * 4}"
+            puts "#{instance.instance_id} [enter_standby]"
+            instance.enter_standby
+
+            puts "#{instance.instance_id} [reboot]"
+            instance.reboot
+
+            sleep REBOOT_TIMEOUT # TODO: Ensure instance is rebooted: how to do this?
+
+            puts "#{instance.instance_id} [exit_standby]"
+            instance.exit_standby
+
+            puts "#{instance.instance_id} [wait_for_healthy]"
+            target_group.wait_for_healthy(instance)
+
+            puts "\n"
+          end
+        ensure
+          restore_configuration
+        end
+      end
+
+      private
+
+      def instances
+        asg.instances
+      end
+
+      def target_group
+        asg.target_groups.first
+      end
+
+      def prepare_configuration
+        @asg_config = Asg::Rebooter::Configuration::AutoScalingGroup.new(asg)
+
+        @target_group_config = asg.target_groups.map do |target_group|
+          Asg::Rebooter::Configuration::TargetGroup.new(target_group)
+        end
+
+        asg_config.prepare
+        target_group_config.each(&:prepare)
+      end
+
+      def restore_configuration
+        asg_config.restore
+        target_group_config.each(&:restore)
+      end
+    end
+  end
+end

data/lib/asg/rebooter/validate_credentials.rb

@@ -0,0 +1,6 @@
+module Asg
+  module Rebooter
+    class ValidateCredentials
+    end
+  end
+end

data/lib/asg/rebooter/version.rb

@@ -0,0 +1,5 @@
+module Asg
+  module Rebooter
+    VERSION = "0.0.1"
+  end
+end

data/lib/asg/rebooter/wrappers/auto_scaling_group.rb

@@ -0,0 +1,59 @@
+module Asg
+  module Rebooter
+    module Wrappers
+      class AutoScalingGroupNameRequiredError < StandardError; end
+
+      class AutoScalingGroup
+        attr_reader :name
+
+        def initialize(name)
+          raise AutoScalingGroupNameRequiredError unless name
+
+          @name = name
+        end
+
+        def suspended_processes
+          asg.suspended_processes
+        end
+
+        def suspend_processes(processes)
+          asg.suspend_processes(scaling_processes: processes)
+        end
+
+        def resume_processes(processes)
+          asg.resume_processes(scaling_processes: processes)
+        end
+
+        def min_size
+          asg.min_size
+        end
+
+        def update_min_size(new_min_size)
+          asg.update(min_size: new_min_size)
+        end
+
+        def target_groups
+          asg.target_group_arns.map do |arn|
+            Asg::Rebooter::Wrappers::TargetGroup.new(arn)
+          end
+        end
+
+        def instances
+          asg.instances.map do |instance|
+            Asg::Rebooter::Wrappers::Instance.new(instance)
+          end
+        end
+
+        def reboot
+          Asg::Rebooter::InstanceRebooter.new(self).execute
+        end
+
+        private
+
+        def asg
+          Aws::AutoScaling::AutoScalingGroup.new(name)
+        end
+      end
+    end
+  end
+end

data/lib/asg/rebooter/wrappers/instance.rb

@@ -0,0 +1,117 @@
+# Example response from AWS for Aws::AutoScaling::Instance:
+#
+#   #<Aws::AutoScaling::Instance:0x00007fc8a2781438
+#    @client=#<Aws::AutoScaling::Client>,
+#    @data=
+#     #<struct Aws::AutoScaling::Types::Instance
+#      instance_id="i-007b9fa2f7df584f7",
+#      instance_type="t3.medium",
+#      availability_zone="us-west-2b",
+#      lifecycle_state="InService",
+#      health_status="Healthy",
+#      launch_configuration_name=
+#       "h1-testserver-lc-asg-rebooter-20191219124104839000000001",
+#      launch_template=nil,
+#      protected_from_scale_in=false,
+#      weighted_capacity=nil>,
+#    @group_name="h1-testserver-asg-asg-rebooter",
+#    @id="i-007b9fa2f7df584f7">
+
+module Asg
+  module Rebooter
+    module Wrappers
+      class Instance
+        LIFECYCLE_STANDBY = 'Standby'
+        LIFECYCLE_IN_SERVICE = 'InService'
+        LIFECYCLE_CHANGE_POLL_TIMEOUT = 4
+
+        HEALTH_STATUS_HEALTHY = 'HEALTHY'
+
+        SLEEP_AFTER_REBOOT = 30
+        SLEEP_AFTER_STANDBY = 10
+
+        POLL_TIMEOUT = 4
+
+        attr_reader :instance
+
+        def initialize(instance)
+          #
+          # @instance is of type Aws::AutoScaling::Instance
+          #
+          # #ec2_instance is of type Aws::EC2::Instance
+          #
+          @instance = instance
+        end
+
+        def method_missing(m, *args, &block)
+          instance.send(m, *args, &block)
+        end
+
+        def enter_standby
+          # raise when in invalid state for action
+
+          instance.enter_standby(should_decrement_desired_capacity: true)
+
+          ensure_lifecycle_state(LIFECYCLE_STANDBY)
+        end
+
+        def exit_standby
+          # raise if in invalid state for action
+
+          instance.exit_standby
+
+          ensure_lifecycle_state(LIFECYCLE_IN_SERVICE)
+        end
+
+        def ensure_lifecycle_state(state)
+          loop do
+            instance.reload
+
+            break if instance.lifecycle_state == state
+
+            sleep LIFECYCLE_CHANGE_POLL_TIMEOUT
+          end
+
+          true
+        end
+
+        def standby?
+          instance.reload
+
+          instance.lifecycle_state == LIFECYCLE_STANDBY
+        end
+
+        def health_status
+          instance.reload.health_status
+        end
+
+        def healthy?
+          health_status == HEALTH_STATUS_HEALTHY
+        end
+
+        def wait_for_healthy
+          #
+          # TODO: Should check health state on the TargetGroup to ensure the instance is ready for work
+          #
+          loop do
+            break if healthy?
+
+            sleep(POLL_TIMEOUT)
+          end
+
+          true
+        end
+
+        def reboot
+          ec2_instance.reboot
+        end
+
+        private
+
+        def ec2_instance
+          Aws::EC2::Instance.new(instance.instance_id)
+        end
+      end
+    end
+  end
+end

data/lib/asg/rebooter/wrappers/target_group.rb

@@ -0,0 +1,72 @@
+module Asg
+  module Rebooter
+    module Wrappers
+      class TargetGroupArnRequiredError < StandardError; end
+
+      class TargetGroup
+        TARGET_GROUP_INSTANCE_HEALTHY = 'healthy'
+        POLL_TIMEOUT = 4
+
+        attr_reader :arn
+
+        def initialize(arn)
+          raise TargetGroupArnRequiredError unless arn
+
+          @arn = arn
+        end
+
+        def attributes
+          target_group_attributes = client.describe_target_group_attributes(target_group_arn: arn)
+          target_group_attributes.attributes.map(&:to_hash)
+        end
+
+        def restore_attributes
+          update_attributes(original_attributes)
+        end
+
+        def update_attributes(attributes_to_update)
+          client.modify_target_group_attributes \
+            target_group_arn: arn,
+            attributes: attributes_to_update
+        end
+
+        def target_health_descriptions
+          client.describe_target_health(target_group_arn: arn).target_health_descriptions
+        end
+
+        def health_for_instance(instance)
+          #
+          # TODO: Write specs and error handling
+          #
+          health_description_for_instance = target_health_descriptions.keep_if do |target_health_description|
+            target_health_description.target.id == instance.id
+          end
+
+          if health_description_for_instance.any?
+            health_description_for_instance.first.target_health.state
+          else
+            'unknown'
+          end
+        end
+
+        def wait_for_healthy(instance)
+          loop do
+            healthy = health_for_instance(instance) == TARGET_GROUP_INSTANCE_HEALTHY
+
+            break if healthy
+
+            sleep(POLL_TIMEOUT)
+          end
+
+          true
+        end
+
+        private
+
+        def client
+          @client ||= Aws::ElasticLoadBalancingV2::Client.new
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,154 @@
+--- !ruby/object:Gem::Specification
+name: asg-rebooter
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Tom de Vries
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2020-02-13 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-autoscaling
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-ec2
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-elasticloadbalancingv2
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.17'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.17'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+description: See summary.
+email:
+- tom@hackerone.com
+executables:
+- asg-rebooter
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- CODE_OF_CONDUCT.md
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- asg-rebooter.gemspec
+- bin/console
+- bin/setup
+- exe/asg-rebooter
+- lib/asg/rebooter.rb
+- lib/asg/rebooter/configuration/auto_scaling_group.rb
+- lib/asg/rebooter/configuration/target_group.rb
+- lib/asg/rebooter/instance_rebooter.rb
+- lib/asg/rebooter/validate_credentials.rb
+- lib/asg/rebooter/version.rb
+- lib/asg/rebooter/wrappers/auto_scaling_group.rb
+- lib/asg/rebooter/wrappers/instance.rb
+- lib/asg/rebooter/wrappers/target_group.rb
+homepage: https://github.com/tomdev/asg-rebooter
+licenses:
+- MIT
+metadata:
+  allowed_push_host: https://rubygems.org
+  homepage_uri: https://github.com/tomdev/asg-rebooter
+  source_code_uri: https://github.com/tomdev/asg-rebooter
+  changelog_uri: https://github.com/tomdev/asg-rebooter
+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: []
+rubygems_version: 3.0.3
+signing_key: 
+specification_version: 4
+summary: Gracefully restart EC2 instances within an Auto Scaling Group.
+test_files: []