rack-remove-param 0.0.1 → 0.0.2

data/lib/rack/remove-param.rb

@@ -1,11 +1,11 @@
 module Rack
   class RemoveParam
-    def initialize(app, param_to_filter)
-        @param_to_filter = param_to_filter
+    def initialize(app, params_to_filter)
+      process_params(params_to_filter)
       @app = app
     end
 
-    def call(env) 
+    def call(env)
       dup._call(env)
     end
 
@@ -13,7 +13,7 @@ module Rack
       @request = Rack::Request.new(env)
 
       if @request.post?
-        delete_param(@request.params, @param_to_filter)
+        delete_param(@request.params)
         env["rack.request.form_hash"] = @request.params
         env["rack.request.form_vars"] = Rack::Utils.build_query(@request.params)
       end
@@ -23,9 +23,16 @@ module Rack
     end
 
     private
-    def delete_param(hash, param)
-      hash.delete(param)
-      hash.each{|k, value| delete_param(value, param) if value.is_a?(Hash) }
+
+    def process_params(params)
+      @params_to_filter = params.is_a?(String) ? Array.new(1, params) : Array.new(params)
+    end
+
+    def delete_param(hash)
+      @params_to_filter.each do |param|
+        hash.delete(param)
+        hash.each{|k, value| delete_param(value, param) if value.is_a?(Hash) }
+      end
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rack-remove-param
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-09-28 00:00:00.000000000 Z
+date: 2013-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &70257290096640 !ruby/object:Gem::Requirement
+  requirement: !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,15 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70257290096640
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: rack
-  requirement: &70257290125080 !ruby/object:Gem::Requirement
+  requirement: !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,7 +37,12 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70257290125080
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Remove POST params from reqs
 email:
 - thomas@mintdigital.com
@@ -62,7 +72,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.11
+rubygems_version: 1.8.23
 signing_key: 
 specification_version: 3
 summary: Filter out unwanted params