deadpool 0.1.1

data/bin/deadpool_admin

@@ -0,0 +1,8 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+
+require File.join(File.dirname(File.dirname(__FILE__)), 'lib', 'deadpool')
+
+Deadpool::Admin.new(ARGV).run
+

data/bin/deadpool_generator

@@ -0,0 +1,8 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+
+require File.join(File.dirname(File.dirname(__FILE__)), 'lib', 'deadpool')
+
+Deadpool::Generator.new(ARGV).run
+

data/bin/deadpool_hosts

@@ -0,0 +1,222 @@
+#!/usr/bin/env ruby
+
+require 'optparse'
+require 'ostruct'
+require 'strscan'
+
+class DeadpoolHosts
+
+  def run
+    @options = self.parse_command_line
+    self.execute_command(@options)
+  end
+
+  def parse_command_line
+    options          = OpenStruct.new
+    options.command_count = 0
+
+    @option_parser = OptionParser.new do |opts|
+      opts.banner = "Usage: deadpool_hosts {help|test|verify|switch} [options]"
+
+      opts.separator "Commands:"
+      opts.on("-h", "--help", "Print this help message.") do |h|
+        options.help         = h
+        options.command_count += 1
+        options.command      = :help
+      end
+      opts.on("--test", "Test that /etc/hosts can be touched.") do |t|
+        options.test         = t
+        options.command_count += 1
+        options.command      = :test
+      end
+      opts.on("--verify", "Verify that /etc/hosts hast the IP address that you think it should.") do |v|
+        options.verify       = v
+        options.command_count += 1
+        options.command      = :verify
+      end
+      opts.on("--switch", "Change the IP address in /etc/hosts") do |s|
+        options.switch       = s
+        options.command_count += 1
+        options.command      = :switch
+      end
+      
+      opts.separator "Options:"
+      opts.on("--host_name=HOSTNAME", String, "Hostname to regex on in /etc/hosts") do |h|
+        options.host_name = h
+      end
+      opts.on("--ip_address=IP_ADDRESS", String, "IP Address") do |ip|
+        options.ip_address = ip
+      end
+      opts.on("--hosts_file=HOST_FILE_PATH", String, "Alternate Path for the Hosts file") do |hosts|
+        options.hosts_file = hosts
+      end
+      opts.on("--verbose", "Verbose Messages.") do |v|
+        options.verbose = v
+      end
+      opts.on("--debug", "Creates a dummy hosts file at /tmp/hosts and operates on that.  Overrides the alternate hosts file option.") do |d|
+        options.debug = d
+      end
+    end
+
+    # Parse and verify
+    remaining_arguments = @option_parser.parse!
+
+    unless remaining_arguments.empty?
+      help "[#{remaining_arguments.join(' ')}] is not understood."
+    end
+    
+    if options.command_count == 0
+      help "You must specify a command."
+    end
+
+    option_sum = (options.help ? 1 : 0) + (options.test ? 1 : 0) + (options.verify ? 1 : 0) + (options.switch ? 1 : 0)
+    if option_sum > 1
+      help "Only one command can be specified."
+    end
+
+    if options.verify
+      options_valid = true
+
+      if options.ip_address.nil?
+        puts "The verify command requires an ip_address argument."
+        options_valid = false
+      elsif ! options.ip_address.match(/\d+\.\d+\.\d+\.\d+/)
+        puts "ip_address doesn't look valid."
+        options_valid = false
+      end
+
+      if options.host_name.nil?
+        puts "The verify command requires a host_name argument."
+        options_valid = false
+      elsif options.host_name.size < 4
+        puts "The host_name parameter must be longer than 4 characters."
+        options_valid = false
+      end
+
+      unless options_valid
+        help
+      end
+    end
+
+
+    return options
+  end
+
+  def execute_command(options)
+    case options.command
+    when :help
+      help
+    when :test
+      test options
+    when :verify
+      verify options
+    when :switch
+      switch options
+    else
+      help
+    end
+  end
+
+  def help(message=nil)
+    unless message.nil?
+      puts message
+    end
+    puts @option_parser.help
+    exit 4
+  end
+
+  def test(options)
+    path = host_file_path(options)
+    out = `touch #{path} 2> /dev/null`
+
+    if $? == 0
+      puts "OK - #{path} is writable"
+      exit 0
+    else
+      puts "ERROR - #{path} is NOT writable"
+      exit 1
+    end
+  end
+
+  def verify(options)
+    scanner  = StringScanner.new File.read(host_file_path(options))
+    verified = scanner.exist?(/#{options.ip_address}\s+#{options.host_name}\s+/)
+
+    if verified
+      puts "OK - #{options.host_name} was verified to point at #{options.ip_address}"
+      exit 0
+    else
+      puts "ERROR - #{options.host_name} does not point at #{options.ip_address}"
+      exit 1
+    end
+  end
+
+  def switch(options)
+    path = host_file_path(options)
+    scanner  = StringScanner.new File.read(path)
+    host_exists = scanner.exist?(/\s+#{options.host_name}\s+/)
+
+    if host_exists
+      scanner.reset
+      scanner.scan_until(/\n.+#{options.host_name}\s*\n/)
+      pre_match  = scanner.pre_match
+      matched    = scanner.matched
+      post_match = scanner.post_match
+      verbose_message "\nPreceding lines: \n#{pre_match}"
+      verbose_message "\nMatched host definition: \n#{matched}"
+      verbose_message "\nFollowing lines: \n#{post_match}"
+      replacement_file = pre_match
+      replacement_file += "\n#{options.ip_address}  #{options.host_name}\n"
+      replacement_file += post_match
+      verbose_message "Replacement hosts file:\n#{replacement_file}"
+      verbose_message "Writing hosts file..."
+      hosts_file = File.open path, 'w'
+      hosts_file.write replacement_file
+      hosts_file.close
+      puts "OK - Host definition replaced successfully."
+      exit 0
+    else
+      replacement_file = File.read(path)
+      replacement_file += "\n#{options.ip_address}  #{options.host_name}\n"
+      verbose_message "Replacement hosts file:\n#{replacement_file}"
+      verbose_message "Writing hosts file..."
+      hosts_file = File.open path, 'w'
+      hosts_file.write replacement_file
+      hosts_file.close
+      puts "OK - Host definition added successfully."
+      exit 0
+    end
+  end
+
+  def host_file_path(options)
+    if options.debug
+      file = File.open '/tmp/hosts', 'w'
+      file.write DATA.read
+      file.close
+      return '/tmp/hosts'
+    elsif options.hosts_file
+      return options.hosts_file
+    else
+      return '/etc/hosts'
+    end
+  end
+
+  def verbose_message(message)
+    if @options.verbose
+      puts message
+    end
+  end
+
+end
+
+EtcHostsSwitch.new.run
+
+__END__
+127.0.0.1           localhost localhost.localdomain
+184.106.205.117     web-01
+
+123.456.78.9  test_host_name
+123.456.78.9  other_service_that_shouldnt_be.touched
+123.456.78.9  test_host_name.similar
+
+

data/config/default_environment.yml

@@ -0,0 +1,7 @@
+
+pid_file: '/var/run/deadpool.pid'
+log_file: '/var/log/deadpool.log'
+log_level: INFO
+admin_hostname: 'localhost'
+admin_port: 5507
+system_check_interval: 30

data/doc/init/deadpool.conf

@@ -0,0 +1,17 @@
+# Deadpool Service
+
+description     "Deadpool Server"
+author          "Kirt Fitzpatrick <kirt.fitzpatrick@akqa.com>"
+
+start on (net-device-up
+          and local-filesystems)
+
+stop on runlevel [016]
+
+respawn
+
+# env HOME=/usr/local/bin/deadpool
+env RUBY=/usr/local/bin/ruby
+umask 007
+
+exec $RUBY $HOME/bin/deadpool_admin --start

data/lib/deadpool.rb

@@ -0,0 +1,45 @@
+
+require 'rubygems'
+require 'eventmachine'
+require 'logger'
+require 'yaml'
+
+$:.unshift File.dirname(__FILE__)
+
+
+module Deadpool
+  OK       = 0
+  WARNING  = 1
+  CRITICAL = 2
+  UNKNOWN  = 3
+
+  autoload :Admin,                'deadpool/admin'
+  autoload :AdminServer,          'deadpool/admin_server'
+  autoload :CommandLineServer,    'deadpool/command_line_server'
+  autoload :Daemonizer,           'deadpool/daemonizer'
+  autoload :Generator,            'deadpool/generator'
+  autoload :Handler,              'deadpool/handler'
+  autoload :Helper,               'deadpool/helper'
+  autoload :Options,              'deadpool/options'
+  autoload :State,                'deadpool/state'
+  autoload :StateSnapshot,        'deadpool/state_snapshot'
+  autoload :Server,               'deadpool/server'
+
+
+  module FailoverProtocol
+    autoload :Base,               'deadpool/failover_protocol'
+  end
+
+  module Monitor
+    autoload :Base,               'deadpool/monitor/base'
+    autoload :Mysql,              'deadpool/monitor/mysql'
+    autoload :GenericNagios,      'deadpool/monitor/generic_nagios'
+  end
+
+  module FailoverProtocol
+    autoload :EtcHosts,           'deadpool/failover_protocol/etc_hosts'
+    autoload :ExecRemoteCommand,  'deadpool/failover_protocol/exec_remote_command'
+  end
+
+  class DeadpoolError < StandardError; end
+end

data/lib/deadpool/admin.rb

@@ -0,0 +1,197 @@
+
+require 'optparse'
+require 'ostruct'
+require 'strscan'
+require 'socket'
+require 'json'
+
+
+module Deadpool
+
+  class Admin
+
+    def initialize(argv)
+      @argv     = argv
+    end
+
+    def run
+      @options = self.parse_command_line
+      @config  = Deadpool::Helper.configure @options
+
+      self.execute_command(@options)
+    end
+
+    def parse_command_line
+      options               = Hash.new
+      options[:command_count] = 0
+      options[:config_path]     = '/etc/deadpool'
+
+      @option_parser = OptionParser.new do |opts|
+        opts.banner = "Usage: deadpool_hosts --command [options]"
+
+        opts.separator "Commands:"
+        opts.on("-h", "--help", "Print this help message.") do |help|
+          options[:command_count] += 1
+          options[:command]        = :help
+        end
+        opts.on("--full_report", "Give the full system report.") do |full_report|
+          options[:command_count] += 1
+          options[:command]        = :full_report
+        end
+        opts.on("--nagios_report", "Report system state in Nagios plugin format.") do |nagios_report|
+          options[:command_count] += 1
+          options[:command]        = :nagios_report
+        end
+        opts.on("--promote_server", "Promote specified server to the master.") do |nagios_report|
+          options[:command_count] += 1
+          options[:command]        = :promote_server
+        end
+        opts.on("--stop", "Stop the server.") do |stop|
+          options[:command_count] += 1
+          options[:command]        = :stop
+        end
+        opts.on("--start", "Start the server in the background.") do |stop|
+          options[:command_count] += 1
+          options[:command]        = :start
+        end
+        opts.on("--foreground", "Start the server in the foreground.") do |stop|
+          options[:command_count] += 1
+          options[:command]        = :foreground
+        end
+
+        opts.separator "Options:"
+        opts.on("--server=SERVER_LABEL", String, "primary_host or secondary_host.") do |server|
+          options[:server] = server
+        end
+        opts.on("--pool=POOL_NAME", String, "Deadpool name to operate on.") do |pool|
+          options[:pool] = pool
+        end
+        opts.on("--config_path=PATH", String, 
+            "Path to configs and custom plugins. #{options[:config_path]} by default.") do |config_path|
+          options[:config_path] = config_path
+        end
+      end
+
+      remaining_arguments = @option_parser.parse! @argv
+
+      unless remaining_arguments.empty?
+        help "[#{remaining_arguments.join(' ')}] is not understood."
+      end
+
+      if options[:command_count] == 0
+        help "You must specify a command."
+      end
+
+      return options
+    end
+
+    def execute_command(options)
+      case options[:command]
+      when :help
+        help
+      when :full_report
+        full_report options
+      when :nagios_report
+        nagios_report options
+      when :promote_server
+        promote_server options
+      when :stop
+        stop options
+      when :start
+        start options
+      when :foreground
+        foreground options
+      else
+        help
+      end
+    end
+
+    def help(message=nil)
+      unless message.nil?
+        puts message
+      end
+      puts @option_parser.help
+      exit 4
+    end
+
+    def full_report(options)
+      puts send_command_to_deadpool_server :command => 'full_report'
+    end
+
+    def nagios_report(options)
+      response = send_command_to_deadpool_server :command => 'nagios_report'
+
+      if (response.to_s =~ /^OK/) != nil
+        puts response.to_s
+        exit OK
+      elsif (response.to_s =~ /^WARNING/) != nil
+        puts response.to_s
+        exit WARNING
+      elsif (response.to_s =~ /^CRITICAL/) != nil
+        puts response.to_s
+        exit CRITICAL
+      elsif (response.to_s =~ /^UNKNOWN/) != nil
+        puts response.to_s
+        exit UNKNOWN
+      else
+        puts "UNKNOWN - #{response.to_s}"
+        exit UNKNOWN
+      end
+    end
+
+    def promote_server(options)
+      error_messages = []
+
+      if options[:pool].nil?
+        error_messages << "Promoting server requires --pool argument."
+      end
+
+      if options[:server].nil?
+        error_messages << "Promoting server requires --server argument."
+      end
+      
+      unless error_messages.empty?
+        help error_messages.join "\n"
+      end
+
+      puts send_command_to_deadpool_server :command => 'promote_server', :pool => options[:pool], :server => options[:server]
+    end
+
+    def stop(options)
+      puts send_command_to_deadpool_server :command => 'stop'
+    end
+
+    def start(options)
+      puts Deadpool::Server.new(options).run(true)
+    end
+
+    def foreground(options)
+      puts Deadpool::Server.new(options).run(false)
+    end
+
+    def send_command_to_deadpool_server(options)
+      output = ''
+
+      begin
+        socket = TCPSocket.open(@config[:admin_hostname], @config[:admin_port])
+      rescue
+        return "Couldn't connect to deadpool server.  Is it running?"
+      end
+
+      if socket
+        socket.puts JSON.dump(options)
+        while line = socket.gets
+          output += line
+        end
+        socket.close
+      else
+        return "Couldn't connect to deadpool server."
+      end
+
+      return output
+    end
+
+  end
+
+end
+