travelynx/lib/Travelynx/Model/InTransit.pm

942 lines
22 KiB
Perl
Raw Normal View History

package Travelynx::Model::InTransit;
2023-07-03 15:59:25 +00:00
# Copyright (C) 2020-2023 Birte Kristina Friesel
2020-11-27 21:12:56 +00:00
#
2021-01-29 17:32:13 +00:00
# SPDX-License-Identifier: AGPL-3.0-or-later
use strict;
use warnings;
use 5.020;
use DateTime;
use JSON;
my %visibility_itoa = (
100 => 'public',
80 => 'travelynx',
60 => 'followers',
30 => 'unlisted',
10 => 'private',
);
my %visibility_atoi = (
public => 100,
travelynx => 80,
followers => 60,
unlisted => 30,
private => 10,
);
2023-08-13 10:51:15 +00:00
sub _epoch {
my ($dt) = @_;
return $dt ? $dt->epoch : 0;
}
sub epoch_to_dt {
my ($epoch) = @_;
# Bugs (and user errors) may lead to undefined timestamps. Set them to
# 1970-01-01 to avoid crashing and show obviously wrong data instead.
$epoch //= 0;
return DateTime->from_epoch(
epoch => $epoch,
time_zone => 'Europe/Berlin',
locale => 'de-DE',
);
}
sub epoch_or_dt_to_dt {
my ($input) = @_;
if ( ref($input) eq 'DateTime' ) {
return $input;
}
return epoch_to_dt($input);
}
sub new {
my ( $class, %opt ) = @_;
return bless( \%opt, $class );
}
# merge [name, eva, data] from old_route into [name, undef, undef] from new_route.
# If new_route already has eva/data, it is kept as-is.
# changes new_route.
sub _merge_old_route {
my ( $self, %opt ) = @_;
my $db = $opt{db};
my $uid = $opt{uid};
my $new_route = $opt{route};
my $res_h = $db->select( 'in_transit', ['route'], { user_id => $uid } )
->expand->hash;
my $old_route = $res_h ? $res_h->{route} : [];
for my $i ( 0 .. $#{$new_route} ) {
if ( $old_route->[$i] and $old_route->[$i][0] eq $new_route->[$i][0] ) {
$new_route->[$i][1] //= $old_route->[$i][1];
if ( not keys %{ $new_route->[$i][2] // {} } ) {
$new_route->[$i][2] = $old_route->[$i][2];
}
}
}
return $new_route;
}
sub add {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
Multi-backend support Squashed commit of the following: commit 92518024ba295456358618c0e8180bd8e996fdf1 Author: Birte Kristina Friesel <birte.friesel@uos.de> Date: Fri Jul 26 18:39:46 2024 +0200 add_or_update station: remove superfluos 'new backend id := old backend id' commit df21c20c6e4c86454f8a9ac69121404415217f2a Author: Birte Kristina Friesel <birte.friesel@uos.de> Date: Fri Jul 26 18:35:51 2024 +0200 revert connection targets min_count to 3 commit be335cef07d0b42874f5fc1de4a1d13396e8e807 Author: Birte Kristina Friesel <birte.friesel@uos.de> Date: Fri Jul 26 18:20:05 2024 +0200 mention backend selection in API documentation commit 9f41828fb4f18fd707e0087def3032e8d4c8d7d8 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 20:19:23 2024 +0200 use_history: not all backends provide route data in departure monitor commit 09714b4d89684b8331d0e96f564a4c7432318f70 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 20:11:44 2024 +0200 disambiguation: pass correct hafas parameter commit 8cdf1120fc32155dc6525be64601b7c10a9c7f52 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 20:11:28 2024 +0200 _checked_in: hide Zuglauf link for non-db checkins commit 7455653f541198e0e0a6d11aed421487ffdb6285 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 20:01:47 2024 +0200 debug output commit b9cda07f85601a58ea32dbdacdd5399f302db52b Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 19:09:07 2024 +0200 fix remaining get_connection_targets / get_connecting_trains_p invocations commit 2759d7258c37c7498905cfe19f6b4c4f6d16bd21 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Wed Jul 24 20:50:12 2024 +0200 support non-DB HAFAS backends (WiP)
2024-07-26 16:55:58 +00:00
my $backend_id = $opt{backend_id};
my $train = $opt{train};
2023-08-13 10:51:15 +00:00
my $journey = $opt{journey};
my $stop = $opt{stop};
my $checkin_station_id = $opt{departure_eva};
my $route = $opt{route};
my $data = $opt{data};
my $json = JSON->new;
2023-08-13 10:51:15 +00:00
if ($train) {
$db->insert(
'in_transit',
{
user_id => $uid,
cancelled => $train->departure_is_cancelled ? 1
2023-08-13 10:51:15 +00:00
: 0,
checkin_station_id => $checkin_station_id,
checkin_time => DateTime->now( time_zone => 'Europe/Berlin' ),
dep_platform => $train->platform,
train_type => $train->type,
train_line => $train->line_no,
train_no => $train->train_no,
train_id => $train->train_id,
sched_departure => $train->sched_departure,
real_departure => $train->departure,
route => $json->encode($route),
messages => $json->encode(
[ map { [ $_->[0]->epoch, $_->[1] ] } $train->messages ]
),
data => JSON->new->encode(
{
rt => $train->departure_has_realtime ? 1
: 0,
%{ $data // {} }
}
),
backend_id => $backend_id,
2023-08-13 10:51:15 +00:00
}
);
}
elsif ( $journey and $stop ) {
my @route;
my $product = $journey->product_at( $stop->loc->eva )
// $journey->product;
2023-08-13 10:51:15 +00:00
for my $j_stop ( $journey->route ) {
push(
@route,
[
$j_stop->loc->name,
$j_stop->loc->eva,
2023-08-13 10:51:15 +00:00
{
sched_arr => _epoch( $j_stop->sched_arr ),
sched_dep => _epoch( $j_stop->sched_dep ),
rt_arr => _epoch( $j_stop->rt_arr ),
rt_dep => _epoch( $j_stop->rt_dep ),
arr_delay => $j_stop->arr_delay,
dep_delay => $j_stop->dep_delay,
load => $j_stop->load,
lat => $j_stop->loc->lat,
lon => $j_stop->loc->lon,
2023-08-13 10:51:15 +00:00
}
]
);
if ( defined $j_stop->tz_offset ) {
$route[-1][2]{tz_offset} = $j_stop->tz_offset;
}
}
2023-08-13 10:51:15 +00:00
$db->insert(
'in_transit',
{
user_id => $uid,
cancelled => $stop->{dep_cancelled}
? 1
: 0,
checkin_station_id => $stop->loc->eva,
2023-08-13 10:51:15 +00:00
checkin_time => DateTime->now( time_zone => 'Europe/Berlin' ),
dep_platform => $stop->{platform},
train_type => $product->type // q{},
train_line => $product->line_no,
train_no => $product->number // q{},
2023-08-13 10:51:15 +00:00
train_id => $journey->id,
sched_departure => $stop->{sched_dep},
real_departure => $stop->{rt_dep} // $stop->{sched_dep},
route => $json->encode( \@route ),
data => JSON->new->encode(
{
rt => $stop->{rt_dep} ? 1 : 0,
%{ $data // {} }
}
),
backend_id => $backend_id,
2023-08-13 10:51:15 +00:00
}
);
}
else {
die('neither train nor journey specified');
}
}
sub add_from_journey {
my ( $self, %opt ) = @_;
my $journey = $opt{journey};
my $db = $opt{db} // $self->{pg}->db;
$db->insert( 'in_transit', $journey );
}
sub delete {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
$db->delete( 'in_transit', { user_id => $uid } );
}
sub delete_incomplete_checkins {
my ( $self, %opt ) = @_;
my $db = $opt{db} // $self->{pg}->db;
return $db->delete( 'in_transit',
{ checkin_time => { '<', $opt{earlier_than} } } )->rows;
}
sub postprocess {
my ( $self, $ret ) = @_;
my $now = DateTime->now( time_zone => 'Europe/Berlin' );
my $epoch = $now->epoch;
my @route = @{ $ret->{route} // [] };
my @route_after;
my $dep_info;
my $is_after = 0;
for my $station (@route) {
if ($is_after) {
push( @route_after, $station );
}
# Note that the departure stop may be present more than once in @route,
# e.g. when traveling along ring lines such as S41 / S42 in Berlin.
if (
$ret->{dep_name}
and $station->[0] eq $ret->{dep_name}
and not($station->[2]{sched_dep}
and $station->[2]{sched_dep} < $ret->{sched_dep_ts} )
)
{
$is_after = 1;
if ( @{$station} > 1 and not $dep_info ) {
$dep_info = $station->[2];
}
}
}
my $ts = $ret->{checkout_ts} // $ret->{checkin_ts};
my $action_time = epoch_to_dt($ts);
$ret->{checked_in} = !$ret->{cancelled};
$ret->{timestamp} = $action_time;
$ret->{timestamp_delta} = $now->epoch - $action_time->epoch;
$ret->{boarding_countdown} = -1;
$ret->{sched_departure} = epoch_to_dt( $ret->{sched_dep_ts} );
$ret->{real_departure} = epoch_to_dt( $ret->{real_dep_ts} );
$ret->{sched_arrival} = epoch_to_dt( $ret->{sched_arr_ts} );
$ret->{real_arrival} = epoch_to_dt( $ret->{real_arr_ts} );
$ret->{route_after} = \@route_after;
$ret->{extra_data} = $ret->{data};
$ret->{comment} = $ret->{user_data}{comment};
$ret->{visibility_str}
= $ret->{visibility}
? $visibility_itoa{ $ret->{visibility} }
: 'default';
$ret->{effective_visibility_str}
= $visibility_itoa{ $ret->{effective_visibility} };
my @parsed_messages;
for my $message ( @{ $ret->{messages} // [] } ) {
my ( $ts, $msg ) = @{$message};
push( @parsed_messages, [ epoch_to_dt($ts), $msg ] );
}
$ret->{messages} = [ reverse @parsed_messages ];
@parsed_messages = ();
for my $message ( @{ $ret->{extra_data}{qos_msg} // [] } ) {
my ( $ts, $msg ) = @{$message};
push( @parsed_messages, [ epoch_to_dt($ts), $msg ] );
}
$ret->{extra_data}{qos_msg} = [@parsed_messages];
if ( $dep_info and $dep_info->{sched_arr} ) {
$dep_info->{sched_arr}
= epoch_to_dt( $dep_info->{sched_arr} );
$dep_info->{rt_arr} = epoch_to_dt( $dep_info->{rt_arr} );
$dep_info->{rt_arr_countdown} = $ret->{boarding_countdown}
= $dep_info->{rt_arr}->epoch - $epoch;
}
for my $station (@route_after) {
if ( @{$station} > 1 ) {
# Note: $station->[2]{sched_arr} may already have been
# converted to a DateTime object. This can happen when a
# station is present several times in a train's route, e.g.
# for Frankfurt Flughafen in some nightly connections.
my $times = $station->[2] // {};
for my $key (qw(sched_arr rt_arr sched_dep rt_dep)) {
if ( $times->{$key} ) {
$times->{$key}
= epoch_or_dt_to_dt( $times->{$key} );
}
}
if ( $times->{sched_arr} and $times->{rt_arr} ) {
$times->{arr_delay}
= $times->{rt_arr}->epoch - $times->{sched_arr}->epoch;
}
if ( $times->{sched_arr} or $times->{rt_arr} ) {
$times->{arr} = $times->{rt_arr} || $times->{sched_arr};
$times->{arr_countdown} = $times->{arr}->epoch - $epoch;
}
if ( $times->{sched_dep} and $times->{rt_dep} ) {
$times->{dep_delay}
= $times->{rt_dep}->epoch - $times->{sched_dep}->epoch;
}
if ( $times->{sched_dep} or $times->{rt_dep} ) {
$times->{dep} = $times->{rt_dep} || $times->{sched_dep};
$times->{dep_countdown} = $times->{dep}->epoch - $epoch;
}
}
}
$ret->{departure_countdown} = $ret->{real_departure}->epoch - $now->epoch;
if ( $ret->{real_arr_ts} ) {
$ret->{arrival_countdown} = $ret->{real_arrival}->epoch - $now->epoch;
$ret->{journey_duration}
= $ret->{real_arrival}->epoch - $ret->{real_departure}->epoch;
$ret->{journey_completion}
= $ret->{journey_duration}
? 1 - ( $ret->{arrival_countdown} / $ret->{journey_duration} )
: 1;
if ( $ret->{journey_completion} > 1 ) {
$ret->{journey_completion} = 1;
}
elsif ( $ret->{journey_completion} < 0 ) {
$ret->{journey_completion} = 0;
}
}
else {
$ret->{arrival_countdown} = undef;
$ret->{journey_duration} = undef;
$ret->{journey_completion} = undef;
}
return $ret;
}
sub get {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
my $table = 'in_transit';
if ( $opt{with_timestamps} ) {
$table = 'in_transit_str';
}
my $res = $db->select( $table, '*', { user_id => $uid } );
my $ret;
if ( $opt{with_data} ) {
$ret = $res->expand->hash;
}
else {
$ret = $res->hash;
}
if ( $opt{with_visibility} and $ret ) {
$ret->{visibility_str}
= $ret->{visibility}
? $visibility_itoa{ $ret->{visibility} }
: 'default';
$ret->{effective_visibility_str}
= $visibility_itoa{ $ret->{effective_visibility} };
}
if ( $opt{postprocess} and $ret ) {
return $self->postprocess($ret);
}
return $ret;
}
2023-07-15 06:36:26 +00:00
sub get_timeline {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
my $where = {
follower_id => $uid,
effective_visibility => { '>=', 60 }
2023-07-15 06:36:26 +00:00
};
if ( $opt{short} ) {
return $db->select(
'follows_in_transit',
[
qw(followee_name train_type train_line train_no train_id dep_eva dep_name arr_eva arr_name)
2023-07-15 06:36:26 +00:00
],
$where
)->hashes->each;
}
my $res = $db->select( 'follows_in_transit', '*', $where );
my $ret;
if ( $opt{with_data} ) {
return map { $self->postprocess($_) } $res->expand->hashes->each;
2023-07-15 06:36:26 +00:00
}
else {
return $res->hashes->each;
}
}
sub get_all_active {
my ( $self, %opt ) = @_;
my $db = $opt{db} // $self->{pg}->db;
return $db->select( 'in_transit_str', '*', { cancelled => 0 } )
->hashes->each;
}
Multi-backend support Squashed commit of the following: commit 92518024ba295456358618c0e8180bd8e996fdf1 Author: Birte Kristina Friesel <birte.friesel@uos.de> Date: Fri Jul 26 18:39:46 2024 +0200 add_or_update station: remove superfluos 'new backend id := old backend id' commit df21c20c6e4c86454f8a9ac69121404415217f2a Author: Birte Kristina Friesel <birte.friesel@uos.de> Date: Fri Jul 26 18:35:51 2024 +0200 revert connection targets min_count to 3 commit be335cef07d0b42874f5fc1de4a1d13396e8e807 Author: Birte Kristina Friesel <birte.friesel@uos.de> Date: Fri Jul 26 18:20:05 2024 +0200 mention backend selection in API documentation commit 9f41828fb4f18fd707e0087def3032e8d4c8d7d8 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 20:19:23 2024 +0200 use_history: not all backends provide route data in departure monitor commit 09714b4d89684b8331d0e96f564a4c7432318f70 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 20:11:44 2024 +0200 disambiguation: pass correct hafas parameter commit 8cdf1120fc32155dc6525be64601b7c10a9c7f52 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 20:11:28 2024 +0200 _checked_in: hide Zuglauf link for non-db checkins commit 7455653f541198e0e0a6d11aed421487ffdb6285 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 20:01:47 2024 +0200 debug output commit b9cda07f85601a58ea32dbdacdd5399f302db52b Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 19:09:07 2024 +0200 fix remaining get_connection_targets / get_connecting_trains_p invocations commit 2759d7258c37c7498905cfe19f6b4c4f6d16bd21 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Wed Jul 24 20:50:12 2024 +0200 support non-DB HAFAS backends (WiP)
2024-07-26 16:55:58 +00:00
sub get_checkout_ids {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
Multi-backend support Squashed commit of the following: commit 92518024ba295456358618c0e8180bd8e996fdf1 Author: Birte Kristina Friesel <birte.friesel@uos.de> Date: Fri Jul 26 18:39:46 2024 +0200 add_or_update station: remove superfluos 'new backend id := old backend id' commit df21c20c6e4c86454f8a9ac69121404415217f2a Author: Birte Kristina Friesel <birte.friesel@uos.de> Date: Fri Jul 26 18:35:51 2024 +0200 revert connection targets min_count to 3 commit be335cef07d0b42874f5fc1de4a1d13396e8e807 Author: Birte Kristina Friesel <birte.friesel@uos.de> Date: Fri Jul 26 18:20:05 2024 +0200 mention backend selection in API documentation commit 9f41828fb4f18fd707e0087def3032e8d4c8d7d8 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 20:19:23 2024 +0200 use_history: not all backends provide route data in departure monitor commit 09714b4d89684b8331d0e96f564a4c7432318f70 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 20:11:44 2024 +0200 disambiguation: pass correct hafas parameter commit 8cdf1120fc32155dc6525be64601b7c10a9c7f52 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 20:11:28 2024 +0200 _checked_in: hide Zuglauf link for non-db checkins commit 7455653f541198e0e0a6d11aed421487ffdb6285 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 20:01:47 2024 +0200 debug output commit b9cda07f85601a58ea32dbdacdd5399f302db52b Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 19:09:07 2024 +0200 fix remaining get_connection_targets / get_connecting_trains_p invocations commit 2759d7258c37c7498905cfe19f6b4c4f6d16bd21 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Wed Jul 24 20:50:12 2024 +0200 support non-DB HAFAS backends (WiP)
2024-07-26 16:55:58 +00:00
my $status = $db->select(
'in_transit',
[ 'checkout_station_id', 'backend_id' ],
{ user_id => $uid }
)->hash;
if ($status) {
Multi-backend support Squashed commit of the following: commit 92518024ba295456358618c0e8180bd8e996fdf1 Author: Birte Kristina Friesel <birte.friesel@uos.de> Date: Fri Jul 26 18:39:46 2024 +0200 add_or_update station: remove superfluos 'new backend id := old backend id' commit df21c20c6e4c86454f8a9ac69121404415217f2a Author: Birte Kristina Friesel <birte.friesel@uos.de> Date: Fri Jul 26 18:35:51 2024 +0200 revert connection targets min_count to 3 commit be335cef07d0b42874f5fc1de4a1d13396e8e807 Author: Birte Kristina Friesel <birte.friesel@uos.de> Date: Fri Jul 26 18:20:05 2024 +0200 mention backend selection in API documentation commit 9f41828fb4f18fd707e0087def3032e8d4c8d7d8 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 20:19:23 2024 +0200 use_history: not all backends provide route data in departure monitor commit 09714b4d89684b8331d0e96f564a4c7432318f70 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 20:11:44 2024 +0200 disambiguation: pass correct hafas parameter commit 8cdf1120fc32155dc6525be64601b7c10a9c7f52 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 20:11:28 2024 +0200 _checked_in: hide Zuglauf link for non-db checkins commit 7455653f541198e0e0a6d11aed421487ffdb6285 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 20:01:47 2024 +0200 debug output commit b9cda07f85601a58ea32dbdacdd5399f302db52b Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Thu Jul 25 19:09:07 2024 +0200 fix remaining get_connection_targets / get_connecting_trains_p invocations commit 2759d7258c37c7498905cfe19f6b4c4f6d16bd21 Author: Birte Kristina Friesel <derf@finalrewind.org> Date: Wed Jul 24 20:50:12 2024 +0200 support non-DB HAFAS backends (WiP)
2024-07-26 16:55:58 +00:00
return $status->{checkout_station_id}, $status->{backend_id};
}
return;
}
sub set_cancelled_destination {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
my $cancelled_destination = $opt{cancelled_destination};
my $res_h = $db->select( 'in_transit', ['data'], { user_id => $uid } )
->expand->hash;
my $data = $res_h ? $res_h->{data} : {};
$data->{cancelled_destination} = $cancelled_destination;
$db->update(
'in_transit',
{
checkout_station_id => undef,
checkout_time => undef,
arr_platform => undef,
sched_arrival => undef,
real_arrival => undef,
data => JSON->new->encode($data),
},
{ user_id => $uid }
);
}
sub set_arrival {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
my $train = $opt{train};
my $route = $opt{route};
$route = $self->_merge_old_route(
db => $db,
uid => $uid,
route => $route
);
my $json = JSON->new;
$db->update(
'in_transit',
{
checkout_time => DateTime->now( time_zone => 'Europe/Berlin' ),
arr_platform => $train->platform,
sched_arrival => $train->sched_arrival,
real_arrival => $train->arrival,
route => $json->encode($route),
messages => $json->encode(
[ map { [ $_->[0]->epoch, $_->[1] ] } $train->messages ]
)
},
{ user_id => $uid }
);
}
sub set_arrival_eva {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
my $checkout_station_id = $opt{arrival_eva};
$db->update(
'in_transit',
{
checkout_station_id => $checkout_station_id,
},
{ user_id => $uid }
);
}
sub set_arrival_times {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
my $sched_arr = $opt{sched_arrival};
my $rt_arr = $opt{rt_arrival};
$db->update(
'in_transit',
{
sched_arrival => $sched_arr,
real_arrival => $rt_arr
},
{ user_id => $uid }
);
}
sub set_polyline {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
my $polyline = $opt{polyline};
my $old_id = $opt{old_id};
my $coords = $polyline->{coords};
my $from_eva = $polyline->{from_eva};
my $to_eva = $polyline->{to_eva};
my $polyline_str = JSON->new->encode($coords);
my $pl_res = $db->select(
'polylines',
['id'],
{
origin_eva => $from_eva,
destination_eva => $to_eva,
polyline => $polyline_str,
},
{ limit => 1 }
);
my $polyline_id;
if ( my $h = $pl_res->hash ) {
$polyline_id = $h->{id};
}
else {
eval {
$polyline_id = $db->insert(
'polylines',
{
origin_eva => $from_eva,
destination_eva => $to_eva,
polyline => $polyline_str
},
{ returning => 'id' }
)->hash->{id};
};
if ($@) {
$self->{log}->warn("add_route_timestamps: insert polyline: $@");
}
}
if ( $polyline_id and ( not defined $old_id or $polyline_id != $old_id ) ) {
$self->set_polyline_id(
uid => $uid,
db => $db,
polyline_id => $polyline_id,
train_id => $opt{train_id},
);
}
}
sub set_polyline_id {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
my $polyline_id = $opt{polyline_id};
my %where = ( user_id => $uid );
if ( $opt{train_id} ) {
$where{train_id} = $opt{train_id};
}
$db->update( 'in_transit', { polyline_id => $polyline_id }, \%where );
}
sub set_route_data {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
my $route = $opt{route};
my $delay_msg = $opt{delay_messages};
my $qos_msg = $opt{qos_messages};
my $him_msg = $opt{him_messages};
my %where = ( user_id => $uid );
if ( $opt{train_id} ) {
$where{train_id} = $opt{train_id};
}
my $res_h = $db->select( 'in_transit', ['data'], { user_id => $uid } )
->expand->hash;
my $data = $res_h ? $res_h->{data} : {};
$data->{delay_msg} = $opt{delay_messages};
$data->{qos_msg} = $opt{qos_messages};
$data->{him_msg} = $opt{him_messages};
# no need to merge $route, it already contains HAFAS data
$db->update(
'in_transit',
{
route => JSON->new->encode($route),
data => JSON->new->encode($data)
},
\%where
);
}
sub unset_arrival_data {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
$db->update(
'in_transit',
{
checkout_time => undef,
arr_platform => undef,
sched_arrival => undef,
real_arrival => undef,
},
{ user_id => $uid }
);
}
sub update_departure {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
my $dep_eva = $opt{dep_eva};
my $arr_eva = $opt{arr_eva};
my $train = $opt{train};
my $route = $opt{route};
my $json = JSON->new;
$route = $self->_merge_old_route(
db => $db,
uid => $uid,
route => $route
);
# selecting on user_id and train_no avoids a race condition if a user checks
# into a new train while we are fetching data for their previous journey. In
# this case, the new train would receive data from the previous journey.
$db->update(
'in_transit',
{
dep_platform => $train->platform,
real_departure => $train->departure,
route => $json->encode($route),
messages => $json->encode(
[ map { [ $_->[0]->epoch, $_->[1] ] } $train->messages ]
),
},
{
user_id => $uid,
train_no => $train->train_no,
checkin_station_id => $dep_eva,
checkout_station_id => $arr_eva,
}
);
}
sub update_departure_cancelled {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
my $dep_eva = $opt{dep_eva};
my $arr_eva = $opt{arr_eva};
my $train = $opt{train};
# depending on the amount of users in transit, some time may
# have passed between fetching $entry from the database and
# now. Ensure that the user is still checked into this train
# by selecting on uid, train no, and checkin/checkout station ID.
my $rows = $db->update(
'in_transit',
{
cancelled => 1,
},
{
user_id => $uid,
train_no => $train->train_no,
checkin_station_id => $dep_eva,
checkout_station_id => $arr_eva,
}
)->rows;
return $rows;
}
2023-08-13 10:51:15 +00:00
sub update_departure_hafas {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
my $dep_eva = $opt{dep_eva};
my $arr_eva = $opt{arr_eva};
my $journey = $opt{journey};
my $stop = $opt{stop};
my $json = JSON->new;
# selecting on user_id and train_no avoids a race condition if a user checks
# into a new train while we are fetching data for their previous journey. In
# this case, the new train would receive data from the previous journey.
$db->update(
'in_transit',
{
real_departure => $stop->{rt_dep},
},
{
user_id => $uid,
train_id => $journey->id,
checkin_station_id => $dep_eva,
checkout_station_id => $arr_eva,
}
);
}
sub update_arrival {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
my $dep_eva = $opt{dep_eva};
my $arr_eva = $opt{arr_eva};
my $train = $opt{train};
my $route = $opt{route};
my $json = JSON->new;
$route = $self->_merge_old_route(
db => $db,
uid => $uid,
route => $route
);
# selecting on user_id, train_no and checkout_station_id avoids a
# race condition when a user checks into a new train or changes
# their destination station while we are fetching times based on no
# longer valid database entries.
my $rows = $db->update(
'in_transit',
{
arr_platform => $train->platform,
sched_arrival => $train->sched_arrival,
real_arrival => $train->arrival,
route => $json->encode($route),
messages => $json->encode(
[ map { [ $_->[0]->epoch, $_->[1] ] } $train->messages ]
),
},
{
user_id => $uid,
train_no => $train->train_no,
checkin_station_id => $dep_eva,
checkout_station_id => $arr_eva,
}
)->rows;
return $rows;
}
2023-08-13 10:51:15 +00:00
sub update_arrival_hafas {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
my $dep_eva = $opt{dep_eva};
my $arr_eva = $opt{arr_eva};
my $journey = $opt{journey};
my $stop = $opt{stop};
my $json = JSON->new;
my @route;
for my $j_stop ( $journey->route ) {
push(
@route,
[
$j_stop->loc->name,
$j_stop->loc->eva,
2023-08-13 10:51:15 +00:00
{
sched_arr => _epoch( $j_stop->sched_arr ),
sched_dep => _epoch( $j_stop->sched_dep ),
rt_arr => _epoch( $j_stop->rt_arr ),
rt_dep => _epoch( $j_stop->rt_dep ),
arr_delay => $j_stop->arr_delay,
dep_delay => $j_stop->dep_delay,
load => $j_stop->load,
lat => $j_stop->loc->lat,
lon => $j_stop->loc->lon,
2023-08-13 10:51:15 +00:00
}
]
);
if ( defined $j_stop->tz_offset ) {
$route[-1][2]{tz_offset} = $j_stop->tz_offset;
}
2023-08-13 10:51:15 +00:00
}
my $res_h = $db->select( 'in_transit', ['route'], { user_id => $uid } )
->expand->hash;
my $old_route = $res_h ? $res_h->{route} : [];
for my $i ( 0 .. $#route ) {
if ( $old_route->[$i] and $old_route->[$i][1] == $route[$i][1] ) {
for my $k (qw(rt_arr rt_dep arr_delay dep_delay)) {
$route[$i][2]{$k} //= $old_route->[$i][2]{$k};
}
}
}
# selecting on user_id and train_no avoids a race condition if a user checks
# into a new train while we are fetching data for their previous journey. In
# this case, the new train would receive data from the previous journey.
$db->update(
'in_transit',
{
real_arrival => $stop->{rt_arr},
route => $json->encode( [@route] ),
},
{
user_id => $uid,
train_id => $journey->id,
checkin_station_id => $dep_eva,
checkout_station_id => $arr_eva,
}
);
}
sub update_data {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
my $new_data = $opt{data} // {};
my %where = ( user_id => $uid );
if ( $opt{train_id} ) {
$where{train_id} = $opt{train_id};
}
my $res_h = $db->select( 'in_transit', ['data'], { user_id => $uid } )
->expand->hash;
my $data = $res_h ? $res_h->{data} : {};
while ( my ( $k, $v ) = each %{$new_data} ) {
$data->{$k} = $v;
}
$db->update( 'in_transit', { data => JSON->new->encode($data) }, \%where );
}
sub update_user_data {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
my $new_data = $opt{user_data} // {};
my %where = ( user_id => $uid );
if ( $opt{train_id} ) {
$where{train_id} = $opt{train_id};
}
my $res_h = $db->select( 'in_transit', ['user_data'], { user_id => $uid } )
->expand->hash;
my $data = $res_h ? $res_h->{user_data} : {};
while ( my ( $k, $v ) = each %{$new_data} ) {
$data->{$k} = $v;
}
$db->update( 'in_transit',
{ user_data => JSON->new->encode($data) }, \%where );
}
sub update_visibility {
my ( $self, %opt ) = @_;
my $uid = $opt{uid};
my $db = $opt{db} // $self->{pg}->db;
my $visibility;
if ( $opt{visibility} and $visibility_atoi{ $opt{visibility} } ) {
$visibility = $visibility_atoi{ $opt{visibility} };
}
$db->update(
'in_transit',
{ visibility => $visibility },
{ user_id => $uid }
);
}
1;