wlmonitor/Haltestelle.rb at master · Sevyls/wlmonitor · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
require 'zlib'

class Haltestelle
  attr_accessor :id, :lat, :lon, :typ, :diva, :name, :gemeinde, :gemeinde_id, :steige, :json, :url, :linien, :steige

  # csv format: "HALTESTELLEN_ID";"TYP";"DIVA";"NAME";"GEMEINDE";"GEMEINDE_ID";"WGS84_LAT";"WGS84_LON"
  def initialize(csv_row)
    @id   = csv_row.field("HALTESTELLEN_ID").to_i
    @lat  = csv_row.field("WGS84_LAT")
    @lon  = csv_row.field("WGS84_LON")
    @typ  = csv_row.field("TYP")
    @diva = csv_row.field("DIVA") # Haltestellennummer in der elektonischen Fahrplanauskunft
    @name = csv_row.field("NAME")
    @gemeinde = csv_row.field("GEMEINDE")
    @gemeinde_id = csv_row.field("GEMEINDE_ID")
    @steige = Set.new
    @linien = Set.new
  end

  def refresh_monitors
    unless @steige.empty?
      rbl_nrs = []

      @steige.each do |steig_id|
        s = App.data.steige[steig_id]

        # manche Steige haben keine RBL_NR im CSV...
        unless s.rbl_nr.empty?
          rbl_nrs << "rbl=#{s.rbl_nr}"
        else
          nil
        end
      end
      rbl_nrs = rbl_nrs.join '&'
      App.logger.debug "haltestelle id: #{@id}, rbl_nrs: '#{rbl_nrs}'"
      url = "http://www.wienerlinien.at/ogd_realtime/monitor?#{rbl_nrs}&sender=#{App.settings.wlsender}"
      App.logger.debug "rbl_nrs: #{rbl_nrs}"

      unless rbl_nrs.empty?
        App.logger.info "Sending GET request to #{url}"
        begin
          resp = Net::HTTP.get_response(URI.parse(url))
          if resp.code.eql? '200'
            App.logger.debug "HTTP 200 received"
            if resp.header[ 'Content-Encoding' ].eql?( 'gzip' ) then
              App.logger.debug "Performing gzip decompression for response body."
              sio = StringIO.new( resp.body )
              gz = Zlib::GzipReader.new( sio )
              data = gz.read()
              App.logger.debug "Finished decompressing gzipped response body."
            else
              App.logger.debug "Page is not compressed. Using text response body. "
              data = resp.body
            end

            App.logger.debug "Parse json monitor data"
            @json = JSON.parse(data)
            monitors = @json["data"]["monitors"]
            @steige.each do |steig_id|
              s = App.data.steige[steig_id]
              s.monitor = monitors.select do |monitor|
                monitor['locationStop']['properties']['attributes']['rbl'] == s.rbl_nr.to_i and
                not (monitor['lines'].select {|line| line['direction'] == s.richtung }).empty?
              end
            end
          else
            App.logger.error resp.code
          end
        rescue Zlib::BufError => e
          App.logger.error e
          App.logger.error "Could not load wlmonitor data for #{rbl_nrs}"
        end
      end
    else
      nil
    end
  end

  def to_json
    @steige_objs = []
    @steige.each do |steig_id|
      s = App.data.steige[steig_id]
      @steige_objs << s
    end

    { 'id' => @id,
      'name' => @name,
      'lat' => @lat,
      'lon' => @lon,
      'steige' => @steige_objs,
      'linien' => @linien
    }.to_json
  end
end