summaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rwxr-xr-xlib/Travelynx.pm111
1 files changed, 111 insertions, 0 deletions
diff --git a/lib/Travelynx.pm b/lib/Travelynx.pm
index 8b8623d..9e00d3f 100755
--- a/lib/Travelynx.pm
+++ b/lib/Travelynx.pm
@@ -1483,6 +1483,117 @@ sub startup {
);
$self->helper(
+ 'get_connection_targets' => sub {
+ my ( $self, %opt ) = @_;
+
+ my $uid = $opt{uid} // $self->current_user->{id};
+ my $threshold = $opt{threshold}
+ // DateTime->now( time_zone => 'Europe/Berlin' )
+ ->subtract( weeks => 60 );
+ my $db = $opt{db} // $self->pg->db;
+
+ my $journey = $db->select( 'in_transit', ['checkout_station_id'],
+ { user_id => $uid } )->hash;
+ if ( not $journey ) {
+ $journey = $db->select(
+ 'journeys',
+ ['checkout_station_id'],
+ {
+ user_id => $uid,
+ cancelled => 0
+ },
+ {
+ limit => 1,
+ order_by => { -desc => 'real_dep_ts' }
+ }
+ )->hash;
+ }
+
+ if ( not $journey ) {
+ return;
+ }
+
+ my $res = $db->query(
+ qq{
+ select
+ count(stations.name) as count,
+ stations.name as dest
+ from journeys
+ left outer join stations on checkout_station_id = stations.id
+ where user_id = ?
+ and checkin_station_id = ?
+ and real_departure > ?
+ group by stations.name
+ order by count desc;
+ },
+ $uid,
+ $journey->{checkout_station_id},
+ $threshold
+ );
+ my @destinations = $res->hashes->grep( sub { shift->{count} > 2 } )
+ ->map( sub { shift->{dest} } )->each;
+ return @destinations;
+ }
+ );
+
+ $self->helper(
+ 'get_connecting_trains' => sub {
+ my ( $self, %opt ) = @_;
+
+ my $status = $self->get_user_status;
+
+ if ( not $status->{arr_ds100} ) {
+ return;
+ }
+
+ my @destinations = $self->get_connection_targets(%opt);
+ my $stationboard
+ = $self->get_departures( $status->{arr_ds100}, 0, 60 );
+
+ @destinations = grep { $_ ne $status->{dep_name} } @destinations;
+
+ if ( $stationboard->{errstr} ) {
+ return;
+ }
+ my @results;
+ my %via_count = map { $_ => 0 } @destinations;
+ for my $train ( @{ $stationboard->{results} } ) {
+ if ( not $train->departure ) {
+ next;
+ }
+ if ( $train->departure->epoch < $status->{real_arrival}->epoch )
+ {
+ next;
+ }
+ if ( $train->train_id eq $status->{train_id} ) {
+ next;
+ }
+ my @via = ( $train->route_post, $train->route_end );
+ for my $dest (@destinations) {
+ if ( $via_count{$dest} < 3
+ and List::Util::any { $_ eq $dest } @via )
+ {
+ push( @results, [ $train, $dest ] );
+ $via_count{$dest}++;
+ next;
+ }
+ }
+ }
+
+ @results = map { $_->[0] }
+ sort { $a->[1] <=> $b->[1] }
+ map {
+ [
+ $_,
+ $_->[0]->departure->epoch // $_->[0]->sched_departur->epoch
+ ]
+ } @results;
+
+ return @results;
+ }
+ );
+
+ $self->helper(
'get_user_travels' => sub {
my ( $self, %opt ) = @_;