stubby 0.0.2 → 0.0.3

data/README.md

@@ -129,6 +129,19 @@ interfaces and will shut down the stubby server.
 To contribute a stub, just add your stub to the list above and issue a pull
 request.
 
+## Environment Verification
+
+It may be useful to deny requests that aren't routed by a system using stubby,
+or for a site accessed with stubby to display some information about the
+environment requested.
+
+The HTTP and HTTPS extensions both append request headers to proxied 
+requests. A STUBBY_ENV header contains the name of the stubby environment.
+
+Additionally, stubby requests send a STUBBY_KEY header which contains a hash
+that should be unique over the stubby user and the instruction that the trigger
+executed. If you configure your application to track STUBBY_KEY values, you can
+whitelist requests to a stubby system. 
 
 ## Stubbing
 

data/lib/stubby/extensions/http.rb

@@ -69,65 +69,61 @@ module Extensions
 
       adapter "http-proxy" do
         url.scheme = "http"
-
-        if url.path.empty?
-          # Proxy all requests, preserve incoming path
-          out = url.dup
-          out.path = request.path
-          request = HTTPI::Request.new
-          request.url = out.to_s
-        else
-          # Proxy to the given path
-          request = HTTPI::Request.new
-          request.url = url.to_s
-        end
-
-        response = HTTPI.get(request)
-        response.headers.delete "transfer-encoding"
-        response.headers.delete "connection"
-
-        status(response.code)
-        headers(response.headers)
-        body(response.body)
+        run_proxy
 
       end
 
       adapter "https-proxy" do
         url.scheme = "https"
+        run_proxy
+      end
+
+      %w(get post put patch delete options link unlink).each do |method|
+        send(method, //) do
+          run_handler
+        end
+      end
+
+      private
+      def run_handler
+        if instruction.nil?
+          not_found
+        elsif adapter=self.class.adapters[url.scheme]
+          instance_eval &adapter
+        else
+          instance_eval &self.class.adapters["default"]
+        end
+      end
 
+      def run_proxy
         if url.path.empty?
           # Proxy all requests, preserve incoming path
           out = url.dup
           out.path = request.path
-          request = HTTPI::Request.new
-          request.url = out.to_s
+          r = HTTPI::Request.new
+          r.url = out.to_s
         else
           # Proxy to the given path
-          request = HTTPI::Request.new
-          request.url = url.to_s
+          r = HTTPI::Request.new
+          r.url = url.to_s
         end
 
-        response = HTTPI.get(request)
+        r.headers["STUBBY_ENV"] = settings.stubby_session.environment
+        r.headers["STUBBY_KEY"] = settings.stubby_session.key(instruction)
+        r.headers["STUBBY_USER"] = settings.stubby_session.user_key
+
+        request.body.rewind
+        r.body = request.body.read
+
+        response = HTTPI.request(request.request_method.downcase.to_sym, r)
         response.headers.delete "transfer-encoding"
         response.headers.delete "connection"
-
+        
         status(response.code)
         headers(response.headers)
         body(response.body)
-
       end
 
-      get(//) do
-        if instruction.nil?
-          not_found
-        elsif adapter=self.class.adapters[url.scheme]
-          instance_eval &adapter
-        else
-          instance_eval &self.class.adapters["default"]
-        end
-      end	
-
-      private
       def forbidden
         [403, "Forbidden"]
       end

data/lib/stubby/master.rb

@@ -151,6 +151,18 @@ module Stubby
       @config.environment = environment
     end
 
+    def key(identifier)
+      Digest::MD5.hexdigest(user_key + identifier)
+    end
+
+    def user_key
+      @user_key ||= read_key
+    end
+
+    def environment
+      @config.environment
+    end
+
     def run!(options={})
       run_network do
         run_master do
@@ -160,6 +172,22 @@ module Stubby
     end
 
     private
+    def read_key
+      File.read(keyfile)
+    rescue
+      generate_key
+    end
+
+    def generate_key
+      SecureRandom.hex(50).tap do |key|
+        File.write(keyfile, key)
+      end
+    end
+
+    def keyfile
+      File.expand_path("~/.stubby/key")
+    end
+
     def run_network
       assume_network_interface
       yield

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: stubby
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors: