chore: import bvg

This commit is contained in:
Kierán Meinhardt
2020-10-17 22:14:00 +02:00
15 changed files with 2829 additions and 0 deletions

50
bvg/bvg.rb Executable file
View File

@@ -0,0 +1,50 @@
#!/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