lib/server_manager.rb

   1 ################################################################################
   2 require 'optparse'
   3
   4 ################################################################################
   5 $ferret_server_options = {
   6   'environment' => nil,
   7   'debug'       => nil,
   8   'root'        => nil
   9 }
  10
  11 ################################################################################
  12 OptionParser.new do |optparser|
  13   optparser.banner = "Usage: #{File.basename($0)} [options] {start|stop|run}"
  14
  15   optparser.on('-h', '--help', "This message") do
  16     puts optparser
  17     exit
  18   end
  19
  20   optparser.on('-R', '--root=PATH', 'Set RAILS_ROOT to the given string') do |r|
  21     $ferret_server_options['root'] = r
  22   end
  23
  24   optparser.on('-e', '--environment=NAME', 'Set RAILS_ENV to the given string') do |e|
  25     $ferret_server_options['environment'] = e
  26   end
  27
  28   optparser.on('--debug', 'Include full stack traces on exceptions') do
  29     $ferret_server_options['debug'] = true
  30   end
  31
  32   $ferret_server_action = optparser.permute!(ARGV)
  33   (puts optparser; exit(1)) unless $ferret_server_action.size == 1
  34
  35   $ferret_server_action = $ferret_server_action.first
  36   (puts optparser; exit(1)) unless %w(start stop run).include?($ferret_server_action)
  37 end
  38
  39 ################################################################################
  40 begin
  41   ENV['FERRET_USE_LOCAL_INDEX'] = 'true'
  42   ENV['RAILS_ENV'] = $ferret_server_options['environment']
  43
  44   # determine RAILS_ROOT unless already set
  45   RAILS_ROOT = $ferret_server_options['root'] || File.join(File.dirname(__FILE__), *(['..']*4)) unless defined? RAILS_ROOT
  46   # check if environment.rb is present
  47   rails_env_file = File.join(RAILS_ROOT, 'config', 'environment')
  48   raise "Unable to find Rails environment.rb at \n#{rails_env_file}.rb\nPlease use the --root option of ferret_server to point it to your RAILS_ROOT." unless File.exists?(rails_env_file+'.rb')
  49   # load it
  50   require rails_env_file
  51
  52   require 'acts_as_ferret'
  53   ActsAsFerret::Remote::Server.new.send($ferret_server_action)
  54 rescue Exception => e
  55   $stderr.puts(e.message)
  56   $stderr.puts(e.backtrace.join("\n")) if $ferret_server_options['debug']
  57   exit(1)
  58 end