redsquare 0.1.2 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a033f99e75b30bf2e87811348baa30bd1004d7f8
-  data.tar.gz: df45f16bab78413de85fca722dc830afa5b77a83
+  metadata.gz: 8ac3a33efd6022077555a6159f9dfc6ce4e337ba
+  data.tar.gz: d7a4e5c17dad56350c10e88ac4dc4fb784985c9b
 SHA512:
-  metadata.gz: 8130f86e465fbd7d4474dd55e3bea47a37b10cff9babb458b92f676484b21ed4bbf50ff545b326ccee4e6f8610158c62ad539000838ac449cac935e083e87979
-  data.tar.gz: 3a8af44fcc4ad34f6bc22415486e7f358460a717ae33d41ad57ae76b39f20688aabcf1b9c66c0c4f6e6da84f35ff92a1e5b647a450ec0f6d044a6936f52cb75a
+  metadata.gz: 76e74fecc5a12c2cff7aad2c511309ee4cf6051d1256b0225d03615f309ac2b9715ad41accfe99686800ae08522ad53cbace3cad6b6eddf2c16cb74a4f76457d
+  data.tar.gz: 2e7eb78d0d244e4b37bdb7b9cc9ed1df6784e0f698dea994ba32ac33fd2fcd0cafefb7fb71876bc75cd92be8215e709e628144044271dd491757cc885d312b0d

data/README.md

@@ -35,6 +35,10 @@ Redsquare.configure do |config|
   # pass a config hash that will be passed to Redis.new
   config.redis = { :host => 'localhost', :port => 6379 }
 
+  # indicate redis methods that should not be supported, calls
+  # to these methods results in a 404
+  config.restricted_methods = [:keys, :dbsize]
+
 end
 ```
 

data/lib/redsquare/app.rb

@@ -79,6 +79,7 @@ module Redsquare
 
     POST_COMMANDS.each do |command|
       post "/#{command}" do
+        return 404 if Config.restricted_methods.include?(command)
         content_type :json
         args = params[:args].map { |a| try_to_i a }
         val = Config.redis.send command, *args
@@ -88,6 +89,7 @@ module Redsquare
 
     GET_COMMANDS.each do |command|
       get "/#{command}/?*" do
+        return 404 if Config.restricted_methods.include?(command)
         content_type :json
         args = params[:splat][0].split("/").map { |a| try_to_i a }
         val = Config.redis.send command, *args

data/lib/redsquare/config.rb

@@ -14,5 +14,13 @@ module Redsquare
       @redis ||= Redis.current
     end
 
+    def restricted_methods
+      @restricted_methods || []
+    end
+
+    def restricted_methods=(methods)
+      @restricted_methods = methods
+    end
+
   end
 end

data/lib/redsquare/version.rb

@@ -1,3 +1,3 @@
 module Redsquare
-  VERSION = "0.1.2"
+  VERSION = "0.2.0"
 end

data/spec/app_spec.rb

@@ -194,4 +194,18 @@ describe Redsquare::App do
 
   end
 
+  describe "config" do
+
+    describe "restricted_methods" do
+
+      it "prevents the method from being exposed" do
+        Redsquare::Config.restricted_methods = [:keys]
+        post "/keys"
+        expect(last_response.status).to eq 404
+      end
+
+    end
+
+  end
+
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: redsquare
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.2.0
 platform: ruby
 authors:
 - JC Grubbs