caltrain 0.0.4

data/data/google_transit/stops.txt

@@ -0,0 +1,32 @@
+stop_id,stop_name,stop_desc,stop_lat,stop_lon,zone_id,stop_url
+"22nd Street Caltrain","22nd Street Caltrain","1149 22nd Street,San Francisco","37.757674","-122.392636","1",
+"Atherton Caltrain","Atherton Caltrain","1 Dinkelspiel Station Lane, Atherton","37.464349","-122.198106","3",""
+"Bayshore Caltrain","Bayshore Caltrain","400 Tunnel Avenue,San Francisco","37.709544","-122.401318","1",
+"Belmont Caltrain","Belmont Caltrain","995 El Camino Real, Belmont","37.520504","-122.276075","2",
+"Blossom Hill Caltrain","Blossom Hill Caltrain","5560 Monterey Hwy.,San Jose","37.252801","-121.797369","5",
+"Broadway Caltrain","Broadway Caltrain","1190 California Drive, Burlingame","37.587466","-122.363233","2",""
+"Burlingame Caltrain","Burlingame Caltrain","290 California Drive, Burlingame","37.579719","-122.345266","2",
+"California Ave Caltrain","California Ave Caltrain","101 California Avenue,Palo Alto","37.428835","-122.142703","3",
+"Capitol Caltrain","Capitol Caltrain","3400 Monterey Hwy.,San Jose","37.284359","-121.841589","5",
+"College Park Caltrain","College Park Caltrain","780 Stockton Avenue,San Jose","37.34217","-121.914998","4",
+"Gilroy Caltrain","Gilroy Caltrain","7150 Monterey Street, Gilroy","37.003084","-121.567091","6",
+"Hayward Park Caltrain","Hayward Park Caltrain","401 Concar Drive,San Mateo","37.552346","-122.308916","2",
+"Hillsdale Caltrain","Hillsdale Caltrain","3333 El Camino Real,San Mateo","37.537503","-122.298001","2",
+"Lawrence Caltrain","Lawrence Caltrain","137 San Zeno Way, Sunnyvale","37.371578","-121.996982","4",
+"Menlo Park Caltrain","Menlo Park Caltrain","1120 Merrill Street,Menlo Park","37.454604","-122.182518","3",
+"Millbrae Caltrain","Millbrae Caltrain","100 California Drive, Millbrae","37.600006","-122.386534","2",
+"Morgan Hill Caltrain","Morgan Hill Caltrain","17300 Depot Street,Morgan Hill","37.129081","-121.650721","6",
+"Mountain View Caltrain","Mountain View Caltrain","600 W. Evelyn Avenue, Mountain View","37.393879","-122.076327","3",
+"Palo Alto Caltrain","Palo Alto Caltrain","95 University Avenue,Palo Alto","37.44307","-122.1649","3",
+"Redwood City Caltrain","Redwood City Caltrain","1 James Avenue,Redwood City","37.485412","-122.231957","2",
+"San Antonio Caltrain","San Antonio Caltrain","190 Showers Drive, Mountain View","37.406163","-122.105787","3",
+"San Bruno Caltrain","San Bruno Caltrain","481 Huntington Avenue,San Bruno","37.624256","-122.408444","1",
+"San Carlos Caltrain","San Carlos Caltrain","599 El Camino Real,San Carlos","37.507361","-122.260365","2",
+"San Francisco Caltrain","San Francisco Caltrain","700 4th Street,San Francisco","37.7764393371","-122.394322993","1",
+"San Jose Caltrain","San Jose Caltrain","65 Cahill Street,San Jose","37.330196","-121.901985","4",
+"San Martin Caltrain","San Martin Caltrain","13400 Monterey Hwy.,San Martin","37.085775","-121.610809","6",
+"San Mateo Caltrain","San Mateo Caltrain","385 First Ave.,San Mateo","37.568209","-122.323933","2",
+"Santa Clara Caltrain","Santa Clara Caltrain","1001 Railroad Avenue,Santa Clara","37.352915","-121.936376","4",
+"So. San Francisco Caltrain","So. San Francisco Caltrain","590 Dubuque Avenue,South San Francisco","37.657013","-122.405516","1",
+"Sunnyvale Caltrain","Sunnyvale Caltrain","121 W. Evelyn Avenue, Sunnyvale","37.377589","-122.030116","3",
+"Tamien Caltrain","Tamien Caltrain","1355 Lick Avenue,San Jose","37.31168","-121.882087","4",

data/data/google_transit/trips.txt

@@ -0,0 +1,131 @@
+trip_id,route_id,service_id,trip_headsign,direction_id,block_id,shape_id
+"101_20110701","ct_local_20110701","WD_20110701","San Francisco (Train 101)","0",,"cal_sj_sf"
+"102_20110701","ct_local_20110701","WD_20110701","San Jose (Train 102)","1",,"cal_sf_sj"
+"103_20110701","ct_local_20110701","WD_20110701","San Francisco (Train 103)","0",,"cal_tam_sf"
+"104_20110701","ct_local_20110701","WD_20110701","Tamien (Train 104)","1",,"cal_sf_tam"
+"134_20110701","ct_local_20110701","WD_20110701","San Jose (Train 134)","1",,"cal_sf_sj"
+"135_20110701","ct_local_20110701","WD_20110701","San Francisco (Train 135)","0",,"cal_sj_sf"
+"138_20110701","ct_local_20110701","WD_20110701","San Jose (Train 138)","1",,"cal_sf_sj"
+"139_20110701","ct_local_20110701","WD_20110701","San Francisco (Train 139)","0",,"cal_sj_sf"
+"142_20110701","ct_local_20110701","WD_20110701","San Jose (Train 142)","1",,"cal_sf_sj"
+"143_20110701","ct_local_20110701","WD_20110701","San Francisco (Train 143)","0",,"cal_sj_sf"
+"146_20110701","ct_local_20110701","WD_20110701","San Jose (Train 146)","1",,"cal_sf_sj"
+"147_20110701","ct_local_20110701","WD_20110701","San Francisco (Train 147)","0",,"cal_sj_sf"
+"150_20110701","ct_local_20110701","WD_20110701","San Jose (Train 150)","1",,"cal_sf_sj"
+"151_20110701","ct_local_20110701","WD_20110701","San Francisco (Train 151)","0",,"cal_sj_sf"
+"154_20110701","ct_local_20110701","WD_20110701","San Jose (Train 154)","1",,"cal_sf_sj"
+"155_20110701","ct_local_20110701","WD_20110701","San Francisco (Train 155)","0",,"cal_sj_sf"
+"158_20110701","ct_local_20110701","WD_20110701","Gilroy (Train 158)","1",,"cal_sf_gil"
+"159_20110701","ct_local_20110701","WD_20110701","San Francisco (Train 159)","0",,"cal_sj_sf"
+"189_20110701","ct_local_20110701","WD_20110701","San Francisco (Train 189)","0",,"cal_sj_sf"
+"190_20110701","ct_local_20110701","WD_20110701","San Jose (Train 190)","1",,"cal_sf_sj"
+"191_20110701","ct_local_20110701","WD_20110701","San Francisco (Train 191)","0",,"cal_sj_sf"
+"192_20110701","ct_local_20110701","WD_20110701","Tamien (Train 192)","1",,"cal_sf_tam"
+"193_20110701","ct_local_20110701","WD_20110701","San Francisco (Train 193)","0",,"cal_tam_sf"
+"194_20110701","ct_local_20110701","WD_20110701","Tamien (Train 194)","1",,"cal_sf_tam"
+"195_20110701","ct_local_20110701","WD_20110701","San Francisco (Train 195)","0",,"cal_tam_sf"
+"196_20110701","ct_local_20110701","WD_20110701","San Jose (Train 196)","1",,"cal_sf_sj"
+"197_20110701","ct_local_20110701","WD_20110701","San Francisco (Train 197)","0",,"cal_sj_sf"
+"198_20110701","ct_local_20110701","WD_20110701","San Jose (Train 198)","1",,"cal_sf_sj"
+"206_20110701","ct_limited_20110701","WD_20110701","San Jose (Train 206)","1",,"cal_sf_sj"
+"207_20110701","ct_limited_20110701","WD_20110701","San Francisco (Train 207)","0",,"cal_tam_sf"
+"208_20110701","ct_limited_20110701","WD_20110701","Tamien (Train 208)","1",,"cal_sf_tam"
+"210_20110701","ct_limited_20110701","WD_20110701","Tamien (Train 210)","1",,"cal_sf_tam"
+"211_20110701","ct_limited_20110701","WD_20110701","San Francisco (Train 211)","0",,"cal_sj_sf"
+"215_20110701","ct_limited_20110701","WD_20110701","San Francisco (Train 215)","0",,"cal_sj_sf"
+"216_20110701","ct_limited_20110701","WD_20110701","San Jose (Train 216)","1",,"cal_sf_sj"
+"217_20110701","ct_limited_20110701","WD_20110701","San Francisco (Train 217)","0",,"cal_gil_sf"
+"218_20110701","ct_limited_20110701","WD_20110701","Tamien (Train 218)","1",,"cal_sf_tam"
+"220_20110701","ct_limited_20110701","WD_20110701","Tamien (Train 220)","1",,"cal_sf_tam"
+"221_20110701","ct_limited_20110701","WD_20110701","San Francisco (Train 221)","0",,"cal_gil_sf"
+"225_20110701","ct_limited_20110701","WD_20110701","San Francisco (Train 225)","0",,"cal_sj_sf"
+"226_20110701","ct_limited_20110701","WD_20110701","San Jose (Train 226)","1",,"cal_sf_sj"
+"227_20110701","ct_limited_20110701","WD_20110701","San Francisco (Train 227)","0",,"cal_gil_sf"
+"228_20110701","ct_limited_20110701","WD_20110701","Tamien (Train 228)","1",,"cal_sf_tam"
+"230_20110701","ct_limited_20110701","WD_20110701","Tamien (Train 230)","1",,"cal_sf_tam"
+"231_20110701","ct_limited_20110701","WD_20110701","San Francisco (Train 231)","0",,"cal_sj_sf"
+"233_20110701","ct_limited_20110701","WD_20110701","San Francisco (Train 233)","0",,"cal_tam_sf"
+"260_20110701","ct_limited_20110701","WD_20110701","Tamien (Train 260)","1",,"cal_sf_tam"
+"261_20110701","ct_limited_20110701","WD_20110701","San Francisco (Train 261)","0",,"cal_tam_sf"
+"263_20110701","ct_limited_20110701","WD_20110701","San Francisco (Train 263)","0",,"cal_tam_sf"
+"264_20110701","ct_limited_20110701","WD_20110701","San Jose (Train 264)","1",,"cal_sf_sj"
+"266_20110701","ct_limited_20110701","WD_20110701","San Jose (Train 266)","1",,"cal_sf_sj"
+"267_20110701","ct_limited_20110701","WD_20110701","San Francisco (Train 267)","0",,"cal_tam_sf"
+"270_20110701","ct_limited_20110701","WD_20110701","Gilroy (Train 270)","1",,"cal_sf_gil"
+"271_20110701","ct_limited_20110701","WD_20110701","San Francisco (Train 271)","0",,"cal_tam_sf"
+"274_20110701","ct_limited_20110701","WD_20110701","San Jose (Train 274)","1",,"cal_sf_sj"
+"275_20110701","ct_limited_20110701","WD_20110701","San Francisco (Train 275)","0",,"cal_sj_sf"
+"276_20110701","ct_limited_20110701","WD_20110701","Gilroy (Train 276)","1",,"cal_sf_gil"
+"277_20110701","ct_limited_20110701","WD_20110701","San Francisco (Train 277)","0",,"cal_tam_sf"
+"280_20110701","ct_limited_20110701","WD_20110701","Tamien (Train 280)","1",,"cal_sf_tam"
+"281_20110701","ct_limited_20110701","WD_20110701","San Francisco (Train 281)","0",,"cal_tam_sf"
+"284_20110701","ct_limited_20110701","WD_20110701","San Jose (Train 284)","1",,"cal_sf_sj"
+"285_20110701","ct_limited_20110701","WD_20110701","San Francisco (Train 285)","0",,"cal_tam_sf"
+"287_20110701","ct_limited_20110701","WD_20110701","San Francisco (Train 287)","0",,"cal_sj_sf"
+"288_20110701","ct_limited_20110701","WD_20110701","Tamien (Train 288)","1",,"cal_sf_tam"
+"305_20110701","ct_bullet_20110701","WD_20110701","San Francisco (Train 305)","0",,"cal_sj_sf"
+"309_20110701","ct_bullet_20110701","WD_20110701","San Francisco (Train 309)","0",,"cal_tam_sf"
+"312_20110701","ct_bullet_20110701","WD_20110701","San Jose (Train 312)","1",,"cal_sf_sj"
+"313_20110701","ct_bullet_20110701","WD_20110701","San Francisco (Train 313)","0",,"cal_sj_sf"
+"314_20110701","ct_bullet_20110701","WD_20110701","San Jose (Train 314)","1",,"cal_sf_sj"
+"319_20110701","ct_bullet_20110701","WD_20110701","San Francisco (Train 319)","0",,"cal_tam_sf"
+"322_20110701","ct_bullet_20110701","WD_20110701","San Jose (Train 322)","1",,"cal_sf_sj"
+"323_20110701","ct_bullet_20110701","WD_20110701","San Francisco (Train 323)","0",,"cal_sj_sf"
+"324_20110701","ct_bullet_20110701","WD_20110701","San Jose (Train 324)","1",,"cal_sf_sj"
+"329_20110701","ct_bullet_20110701","WD_20110701","San Francisco (Train 329)","0",,"cal_tam_sf"
+"332_20110701","ct_bullet_20110701","WD_20110701","San Jose (Train 332)","1",,"cal_sf_sj"
+"362_20110701","ct_bullet_20110701","WD_20110701","San Jose (Train 362)","1",,"cal_sf_sj"
+"365_20110701","ct_bullet_20110701","WD_20110701","San Francisco (Train 365)","0",,"cal_sj_sf"
+"368_20110701","ct_bullet_20110701","WD_20110701","Tamien (Train 368)","1",,"cal_sf_tam"
+"369_20110701","ct_bullet_20110701","WD_20110701","San Francisco (Train 369)","0",,"cal_sj_sf"
+"372_20110701","ct_bullet_20110701","WD_20110701","San Jose (Train 372)","1",,"cal_sf_sj"
+"373_20110701","ct_bullet_20110701","WD_20110701","San Francisco (Train 373)","0",,"cal_sj_sf"
+"378_20110701","ct_bullet_20110701","WD_20110701","Tamien (Train 378)","1",,"cal_sf_tam"
+"379_20110701","ct_bullet_20110701","WD_20110701","San Francisco (Train 379)","0",,"cal_sj_sf"
+"382_20110701","ct_bullet_20110701","WD_20110701","San Jose (Train 382)","1",,"cal_sf_sj"
+"383_20110701","ct_bullet_20110701","WD_20110701","San Francisco (Train 383)","0",,"cal_sj_sf"
+"386_20110701","ct_bullet_20110701","WD_20110701","Tamien (Train 386)","1",,"cal_sf_tam"
+"421_20110701","ct_local_20110701","ST_20110701","San Francisco (Train 421)","0",,"cal_sj_sf"
+"422_20110701","ct_local_20110701","WE_20110701","San Jose  (Train 422)","1",,"cal_sf_sj"
+"423_20110701","ct_local_20110701","WE_20110701","San Francisco (Train 423)","0",,"cal_sj_sf"
+"424_20110701","ct_local_20110701","WE_20110701","San Jose  (Train 424)","1",,"cal_sf_sj"
+"425_20110701","ct_local_20110701","WE_20110701","San Francisco (Train 425)","0",,"cal_sj_sf"
+"426_20110701","ct_local_20110701","WE_20110701","San Jose  (Train 426)","1",,"cal_sf_sj"
+"427_20110701","ct_local_20110701","WE_20110701","San Francisco (Train 427)","0",,"cal_sj_sf"
+"428_20110701","ct_local_20110701","WE_20110701","San Jose  (Train 428)","1",,"cal_sf_sj"
+"429_20110701","ct_local_20110701","WE_20110701","San Francisco (Train 429)","0",,"cal_sj_sf"
+"430_20110701","ct_local_20110701","WE_20110701","San Jose  (Train 430)","1",,"cal_sf_sj"
+"431_20110701","ct_local_20110701","WE_20110701","San Francisco (Train 431)","0",,"cal_sj_sf"
+"432_20110701","ct_local_20110701","WE_20110701","San Jose  (Train 432)","1",,"cal_sf_sj"
+"433_20110701","ct_local_20110701","WE_20110701","San Francisco (Train 433)","0",,"cal_sj_sf"
+"434_20110701","ct_local_20110701","WE_20110701","San Jose  (Train 434)","1",,"cal_sf_sj"
+"435_20110701","ct_local_20110701","WE_20110701","San Francisco (Train 435)","0",,"cal_sj_sf"
+"436_20110701","ct_local_20110701","WE_20110701","San Jose  (Train 436)","1",,"cal_sf_sj"
+"437_20110701","ct_local_20110701","WE_20110701","San Francisco (Train 437)","0",,"cal_sj_sf"
+"438_20110701","ct_local_20110701","WE_20110701","San Jose  (Train 438)","1",,"cal_sf_sj"
+"439_20110701","ct_local_20110701","WE_20110701","San Francisco (Train 439)","0",,"cal_sj_sf"
+"440_20110701","ct_local_20110701","WE_20110701","San Jose  (Train 440)","1",,"cal_sf_sj"
+"441_20110701","ct_local_20110701","WE_20110701","San Francisco (Train 441)","0",,"cal_sj_sf"
+"442_20110701","ct_local_20110701","WE_20110701","San Jose  (Train 442)","1",,"cal_sf_sj"
+"443_20110701","ct_local_20110701","WE_20110701","San Francisco (Train 443)","0",,"cal_sj_sf"
+"444_20110701","ct_local_20110701","WE_20110701","San Jose  (Train 444)","1",,"cal_sf_sj"
+"445_20110701","ct_local_20110701","WE_20110701","San Francisco (Train 445)","0",,"cal_sj_sf"
+"446_20110701","ct_local_20110701","WE_20110701","San Jose  (Train 446)","1",,"cal_sf_sj"
+"447_20110701","ct_local_20110701","WE_20110701","San Francisco (Train 447)","0",,"cal_sj_sf"
+"448_20110701","ct_local_20110701","WE_20110701","San Jose  (Train 448)","1",,"cal_sf_sj"
+"449_20110701","ct_local_20110701","WE_20110701","San Francisco (Train 449)","0",,"cal_sj_sf"
+"450_20110701","ct_local_20110701","ST_20110701","San Jose  (Train 450)","1",,"cal_sf_sj"
+"451_20110701","ct_local_20110701","ST_20110701","San Francisco (Train 451)","0",,"cal_sj_sf"
+"454_20110701","ct_local_20110701","ST_20110701","San Jose  (Train 454)","1",,"cal_sf_sj"
+"801_20110701","ct_bullet_20110701","WE_20110701","San Francisco (Train 801)","0",,"cal_sj_sf"
+"802_20110701","ct_bullet_20110701","WE_20110701","San Jose (Train 802)","1",,"cal_sf_sj"
+"803_20110701","ct_bullet_20110701","WE_20110701","San Francisco (Train 803)","0",,"cal_sj_sf"
+"804_20110701","ct_bullet_20110701","WE_20110701","San Jose (Train 804)","1",,"cal_sf_sj"
+"219_20120220","ct_bullet_20120220","ST_20120220","San Francisco (Train 219)","0","","cal_sj_sf"
+"501_20120220","ct_bullet_20120220","ST_20120220","San Francisco (Train 207)","0","","cal_sj_sf"
+"502_20120220","ct_bullet_20120220","ST_20120220","San Jose (Train 502)","1","","cal_sf_sj"
+"503_20120220","ct_bullet_20120220","ST_20120220","San Francisco (Train 503)","0",,"cal_gil_sf"
+"504_20120220","ct_bullet_20120220","ST_20120220","San Jose (Train 504)","1","","cal_sf_gil"
+"505_20120220","ct_bullet_20120220","ST_20120220","San Francisco (Train 505)","0",,"cal_gil_sf"
+"506_20120220","ct_bullet_20120220","ST_20120220","San Jose (Train 506)","1","","cal_sf_sj"
+"508_20120220","ct_bullet_20120220","ST_20120220","San Jose (Train 508)","1","","cal_sf_sj"

data/lib/caltrain.rb

@@ -0,0 +1,125 @@
+class Caltrain
+  TIMES_PATH = File.expand_path('../../data/google_transit/stop_times.txt', __FILE__)
+  TRIPS_PATH = File.expand_path('../../data/google_transit/trips.txt', __FILE__)
+
+  class << self
+    def upcoming_departures(loc, dir)
+      times(loc, dir).select { |time| time > now }.sort
+    end
+
+    def next_departure(loc, dir)
+      times(loc, dir).find { |time| time > now }
+    end
+
+    def times(loc, dir)
+      @times ||= times_for_location(loc).select { |line| trip_ids_for_direction(dir).include?(line[0]) }.map {|i| i[1]}.sort
+    end
+
+    def trip_ids_for_direction(dir)
+      @trip_ids_for_direction ||= trips_for_time_of_week.select {|trip| trip[4] == (dir == :north ? '0' : '1') }.map(&:first)
+    end
+
+    def trips_for_time_of_week
+      @trips_for_time_of_week ||= all_trips.select { |trip| trip[2] =~ (weekend? ? /^WE/ : /^WD/) }
+    end
+
+    def times_for_location(loc)
+      @times_for_location ||= all_times.select { |arr| arr[3] =~ /^#{abbrevs[loc]}/ }
+    end
+
+    def all_times
+      @all_times ||= File.read(TIMES_PATH).split(/[\n\r]+/)[1..-1].map { |line| line.gsub('"', '').split(/,+/) }
+    end
+
+    def all_trips
+      @all_trips ||= File.read(TRIPS_PATH).split(/[\n\r]+/)[1..-1].map { |line| line.gsub('"', '').split(/,+/) }.sort
+    end
+
+    def weekend?
+      @weekend ||= (Time.now.saturday? || Time.now.sunday?)
+    end
+
+    def now
+      @now ||= Time.now.strftime('%H:%M:%S')
+    end
+
+    def abbrevs
+      {
+        :tt  => "22nd Street",
+        :ath => "Atherton",
+        :bsh => "Bayshore",
+        :bel => "Belmont",
+        :bhl => "Blossom Hill",
+        :bdw => "Broadway",
+        :brl => "Burlingame",
+        :cal => "California Ave",
+        :cap => "Capitol",
+        :clp => "College Park",
+        :gil => "Gilroy",
+        :hay => "Hayward Park",
+        :hil => "Hillsdale",
+        :law => "Lawrence",
+        :men => "Menlo Park",
+        :mil => "Millbrae",
+        :mrg => "Morgan Hill",
+        :mv  => "Mountain View",
+        :pa  => "Palo Alto",
+        :rc  => "Redwood City",
+        :sa  => "San Antonio",
+        :sb  => "San Bruno",
+        :scl => "San Carlos",
+        :sf  => "San Francisco",
+        :sj  => "San Jose",
+        :smt => "San Martin",
+        :sm  => "San Mateo",
+        :sc  => "Santa Clara",
+        :ssf => "So. San Francisco",
+        :sv  => "Sunnyvale",
+        :tam => "Tamien"
+      }
+    end
+
+    def pretty_abbrevs
+      max_field_width = abbrevs.values.map(&:size).max
+
+      abbrevs.values.zip(abbrevs.keys).each do |name, abr|
+        puts "%#{max_field_width}s #{abr}" % name
+      end
+    end
+
+    def usage
+      puts "Usage:"
+      puts "  caltrain <action> <location> <direction>"
+      puts "  action => [ next | list ]"
+      puts ""
+      puts "Abbreviations:"
+      pretty_abbrevs
+      exit(1)
+    end
+
+    def actions
+      { :list => :upcoming_departures, :next => :next_departure }
+    end
+
+    def clean!
+      instance_variables.each { |i| instance_variable_set(i, nil) }
+      true
+    end
+
+    def run!(args)
+      begin
+        act, loc, dir = *args
+
+        if dir =~ /^n/
+          puts method(actions[act.to_sym]).call(loc.to_sym, :north)
+        elsif dir =~ /^s/
+          puts method(actions[act.to_sym]).call(loc.to_sym, :south)
+        else
+          raise
+        end
+      rescue
+        usage
+      end
+    end
+  end
+end

data/spec/caltrain_spec.rb

@@ -0,0 +1,95 @@
+require 'minitest/autorun'
+require 'minitest/pride'
+require 'mocha'
+
+require File.expand_path('../lib/caltrain', File.dirname(__FILE__))
+
+describe Caltrain do
+  before { Caltrain.stubs(:weekend?).returns(false) }
+
+  after { Caltrain.clean! }
+
+  describe 'trips' do
+    it 'should parse trips.txt depending on the day of the week' do
+      Caltrain.all_trips.first.must_equal(['101_20110701','ct_local_20110701','WD_20110701','San Francisco (Train 101)','0','cal_sj_sf'])
+    end
+  end
+
+  describe 'times' do
+    it 'should parse stop_times.txt correctly' do
+      line = ['280_20110701','18:50:00','18:50:00','Mountain View Caltrain','10']
+      Caltrain.all_times.must_include(line)
+    end
+
+    it 'should return times for a location' do
+      ['16:58:00', '08:13:00', '19:44:00'].each do |time|
+        Caltrain.times_for_location(:sv).map {|i| i[1]}.must_include(time)
+      end
+    end
+  end
+
+  describe '#next_departure' do
+    it 'should return the next departure for north' do
+      Caltrain.stubs(:now).returns('08:05:12')
+      Caltrain.next_departure(:sv, :north).must_equal('08:13:00')
+    end
+
+    it 'should return the next departure for south' do
+      Caltrain.stubs(:now).returns('18:50:00')
+      Caltrain.next_departure(:sf, :south).must_equal('18:56:00')
+    end
+  end
+
+  describe '#upcoming_departures' do
+    it 'should return the upcoming trips for north' do
+      Caltrain.stubs(:now).returns('08:05:12')
+      upcoming1 = Caltrain.upcoming_departures(:sv, :north)
+      upcoming1.must_include('08:18:00')
+      upcoming1.wont_include('09:55:00')
+      upcoming1.wont_include('09:14:00')
+    end
+
+    it 'should return upcoming trips for south' do
+      Caltrain.stubs(:now).returns('18:30:10')
+      upcoming2 = Caltrain.upcoming_departures(:sf, :south)
+      upcoming2.must_include('18:33:00')
+      upcoming2.wont_include('18:44:00')
+      upcoming2.wont_include('18:59:00')
+    end
+  end
+
+  describe 'weekend versus weekday' do
+    it 'should pick weekday times' do
+      Caltrain.stubs(:now).returns('09:10:10')
+
+      Caltrain.next_departure(:mv, :north).must_equal('09:29:00')
+
+      upcoming = Caltrain.upcoming_departures(:mv, :north)
+      upcoming.must_include('16:37:00')
+      upcoming.wont_include('09:19:00')
+    end
+
+    it 'should pick weekend times' do
+      Caltrain.stubs(:weekend?).returns(true)
+      Caltrain.stubs(:now).returns('07:46:15')
+
+      Caltrain.next_departure(:sj, :north).must_equal('08:00:00')
+
+      upcoming = Caltrain.upcoming_departures(:sj, :south)
+      upcoming.must_include('12:00:00')
+      upcoming.wont_include('07:50:00')
+    end
+  end
+
+  describe 'excluding last stops' do
+    it 'should not include times for last sf stop' do
+      Caltrain.stubs(:now).returns('18:58:30')
+      Caltrain.next_departure(:sf, :south).must_equal('19:30:00')
+    end
+
+    it 'should not include times for last sj stop' do
+      Caltrain.stubs(:now).returns('08:12:10')
+      Caltrain.next_departure(:sj, :north).must_equal('08:22:00')
+    end
+  end
+end

metadata

@@ -0,0 +1,70 @@
+--- !ruby/object:Gem::Specification
+name: caltrain
+version: !ruby/object:Gem::Version
+  version: 0.0.4
+  prerelease: 
+platform: ruby
+authors:
+- Alex Genco
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-03-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: mocha
+  requirement: &2151813220 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *2151813220
+description: A command line tool to easily figure out when the next caltrain leaves.
+email: alexgenco@gmail.com
+executables:
+- caltrain
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/caltrain.rb
+- bin/caltrain
+- data/google_transit/agency.txt
+- data/google_transit/calendar.txt
+- data/google_transit/calendar_dates.txt
+- data/google_transit/fare_attributes.txt
+- data/google_transit/fare_rules.txt
+- data/google_transit/routes.txt
+- data/google_transit/shapes.txt
+- data/google_transit/stop_times.txt
+- data/google_transit/stops.txt
+- data/google_transit/trips.txt
+- spec/caltrain_spec.rb
+homepage: http://github.com/alexgenco/caltrain
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.17
+signing_key: 
+specification_version: 3
+summary: Get the next caltrain
+test_files:
+- spec/caltrain_spec.rb