class AnyCable::Rails::ConnectionFactory

def call(socket, **options)
  connection_class = use_router? ? resolve_connection_class(socket.env) :
                                   ActionCable.server.config.connection_class.call
  AnyCable::Rails::Connection.new(connection_class, socket, **options)
end

def initialize(&block)
  @mappings = []
  @use_router = false
  instance_eval(&block) if block
end

def map(route, &block)
  raise ArgumentError, "Block is required" unless block
  @use_router = true
  mappings << [route, block]
end

def resolve_connection_class(env)
  path = env["PATH_INFO"]
  mappings.each do |(prefix, resolver)|
    return resolver.call if path.starts_with?(prefix)
  end
  raise "No connection class found matching #{path}"
end