headers.rb in api-auth-1.0.1

- old
+ new

@@ -1,15 +1,15 @@
 module ApiAuth
-  
+
   # Builds the canonical string given a request object.
   class Headers
-    
+
     include RequestDrivers
-    
+
     def initialize(request)
       @original_request = request
-      
+
       case request.class.to_s
       when /Net::HTTP/
         @request = NetHttpRequest.new(request)
       when /RestClient/
         @request = RestClientRequest.new(request)
@@ -29,31 +29,52 @@
         raise UnknownHTTPRequest, "#{request.class.to_s} is not yet supported."
       end
       true
     end
     
+    # Returns the request timestamp
+    def timestamp
+       @request.timestamp 
+    end
+
     # Returns the canonical string computed from the request's headers
     def canonical_string
       [ @request.content_type,
         @request.content_md5,
         @request.request_uri,
         @request.timestamp
       ].join(",")
     end
-    
+
     # Returns the authorization header from the request's headers
     def authorization_header
       @request.authorization_header
     end
-    
+
+    def set_date
+      @request.set_date if @request.timestamp.blank?
+    end
+
+    def calculate_md5
+      @request.populate_content_md5 if @request.content_md5.blank?
+    end
+
+    def md5_mismatch?
+      if @request.content_md5.blank?
+        false
+      else
+        @request.md5_mismatch?
+      end
+    end
+
     # Sets the request's authorization header with the passed in value.
     # The header should be the ApiAuth HMAC signature.
     #
     # This will return the original request object with the signed Authorization
     # header already in place.
     def sign_header(header)
       @request.set_auth_header header
     end
-    
+
   end
-  
+
 end