2005-06-16 05:50:24 +00:00
|
|
|
require 'optparse'
|
2010-04-26 15:25:07 +00:00
|
|
|
require 'rbconfig'
|
2005-06-16 05:50:24 +00:00
|
|
|
|
2005-10-14 02:51:33 +00:00
|
|
|
options = { :environment => (ENV['RAILS_ENV'] || "development").dup }
|
2006-09-29 07:45:08 +00:00
|
|
|
code_or_file = nil
|
2005-06-16 05:50:24 +00:00
|
|
|
|
2006-09-26 07:10:08 +00:00
|
|
|
ARGV.clone.options do |opts|
|
2005-06-16 05:50:24 +00:00
|
|
|
script_name = File.basename($0)
|
2010-02-03 00:16:26 +00:00
|
|
|
opts.banner = "Usage: runner [options] ('Some.ruby(code)' or a filename)"
|
2005-06-16 05:50:24 +00:00
|
|
|
|
|
|
|
opts.separator ""
|
|
|
|
|
|
|
|
opts.on("-e", "--environment=name", String,
|
|
|
|
"Specifies the environment for the runner to operate under (test/development/production).",
|
2006-06-28 20:53:00 +00:00
|
|
|
"Default: development") { |v| options[:environment] = v }
|
2005-06-16 05:50:24 +00:00
|
|
|
|
|
|
|
opts.separator ""
|
|
|
|
|
|
|
|
opts.on("-h", "--help",
|
2010-09-19 07:39:42 +00:00
|
|
|
"Show this help message.") { $stdout.puts opts; exit }
|
2006-09-26 07:10:08 +00:00
|
|
|
|
2010-08-24 02:09:35 +00:00
|
|
|
if RbConfig::CONFIG['host_os'] !~ /mswin|mingw/
|
2006-09-26 07:10:08 +00:00
|
|
|
opts.separator ""
|
|
|
|
opts.separator "You can also use runner as a shebang line for your scripts like this:"
|
|
|
|
opts.separator "-------------------------------------------------------------"
|
2010-09-16 04:02:33 +00:00
|
|
|
opts.separator "#!/usr/bin/env #{File.expand_path($0)} runner"
|
2006-09-26 07:10:08 +00:00
|
|
|
opts.separator ""
|
|
|
|
opts.separator "Product.find(:all).each { |p| p.price *= 2 ; p.save! }"
|
|
|
|
opts.separator "-------------------------------------------------------------"
|
|
|
|
end
|
|
|
|
|
2006-09-29 07:45:08 +00:00
|
|
|
opts.order! { |o| code_or_file ||= o } rescue retry
|
2005-06-16 05:50:24 +00:00
|
|
|
end
|
|
|
|
|
2006-09-29 07:45:08 +00:00
|
|
|
ARGV.delete(code_or_file)
|
|
|
|
|
2005-10-12 12:44:03 +00:00
|
|
|
ENV["RAILS_ENV"] = options[:environment]
|
2005-06-16 05:50:24 +00:00
|
|
|
|
2010-04-29 06:39:44 +00:00
|
|
|
require APP_PATH
|
2010-07-01 15:07:48 +00:00
|
|
|
Rails.application.require_environment!
|
2010-02-09 16:40:06 +00:00
|
|
|
|
2008-11-22 19:32:36 +00:00
|
|
|
begin
|
|
|
|
if code_or_file.nil?
|
|
|
|
$stderr.puts "Run '#{$0} -h' for help."
|
|
|
|
exit 1
|
|
|
|
elsif File.exist?(code_or_file)
|
2010-06-22 13:15:58 +00:00
|
|
|
$0 = code_or_file
|
2008-11-22 19:32:36 +00:00
|
|
|
eval(File.read(code_or_file), nil, code_or_file)
|
|
|
|
else
|
|
|
|
eval(code_or_file)
|
|
|
|
end
|
|
|
|
ensure
|
2008-12-11 00:07:02 +00:00
|
|
|
if defined? Rails
|
|
|
|
Rails.logger.flush if Rails.logger.respond_to?(:flush)
|
|
|
|
end
|
2006-09-26 07:10:08 +00:00
|
|
|
end
|