show followee checkins in departure board

2 files changed, 13 insertions, 1 deletions

diff --git a/lib/Travelynx/Controller/Traveling.pm b/lib/Travelynx/Controller/Traveling.pm
index c69d517..fc59b68 100755
--- a/lib/Travelynx/Controller/Traveling.pm
+++ b/lib/Travelynx/Controller/Traveling.pm
@@ -799,6 +799,18 @@ sub station {
 	my ($self)  = @_;
 	my $station = $self->stash('station');
 	my $train   = $self->param('train');
+	my $uid     = $self->current_user->{id};
+
+	my @timeline = $self->in_transit->get_timeline(
+		uid   => $uid,
+		short => 1
+	);
+	my %checkin_by_train;
+	for my $checkin (@timeline) {
+		say $checkin->{train_id};
+		push( @{ $checkin_by_train{ $checkin->{train_id} } }, $checkin );
+	}
+	$self->stash( checkin_by_train => \%checkin_by_train );
 
 	$self->render_later;
 
diff --git a/lib/Travelynx/Model/InTransit.pm b/lib/Travelynx/Model/InTransit.pm
index 0fb8e1d..26f689f 100644
--- a/lib/Travelynx/Model/InTransit.pm
+++ b/lib/Travelynx/Model/InTransit.pm
@@ -316,7 +316,7 @@ sub get_timeline {
 		return $db->select(
 			'follows_in_transit',
 			[
-				qw(followee_name train_type train_line train_no dep_eva dep_name arr_eva arr_name)
+				qw(followee_name train_type train_line train_no train_id dep_eva dep_name arr_eva arr_name)
 			],
 			$where
 		)->hashes->each;