require './auth'
class StreamingApi < DataSiftExample

  def initialize
    super
    @datasift = DataSift::Client.new(@config)
    run
  end

  def run
    begin
      rubyReceived   = 0
      pythonReceived = 0
      ruby           = 'interaction.content contains "ruby"'
      rubyStream     = @datasift.compile ruby

      python       = 'interaction.content contains "python"'
      pythonStream = @datasift.compile python

      on_delete = lambda { |stream, m| puts 'We must delete this to be compliant ==> ' + m }

      on_error = lambda do |stream, e|
        puts 'A serious error has occurred'
        puts e.message
      end

      on_message_ruby = lambda do |message, stream, hash|
        rubyReceived += 1
        puts "Ruby #{rubyReceived}, #{message}"

        if rubyReceived >= 10
          puts 'un-subscribing from ruby stream '+ hash
          stream.unsubscribe hash
        end
      end

      on_message_python = lambda do |message, stream, hash|
        pythonReceived += 1
        puts "python #{pythonReceived}, #{message}"

        if pythonReceived >= 10
          puts 'un-subscribing from python stream '+ hash
          stream.unsubscribe hash
        end
      end

      on_connect = lambda do |stream|
        #
        puts 'subscribing to python stream '+ pythonStream[:data][:hash]
        stream.subscribe(pythonStream[:data][:hash], on_message_python)
        puts 'Subscribed to '+ pythonStream[:data][:hash]
        sleep 1
        #
        puts 'subscribing to ruby stream '+ rubyStream[:data][:hash]
        stream.subscribe(rubyStream[:data][:hash], on_message_ruby)
        puts 'Subscribed to '+ rubyStream[:data][:hash]
      end

      on_close = lambda do |stream|
        puts 'closed'
      end

      on_datasift_message = lambda do |stream, message, hash|
        #not all messages have a hash
        puts "is_success =  #{message[:is_success]}, is_failure =  #{message[:is_failure]}, is_warning =  #{message[:is_warning]}, is_tick =  #{message[:is_tick]}"
        puts "DataSift Message #{hash} ==> #{message}"
      end

      conn                     = DataSift::new_stream(@config, on_delete, on_error, on_connect, on_close)
      conn.on_datasift_message = on_datasift_message
      #can do something else here now...
      puts 'Do some other business stuff...'
      conn.stream.read_thread.join
        #rescue DataSiftError
    rescue DataSiftError => dse
      puts "Error #{dse.message}"
      # Then match specific one to take action - All errors thrown by the client extend DataSiftError
      case dse
        when ConnectionError
          # some connection error
        when AuthError
        when BadRequestError
        else
          # do something else...
      end
    end
  end
end
StreamingApi.new