vendor/rails/railties/lib/commands/runner.rb

   1 require 'optparse'
   2
   3 options = { :environment => (ENV['RAILS_ENV'] || "development").dup }
   4 code_or_file = nil
   5
   6 ARGV.clone.options do |opts|
   7   script_name = File.basename($0)
   8   opts.banner = "Usage: #{$0} [options] ('Some.ruby(code)' or a filename)"
   9
  10   opts.separator ""
  11
  12   opts.on("-e", "--environment=name", String,
  13           "Specifies the environment for the runner to operate under (test/development/production).",
  14           "Default: development") { |v| options[:environment] = v }
  15
  16   opts.separator ""
  17
  18   opts.on("-h", "--help",
  19           "Show this help message.") { $stderr.puts opts; exit }
  20
  21   if RUBY_PLATFORM !~ /mswin/
  22     opts.separator ""
  23     opts.separator "You can also use runner as a shebang line for your scripts like this:"
  24     opts.separator "-------------------------------------------------------------"
  25     opts.separator "#!/usr/bin/env #{File.expand_path($0)}"
  26     opts.separator ""
  27     opts.separator "Product.find(:all).each { |p| p.price *= 2 ; p.save! }"
  28     opts.separator "-------------------------------------------------------------"
  29   end
  30
  31   opts.order! { |o| code_or_file ||= o } rescue retry
  32 end
  33
  34 ARGV.delete(code_or_file)
  35
  36 ENV["RAILS_ENV"] = options[:environment]
  37 RAILS_ENV.replace(options[:environment]) if defined?(RAILS_ENV)
  38
  39 require RAILS_ROOT + '/config/environment'
  40
  41 if code_or_file.nil?
  42   $stderr.puts "Run '#{$0} -h' for help."
  43   exit 1
  44 elsif File.exist?(code_or_file)
  45   eval(File.read(code_or_file), nil, code_or_file)
  46 else
  47   eval(code_or_file)
  48 end