Contents

require 'rack'

module Rack

  module Rewritten

    class Url

      def initialize(app, &block)
        @app = app

        self.instance_eval(&block) if block_given?
      end

      def call(env)
        req = Rack::Request.new(env)

        subdomain = env["SUBDOMAIN"] ? "#{env["SUBDOMAIN"]}:" : ""

        if to = ::Rewritten.includes?("#{subdomain}#{req.path_info}")
          current_path = ::Rewritten.get_current_translation(to)
          current_path = current_path.split(":").last

          if current_path == req.path_info
            # if this is the current path, rewrite path and parameters
            tpath, tparams = split_to_path_params(to)
            req.path_info = tpath
            env['QUERY_STRING'] = Rack::Utils.build_query(tparams.merge(req.params))
            @app.call(req.env) 
          else
            # if this is not the current path, redirect to current path
            # NOTE: assuming redirection is always to non-subdomain-path
            
            r = Rack::Response.new

            new_path = env["rack.url_scheme"].dup
            new_path << "://"
            new_path << env["HTTP_HOST"].dup.sub(/^#{subdomain.chomp(':')}\./, '')
            new_path << current_path
            new_path << '?' << env["QUERY_STRING"] unless (env["QUERY_STRING"]||'').empty?

            r.redirect(new_path, 301)
            a = r.finish
          end
        else
          @app.call(req.env) 
        end
      end

      def split_to_path_params(path_and_query)
        path, query_string = path_and_query.split('?').push('')[0..1]
        [path, Rack::Utils.parse_query(query_string)] 
      end


      private
      
      def add_translation(from,to)
        ::Rewritten.redis = :test unless ::Rewritten.redis == :test
        ::Rewritten.add_translation(from, to)
      end


      
    end
  end

end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
rewritten-0.5.0	lib/rack/url.rb
rewritten-0.4.0	lib/rack/url.rb