Class: Pupil::Stream

9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

# File 'lib/pupil/stream/base.rb', line 9

def initialize key
  @screen_name = key[:screen_name]
  @client = nil
  @config = nil

  @consumer = OAuth::Consumer.new(
  key[:consumer_key],
  key[:consumer_secret],
  :site => TWITTER_API_URL
  )
  @access_token = OAuth::AccessToken.new(
  @consumer,
  key[:access_token],
  key[:access_token_secret]
  )
end

32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

# File 'lib/pupil/stream/base.rb', line 32

def run_get_stream(type, param=nil, &block)
  uri = URI.parse(STREAM_APIS[type] % Pupil.param_serializer(param))
  https = Net::HTTP.new(uri.host, uri.port)
  https.use_ssl = true
  https.verify_mode = OpenSSL::SSL::VERIFY_NONE
  https.verify_depth = 5

  while true do
    begin
      https.start do |https|
        request = Net::HTTP::Get.new(uri.request_uri)
        request["User-Agent"] = "Ruby/#{RUBY_VERSION} Pupil::Stream"
        request.oauth!(https, @consumer, @access_token)
        buf = ""
        https.request(request) do |response|
          response.read_body do |chunk|
            buf << chunk
            while (line = buf[/.+?(\r\n)+/m]) != nil
              begin
                buf.sub!(line,"")
                line.strip!
                status = JSON.parse(line)
              rescue
                break
              end
              
              block.call StreamEvent.new(status)
            end
          end
        end
      end
    rescue => vars
      raise StreamError, "StreamError: #{vars}"
    end
  end
end