# frozen_string_literal: true module ServiceWorker class Middleware REQUEST_METHOD = "REQUEST_METHOD".freeze GET = "GET".freeze HEAD = "HEAD".freeze def initialize(app, opts = {}) @app = app @opts = opts @headers = opts.fetch(:headers, {}).merge(default_headers) @router = opts.fetch(:routes, ServiceWorker::Router.new) end def call(env) case env[REQUEST_METHOD] when GET, HEAD route_match = @router.match_route(env) return respond_to_match(route_match, env) if route_match end @app.call(env) end private def default_headers { "Cache-Control" => "private, max-age=0, no-cache" } end def respond_to_match(route_match, env) env = env.merge("serviceworker.asset_name" => route_match.asset_name) status, headers, body = handler.call(env) [status, headers.merge(@headers).merge(route_match.headers), body] end def info(msg) logger.info "[#{self.class}] - #{msg}" end # TODO # provide non-rails handler as default def handler @handler ||= @opts.fetch(:handler) end def logger @logger ||= @opts.fetch(:logger, Logger.new(STDOUT)) end end end