module Restforce::Concerns::Streaming

def faye
  unless options[:instance_url]
    raise 'Instance URL missing. Call .authenticate! first.'
  end
  url = "#{options[:instance_url]}/cometd/#{options[:api_version]}"
  @faye ||= Faye::Client.new(url).tap do |client|
    client.set_header 'Authorization', "OAuth #{options[:oauth_token]}"
    client.bind 'transport:down' do
      Restforce.log "[COMETD DOWN]"
      client.set_header 'Authorization', "OAuth #{authenticate!.access_token}"
    end
    client.bind 'transport:up' do
      Restforce.log "[COMETD UP]"
    end
    client.add_extension ReplayExtension.new(replay_handlers)
  end
end

def legacy_subscribe(topics, options = {}, &block)
  topics = Array(topics).map { |channel| "/topic/#{channel}" }
  subscription(topics, options, &block)
end

def replay_handlers
  @_replay_handlers ||= {}
end

def subscription(channels, options = {}, &block)
  one_or_more_channels = Array(channels)
  one_or_more_channels.each do |channel|
    replay_handlers[channel] = options[:replay]
  end
  faye.subscribe(one_or_more_channels, &block)
end