require 'active_support/core_ext/hash/except' module ActionController # Rewrites URLs for Base.redirect_to and Base.url_for in the controller. class UrlRewriter #:nodoc: RESERVED_OPTIONS = [:anchor, :params, :only_path, :host, :protocol, :port, :trailing_slash, :skip_relative_url_root] def initialize(request, parameters) @request, @parameters = request, parameters end def rewrite(router, options = {}) options[:host] ||= @request.host_with_port options[:protocol] ||= @request.protocol self.class.rewrite(router, options, @request.symbolized_path_parameters) do |options| process_path_options(options) end end def to_str "#{@request.protocol}, #{@request.host_with_port}, #{@request.path}, #{@parameters[:controller]}, #{@parameters[:action]}, #{@request.parameters.inspect}" end alias_method :to_s, :to_str # ROUTES TODO: Class method code smell def self.rewrite(router, options, path_segments=nil) handle_positional_args(options) rewritten_url = "" # ROUTES TODO: Fix the tests segments = options.delete(:_path_segments) path_segments = path_segments ? path_segments.merge(segments || {}) : segments unless options[:only_path] rewritten_url << (options[:protocol] || "http") rewritten_url << "://" unless rewritten_url.match("://") rewritten_url << rewrite_authentication(options) raise "Missing host to link to! Please provide :host parameter or set default_url_options[:host]" unless options[:host] rewritten_url << options[:host] rewritten_url << ":#{options.delete(:port)}" if options.key?(:port) end path_options = options.except(*RESERVED_OPTIONS) path_options = yield(path_options) if block_given? path = router.generate(path_options, path_segments || {}) # ROUTES TODO: This can be called directly, so script_name should probably be set in the router rewritten_url << (options[:trailing_slash] ? path.sub(/\?|\z/) { "/" + $& } : path) rewritten_url << "##{Rack::Utils.escape(options[:anchor].to_param.to_s)}" if options[:anchor] rewritten_url end protected def self.handle_positional_args(options) return unless args = options.delete(:_positional_args) keys = options.delete(:_positional_keys) keys -= options.keys if args.size < keys.size - 1 # take format into account args = args.zip(keys).inject({}) do |h, (v, k)| h[k] = v h end # Tell url_for to skip default_url_options options.merge!(args) end def self.rewrite_authentication(options) if options[:user] && options[:password] "#{Rack::Utils.escape(options.delete(:user))}:#{Rack::Utils.escape(options.delete(:password))}@" else "" end end # Given a Hash of options, generates a route def process_path_options(options) options = options.symbolize_keys options.update(options[:params].symbolize_keys) if options[:params] if (overwrite = options.delete(:overwrite_params)) options.update(@parameters.symbolize_keys) options.update(overwrite.symbolize_keys) end options end end end