lib/wayfarer/handler.rb in wayfarer-0.4.6 vs lib/wayfarer/handler.rb in wayfarer-0.4.7

@@ -1,15 +1,23 @@
 # frozen_string_literal: true
 
 module Wayfarer
-  class Handler
-    include Wayfarer::Middleware::Controller
+  module Handler
+    extend ActiveSupport::Concern
 
-    api Wayfarer::Middleware::Fetch
-    api Wayfarer::Middleware::Stage
+    included do
+      include Wayfarer::Middleware::Controller
 
-    use Wayfarer::Middleware::Router
-    use Wayfarer::Middleware::Dispatch
+      use Wayfarer::Middleware::ContentType
+      use Wayfarer::Middleware::Router
+      use Wayfarer::Middleware::Dispatch
 
-    singleton_class.undef_method :after_batch
+      api Wayfarer::Middleware::UserAgent
+      api Wayfarer::Middleware::Stage
+
+      singleton_class.undef_method :before_fetch
+      singleton_class.undef_method :around_fetch
+      singleton_class.undef_method :after_fetch
+      singleton_class.undef_method :after_batch
+    end
   end
 end