lib/cotton_tail/middleware.rb in cotton-tail-0.3.0 vs lib/cotton_tail/middleware.rb in cotton-tail-0.4.0

@@ -6,10 +6,12 @@
 module CottonTail
   # Top level namespace for Middleware
   module Middleware
     autoload :Router, 'cotton_tail/middleware/router'
 
-    DEFAULT_STACK = ::Middleware::Builder.new do |b|
-      b.use Router
+    def self.default_stack(app)
+      ::Middleware::Builder.new do |b|
+        b.use Router, handlers: app.routes.handlers
+      end
     end
   end
 end