firehoseio/firehose

View on GitHub
lib/firehose/server/configuration.rb

Summary

Maintainability
A
0 mins
Test Coverage
A
100%
require "set"

module Firehose
  module Server
    # Configuration object for Firehose server.
    class Configuration
      attr_accessor :message_filter, :redis
      attr_reader :deprecated_channels

      def initialize
        @redis = Redis.new
        @message_filter = MessageFilter
        @deprecated_channels = Set.new
        @deprecated_channel_check = -> (_channel) { false }
      end

      def deprecate_channel(channel)
        unless channel_deprecated?(channel)
          Firehose.logger.info "Deprecated channel: #{channel}"
          @deprecated_channels << channel
        end
      end

      def undeprecate_channel(channel)
        if channel_deprecated?(channel)
          Firehose.logger.info "Undeprecated channel: #{channel}"
          @deprecated_channels.delete channel
        end
      end

      def deprecated_channels=(channels)
        @deprecated_channels = Set.new(channels)
      end

      def deprecated_channel(&block)
        @deprecated_channel_check = block
      end

      def channel_deprecated?(channel)
        @deprecated_channels.include?(channel) ||
          !!@deprecated_channel_check.call(channel)
      end
    end
  end
end