firehoseio/firehose

module Firehose
  module Rack
    # Acts as the glue between the HTTP/WebSocket world and the Firehose::Server class,
    # which talks directly to the Redis server. Also dispatches between HTTP and WebSocket
    # transport handlers depending on the clients' request.
    class App
      def initialize
        yield self if block_given?
      end

      def call(env)
        # Cache the parsed request so we don't need to re-parse it when we pass
        # control onto another app.
        req     = env['parsed_request'] ||= ::Rack::Request.new(env)
        method  = req.request_method

        case method
        when 'PUT'
          if Server.configuration.channel_deprecated?(req.path)
            Firehose.logger.warn "Publishing to DEPRECATED Channel: #{req.path}"
          end

          # Firehose::Client::Publisher PUT's payloads to the server.
          publisher.call(env)
        when 'HEAD'
          # HEAD requests are used to prevent sockets from timing out
          # from inactivity
          ping.call(env)
        else
          case req.path
          when "/metrics@firehose"
            metrics_api.call(env)
          else
            # TODO - 'harden' this up with a GET request and throw a "Bad Request"
            # HTTP error code. I'd do it now but I'm in a plane and can't think of it.
            consumer.call(env)
          end
        end
      end

      # The consumer pulls messages off of the backend and passes messages to the
      # connected HTTP or WebSocket client. This can be configured from the initialization
      # method of the rack app.
      def consumer
        @consumer ||= Consumer.new
      end

      private
      def publisher
        @publisher ||= Publisher.new
      end

      def ping
        @ping ||= Ping.new
      end

      def metrics_api
        @metrics_api ||= MetricsAPI.new
      end
    end
  end
end