mbta-realtime 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8b0be0efe563e0c632bc4faa640b03dfe9055229
-  data.tar.gz: 133680e195cbb5aebab0983a116f9a6fb8faf158
+  metadata.gz: 7502a15244bd9b2909ffaa3ffdedc7863cf5fa87
+  data.tar.gz: 7a218c049d825a54f47eddee4c24fcf9ade2f9a7
 SHA512:
-  metadata.gz: d9a228a1af855ccf1e8ae6acca1f3f4497da277deef45d648b720286fb6a25a2d53e015f11fb98edd5a0e4f217a91567327f2bf132ddf913ab549f8950a4f585
-  data.tar.gz: adb8612fd872592d06e463ae681a322ee293650ffe92b9f4fac778f97b3f8570eb45452d06b0233ddc96f8ad430337d1465a57ea866dfea29002fb794d7afeb7
+  metadata.gz: 66b1f0107cb808defb87069af8b73daa75ee99e534bdcebbf8d41709646ae292f3d5c7bfd53c5ac31302a9257dce722432e70fa574acb1bde73ce6c1c3424799
+  data.tar.gz: 5d9601b982ac5ca874db3a8b9cf6fd036cc22c852434a9fb4726598874ef139b339fa64216470f7a50b9b1e20ab0a33c08ec0e06bcf224f95a20cf289c50c348

data/VERSION

@@ -1 +1 @@
-0.1.2
+0.1.3

data/lib/mbta-realtime/client.rb

@@ -75,8 +75,9 @@ module MBTARealtime
     end
 
 
-    def predictions_by_stop(stop_id)
-      raise NotImplementedError
+    def predictions_by_stop(stop_id, flatten=false)
+      ps = self.class.get("/predictionsbystop", url_opts({stop: stop_id})).to_h
+      flatten ? flatten_predictions(ps) : ps
     end
 
 
@@ -85,13 +86,9 @@ module MBTARealtime
     end
 
 
-    def predictions_by_location(location)
-      raise NotImplementedError
-    end
-
-
-    def predictions_by_location_name(intersection)
-      raise NotImplementedError
+    def predictions_by_location_name(intersection, flatten=false)
+      ps = predictions_by_stop( nearest_stop_id_by_location_name(intersection) )
+      flatten ? flatten_predictions( ps ) : ps
     end
 
 
@@ -102,6 +99,29 @@ module MBTARealtime
         { query: @options[:query].merge( args ) }
       end
 
+      def flatten_predictions(predictions_response)
+        @predictions = []
+
+        travel_modes = predictions_response['mode']
+        travel_modes.each do |mode|
+          routes = mode['route']
+          routes.each do |route|
+            route_obj  = OpenStruct.new(id: route['route_id'], name: route['route_name'], trips: [])
+            directions = route['direction']
+            directions.each do |dir|
+              trips = dir['trip']
+              trips.each do |trip|
+                route_obj.trips << OpenStruct.new(id:    trip['trip_id'],
+                                                  time:  trip['pre_away'])
+              end
+            end
+
+            @predictions << route_obj
+          end
+        end
+        @predictions
+      end
+
   end
 end
 

data/lib/mbta-realtime/version.rb

@@ -1,3 +1,3 @@
 module MBTARealtime
-  VERSION = '0.1.2'
+  VERSION = '0.1.3'
 end

data/mbta-realtime.gemspec

@@ -2,11 +2,11 @@
 # DO NOT EDIT THIS FILE DIRECTLY
 # Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
 # -*- encoding: utf-8 -*-
-# stub: mbta-realtime 0.1.2 ruby lib
+# stub: mbta-realtime 0.1.3 ruby lib
 
 Gem::Specification.new do |s|
   s.name = "mbta-realtime"
-  s.version = "0.1.2"
+  s.version = "0.1.3"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.require_paths = ["lib"]

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mbta-realtime
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Matt Cloyd