lib/spire/error.rb in spire-0.2.3 vs lib/spire/error.rb in spire-0.2.4

@@ -1,8 +1,11 @@
 module Spire
     class Error
-        def initialize(message, status='200')
+        def initialize(opts={})
+            status = opts[:status] or 200
+            message = opts[:message] or "Unknown Error"
+            
             case status
             when 200
                 self.return_error(message, status)
             when 404
                 self.return_404
@@ -10,10 +13,12 @@
                 self.return_401
             when 204
                 self.return_204
             when 301
                 self.return_301
+            when 444
+                self.return_444
             end
         end
 
         def return_404
             @return = Response.new("404 // Page not found", 'text/html;', 404)
@@ -32,10 +37,14 @@
         end
 
         def return_error(message, status)
             @return = Response.new(message, 'text/html;', status)
         end
-
+          
+        def return_444
+            @return = Response.new("No response made by the server, check for a valid response", 'text/html;', 404)
+        end
+        
         def to_rack
             @return.to_rack
         end
     end
 end
\ No newline at end of file