Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Fetching contributors…

Cannot retrieve contributors at this time

145 lines (124 sloc) 4.517 kB
require 'goliath/goliath'
require 'goliath/runner'
require 'goliath/rack'
# Pre-load the goliath environment so it's available as we try to parse the class.
# This means we can use Goliath.env?(:development) or Goliath.env?(:produdction) in the use statements.
#
# Note, as implmented, you have to have -e as it's own flag, you can't do -sve dev
# as it won't pickup the e flag.
env = ENV['RACK_ENV']
env ||= begin
if ((i = ARGV.index('-e')) || (i = ARGV.index('--environment')))
ARGV[i + 1]
end
end
Goliath.env = env if env
module Goliath
# The main execution class for Goliath. This will execute in the at_exit
# handler to run the server.
#
# @private
class Application
# Most of this stuff is straight out of sinatra.
# Set of caller regex's to be skipped when looking for our API file
CALLERS_TO_IGNORE = [ # :nodoc:
/\/goliath(\/application)?\.rb$/, # all goliath code
/\/goliath(\/(rack|validation|plugins)\/)/, # all goliath code
/rubygems\/custom_require\.rb$/, # rubygems require hacks
/bundler(\/runtime)?\.rb/, # bundler require hacks
/<internal:/ # internal in ruby >= 1.9.2
]
# @todo add rubinius (and hopefully other VM impls) ignore patterns ...
CALLERS_TO_IGNORE.concat(RUBY_IGNORE_CALLERS) if defined?(RUBY_IGNORE_CALLERS)
# Like Kernel#caller but excluding certain magic entries and without
# line / method information; the resulting array contains filenames only.
def self.caller_files
caller_locations.map { |file, line| file }
end
# Like caller_files, but containing Arrays rather than strings with the
# first element being the file, and the second being the line.
def self.caller_locations
caller(1).
map { |line| line.split(/:(?=\d|in )/)[0,2] }.
reject { |file, line| CALLERS_TO_IGNORE.any? { |pattern| file =~ pattern } }
end
# Find the app_file that was used to execute the application
#
# @return [String] The app file
def self.app_file
c = caller_files.first
c = $0 if !c || c.empty?
c
end
# Returns the userland class which inherits the Goliath API
#
# @return [String] The app class
def self.app_class
@app_class
end
# Sets the userland class that should use the Goliath API
#
# @param app_class [String|Symbol|Constant] The new app class
# @return [String] app_class The new app class
def self.app_class=(app_class)
@app_class = app_class.to_s
end
# Retrive the base directory for the API before we've changed directories
#
# @note Note sure of a better way to handle this. Goliath will do a chdir
# when the runner is executed. If you need the +root_path+ before
# the runner is executing (like, in a use statement) you need this method.
#
# @param args [Array] Any arguments to append to the path
# @return [String] path for the given arguments
def self.app_path(*args)
@app_path ||= File.expand_path(File.dirname(app_file))
File.join(@app_path, *args)
end
# Retrieve the base directory for the API
#
# @param args [Array] Any arguments to append to the path
# @return [String] path for the given arguments
def self.root_path(*args)
return app_path(args) if Goliath.env?(:test)
@root_path ||= File.expand_path("./")
File.join(@root_path, *args)
end
# Execute the application
#
# @return [Nil]
def self.run!
unless @app_class
file = File.basename(app_file, '.rb')
@app_class = camel_case(file)
end
begin
klass = Kernel
@app_class.split('::').each do |con|
klass = klass.const_get(con)
end
rescue NameError
raise NameError, "Class #{@app_class} not found."
end
api = klass.new
runner = Goliath::Runner.new(ARGV, api)
runner.app = Goliath::Rack::Builder.build(klass, api)
runner.load_plugins(klass.plugins)
runner.run
end
private
# Convert a string to camel case
#
# @param str [String] The string to convert
# @return [String] The camel cased string
def self.camel_case(str)
return str if str !~ /_/ && str =~ /[A-Z]+.*/
str.split('_').map { |e| e.capitalize }.join
end
end
at_exit do
if $!.nil? && $0 == Goliath::Application.app_file
Application.run!
end
end
end
Jump to Line
Something went wrong with that request. Please try again.