railtie.rb in serviceworker-rails-0.2.0

- old
+ new

@@ -1,21 +1,20 @@
 require "rails"
 require "rails/railtie"
+require "serviceworker"
 
 module ServiceWorker
   class Railtie < ::Rails::Railtie
+    config.serviceworker = ActiveSupport::OrderedOptions.new
+    config.serviceworker.headers = {}
+    config.serviceworker.routes = ServiceWorker::Router.new
+    config.serviceworker.handler = ServiceWorker::Rails::Handler.new
+
     initializer "serviceworker-rails.configure_rails_initialization" do
-      insert_middleware
-      ServiceWorker.logger = ::Rails.logger
-      ServiceWorker.root = ::Rails.root.to_s
-    end
+      config.serviceworker.logger ||= ::Rails.logger
+      config.serviceworker.routes.draw_default unless config.serviceworker.routes.any?
 
-    def insert_middleware
-      if defined? ::Rack::SendFile
-        app.middleware.insert_after ::Rack::Sendfile, ServiceWorker::Middleware
-      else
-        app.middleware.use ServiceWorker::Middleware
-      end
+      app.middleware.use ServiceWorker::Middleware, config.serviceworker
     end
 
     def app
       ::Rails.application
     end