lib/gtfs/realtime/stop.rb in gtfs-realtime-0.2.1 vs lib/gtfs/realtime/stop.rb in gtfs-realtime-0.2.2

@@ -16,8 +16,26 @@
       ]
       many_through_many :active_routes, class: GTFS::Realtime::Route, through: [
         [:stop_time_updates, :stop_id, :trip_update_id],
         [:trip_updates, :id, :route_id]
       ]
+
+      def stop_times_schedule_for(date)
+        # TODO: .all.first is a weird syntax to do eager loading correctly. Maybe there's a better way?
+        self_with_eager_loads = GTFS::Realtime::Stop.where(id: id).eager(stop_times: {trip: [:calendar_dates, :route, :shapes]}).all.first
+        self_with_eager_loads.stop_times.select{|st| st.trip.active?(Date.today)}.sort_by{|st| st.departure_time}
+      end
+
+      def stop_times_for_today
+        stop_times = stop_times_schedule_for(Date.today)
+        stop_time_updates.each do |stu|
+          # find a matching existing record in the schedule
+          stop_time = stop_times.find{|st| st.trip_id == stu.trip_update.trip_id}
+
+          # update its info
+          stop_time.set(stu)
+        end
+        stop_times
+      end
     end
   end
 end
\ No newline at end of file