whenever-elasticbeanstalk 1.0.0

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in whenever-elasticbeanstalk.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Chad McGimpsey
+
+MIT License
+
+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,106 @@
+# Whenever::Elasticbeanstalk
+
+Whenever-elasticbeanstalk is an extension gem to [Whenever](https://github.com/javan/whenever) that automatically ensures that one instance in an AWS Elastic Beanstalk environment is set as leader. This allows you to run cron jobs on all instances, or just on the leader. This is required since Elastic Beanstalk may start or stop any instance as it scales up or down.
+
+## Installation
+
+**Whenever-elasticbeanstalk is still under development and not packaged as a gem yet**
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'whenever-elasticbeanstalk'
+```
+
+And then execute:
+
+```bash
+$ bundle
+```
+
+Or install it yourself as:
+
+```bash
+$ gem install whenever-elasticbeanstalk
+```
+
+## Getting started
+```bash
+$ cd /apps/my-great-project
+$ wheneverize-eb .
+```
+
+This will create an initial `config/schedule.rb` file for you with the `ensure_leader` job set to run every minute. It will also create a `.ebextensions/cron.config` file that will automatically choose a leader on environment initialization, and start up Whenever with the correct `leader` role.
+
+### Manually updating schedule
+
+If you are already using Whenever, running `wheneverize-eb .` won't overwrite your `config/schedule.rb` file. You'll need to add the following lines in order for your environment to always have one leader.
+
+```ruby
+every 1.minute do
+  command "cd /var/app/current && bundle exec ensure_one_cron_leader"
+end
+```
+
+### Create AWS IAM user
+
+In order for the scripts to work, you need to supply AWS credentials for a user with access to EC2 instances and tags. It is recommended to create a new user with limited access.
+
+Example policy:
+```json
+{
+  "Version": "2012-10-17",
+  "Statement": [
+    {
+      "Action": [
+        "ec2:DescribeInstanceAttribute",
+        "ec2:DescribeInstanceStatus",
+        "ec2:DescribeInstances",
+        "ec2:DescribeTags",
+        "ec2:CreateTags"
+      ],
+      "Resource": [
+        "*"
+      ],
+      "Effect": "Allow"
+    }
+  ]
+}
+```
+
+Then add the credentials to your `config/whenever-elasticbeanstalk.yml` file.
+```yaml
+staging:
+	access_key_id: 'your access key'
+	secret_access_key: 'your secret access key'
+```
+
+## Usage
+
+For `config/schedule.rb` usage, please see the documentation for the [Whenever gem](https://github.com/javan/whenever).
+
+### Run a task on only one instance
+
+To run a task only on one instance, assign the task to the `leader` role.
+```ruby
+every :day, :at => "12:30am", :roles => [:leader] do
+	runner "MyModel.task_to_run_nightly_only_on_one_instance"
+end
+```
+
+### Run a task on all instances
+
+To run a task on all instance, omit the `roles` option.
+```ruby
+every 1.minute do
+	command "touch /opt/elasticbeanstalk/support/.cron_check"
+end
+```
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/bin/create_cron_leader

@@ -0,0 +1,51 @@
+#!/usr/bin/ruby
+
+require 'optparse'
+require 'rubygems'
+gem 'aws-sdk'
+require 'aws-sdk'
+
+options = {}
+
+optparse = OptionParser.new do |opts|
+  opts.banner = "Usage: #{File.basename($0)} [options]"
+ 
+	# Define the options, and what they do
+	options[:no_update] = false
+	opts.on( '--no-update', 'Do not update crontab after making leader' ) do
+	 options[:no_update] = true
+	end
+
+	# This displays the help screen, all programs are
+	# assumed to have this option.
+	opts.on( '-h', '--help', 'Display this screen' ) do
+	 puts opts
+	 exit
+	end
+end
+
+optparse.parse!
+
+this_instance_id = `/opt/aws/bin/ec2-metadata -i | awk '{print $2}'`.strip
+environment = ENV["RAILS_ENV"]
+
+AWS_CREDENTIALS = YAML.load_file("config/whenever-elasticbeanstalk.yml")[environment]
+AWS.config(AWS_CREDENTIALS)
+ec2 = AWS::EC2.new
+
+environment_name = ec2.instances[this_instance_id].tags["elasticbeanstalk:environment-name"]
+
+leader_instances = ec2.instances.inject([]) do |m, i|
+	m << i.id if i.tags["elasticbeanstalk:environment-name"] == environment_name &&
+		i.status == :running &&
+		i.tags["leader"] == "true"
+	m
+end
+
+if leader_instances.count < 1
+	ec2.instances[this_instance_id].tags["leader"] = "true"
+end
+
+unless options[:no_update]
+	`bundle exec setup_cron`
+end

data/bin/ensure_one_cron_leader

@@ -0,0 +1,27 @@
+#!/usr/bin/ruby
+require 'rubygems'
+gem 'aws-sdk'
+
+require 'aws-sdk'
+
+this_instance_id = `/opt/aws/bin/ec2-metadata -i | awk '{print $2}'`.strip
+environment = ENV["RAILS_ENV"]
+
+AWS_CREDENTIALS = YAML.load_file("config/whenever-elasticbeanstalk.yml")[environment]
+AWS.config(AWS_CREDENTIALS)
+ec2 = AWS::EC2.new
+
+environment_name = ec2.instances[this_instance_id].tags["elasticbeanstalk:environment-name"]
+
+leader_instances = ec2.instances.inject([]) do |m, i|
+	m << i.id if i.tags["elasticbeanstalk:environment-name"] == environment_name &&
+		i.status == :running &&
+		i.tags["leader"] == "true"
+	m
+end
+
+if leader_instances.count < 1
+	`bundle exec create_cron_leader`
+elsif leader_instances.count > 1 && leader_instances.includes? this_instance_id
+	`bundle exec remove_cron_leader`
+end

data/bin/remove_cron_leader

@@ -0,0 +1,27 @@
+#!/usr/bin/ruby
+require 'rubygems'
+gem 'aws-sdk'
+
+require 'aws-sdk'
+
+this_instance_id = `/opt/aws/bin/ec2-metadata -i | awk '{print $2}'`.strip
+environment = ENV["RAILS_ENV"]
+
+AWS_CREDENTIALS = YAML.load_file("config/whenever-elasticbeanstalk.yml")[environment]
+AWS.config(AWS_CREDENTIALS)
+ec2 = AWS::EC2.new
+
+environment_name = ec2.instances[this_instance_id].tags["elasticbeanstalk:environment-name"]
+
+leader_instances = ec2.instances.inject([]) do |m, i|
+	m << i.id if i.tags["elasticbeanstalk:environment-name"] == environment_name &&
+		i.status == :running &&
+		i.tags["leader"] == "true"
+	m
+end
+
+if leader_instances.count > 1 && leader_instances.includes? this_instance_id
+	ec2.instances[this_instance_id].tags["leader"] = "false"
+end
+
+`bundle exec setup_cron`

data/bin/setup_cron

@@ -0,0 +1,18 @@
+#!/usr/bin/ruby
+require 'rubygems'
+gem 'aws-sdk'
+
+require 'aws-sdk'
+
+this_instance_id = `/opt/aws/bin/ec2-metadata -i | awk '{print $2}'`.strip
+environment = ENV["RAILS_ENV"]
+
+AWS_CREDENTIALS = YAML.load_file("config/whenever-elasticbeanstalk.yml")[environment]
+AWS.config(AWS_CREDENTIALS)
+ec2 = AWS::EC2.new
+
+if ec2.instances[this_instance_id].tags["leader"] == "true"
+	`bundle exec whenever --roles leader --set 'environment=#{environment}&path=/var/app/current' --update-crontab`
+else
+	`bundle exec whenever --set 'environment=#{environment}&path=/var/app/current' --update-crontab`
+end

data/bin/wheneverize-eb

@@ -0,0 +1,132 @@
+#!/usr/bin/env ruby
+
+# This file is based heavily on Whenever's 'wheneverize' command
+
+require 'optparse'
+require 'fileutils'
+
+OptionParser.new do |opts|
+  opts.banner = "Usage: #{File.basename($0)} [path]"
+
+  begin
+    opts.parse!(ARGV)
+  rescue OptionParser::ParseError => e
+    warn e.message
+    puts opts
+    exit 1
+  end
+end
+
+unless ARGV.empty?
+  if !File.exists?(ARGV.first)
+    abort "`#{ARGV.first}' does not exist."
+  elsif !File.directory?(ARGV.first)
+    abort "`#{ARGV.first}' is not a directory."
+  elsif ARGV.length > 1
+    abort "Too many arguments; please specify only the directory to wheneverize."
+  end
+end
+
+base = ARGV.empty? ? '.' : ARGV.shift
+
+schedule_content = <<-FILE
+# Use this file to easily define all of your cron jobs.
+#
+# It's helpful, but not entirely necessary to understand cron before proceeding.
+# http://en.wikipedia.org/wiki/Cron
+
+# Example:
+#
+# set :output, "/path/to/my/cron_log.log"
+#
+# every 2.hours do
+#   command "/usr/bin/some_great_command"
+#   runner "MyModel.some_method"
+#   rake "some:great:rake:task"
+# end
+#
+# every 4.days do
+#   runner "AnotherModel.prune_old_records"
+# end
+
+# Learn more: http://github.com/javan/whenever
+
+every 1.minute do
+  command "cd /var/app/current && bundle exec ensure_one_cron_leader"
+end
+FILE
+
+file = 'config/schedule.rb'
+file = File.join(base, file)
+if File.exists?(file)
+  warn "[skip] `#{file}' already exists"
+elsif File.exists?(file.downcase)
+  warn "[skip] `#{file.downcase}' exists, which could conflict with `#{file}'"
+elsif !File.exists?(File.dirname(file))
+  warn "[skip] directory `#{File.dirname(file)}' does not exist"
+else
+  puts "[add] writing `#{file}'"
+  File.open(file, "w") { |f| f.write(schedule_content) }
+end
+
+
+eb_config_content = <<-FILE
+files:
+  /opt/elasticbeanstalk/hooks/appdeploy/post/10_reload_cron.sh:
+    mode: "00700"
+    owner: root
+    group: root
+    content: |
+      #!/usr/bin/env bash
+      . /opt/elasticbeanstalk/support/envvars
+      cd $EB_CONFIG_APP_CURRENT
+      bundle exec setup_cron
+
+container_commands:
+  cron_01_set_leader:
+    test: test ! -f /opt/elasticbeanstalk/support/.cron-setup-complete
+    leader_only: true
+    cwd: /var/app/ondeck
+    command: bundle exec create_cron_leader --no-update
+  cron_02_write_cron_setup_complete_file:
+    cwd: /opt/elasticbeanstalk/support
+    command: touch .cron-setup-complete
+FILE
+
+file = '.ebextensions/cron.config'
+file = File.join(base, file)
+if File.exists?(file)
+  warn "[skip] `#{file}' already exists"
+elsif File.exists?(file.downcase)
+  warn "[skip] `#{file.downcase}' exists, which could conflict with `#{file}'"
+elsif !File.exists?(File.dirname(file))
+  warn "[skip] directory `#{File.dirname(file)}' does not exist"
+else
+  puts "[add] writing `#{file}'"
+  File.open(file, "w") { |f| f.write(eb_config_content) }
+end
+
+
+aws_credentials_content = <<-FILE
+staging:
+  access_key_id: ''
+  secret_access_key: ''
+production:
+  access_key_id: ''
+  secret_access_key: ''
+FILE
+
+file = 'config/whenever-elasticbeanstalk.yml'
+file = File.join(base, file)
+if File.exists?(file)
+  warn "[skip] `#{file}' already exists"
+elsif File.exists?(file.downcase)
+  warn "[skip] `#{file.downcase}' exists, which could conflict with `#{file}'"
+elsif !File.exists?(File.dirname(file))
+  warn "[skip] directory `#{File.dirname(file)}' does not exist"
+else
+  puts "[add] writing `#{file}'"
+  File.open(file, "w") { |f| f.write(aws_credentials_content) }
+end
+
+puts "[done] wheneverized for Elastic Beanstalk!"

data/lib/whenever-elasticbeanstalk.rb

@@ -0,0 +1,7 @@
+require "whenever-elasticbeanstalk/version"
+
+module Whenever
+  module Elasticbeanstalk
+    # Your code goes here...
+  end
+end

data/lib/whenever-elasticbeanstalk/version.rb

@@ -0,0 +1,5 @@
+module Whenever
+  module Elasticbeanstalk
+    VERSION = "1.0.0"
+  end
+end

data/whenever-elasticbeanstalk.gemspec

@@ -0,0 +1,23 @@
+# -*- encoding: utf-8 -*-
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'whenever-elasticbeanstalk/version'
+
+Gem::Specification.new do |gem|
+  gem.name          = "whenever-elasticbeanstalk"
+  gem.version       = Whenever::Elasticbeanstalk::VERSION
+  gem.platform      = Gem::Platform::RUBY
+  gem.authors       = ["Chad McGimpsey"]
+  gem.email         = ["chad.mcgimpsey@gmail.com"]
+  gem.description   = %q{Use Whenever on AWS Elastic Beanstalk}
+  gem.summary       = %q{Allows you to run cron jobs easily on one or all AWS Elastic Beanstalk instances.}
+  gem.homepage      = "https://github.com/dignoe/whenever-elasticbeanstalk"
+
+  gem.add_dependency('whenever')
+  gem.add_dependency('aws-sdk')
+
+  gem.files         = `git ls-files`.split($/)
+  gem.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.require_paths = ["lib"]
+end

metadata

@@ -0,0 +1,95 @@
+--- !ruby/object:Gem::Specification
+name: whenever-elasticbeanstalk
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+  prerelease: 
+platform: ruby
+authors:
+- Chad McGimpsey
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-06-22 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: whenever
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Use Whenever on AWS Elastic Beanstalk
+email:
+- chad.mcgimpsey@gmail.com
+executables:
+- create_cron_leader
+- ensure_one_cron_leader
+- remove_cron_leader
+- setup_cron
+- wheneverize-eb
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/create_cron_leader
+- bin/ensure_one_cron_leader
+- bin/remove_cron_leader
+- bin/setup_cron
+- bin/wheneverize-eb
+- lib/whenever-elasticbeanstalk.rb
+- lib/whenever-elasticbeanstalk/version.rb
+- whenever-elasticbeanstalk.gemspec
+homepage: https://github.com/dignoe/whenever-elasticbeanstalk
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.25
+signing_key: 
+specification_version: 3
+summary: Allows you to run cron jobs easily on one or all AWS Elastic Beanstalk instances.
+test_files: []