joshbuddy-rack-rewrite 0.0.2

data/README.rdoc

@@ -0,0 +1,41 @@
+= rack-rewrite
+
+Rack middleware for request rewriting.
+
+== Example
+
+  use Rack::Rewrite do
+    
+    on :method => 'get' do
+      
+      # When it sees /api or /test, it calls whatever is in the act block
+      #
+      on :uri => %r{/(api|test)} do
+        act { puts "hey way to go!" }
+        pass
+      end
+  
+      # When it sees /google, it makes a redirect to google with the value of the query string as the search
+      #
+      on :uri => %r{/google.*} do
+        redirect { "http://google.com/search?q=#{CGI.escape(query_string)}" }
+      end
+  
+      # If a URI starts with /valid_place, it strips it off and keeps on truckin'
+      #
+      on :uri => %r{/valid_place/.*} do
+        set(:uri) { uri[%r{/valid_place(/.*)}, 1] }
+        pass
+      end
+      
+      # If a URI starts with /hello_kitty, add a kitty header equal to your query string
+      #
+      on :uri => %r{/hello_kitty/.*} do
+        act { header['kitty'] = query_string }
+        pass
+      end
+      
+      fail
+    end
+    
+  end

data/VERSION.yml

@@ -0,0 +1,4 @@
+--- 
+:major: 0
+:minor: 0
+:patch: 2

data/lib/rack_rewrite/actions.rb

@@ -0,0 +1,142 @@
+module Rack
+  class Rewrite
+    class Action
+      
+      def uri
+        @env['PATH_INFO']
+      end
+      
+      def uri=(uri)
+        @env['PATH_INFO'] = uri
+        update_uri_qs
+      end
+      
+      def query_string
+        @env['QUERY_STRING']
+      end
+
+      def query_string=(query_string)
+        @env['QUERY_STRING'] = query_string
+        update_uri_qs
+      end
+      
+      def update_uri_qs
+        @env['REQUEST_PATH'] = uri
+        @env['REQUEST_URI'] = query_string.empty? ? uri : "#{uri}?#{query_string}"
+      end
+      
+      def method
+        @env['REQUEST_METHOD']
+      end
+      
+      def method=(method)
+        @env['REQUEST_METHOD'] = method.to_s.upcase
+      end
+      
+      def host
+        @env['HTTP_HOST']
+      end
+      
+      def host=(host)
+        @env['HTTP_HOST'] = host
+      end
+      
+      def port
+        @env['SERVER_PORT']
+      end
+      
+      def port=(port)
+        @env['SERVER_PORT'] = port.to_s
+      end
+      
+      def scheme
+        @env['rack.url_scheme']
+      end
+      
+      def scheme=(scheme)
+        @env['rack.url_scheme'] = scheme
+      end
+      
+      def setup(env)
+        @env = env
+      end
+
+      class Pass < Action
+        def call(env)
+          throw :pass
+        end
+      end
+
+      class Fail < Action
+        def call(env)
+          raise
+        end
+      end
+
+      class Set < Action
+
+        def initialize(variable, action)
+          @variable = variable
+          @action = action
+        end
+
+        def call(env)
+          setup(env)
+          self.send :"#{@variable}=", case @action
+          when Proc
+            instance_eval(&@action)
+          else
+            instance_eval(@action)
+          end
+          env
+        end
+
+      end
+
+      class Do < Action
+        def initialize(caller, action)
+          @caller = caller
+          @action = action
+        end
+
+        def method_missing(method, *args, &block)
+          @caller.send(method, *args, &block)
+        end
+
+        def call(env)
+          setup(env)
+          case @action
+          when Proc
+            instance_eval(&@action)
+          else
+            instance_eval(@action)
+          end
+          env
+        end
+
+      end
+
+      class Redirect < Action
+        def initialize(caller, action)
+          @caller = caller
+          @action = action
+        end
+
+        def call(env)
+          setup(env)
+          @caller.redirect = [ 302, {'Location'=> case @action
+          when Proc
+            instance_eval(&@action)
+          else
+            instance_eval(@action)
+          end}, []]
+          throw :pass
+        end
+      end
+      
+    end
+
+  end
+end
+    
+    

data/lib/rack_rewrite.rb

@@ -0,0 +1,108 @@
+$:.unshift(File.dirname(__FILE__))
+require 'rack_rewrite/actions'
+
+module Rack
+  class Rewrite
+    
+    attr_accessor :path, :redirect, :headers
+    
+    class ConditionSet
+      
+      attr_reader :parent_set, :actions, :conditions
+      
+      def initialize(parent_set, conditions = nil)
+        @parent_set = parent_set
+        @conditions = conditions
+        @actions = []
+      end
+
+      def satisfied?(env)
+        if conditions
+          
+          uri_ok = conditions.key?(:uri) ? conditions[:uri] === env['PATH_INFO'] : true
+          method_ok = conditions.key?(:method) ? conditions[:method] === env['REQUEST_METHOD'].downcase : true
+          host_ok = conditions.key?(:host) ? conditions[:host] === env['HTTP_HOST'] : true
+          port_ok = conditions.key?(:port) ? conditions[:port] === env['SERVER_PORT'].to_i : true
+          scheme_ok = conditions.key?(:scheme) ? conditions[:scheme] === env['rack.url_scheme'] : true
+          
+          #puts "uri_ok: #{uri_ok} method_ok #{method_ok} host_ok #{host_ok} port_ok #{port_ok} scheme_ok #{scheme_ok} for conditions #{conditions.inspect}"
+          
+          uri_ok && method_ok && host_ok && port_ok && scheme_ok
+        else
+          true
+        end
+      end
+      
+      def inspect
+        "#{id} .. conditions=#{conditions.inspect}\nactions=#{actions.inspect}"
+      end
+      
+    end
+    
+    def initialize(app, options = {}, &block)
+      @app = app
+      @options = []
+      @current_condition_set = @root = ConditionSet.new(nil)
+      instance_eval(&block)
+    end
+    
+    def on(conditions = {}, &block)
+      @current_condition_set.actions << ConditionSet.new(@current_condition_set, conditions)
+      @current_condition_set = @current_condition_set.actions.last
+      instance_eval(&block)
+      @current_condition_set = @current_condition_set.parent_set
+    end
+
+    def set(variable, pattern = nil, &block)
+      @current_condition_set.actions << Action::Set.new(variable, pattern || block)
+    end
+
+    def act(&block)
+      @current_condition_set.actions << Action::Do.new(self, block)
+    end
+
+    def pass
+      @current_condition_set.actions << Action::Pass.new
+    end
+    
+    def fail
+      @current_condition_set.actions << Action::Fail.new
+    end
+    
+    def redirect(pattern = nil, &block)
+      @current_condition_set.actions << Action::Redirect.new(self, pattern || block)
+    end
+    
+    def call(env)
+      @headers = {}
+      catch(:pass) {
+        env = call_conditions(env, @root)
+        raise
+      }
+      if @redirect
+        redirect = @redirect
+        @redirect = nil
+        redirect
+      else
+        (status, headers, body) = @app.call(env)
+        [status, headers.merge(@headers), body]
+      end
+    end
+    
+    def call_conditions(env, conditions_set)
+      if conditions_set.satisfied?(env)
+        conditions_set.actions.each_with_index do |act, index|
+          break if @done
+          case act
+          when ConditionSet
+            call_conditions(env, act)
+          else
+            env = act.call(env)
+          end
+        end
+      end
+      
+    end
+    
+  end
+end

data/spec/conditions_spec.rb

@@ -0,0 +1,22 @@
+require 'spec_helper'
+
+describe "Rack::Rewrite Conditions" do
+
+  ['get', 'post', 'put', 'delete'].each do |method|
+
+    it "should detect method #{method}" do
+      env = Rack::MockRequest.env_for('/test', :method => method)
+      app = mock('app')
+      app.should_receive(:call).with(env).and_return([200, {}, ["body"]])
+      Rack::Rewrite.new(app) { on(:method => method) { pass } }.call(env)
+    end
+  end
+  
+  it "should detect a simple uri" do
+    env = Rack::MockRequest.env_for('/test', :method => 'get')
+    app = mock('app')
+    app.should_receive(:call).with(env).and_return([200, {}, ["body"]])
+    Rack::Rewrite.new(app) { on(:uri => '/test') { pass } }.call(env)
+  end
+
+end

data/spec/rewrite_spec.rb

@@ -0,0 +1,24 @@
+require 'spec_helper'
+
+describe "Rack::Rewrite Rewriting" do
+
+  it "should rewrite a uri" do
+    env = Rack::MockRequest.env_for('/test', :method => 'get')
+    app = mock('app')
+    app.should_receive(:call) { |resp|
+      resp['PATH_INFO'].should == '/test/test'
+      [200, {}, ["body"]]
+    }
+    
+    Rack::Rewrite.new(app) { on(:uri => '/test') { set(:uri) { "/test#{uri}" }; pass } }.call(env)
+  end
+
+  it "should arbitrarily add a new header" do
+    env = Rack::MockRequest.env_for('/test?Happy-Land', :method => 'get')
+    app = mock('app')
+    app.should_receive(:call).and_return([200, {'Content-type' => 'text/html'}, ['mybody']])
+    response = Rack::Rewrite.new(app) { on(:uri => '/test') { act{ headers['My-special-header'] = query_string }; pass } }.call(env)
+    response[1]['My-special-header'].should == 'Happy-Land'
+  end
+
+end

data/spec/spec_helper.rb

@@ -0,0 +1,11 @@
+require 'spec'
+require 'rubygems'
+require 'rack'
+
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'rack_rewrite'
+
+Spec::Runner.configure do |config|
+  
+end

metadata

@@ -0,0 +1,61 @@
+--- !ruby/object:Gem::Specification 
+name: joshbuddy-rack-rewrite
+version: !ruby/object:Gem::Version 
+  version: 0.0.2
+platform: ruby
+authors: 
+- Joshua Hull
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-05-04 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: 
+email: joshbuddy@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README.rdoc
+files: 
+- README.rdoc
+- VERSION.yml
+- lib/rack_rewrite
+- lib/rack_rewrite/actions.rb
+- lib/rack_rewrite.rb
+- spec/conditions_spec.rb
+- spec/rewrite_spec.rb
+- spec/spec_helper.rb
+has_rdoc: true
+homepage: http://github.com/joshbuddy/rack-rewrite
+post_install_message: 
+rdoc_options: 
+- --inline-source
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 2
+summary: TODO
+test_files: []
+