lib/shogun/server.rb in shogun-2.0.0 vs lib/shogun/server.rb in shogun-2.1.0

@@ -1,25 +1,26 @@
 module Shogun
   module Server
     def serve
       @rack.use(Rack::Log, @logger)
       @rack.use(Rack::Runtime)
-      @rack.use(ActiveRecord::ConnectionAdapters::ConnectionManagement)
-      @rack.use(ActiveRecord::QueryCache)
-      @rack.use(Rack::Chunker)
-      @rack.use(Rack::AcceptSetter, accept_types)
-      @rack.use(Rack::ContentLengthSetter)
-      @rack.use(Rack::AuthenticationBearer, &bearer)
-      @rack.use(Rack::BodyDeserializer, deserialization_map)
-      @rack.use(Rack::BodySerializer, serialization_map, default_serializer)
       @rack.use(Rack::Protection::RemoteReferrer)
       @rack.use(Rack::Protection::HttpOrigin)
       @rack.use(Rack::Protection::EscapedParams)
       @rack.use(Rack::Protection::XSSHeader)
       @rack.use(Rack::Protection::FrameOptions)
       @rack.use(Rack::Protection::PathTraversal)
       @rack.use(Rack::Protection::IPSpoofing)
+      @rack.use(Rack::Chunker)
+      @rack.use(Rack::AcceptSetter, accept_types)
+      @rack.use(Rack::ContentLengthSetter)
+      @rack.use(Rack::AuthenticationBearer, &bearer)
+      @rack.use(Rack::BodyDeserializer, deserialization_map)
+      @rack.use(Rack::BodySerializer, serialization_map, default_serializer)
+      @rack.use(ActiveRecord::ConnectionAdapters::ConnectionManagement)
+      @rack.use(ActiveRecord::QueryCache)
+      @rack.use(Rack::Cors, &cors)
       @rack.run(Shogun::Dispatch.new(logger: @logger, &endpoints))
     end
 
     private def endpoints
       self.class.const_get("ENDPOINTS")
@@ -41,8 +42,12 @@
       self.class.const_get("DEFAULT_SERIALIZER")
     end
 
     private def accept_types
       self.class.const_get("ACCEPT_TYPES")
+    end
+
+    private def cors
+      self.class.const_get("CORS")
     end
   end
 end