lib/rtml/environment.rb in rtml-2.0.3 vs lib/rtml/environment.rb in rtml-2.0.4

@@ -17,12 +17,24 @@
 end
 
 require File.join(path, 'builtin/rtml_helpers/rtml/get_state')
 require File.join(path, 'builtin/rtml_helpers/rtml/rescue_action')
 
-Dir[File.join(path, 'lib/extensions/**/*.rb')].each { |fi| require fi if File.file?(fi) }
+if defined?(Rails)
+  Rails.configuration.after_initialize do
+    Dir[File.join(path, 'lib/extensions/**/*.rb')].each { |fi| require fi if File.file?(fi) }
+  end
+else
+  Dir[File.join(path, 'lib/extensions/**/*.rb')].each { |fi| require fi if File.file?(fi) }
+end
 
 # Add RTML routes
 routes_file = File.join(Rtml.root, 'lib/rtml_routes.rb')
 unless ActionController::Routing::Routes.configuration_files.include? routes_file
   ActionController::Routing::Routes.configuration_files << routes_file
 end
+
+Mime::TML = Mime::Type.lookup("text/tml")
+require File.join(path, "lib/rtml/rules/tml_param_parser")
+ActionController::Base.param_parsers[Mime::TML] = proc { |data|
+  Rtml::Rules::TmlParamParser.new(data).params
+}