national-rail 0.3.0 → 0.3.1

data/lib/national-rail.rb

@@ -1,3 +1,4 @@
 require 'national-rail/journey_planner'
 require 'national-rail/live_departure_boards'
-require 'national-rail/station_list'
+require 'national-rail/station_list'
+require 'national-rail/virgin_live_departure_boards'

data/lib/national-rail/virgin_live_departure_boards.rb

@@ -0,0 +1,100 @@
+require 'mechanize'
+require 'hpricot'
+
+module NationalRail
+
+  class VirginLiveDepartureBoards
+
+    module CellParser
+      def cell_text(td)
+        td.inner_html.gsub(" ", "")
+      end
+    end
+
+    class SummaryRow
+
+      include CellParser
+
+      attr_reader :attributes
+
+      def initialize(agent, details_link, attributes)
+        @agent, @details_link, @attributes = agent, details_link, attributes
+      end
+
+      def details
+        @agent.transact do
+          page = @details_link.click
+          will_call_at = []
+          table = page.doc/"table[@summary='Will call at']"
+          if table.any?
+            (table/"tbody tr").each do |tr|
+              tds = tr/"td"
+              next unless tds.length == 3
+              will_call_at << {
+                :station => cell_text(tds[0]),
+                :timetabled_arrival => cell_text(tds[1]),
+                :expected_arrival => cell_text(tds[2])
+              }
+            end
+          end
+          previous_calling_points = []
+          table = page.doc/"table[@summary='Previous calling points']"
+          if table.any?
+            (table/"tbody tr").each do |tr|
+              tds = tr/"td"
+              next unless tds.length == 4
+              previous_calling_points << {
+                :station => cell_text(tds[0]),
+                :timetabled_departure => cell_text(tds[1]),
+                :expected_departure => cell_text(tds[2]),
+                :actual_departure => cell_text(tds[3])
+              }
+            end
+          end
+          { :will_call_at => will_call_at, :previous_calling_points => previous_calling_points }
+        end
+      end
+    end
+
+    class HpricotParser < Mechanize::Page
+      attr_reader :doc
+      def initialize(uri = nil, response = nil, body = nil, code = nil)
+        @doc = Hpricot(body)
+        super(uri, response, body, code)
+      end
+    end
+
+    include CellParser
+
+    def initialize
+      @agent = Mechanize.new
+      @agent.pluggable_parser.html = HpricotParser
+      @agent.user_agent_alias = "Mac FireFox"
+    end
+
+    def summary(station_code)
+      summary_rows = []
+      @agent.get("http://realtime.nationalrail.co.uk/virgintrains/summary.aspx?T=#{station_code}") do |page|
+        encoding = 'UTF-8'
+        summary_rows = (page.doc/"table#TrainTable tbody tr")[2..-1].map do |tr|
+          tds = tr/"td"
+          details_href = (tds[0]/"a").first["href"]
+          details_link = page.links.detect { |l| l.attributes["href"] == details_href }
+          SummaryRow.new(@agent, details_link, {
+            :from => (tds[0]/"a").inner_text,
+            :timetabled_arrival => cell_text(tds[1]),
+            :expected_arrival => cell_text(tds[2]),
+            :platform => cell_text(tds[3]),
+            :to => (tds[4]/"a").inner_text,
+            :timetabled_departure => cell_text(tds[5]),
+            :expected_departure => cell_text(tds[6]),
+            :details_url => "http://realtime.nationalrail.co.uk/virgintrains/#{details_href}"
+          })
+        end
+      end
+      summary_rows
+    end
+
+  end
+
+end

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 3
-  - 0
-  version: 0.3.0
+  - 1
+  version: 0.3.1
 platform: ruby
 authors: 
 - James Mead
@@ -14,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-11-04 00:00:00 +00:00
+date: 2010-11-05 00:00:00 +00:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -73,6 +73,7 @@ files:
 - lib/national-rail/live_departure_boards.rb
 - lib/national-rail/station_list.rb
 - lib/national-rail/stations.kml
+- lib/national-rail/virgin_live_departure_boards.rb
 - lib/national-rail.rb
 has_rdoc: true
 homepage: