bin/god

   1 #!/usr/bin/env ruby
   2
   3 $:.unshift File.join(File.dirname(__FILE__), *%w[.. lib])
   4
   5 require 'rubygems'
   6 require 'optparse'
   7 require 'drb'
   8
   9 options = {:daemonize => true, :port => 17165}
  10
  11 OptionParser.new do |opts|
  12   opts.banner = <<-EOF
  13 Usage: god [command] [options]
  14
  15   Commands:
  16     start <watch or group name>
  17     stop <watch or group name>
  18     monitor <watch or group name>
  19     unmonitor <watch or group name>
  20     load <file>
  21     log <watch name>
  22     status
  23     terminate
  24
  25   Options:
  26 EOF
  27
  28   opts.on("-cCONFIG", "--config-file CONFIG", "Configuration file") do |x|
  29     options[:config] = x
  30   end
  31
  32   opts.on("-pPORT", "--port PORT", "Communications port") do |x|
  33     options[:port] = x
  34   end
  35
  36   opts.on("-PFILE", "--pid FILE", "Where to write the PID file") do |x|
  37     options[:pid] = x
  38   end
  39
  40   opts.on("-lFILE", "--log FILE", "Where to write the log file") do |x|
  41     options[:log] = x
  42   end
  43
  44   opts.on("-D", "--no-daemonize", "Don't daemonize") do
  45     options[:daemonize] = false
  46   end
  47
  48   opts.on("-v", "--version", "Print the version number and exit") do
  49     options[:version] = true
  50   end
  51
  52   opts.on("-V", "Print extended version and build information") do
  53     options[:info] = true
  54   end
  55 end.parse!
  56
  57 if options[:version]
  58   require 'god'
  59
  60   # print version
  61   puts "Version #{God::VERSION}"
  62   exit!(0)
  63 elsif options[:info]
  64   require 'god'
  65
  66   puts "Version: #{God::VERSION}"
  67   puts "Polls: enabled"
  68   puts "Events: " + God::EventHandler.event_system
  69
  70   exit!(0)
  71 elsif command = ARGV[0]
  72   require 'god'
  73
  74   # a command was specified
  75
  76   # connect to remote drb
  77     DRb.start_service
  78     server = DRbObject.new nil, "druby://localhost:#{options[:port]}"
  79
  80   if command == 'load'
  81     file = ARGV[1]
  82
  83     puts "Sending '#{command}' command"
  84
  85     code = File.read(file)
  86
  87     watches = server.running_load(code)
  88
  89     # output response
  90     puts 'The following watches were affected:'
  91     watches.each do |w|
  92       puts '  ' + w.name
  93     end
  94
  95     puts "Done"
  96   elsif command == 'status'
  97     puts server.status
  98   elsif command == 'log'
  99     begin
 100       Signal.trap('INT') { exit!(0) }
 101       name = ARGV[1]
 102       t = Time.at(0)
 103       loop do
 104         print server.running_log(name, t)
 105         t = Time.now
 106         sleep 1
 107       end
 108     rescue God::NoSuchWatchError
 109       puts "No such watch"
 110     rescue DRb::DRbConnError
 111       puts "The server went away"
 112     rescue => e
 113       puts e.message
 114       puts e.backtrace.join("\n")
 115     ensure
 116       exit!(0)
 117     end
 118   elsif command == 'terminate'
 119     if server.stop_all
 120       puts 'Stopped all watches'
 121     else
 122       puts 'Could not stop all watches within 10 seconds'
 123     end
 124
 125     begin
 126       server.terminate
 127       abort 'Could not stop god'
 128     rescue DRb::DRbConnError
 129       puts 'Stopped god'
 130       exit!(0)
 131     end
 132   else
 133     # get the name of the watch/group
 134     name = ARGV[1]
 135
 136     begin
 137       puts "Sending '#{command}' command"
 138
 139       # send command
 140       watches = server.control(name, command)
 141
 142       # output response
 143       puts 'The following watches were affected:'
 144       watches.each do |w|
 145         puts '  ' + w.name
 146       end
 147     rescue God::InvalidCommandError
 148       abort "Command '#{command}' is not valid. Run 'god --help' for usage"
 149     end
 150   end
 151
 152   exit!(0)
 153 else
 154   # start god
 155   if !options[:daemonize]
 156     require 'god'
 157
 158     if options[:port]
 159       God.port = options[:port]
 160     end
 161
 162     load File.expand_path(options[:config])
 163   else
 164     pid = fork do
 165       begin
 166         require 'god'
 167
 168         log_file = options[:log] || "/dev/null"
 169
 170         STDIN.reopen "/dev/null"
 171         STDOUT.reopen(log_file, "a")
 172         STDERR.reopen STDOUT
 173
 174         puts "Starting god"
 175
 176         unless God::EventHandler.loaded?
 177           puts
 178           puts "***********************************************************************"
 179           puts "*"
 180           puts "* Event conditions are not available for your installation of god."
 181           puts "* You may still use and write custom conditions using the poll system"
 182           puts "*"
 183           puts "***********************************************************************"
 184           puts
 185         end
 186
 187         puts "Resetting file descriptors"
 188
 189         puts "Loading config"
 190
 191         if options[:port]
 192           God.port = options[:port]
 193         end
 194
 195         load File.expand_path(options[:config])
 196
 197         Signal.trap('HUP') {}
 198       rescue => e
 199         File.open('god.log', 'a') { |f| f.puts e.message + "\n" + e.backtrace }
 200         abort "!!! ERROR - See god.log !!!"
 201       end
 202     end
 203
 204     if options[:pid]
 205       File.open(options[:pid], 'w') { |f| f.write pid }
 206     end
 207
 208     ::Process.detach pid
 209
 210     exit!(0)
 211   end
 212 end