lib/jsonapionify/api/base/app_builder.rb in jsonapionify-0.0.1.pre vs lib/jsonapionify/api/base/app_builder.rb in jsonapionify-0.9.0

@@ -1,8 +1,19 @@
 module JSONAPIonify::Api
   module Base::AppBuilder
 
+    def self.extended(klass)
+      klass.class_eval do
+        extend JSONAPIonify::InheritedAttributes
+        inherited_array_attribute :middleware
+      end
+    end
+
+    def use(*args, &block)
+      middleware << [args, block]
+    end
+
     def call(env)
       app.call(env)
     end
 
     private
@@ -13,19 +24,23 @@
         use Rack::ShowExceptions
         use Rack::CommonLogger
         use Base::Reloader unless ENV['RACK_ENV'] == 'production'
         map "/docs" do
           run ->(env) {
-            request    = JSONAPIonify::Api::Server::Request.new env
+            request = JSONAPIonify::Api::Server::Request.new env
             if request.path_info.present?
               return [301, { 'location' => request.path.chomp(request.path_info) }, []]
             end
-            response   = Rack::Response.new
+            response = Rack::Response.new
             response.write api.documentation_output(request)
             response.finish
           }
         end
         map "/" do
+          use Rack::MethodOverride
+          api.middleware.each do |args, block|
+            use *args, &block
+          end
           run JSONAPIonify::Api::Server.new(api)
         end
       end
     end
   end