capistrano-monit 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 capistrano-monit.gemspec
+gemspec

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Andrew Eberbach
+
+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,29 @@
+# Capistrano::Monit
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'capistrano-monit'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install capistrano-monit
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,2 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"

data/capistrano-monit.gemspec

@@ -0,0 +1,17 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/capistrano-monit/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Andrew Eberbach"]
+  gem.email         = ["andrew@ebertech.ca"]
+  gem.description   = %q{Monit deployment helper}
+  gem.summary       = %q{Monit deployment helper}
+  gem.homepage      = ""
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "capistrano-monit"
+  gem.require_paths = ["lib"]
+  gem.version       = Capistrano::Monit::VERSION
+end

data/lib/capistrano-monit.rb

@@ -0,0 +1,5 @@
+module Capistrano
+  module Monit
+    autoload :VERSION, "capistrano-monit/version"
+  end
+end

data/lib/capistrano-monit/version.rb

@@ -0,0 +1,5 @@
+module Capistrano
+  module Monit
+    VERSION = "1.0.0"
+  end
+end

data/lib/capistrano/monit.rb

@@ -0,0 +1,5 @@
+Capistrano::Configuration.instance(:must_exist).load do
+  Dir.glob(File.join(File.dirname(__FILE__), "..", "..",  "recipes", "*.rb")).each do |file|
+    load file
+  end
+end

data/lib/monit.rb

@@ -0,0 +1,179 @@
+require 'erb'
+class Monit
+  class MonitoredItem
+    attr_accessor :depends_on
+    attr_accessor :base_path
+    attr_accessor :monit
+    attr_accessor :name
+
+    def initialize(name, monit, &block)
+      self.name = name
+      self.depends_on = []
+      self.monit = monit
+      instance_eval(&block)
+    end  
+    
+    def each_dependency(&block)
+      depends_on.each do |dependency_name|
+        yield monitored.detect{|m| m.name == dependency_name}.unique_name
+      end
+    end
+    
+    def conf_name
+      "#{unique_name}.conf"
+    end
+    
+    def conf_path
+      "#{home_dir}/etc/monit.d/#{conf_name}"
+    end
+    
+    def env_path
+      "#{home_dir}/etc/environments/#{env_name}"
+    end
+    
+    def env_name
+      "#{unique_name}.env"      
+    end
+    
+    def to_monit   
+      ERB.new(File.read(template_path)).result(binding)
+    end
+    
+    def unique_name
+      "#{application}.#{stage}.#{name}"
+    end
+    
+    def to_environment
+      ERB.new(File.read(environment_template_path)).result(binding)
+    end  
+
+    def method_missing(method, *args)
+      monit.send(method, *args)
+    end
+    
+    def needs_environment?
+      false
+    end
+  end
+
+  class MonitoredFile < MonitoredItem
+    attr_accessor :path
+    
+    def template_path
+      File.expand_path("../../templates/file.conf.erb", __FILE__)      
+    end
+    
+    def file_path
+      File.join(current_path, path)
+    end
+  end
+
+  class MonitoredProgram < MonitoredItem
+    attr_accessor :command
+    attr_accessor :log_file
+    attr_accessor :pid_file
+    attr_accessor :bundler
+    
+    def initialize(*args)
+      self.bundler = true
+      super
+    end
+    
+    def start_command
+      %Q{#{monit_rvm_shell_path} #{env_path} start 'cd #{current_path} && #{bundle} #{command}'}
+    end
+    
+    def stop_command
+      %Q{#{monit_rvm_shell_path} #{env_path} stop 'cd #{current_path} && #{bundle}'}      
+    end
+    
+    def bundle
+      bundler ? "bundle exec" : ""
+    end
+    
+    def pidfile_path
+      File.join(current_path, pid_file)
+    end
+    
+    def logfile_path
+      File.join(current_path, log_file)      
+    end
+    
+    def gem_base_dir
+      capture(%Q{cd #{current_path} && #{bundle} ruby -e 'puts Gem.loaded_specs["capistrano-monit"].gem_dir'}).strip
+    end
+    
+    def monit_rvm_shell_path
+      File.join(gem_base_dir, "scripts/rvm-monit-shell")
+    end
+    
+    def program_search_path
+      "/usr/local/bin:/bin:/usr/bin:/sbin:/usr/sbin"
+    end
+    
+    def template_path
+      File.expand_path("../../templates/program.conf.erb", __FILE__)
+    end
+    
+    def environment_template_path
+      File.expand_path("../../templates/environment.erb", __FILE__)      
+    end    
+    
+    def needs_environment?
+      true
+    end
+    
+  end
+
+  TYPES = {
+    :file => MonitoredFile,
+    :program => MonitoredProgram
+  }
+
+  class << self
+    def configure(&block)
+      new(&block)
+    end  
+    
+    def load_from(file, context)
+      self.instance_eval(File.read(file), file).tap do |monit|
+        monit.context = context
+      end
+    end
+  end
+  
+  def template_path
+    File.expand_path("../../templates/monit.conf.erb", __FILE__)
+  end
+  
+  def to_monit   
+    ERB.new(File.read(template_path)).result(binding)
+  end
+  
+  def monitd_path
+    File.join(home_dir, "etc/monit.d")
+  end
+  
+  attr_accessor :port
+  attr_accessor :monitored
+  attr_accessor :context
+  
+  def method_missing(method, *args)
+    if context && context.respond_to?(method)
+      context.send(method, *args)
+    else
+      super
+    end
+  end
+
+  def initialize(&block)
+    self.monitored = []
+    self.port = 1192
+    instance_eval(&block)
+  end
+
+  def monitors(options, &block)
+    type, name = options.first
+    self.monitored << TYPES[type].new(name, self, &block)
+  end
+end

data/recipes/monit.rb

@@ -0,0 +1,64 @@
+require 'monit'
+
+_cset(:home_dir){ capture("echo $HOME").strip }
+_cset(:monit_config_path) { File.join(Dir.pwd, "config", "monit", "#{stage}.rb") }
+namespace :monit do
+
+  task :make_directories do
+    home_dir = 
+    %W{etc/
+      etc/monit.d
+      etc/environments
+      var/
+      var/run
+      var/log
+      var/monit
+      /etc/cron.d/reboot/
+    }.map do |dir|
+      File.join(fetch(:home_dir), dir)
+    end.tap{|dirs| run %Q{mkdir -p #{dirs.join(" ")}}}    
+  end
+
+  task :upload_config do      
+    if File.exists?(monit_config_path)
+      ::Monit.load_from(monit_config_path, self).tap do |monit|
+        put monit.to_monit, "#{home_dir}/etc/monit.conf"
+        run "chmod 600 #{"#{home_dir}/etc/monit.conf"}"        
+      end
+    end
+  end
+
+  task :setup do 
+    make_directories
+    upload_config  
+  end
+  
+  desc "Tell monit to reread its configuration"
+  task :reload do
+    run "monit -c #{home_dir}/etc/monit.conf"            
+    run "monit -c #{home_dir}/etc/monit.conf reload"    
+  end
+  
+  task :restart do
+    run "monit -c #{home_dir}/etc/monit.conf"            
+    run "monit -c #{home_dir}/etc/monit.conf restart all"      
+  end
+
+  task :deploy do
+    if File.exists?(monit_config_path)
+      ::Monit.load_from(monit_config_path, self).tap do |monit|
+        monit.monitored.each do |monitored_item|
+          put monitored_item.to_monit, monitored_item.conf_path
+          if monitored_item.needs_environment? 
+            put monitored_item.to_environment, monitored_item.env_path
+          end
+        end
+      end
+    end
+  end
+end
+
+after "deploy:setup", "monit:setup"
+after "deploy:update_code", "monit:deploy"
+after "monit:deploy", "monit:restart"
+after "monit:setup", "monit:reload"

data/scripts/rvm-monit-shell

@@ -0,0 +1,114 @@
+#!/bin/bash
+eval HOME=~`whoami`
+
+RC_FILE=$1
+shift
+
+start() {
+    shift
+    if ! mkdir -p "$(dirname "$PIDFILE")"
+            then
+            echo "Failed making $(dirname "$PIDFILE")"
+            exit 3
+    fi
+
+    if ! mkdir -p "$(dirname "$LOGFILE")"
+            then
+            echo "Failed making $(dirname "$LOGFILE")"
+            exit 4
+    fi
+
+    echo $$ > "$PIDFILE";
+
+    if [ -z "$1" ]
+            then
+            echo "You must pass in a command to execute"
+            usage
+    fi
+    echo "Writing to $LOGFILE"
+
+    exec bash -c "$@ >> $LOGFILE 2>&1"
+}
+
+stop() {
+	if [ -f "$PIDFILE" ]
+		then
+		echo "killing $PIDFILE"
+		if pkill -TERM -P $(cat "$PIDFILE")
+			then		
+			rm "$PIDFILE"
+		else
+			echo "Failed to kill, leaving pidfile."
+			exit 2
+		fi		
+	else
+		echo "No such pid file $PIDFILE"
+		exit 3
+	fi
+}
+
+usage() {
+	echo "usage: $0 [PATH_TO_RC] {start|stop|restart} COMMAND [ARG1, ARG2...]"
+	exit 1
+}
+
+if [ -z "$RC_FILE" ]
+	then
+	RC_FILE=$HOME/.rvmmonitrc
+fi
+
+if [ -f "$RC_FILE" ]
+	then
+	source "$RC_FILE"
+else
+	echo "The first argument needs to be a path to a file that will be sourced before running since $RC_FILE does not exist"
+	usage
+fi
+
+# Load RVM into a shell session *as a function*
+if [[ -s "$HOME/.rvm/scripts/rvm" ]] ; then
+	# First try to load from a user install
+	source "$HOME/.rvm/scripts/rvm"
+elif [[ -s "/usr/local/rvm/scripts/rvm" ]] ; then
+	# Then try to load from a root install
+	source "/usr/local/rvm/scripts/rvm"
+else
+	printf "ERROR: An RVM installation was not found.\n"
+	exit 2
+fi
+
+if [ -z "$LOGFILE" ] 
+	then
+	echo "You must specify a LOGFILE environment variable. You can put it in $RC_FILE"
+	usage
+fi
+
+if [ -z "$PIDFILE" ] 
+	then
+	echo "You must specify a PIDFILE environment variable. You can put it in $RC_FILE"
+	usage
+fi
+
+if [ -z "$RVM_STRING" ] 
+	then
+	echo "You must specify a RVM_STRING environment variable. You can put it in $RC_FILE"
+	usage
+fi
+
+rvm use "$RVM_STRING" > /dev/null 2>&1
+
+case $1 in
+	start)
+		start $*
+	;;
+	stop)  
+		stop
+	;;
+	restart)  
+		stop
+		start $*
+	;;	
+	*)  
+	usage ;;
+esac
+exit 0

data/templates/environment.erb

@@ -0,0 +1,5 @@
+export PIDFILE=<%= pidfile_path %>
+export LOGFILE=<%= logfile_path %>
+export PATH=<%= program_search_path %>
+export RVM_STRING=<%= rvm_ruby_string %>
+export RAILS_ENV=<%= stage %>

data/templates/file.conf.erb

@@ -0,0 +1,2 @@
+check file <%= unique_name%> path <%= file_path%>
+    if changed timestamp then restart

data/templates/monit.conf.erb

@@ -0,0 +1,25 @@
+set daemon  5
+set logfile <%= home_dir %>/var/log/monit.log
+set idfile <%= home_dir %>/var/monit/id
+set statefile <%= home_dir %>/var/monit/state
+
+set eventqueue
+basedir <%= home_dir %>/var/monit/eventqueue  # set the base directory where events will be stored
+slots 1000           # optionally limit the queue size
+
+set mail-format { 
+from: monit@ebertech.ca 
+subject: [monit] $SERVICE $EVENT	
+message: $EVENT Service $SERVICE
+Host:        $HOST 
+Date:        $DATE
+Action:      $ACTION 
+Description: $DESCRIPTION 
+}
+set mmonit https://monit:monit@mmonit.ebertech.ca/collector
+
+set httpd port <%= port %> and
+allow localhost
+allow mmonit.ebertech.ca
+
+Include <%= monitd_path %>/*.conf

data/templates/program.conf.erb

@@ -0,0 +1,7 @@
+check process <%= unique_name %>
+    with pidfile <%= pidfile_path %>
+        start program = "<%= start_command %>" 
+        stop program = "<%= stop_command %>" 
+		<% each_dependency do |dependency| %>
+			depends on <%= dependency %>
+		<% end %>

metadata

@@ -0,0 +1,61 @@
+--- !ruby/object:Gem::Specification
+name: capistrano-monit
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+  prerelease: 
+platform: ruby
+authors:
+- Andrew Eberbach
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-06-18 00:00:00.000000000 Z
+dependencies: []
+description: Monit deployment helper
+email:
+- andrew@ebertech.ca
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- capistrano-monit.gemspec
+- lib/capistrano-monit.rb
+- lib/capistrano-monit/version.rb
+- lib/capistrano/monit.rb
+- lib/monit.rb
+- recipes/monit.rb
+- scripts/rvm-monit-shell
+- templates/environment.erb
+- templates/file.conf.erb
+- templates/monit.conf.erb
+- templates/program.conf.erb
+homepage: ''
+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.24
+signing_key: 
+specification_version: 3
+summary: Monit deployment helper
+test_files: []