51 lines
1.4 KiB
Ruby
51 lines
1.4 KiB
Ruby
|
#!/usr/bin/env ruby
|
||
|
|
|
||
|
|
require "open-uri"
|
||
|
|
require "nokogiri"
|
||
|
|
require "rainbow"
|
||
|
|
|
||
|
|
Departure = Struct.new(:time, :line, :terminus) do
|
||
|
|
def self.from_row(row)
|
||
|
|
Departure.new(row[0].text.strip, row[1].text.strip.gsub(/\s+/, " "), row[2].text.strip)
|
||
|
|
end
|
||
|
|
|
||
|
|
def log
|
||
|
|
time_string = if time[-1] == "*" then Rainbow(time[0..4]).inverse else time end
|
||
|
|
line_string = {
|
||
|
|
'B' => Rainbow(line).magenta,
|
||
|
|
'R' => Rainbow(line).white,
|
||
|
|
'U' => Rainbow(line).blue,
|
||
|
|
'S' => Rainbow(line).green,
|
||
|
|
'T' => Rainbow(line).red,
|
||
|
|
}[line[0]] || line
|
||
|
|
|
||
|
|
printf("%s %-20s %s\n", time_string, line_string, terminus)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def request_url(input)
|
||
|
|
"http://mobil.bvg.de/Fahrinfo/bin/stboard.bin/dox?" + URI.encode_www_form(
|
||
|
|
"input" => URI.encode_www_form_component(input),
|
||
|
|
"start" => "yes",
|
||
|
|
"boardType" => "depRT",
|
||
|
|
)
|
||
|
|
end
|
||
|
|
|
||
|
|
url = request_url(ARGV[0])
|
||
|
|
|
||
|
|
document = Nokogiri::HTML(open(url))
|
||
|
|
|
||
|
|
title = document.xpath("//div[@id=\"ivu_overview_input\"]/strong")
|
||
|
|
puts Rainbow(title.text).bright if title
|
||
|
|
|
||
|
|
document.xpath("//span[@class=\"select\"]/a").each do |option|
|
||
|
|
station_name = option.text.strip
|
||
|
|
station_code = option.xpath("@href").text[/input=(\d+)/, 1]
|
||
|
|
print station_code, " ", Rainbow(station_name).yellow, "\n"
|
||
|
|
end
|
||
|
|
|
||
|
|
document.xpath("//table/tbody/tr").each do |row|
|
||
|
|
columns = row.xpath("td")
|
||
|
|
Departure.from_row(columns).log unless columns.length < 3
|
||
|
|
end
|