require 'action_dispatch/http/request' require 'active_support/core_ext/uri' require 'rack/utils' module ActionDispatch module Routing class Redirect # :nodoc: attr_reader :status, :block def initialize(status, block) @status = status @block = block end def call(env) req = Request.new(env) uri = URI.parse(path(req.symbolized_path_parameters, req)) uri.scheme ||= req.scheme uri.host ||= req.host uri.port ||= req.port unless req.standard_port? body = %(You are being redirected.) headers = { 'Location' => uri.to_s, 'Content-Type' => 'text/html', 'Content-Length' => body.length.to_s } [ status, headers, [body] ] end def path(params, request) block.call params, request end end class OptionRedirect < Redirect # :nodoc: alias :options :block def path(params, request) url_options = { :protocol => request.protocol, :host => request.host, :port => request.optional_port, :path => request.path, :params => request.query_parameters }.merge options if !params.empty? && url_options[:path].match(/%\{\w*\}/) url_options[:path] = (url_options[:path] % escape_path(params)) end ActionDispatch::Http::URL.url_for url_options end private def escape_path(params) Hash[params.map{ |k,v| [k, URI.parser.escape(v)] }] end end module Redirection # Redirect any path to another path: # # match "/stories" => redirect("/posts") # # You can also use interpolation in the supplied redirect argument: # # match 'docs/:article', :to => redirect('/wiki/%{article}') # # Alternatively you can use one of the other syntaxes: # # The block version of redirect allows for the easy encapsulation of any logic associated with # the redirect in question. Either the params and request are supplied as arguments, or just # params, depending of how many arguments your block accepts. A string is required as a # return value. # # match 'jokes/:number', :to => redirect { |params, request| # path = (params[:number].to_i.even? ? "wheres-the-beef" : "i-love-lamp") # "http://#{request.host_with_port}/#{path}" # } # # The options version of redirect allows you to supply only the parts of the url which need # to change, it also supports interpolation of the path similar to the first example. # # match 'stores/:name', :to => redirect(:subdomain => 'stores', :path => '/%{name}') # match 'stores/:name(*all)', :to => redirect(:subdomain => 'stores', :path => '/%{name}%{all}') # # Finally, an object which responds to call can be supplied to redirect, allowing you to reuse # common redirect routes. The call method must accept two arguments, params and request, and return # a string. # # match 'accounts/:name' => redirect(SubdomainRedirector.new('api')) # def redirect(*args, &block) options = args.last.is_a?(Hash) ? args.pop : {} status = options.delete(:status) || 301 return OptionRedirect.new(status, options) if options.any? path = args.shift block = lambda { |params, request| (params.empty? || !path.match(/%\{\w*\}/)) ? path : (path % escape(params)) } if String === path block = path if path.respond_to? :call # :FIXME: remove in Rails 4.0 if block && block.respond_to?(:arity) && block.arity < 2 msg = "redirect blocks with arity of #{block.arity} are deprecated. Your block must take 2 parameters: the environment, and a request object" ActiveSupport::Deprecation.warn msg deprecated_block = block block = lambda { |params, _| deprecated_block.call(params) } end raise ArgumentError, "redirection argument not supported" unless block Redirect.new status, block end private def escape(params) Hash[params.map{ |k,v| [k, Rack::Utils.escape(v)] }] end end end end