require 'buftok'
require 'http'
require 'json'
require 'twitter/error'

module Twitter
  module Streaming
    class Response
      # Initializes a new Response object
      #
      # @return [Twitter::Streaming::Response]
      def initialize(&block)
        @block     = block
        @parser    = Http::Parser.new(self)
        @tokenizer = BufferedTokenizer.new("\r\n")
      end

      def <<(data)
        @parser << data
      end

      def on_headers_complete(_headers)
        error = Twitter::Error::ERRORS[@parser.status_code]
        raise error if error
      end

      def on_body(data)
        @tokenizer.extract(data).each do |line|
          next if line.empty?
          @block.call(JSON.parse(line, symbolize_names: true))
        end
      end
    end
  end
end