Contents

# frozen_string_literal: true

require "serviceworker/handlers"

module ServiceWorker
  class Middleware
    REQUEST_METHOD = "REQUEST_METHOD"
    GET = "GET"
    HEAD = "HEAD"

    # Initialize the Rack middleware for responding to serviceworker asset
    # requests
    #
    # @app [#call] middleware stack
    # @opts [Hash] options to inject
    # @param opts [#match_route] :routes matches routes on PATH_INFO
    # @param opts [Hash] :headers default headers to use for matched routes
    # @param opts [#call] :handler resolves response from matched asset name
    # @param opts [#info] :logger logs requests
    def initialize(app, opts = {})
      @app = app
      @opts = opts
      @headers = default_headers.merge(opts.fetch(:headers, {}))
      @router = opts.fetch(:routes, ServiceWorker::Router.new)
      @handler = Handlers.build(@opts.fetch(:handler, nil))
    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_for_route_match(route_match).call(env)

      [status, headers.merge(@headers).merge(route_match.headers), body]
    end

    def handler_for_route_match(route_match)
      Handlers.handler_for_route_match(route_match) || @handler
    end

    def info(msg)
      logger.info "[#{self.class}] - #{msg}"
    end

    def logger
      @logger ||= @opts.fetch(:logger, Logger.new(STDOUT))
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
serviceworker-rails-0.6.0	lib/serviceworker/middleware.rb