pre-parse out -e or --environment for use in use statements

Author: dj2

examples/env_use_statements.rb

@@ -0,0 +1,17 @@
+#!/usr/bin/env ruby
+$:<< '../lib' << 'lib'
+
+require'goliath'
+require'yajl'
+
+classEnvUseStatements < Goliath::API
+ifGoliath.dev?
+useGoliath::Rack::Render,'json'
+elsifGoliath.prod?
+useGoliath::Rack::Render,'xml'
+end
+
+defresponse(env)
+[200,{},{'Test'=>'Response'}]
+end
+end

lib/goliath/application.rb

@@ -2,6 +2,19 @@
 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.dev? or Goliath.prod? 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=envifenv
+
 moduleGoliath
 # The main execution class for Goliath. This will execute in the at_exit
 # handler to run the server.
@@ -44,14 +57,14 @@ def self.app_file
 c= $0 if !c || c.empty?
 c
 end
-
+
 # Returns the userland class which inherits the Goliath API
 #
 # @return [String] The app class
 defself.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