travelynx/lib/Travelynx.pm

1631 lines
40 KiB
Perl
Raw Normal View History

package Travelynx;
use Mojo::Base 'Mojolicious';
2019-04-17 11:10:49 +00:00
use Mojo::Pg;
use Mojolicious::Plugin::Authentication;
use Cache::File;
use Crypt::Eksblowfish::Bcrypt qw(bcrypt en_base64);
use DateTime;
use DBI;
use Encode qw(decode encode);
use Geo::Distance;
2019-04-07 14:55:35 +00:00
use JSON;
use List::Util qw(first);
use List::MoreUtils qw(after_incl before_incl);
use Travel::Status::DE::IRIS;
use Travel::Status::DE::IRIS::Stations;
2019-04-02 18:10:48 +00:00
use Travelynx::Helper::Sendmail;
sub check_password {
my ( $password, $hash ) = @_;
if ( bcrypt( $password, $hash ) eq $hash ) {
return 1;
}
return 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'
);
}
sub get_station {
my ($station_name) = @_;
my @candidates
= Travel::Status::DE::IRIS::Stations::get_station($station_name);
if ( @candidates == 1 ) {
return $candidates[0];
}
return undef;
}
sub startup {
my ($self) = @_;
push( @{ $self->commands->namespaces }, 'Travelynx::Command' );
2019-03-22 15:57:24 +00:00
$self->defaults( layout => 'default' );
2019-03-22 15:57:24 +00:00
$self->types->type( json => 'application/json; charset=utf-8' );
$self->plugin('Config');
if ( $self->config->{secrets} ) {
$self->secrets( $self->config->{secrets} );
}
2019-03-22 15:57:24 +00:00
$self->plugin(
authentication => {
autoload_user => 1,
fail_render => { template => 'login' },
load_user => sub {
my ( $self, $uid ) = @_;
return $self->get_user_data($uid);
},
validate_user => sub {
my ( $self, $username, $password, $extradata ) = @_;
my $user_info = $self->get_user_password($username);
if ( not $user_info ) {
return undef;
}
if ( $user_info->{status} != 1 ) {
return undef;
}
if ( check_password( $password, $user_info->{password_hash} ) )
{
return $user_info->{id};
}
return undef;
2019-03-22 15:57:24 +00:00
},
}
);
$self->sessions->default_expiration( 60 * 60 * 24 * 180 );
$self->defaults( layout => 'default' );
$self->attr(
cache_iris_main => sub {
my ($self) = @_;
return Cache::File->new(
cache_root => $self->app->config->{cache}->{schedule},
default_expires => '6 hours',
lock_level => Cache::File::LOCK_LOCAL(),
);
}
);
$self->attr(
cache_iris_rt => sub {
my ($self) = @_;
return Cache::File->new(
cache_root => $self->app->config->{cache}->{realtime},
default_expires => '70 seconds',
lock_level => Cache::File::LOCK_LOCAL(),
);
}
);
2019-03-22 15:57:24 +00:00
$self->attr(
action_type => sub {
return {
checkin => 1,
checkout => 2,
cancelled_from => 4,
cancelled_to => 5,
};
}
);
$self->attr(
action_types => sub {
return [qw(checkin checkout undo cancelled_from cancelled_to)];
}
);
$self->attr(
token_type => sub {
return {
status => 1,
history => 2,
action => 3,
};
}
);
$self->attr(
token_types => sub {
return [qw(status history action)];
}
);
$self->attr(
dbh => sub {
my ($self) = @_;
my $config = $self->app->config;
my $dbname = $config->{db}->{database};
my $host = $config->{db}->{host} // 'localhost';
my $port = $config->{db}->{port} // 5432;
my $user = $config->{db}->{user};
my $pw = $config->{db}->{password};
2019-03-22 15:56:49 +00:00
2019-03-25 16:04:39 +00:00
return DBI->connect(
"dbi:Pg:dbname=${dbname};host=${host};port=${port}",
$user, $pw, { AutoCommit => 1 } );
2019-03-22 15:57:24 +00:00
}
);
$self->attr(
get_all_actions_query => sub {
my ($self) = @_;
2019-03-22 15:57:24 +00:00
return $self->app->dbh->prepare(
qq{
select user_actions.id, action_id, extract(epoch from action_time),
stations.ds100, stations.name,
train_type, train_line, train_no, train_id,
2019-03-25 16:04:39 +00:00
extract(epoch from sched_time), extract(epoch from real_time),
route, messages, edited
from user_actions
left outer join stations on station_id = stations.id
where user_id = ?
order by action_time desc
}
2019-03-22 15:57:24 +00:00
);
}
);
$self->attr(
get_last_actions_query => sub {
my ($self) = @_;
2019-03-22 15:57:24 +00:00
return $self->app->dbh->prepare(
qq{
select user_actions.id, action_id, extract(epoch from action_time),
stations.ds100, stations.name,
train_type, train_line, train_no, train_id,
2019-03-25 16:04:39 +00:00
extract(epoch from sched_time), extract(epoch from real_time),
route, messages, edited
from user_actions
left outer join stations on station_id = stations.id
where user_id = ?
order by action_time desc
limit 10
}
2019-03-22 15:57:24 +00:00
);
}
);
2019-03-27 20:20:59 +00:00
$self->attr(
get_interval_actions_query => sub {
my ($self) = @_;
# Note: Selecting on real_time would be more intuitive, but is not
# possible at the moment -- non-realtime checkouts
2019-03-27 20:20:59 +00:00
# lack both sched_time and real_time.
return $self->app->dbh->prepare(
qq{
select user_actions.id, action_id, extract(epoch from action_time),
stations.ds100, stations.name,
2019-03-27 20:20:59 +00:00
train_type, train_line, train_no, train_id,
extract(epoch from sched_time), extract(epoch from real_time),
route, messages, edited
2019-03-27 20:20:59 +00:00
from user_actions
left outer join stations on station_id = stations.id
where user_id = ?
and action_time >= to_timestamp(?)
and action_time < to_timestamp(?)
order by action_time desc
}
);
}
);
2019-03-22 15:57:24 +00:00
$self->attr(
get_journey_actions_query => sub {
my ($self) = @_;
2019-03-22 15:57:24 +00:00
return $self->app->dbh->prepare(
qq{
select user_actions.id, action_id, extract(epoch from action_time),
stations.ds100, stations.name,
train_type, train_line, train_no, train_id,
2019-03-25 16:04:39 +00:00
extract(epoch from sched_time), extract(epoch from real_time),
route, messages, edited
from user_actions
left outer join stations on station_id = stations.id
where user_id = ?
2019-03-31 17:32:41 +00:00
and user_actions.id <= ?
order by action_time desc
limit 2
}
2019-03-22 15:57:24 +00:00
);
}
);
$self->attr(
get_api_tokens_query => sub {
my ($self) = @_;
2019-03-22 15:57:24 +00:00
return $self->app->dbh->prepare(
qq{
select
type, token
from tokens where user_id = ?
}
2019-03-22 15:57:24 +00:00
);
}
);
$self->attr(
get_api_token_query => sub {
my ($self) = @_;
2019-03-22 15:57:24 +00:00
return $self->app->dbh->prepare(
qq{
select
token
from tokens where user_id = ? and type = ?
}
2019-03-22 15:57:24 +00:00
);
}
);
$self->attr(
drop_api_token_query => sub {
my ($self) = @_;
2019-03-22 15:57:24 +00:00
return $self->app->dbh->prepare(
qq{
delete from tokens where user_id = ? and type = ?
}
2019-03-22 15:57:24 +00:00
);
}
);
$self->attr(
set_api_token_query => sub {
my ($self) = @_;
2019-03-22 15:57:24 +00:00
return $self->app->dbh->prepare(
qq{
2019-03-25 16:04:39 +00:00
insert into tokens
(user_id, type, token)
values
(?, ?, ?)
2019-03-25 16:04:39 +00:00
on conflict (user_id, type)
do update set token = EXCLUDED.token
}
2019-03-22 15:57:24 +00:00
);
}
);
2019-04-17 11:10:49 +00:00
$self->helper(
sendmail => sub {
state $sendmail = Travelynx::Helper::Sendmail->new(
config => ( $self->config->{mail} // {} ),
log => $self->log
2019-03-22 15:57:24 +00:00
);
2019-04-17 11:10:49 +00:00
}
2019-03-22 15:57:24 +00:00
);
2019-04-04 16:26:53 +00:00
$self->helper(
2019-04-17 11:10:49 +00:00
pg => sub {
my ($self) = @_;
my $config = $self->app->config;
my $dbname = $config->{db}->{database};
my $host = $config->{db}->{host} // 'localhost';
my $port = $config->{db}->{port} // 5432;
my $user = $config->{db}->{user};
my $pw = $config->{db}->{password};
state $pg
= Mojo::Pg->new("postgresql://${user}\@${host}:${port}/${dbname}")
->password($pw);
2019-04-13 15:09:10 +00:00
}
2019-04-04 16:26:53 +00:00
);
2019-04-02 18:10:48 +00:00
$self->helper(
'numify_skipped_stations' => sub {
my ( $self, $count ) = @_;
if ( $count == 0 ) {
return 'INTERNAL ERROR';
}
if ( $count == 1 ) {
return
'Eine Station ohne Geokoordinaten wurde nicht berücksichtigt.';
}
return
"${count} Stationen ohne Geookordinaten wurden nicht berücksichtigt.";
}
);
2019-03-22 15:57:24 +00:00
$self->helper(
'get_departures' => sub {
my ( $self, $station, $lookbehind ) = @_;
$lookbehind //= 180;
my @station_matches
= Travel::Status::DE::IRIS::Stations::get_station($station);
if ( @station_matches == 1 ) {
$station = $station_matches[0][0];
my $status = Travel::Status::DE::IRIS->new(
station => $station,
main_cache => $self->app->cache_iris_main,
realtime_cache => $self->app->cache_iris_rt,
2019-03-22 15:57:24 +00:00
lookbehind => 20,
datetime => DateTime->now( time_zone => 'Europe/Berlin' )
->subtract( minutes => $lookbehind ),
lookahead => $lookbehind + 10,
);
return {
results => [ $status->results ],
errstr => $status->errstr,
station_ds100 => (
$status->station ? $status->station->{ds100} : 'undef'
),
station_name =>
( $status->station ? $status->station->{name} : 'undef' ),
};
}
elsif ( @station_matches > 1 ) {
return {
results => [],
errstr => 'Ambiguous station name',
};
}
else {
return {
results => [],
errstr => 'Unknown station name',
};
}
}
);
# Returns (checkin id, checkout id, error)
# Must be called during a transaction.
# Must perform a rollback on error.
$self->helper(
'add_journey' => sub {
my ( $self, %opt ) = @_;
my $user_status = $self->get_user_status;
if ( $user_status->{checked_in} or $user_status->{cancelled} ) {
# TODO: change database schema to one row per journey instead of two
return ( undef, undef,
'Während einer Zugfahrt können momentan keine manuellen Einträge vorgenommen werden. Klingt komisch, ist aber so.'
);
}
my $uid = $self->current_user->{id};
my $dep_station = get_station( $opt{dep_station} );
my $arr_station = get_station( $opt{arr_station} );
if ( not $dep_station ) {
return ( undef, undef, 'Unbekannter Startbahnhof' );
}
if ( not $arr_station ) {
return ( undef, undef, 'Unbekannter Zielbahnhof' );
}
my $checkin_id;
my $checkout_id;
eval {
$checkin_id = $self->pg->db->insert(
'user_actions',
{
user_id => $uid,
action_id => $self->app->action_type->{checkin},
station_id => $self->get_station_id(
ds100 => $dep_station->[0],
name => $dep_station->[1],
),
action_time =>
DateTime->now( time_zone => 'Europe/Berlin' ),
edited => 0x0f,
train_type => $opt{train_type},
train_line => $opt{train_line},
train_no => $opt{train_no},
sched_time => $opt{sched_departure},
real_time => $opt{rt_departure},
},
{ returning => 'id' }
)->hash->{id};
};
if ($@) {
$self->app->log->error(
"add_journey($uid, checkin): INSERT failed: $@");
return ( undef, undef, 'INSERT failed: ' . $@ );
}
eval {
$checkout_id = $self->pg->db->insert(
'user_actions',
{
user_id => $uid,
action_id => $self->app->action_type->{checkout},
station_id => $self->get_station_id(
ds100 => $arr_station->[0],
name => $arr_station->[1],
),
action_time =>
DateTime->now( time_zone => 'Europe/Berlin' ),
edited => 0x0f,
train_type => $opt{train_type},
train_line => $opt{train_line},
train_no => $opt{train_no},
sched_time => $opt{sched_arrival},
real_time => $opt{rt_arrival},
},
{ returnning => 'id' }
)->hash->{id};
};
if ($@) {
$self->app->log->error(
"add_journey($uid, checkout): INSERT failed: $@");
return ( undef, undef, 'INSERT failed: ' . $@ );
}
return ( $checkin_id, $checkout_id, undef );
}
);
2019-03-22 15:57:24 +00:00
$self->helper(
'checkin' => sub {
my ( $self, $station, $train_id, $action_id ) = @_;
2019-03-22 15:57:24 +00:00
$action_id //= $self->app->action_type->{checkin};
2019-03-22 15:57:24 +00:00
my $status = $self->get_departures($station);
if ( $status->{errstr} ) {
return ( undef, $status->{errstr} );
}
else {
my ($train)
= first { $_->train_id eq $train_id } @{ $status->{results} };
if ( not defined $train ) {
return ( undef, "Train ${train_id} not found" );
}
else {
2019-03-22 15:57:24 +00:00
my $user = $self->get_user_status;
if ( $user->{checked_in} ) {
# If a user is already checked in, we assume that they forgot to
# check out and do it for them.
2019-03-22 15:57:24 +00:00
$self->checkout( $station, 1 );
}
elsif ( $user->{cancelled} ) {
# Same
2019-04-06 09:20:10 +00:00
$self->checkout( $station, 1,
$self->app->action_type->{cancelled_to} );
2019-03-22 15:57:24 +00:00
}
eval {
$self->pg->db->insert(
'user_actions',
{
user_id => $self->current_user->{id},
action_id => $action_id,
station_id => $self->get_station_id(
ds100 => $status->{station_ds100},
name => $status->{station_name}
),
action_time =>
DateTime->now( time_zone => 'Europe/Berlin' ),
edited => 0,
train_type => $train->type,
train_line => $train->line_no,
train_no => $train->train_no,
train_id => $train->train_id,
sched_time => $train->sched_departure,
real_time => $train->departure,
route => join( '|', $train->route ),
messages => join(
'|',
map {
( $_->[0] ? $_->[0]->epoch : q{} ) . ':'
. $_->[1]
} $train->messages
)
}
);
};
if ($@) {
2019-03-25 17:12:01 +00:00
my $uid = $self->current_user->{id};
$self->app->log->error(
"Checkin($uid, $action_id): INSERT failed: $@");
return ( undef, 'INSERT failed: ' . $@ );
2019-03-22 15:57:24 +00:00
}
return ( $train, undef );
2019-03-22 15:57:24 +00:00
}
}
}
);
$self->helper(
'undo' => sub {
my ( $self, $action_id ) = @_;
2019-03-22 15:57:24 +00:00
my $status = $self->get_user_status;
2019-03-22 15:57:24 +00:00
if ( $action_id < 1 or $status->{action_id} != $action_id ) {
return
"Invalid action ID: $action_id != $status->{action_id}. Note that you can only undo your latest action.";
}
2019-04-17 11:10:49 +00:00
eval {
$self->pg->db->delete( 'user_actions', { id => $action_id } );
};
if ($@) {
2019-03-25 17:12:01 +00:00
my $uid = $self->current_user->{id};
2019-04-17 11:10:49 +00:00
$self->app->log->error(
"Undo($uid, $action_id): DELETE failed: $@");
return 'DELETE failed: ' . $@;
}
2019-04-17 11:10:49 +00:00
return;
}
2019-03-22 15:57:24 +00:00
);
$self->helper(
'invalidate_stats_cache' => sub {
my ( $self, $ts ) = @_;
my $uid = $self->current_user->{id};
$ts //= DateTime->now( time_zone => 'Europe/Berlin' );
# ts is the checkout timestamp or (for manual entries) the
# time of arrival. As the journey may span a month or year boundary,
# there is a total of five cache entries we need to invalidate:
# * year, month
# * year
# * (year, month) - 1 month (with wraparound)
# * (year) - 1 year
# * total stats
2019-04-17 11:10:49 +00:00
$self->pg->db->delete(
'journey_stats',
{
user_id => $uid,
year => $ts->year,
month => $ts->month,
}
);
$self->pg->db->delete(
'journey_stats',
{
user_id => $uid,
year => $ts->year,
month => 0,
}
);
$ts->subtract( months => 1 );
2019-04-17 11:10:49 +00:00
$self->pg->db->delete(
'journey_stats',
{
user_id => $uid,
year => $ts->year,
month => $ts->month,
}
);
$ts->subtract( months => 11 );
2019-04-17 11:10:49 +00:00
$self->pg->db->delete(
'journey_stats',
{
user_id => $uid,
year => $ts->year,
month => 0,
}
);
$self->pg->db->delete(
'journey_stats',
{
user_id => $uid,
year => 0,
month => 0,
}
);
}
);
2019-03-22 15:57:24 +00:00
$self->helper(
'checkout' => sub {
my ( $self, $station, $force, $action_id ) = @_;
2019-03-22 15:57:24 +00:00
$action_id //= $self->app->action_type->{checkout};
2019-03-22 15:57:24 +00:00
my $status = $self->get_departures( $station, 180 );
my $user = $self->get_user_status;
my $train_id = $user->{train_id};
2019-03-22 15:57:24 +00:00
if ( not $user->{checked_in} and not $user->{cancelled} ) {
return 'You are not checked into any train';
}
if ( $status->{errstr} and not $force ) {
return $status->{errstr};
}
my $now = DateTime->now( time_zone => 'Europe/Berlin' );
2019-03-22 15:57:24 +00:00
my ($train)
= first { $_->train_id eq $train_id } @{ $status->{results} };
if ( not defined $train ) {
if ($force) {
eval {
$self->pg->db->insert(
'user_actions',
{
user_id => $self->current_user->{id},
action_id => $action_id,
station_id => $self->get_station_id(
ds100 => $status->{station_ds100},
name => $status->{station_name}
),
action_time => $now,
edited => 0
}
);
};
if ($@) {
2019-03-25 17:12:01 +00:00
my $uid = $self->current_user->{id};
$self->app->log->error(
"Force checkout($uid, $action_id): INSERT failed: $@"
2019-03-25 17:12:01 +00:00
);
return 'INSERT failed: ' . $@;
2019-03-22 15:57:24 +00:00
}
$self->invalidate_stats_cache;
return;
2019-03-22 15:57:24 +00:00
}
else {
return "Train ${train_id} not found";
}
}
else {
eval {
$self->pg->db->insert(
'user_actions',
{
user_id => $self->current_user->{id},
action_id => $action_id,
station_id => $self->get_station_id(
ds100 => $status->{station_ds100},
name => $status->{station_name}
),
action_time => $now,
edited => 0,
train_type => $train->type,
train_line => $train->line_no,
train_no => $train->train_no,
train_id => $train->train_id,
sched_time => $train->sched_arrival,
real_time => $train->arrival,
route => join( '|', $train->route ),
messages => join(
'|',
map {
( $_->[0] ? $_->[0]->epoch : q{} ) . ':'
. $_->[1]
} $train->messages
)
}
);
};
if ($@) {
2019-03-25 17:12:01 +00:00
my $uid = $self->current_user->{id};
$self->app->log->error(
"Checkout($uid, $action_id): INSERT failed: $@");
return 'INSERT failed: ' . $@;
2019-03-22 15:57:24 +00:00
}
$self->invalidate_stats_cache;
return;
2019-03-22 15:57:24 +00:00
}
}
);
$self->helper(
'update_journey_part' => sub {
my ( $self, $checkin_id, $checkout_id, $key, $value ) = @_;
my $rows;
eval {
my $db = $self->pg->db;
if ( $key eq 'sched_departure' ) {
$rows = $db->update(
'user_actions',
{
sched_time => $value,
},
{
id => $checkin_id,
action_id => $self->app->action_type->{checkin},
}
)->rows;
}
elsif ( $key eq 'rt_departure' ) {
$rows = $db->update(
'user_actions',
{
real_time => $value,
},
{
id => $checkin_id,
action_id => $self->app->action_type->{checkin},
}
)->rows;
}
elsif ( $key eq 'sched_arrival' ) {
$rows = $db->update(
'user_actions',
{
sched_time => $value,
},
{
id => $checkout_id,
action_id => $self->app->action_type->{checkout},
}
)->rows;
}
elsif ( $key eq 'rt_arrival' ) {
$rows = $db->update(
'user_actions',
{
real_time => $value,
},
{
id => $checkout_id,
action_id => $self->app->action_type->{checkout},
}
)->rows;
}
else {
$self->app->log->error(
"update_journey_part($checkin_id, $checkout_id): Invalid key $key"
);
}
};
if ($@) {
$self->app->log->error(
"update_journey_part($checkin_id, $checkout_id): UPDATE failed: $@"
);
return 'UPDATE failed: ' . $@;
}
if ( $rows == 1 ) {
return undef;
}
return 'UPDATE failed: did not match any journey part';
}
);
$self->helper(
'journey_sanity_check' => sub {
my ( $self, $journey ) = @_;
if ( $journey->{sched_duration} and $journey->{sched_duration} < 0 )
{
2019-04-09 16:11:55 +00:00
return
'Die geplante Dauer dieser Zugfahrt ist negativ. Zeitreisen werden aktuell nicht unterstützt.';
}
if ( $journey->{rt_duration} and $journey->{rt_duration} < 0 ) {
2019-04-09 16:11:55 +00:00
return
'Die Dauer dieser Zugfahrt ist negativ. Zeitreisen werden aktuell nicht unterstützt.';
}
if ( $journey->{sched_duration}
and $journey->{sched_duration} > 60 * 60 * 24 )
{
2019-04-09 16:11:55 +00:00
return 'Die Zugfahrt ist länger als 24 Stunden.';
}
if ( $journey->{rt_duration}
and $journey->{rt_duration} > 60 * 60 * 24 )
{
2019-04-09 16:11:55 +00:00
return 'Die Zugfahrt ist länger als 24 Stunden.';
}
return undef;
}
);
2019-03-22 15:57:24 +00:00
$self->helper(
'get_station_id' => sub {
my ( $self, %opt ) = @_;
2019-04-22 07:58:39 +00:00
my $res = $self->pg->db->select( 'stations', ['id'],
{ ds100 => $opt{ds100} } );
my $res_h = $res->hash;
if ($res_h) {
$res->finish;
return $res_h->{id};
2019-03-22 15:57:24 +00:00
}
2019-04-22 07:58:39 +00:00
$self->pg->db->insert(
'stations',
{
ds100 => $opt{ds100},
name => $opt{name},
}
);
$res = $self->pg->db->select( 'stations', ['id'],
{ ds100 => $opt{ds100} } );
my $id = $res->hash->{id};
$res->finish;
return $id;
2019-03-22 15:57:24 +00:00
}
);
2019-03-22 15:57:24 +00:00
$self->helper(
'get_user_token' => sub {
my ( $self, $uid ) = @_;
my $res = $self->pg->db->select(
'users',
[ 'name', 'status', 'token' ],
{ id => $uid }
);
if ( my $ret = $res->array ) {
return @{$ret};
2019-03-22 15:57:24 +00:00
}
return;
}
);
2019-03-22 15:57:24 +00:00
# This helper should only be called directly when also providing a user ID.
# If you don't have one, use current_user() instead (get_user_data will
# delegate to it anyways).
$self->helper(
'get_user_data' => sub {
my ( $self, $uid ) = @_;
2019-03-22 15:57:24 +00:00
$uid //= $self->current_user->{id};
2019-04-22 07:58:39 +00:00
my $user_data = $self->pg->db->select(
'users',
'id, name, status, public_level, email, '
. 'extract(epoch from registered_at) as registered_at_ts, '
. 'extract(epoch from last_login) as last_login_ts, '
. 'extract(epoch from deletion_requested) as deletion_requested_ts',
{ id => $uid }
)->hash;
if ($user_data) {
2019-03-22 15:57:24 +00:00
return {
2019-04-22 07:58:39 +00:00
id => $user_data->{id},
name => $user_data->{name},
status => $user_data->{status},
is_public => $user_data->{public_level},
email => $user_data->{email},
2019-03-22 15:57:24 +00:00
registered_at => DateTime->from_epoch(
2019-04-22 07:58:39 +00:00
epoch => $user_data->{registered_at_ts},
2019-03-22 15:57:24 +00:00
time_zone => 'Europe/Berlin'
),
last_seen => DateTime->from_epoch(
2019-04-22 07:58:39 +00:00
epoch => $user_data->{last_login_ts},
2019-03-22 15:57:24 +00:00
time_zone => 'Europe/Berlin'
),
2019-04-22 07:58:39 +00:00
deletion_requested => $user_data->{deletion_requested_ts}
2019-03-22 15:57:24 +00:00
? DateTime->from_epoch(
2019-04-22 07:58:39 +00:00
epoch => $user_data->{deletion_requested_ts},
2019-03-22 15:57:24 +00:00
time_zone => 'Europe/Berlin'
)
: undef,
};
}
2019-03-22 15:57:24 +00:00
return undef;
}
2019-03-22 15:57:24 +00:00
);
$self->helper(
'get_api_token' => sub {
my ( $self, $uid ) = @_;
$uid //= $self->current_user->{id};
$self->app->get_api_tokens_query->execute($uid);
my $rows = $self->app->get_api_tokens_query->fetchall_arrayref;
my $token = {};
for my $row ( @{$rows} ) {
$token->{ $self->app->token_types->[ $row->[0] - 1 ] }
= $row->[1];
}
return $token;
}
2019-03-22 15:57:24 +00:00
);
$self->helper(
'get_user_password' => sub {
my ( $self, $name ) = @_;
2019-04-22 07:58:39 +00:00
my $res_h = $self->pg->db->select(
'users',
'id, name, status, password as password_hash',
{ name => $name }
)->hash;
return $res_h;
}
2019-03-22 15:57:24 +00:00
);
2019-03-22 15:57:24 +00:00
$self->helper(
'add_user' => sub {
my ( $self, $user_name, $email, $token, $password ) = @_;
2019-04-22 07:58:39 +00:00
my $now = DateTime->now( time_zone => 'Europe/Berlin' );
2019-04-22 07:58:39 +00:00
my $res = $self->pg->db->insert(
'users',
{
name => $user_name,
status => 0,
public_level => 0,
email => $email,
token => $token,
password => $password,
registered_at => $now,
last_login => $now,
},
{ returning => 'id' }
);
return $res->hash->{id};
}
);
2019-04-22 07:58:39 +00:00
$self->helper(
'flag_user_deletion' => sub {
my ( $self, $uid ) = @_;
my $now = DateTime->now( time_zone => 'Europe/Berlin' );
$self->pg->db->update(
'users',
{ deletion_requested => $now },
{
id => $uid,
2019-03-22 15:57:24 +00:00
}
2019-04-22 07:58:39 +00:00
);
}
);
$self->helper(
'unflag_user_deletion' => sub {
my ( $self, $uid ) = @_;
$self->pg->db->update(
'users',
{
deletion_requested => undef,
},
{
id => $uid,
2019-03-22 15:57:24 +00:00
}
2019-04-22 07:58:39 +00:00
);
2019-03-22 15:57:24 +00:00
}
);
2019-04-06 09:20:10 +00:00
$self->helper(
'set_user_password' => sub {
my ( $self, $uid, $password ) = @_;
2019-04-22 07:58:39 +00:00
$self->pg->db->update(
'users',
{ password => $password },
{ id => $uid }
);
2019-04-06 09:20:10 +00:00
}
);
2019-03-22 15:57:24 +00:00
$self->helper(
'check_if_user_name_exists' => sub {
my ( $self, $user_name ) = @_;
2019-04-22 07:58:39 +00:00
my $count = $self->pg->db->select(
'users',
'count(*) as count',
{ name => $user_name }
)->hash->{count};
2019-04-22 07:58:39 +00:00
if ($count) {
2019-03-22 15:57:24 +00:00
return 1;
}
return 0;
}
2019-03-22 15:57:24 +00:00
);
$self->helper(
'check_if_mail_is_blacklisted' => sub {
my ( $self, $mail ) = @_;
2019-04-22 07:58:39 +00:00
my $count = $self->pg->db->select(
'users',
'count(*) as count',
{
email => $mail,
status => 0,
}
)->hash->{count};
if ($count) {
2019-03-22 15:57:24 +00:00
return 1;
}
2019-04-22 07:58:39 +00:00
$count = $self->pg->db->select(
'pending_mails',
'count(*) as count',
{
email => $mail,
num_tries => { '>', 1 },
}
)->hash->{count};
if ($count) {
2019-03-22 15:57:24 +00:00
return 1;
}
return 0;
}
2019-03-22 15:57:24 +00:00
);
2019-04-04 16:26:53 +00:00
$self->helper(
'delete_journey' => sub {
my ( $self, $checkin_id, $checkout_id, $checkin_epoch,
$checkout_epoch )
= @_;
my $uid = $self->current_user->{id};
my @journeys = $self->get_user_travels(
uid => $uid,
checkout_id => $checkout_id
);
if ( @journeys == 0 ) {
return 'Journey not found';
}
my $journey = $journeys[0];
# Double-check (comparing both ID and action epoch) to make sure we
# are really deleting the right journey and the user isn't just
# playing around with POST requests.
if ( $journey->{ids}[0] != $checkin_id
or $journey->{ids}[1] != $checkout_id
or $journey->{checkin}->epoch != $checkin_epoch
or $journey->{checkout}->epoch != $checkout_epoch )
{
return 'Invalid journey data';
}
my $rows;
eval {
$rows = $self->pg->db->delete(
'user_actions',
{
user_id => $uid,
id => [ $checkin_id, $checkout_id ]
}
)->rows;
};
if ($@) {
$self->app->log->error(
"Delete($uid, $checkin_id, $checkout_id): DELETE failed: $@"
);
return 'DELETE failed: ' . $@;
}
if ( $rows == 2 ) {
$self->invalidate_stats_cache( $journey->{checkout} );
return undef;
2019-04-04 16:26:53 +00:00
}
return sprintf( 'Deleted %d rows, expected 2', $rows );
2019-04-04 16:26:53 +00:00
}
);
2019-04-07 14:55:35 +00:00
$self->helper(
'get_journey_stats' => sub {
my ( $self, %opt ) = @_;
if ( $opt{cancelled} ) {
$self->app->log->warning(
'get_journey_stats called with illegal option cancelled => 1'
);
return {};
}
my $uid = $self->current_user->{id};
my $year = $opt{year} // 0;
my $month = $opt{month} // 0;
# Assumption: If the stats cache contains an entry it is up-to-date.
# -> Cache entries must be explicitly invalidated whenever the user
# checks out of a train or manually edits/adds a journey.
2019-04-22 07:58:39 +00:00
my $res = $self->pg->db->select(
'journey_stats',
['data'],
{
user_id => $uid,
year => $year,
month => $month
}
);
my $res_h = $res->expand->hash;
2019-04-07 14:55:35 +00:00
2019-04-22 07:58:39 +00:00
if ($res_h) {
$res->finish;
return $res_h->{data};
2019-04-07 14:55:35 +00:00
}
my $interval_start = DateTime->new(
time_zone => 'Europe/Berlin',
year => 2000,
month => 1,
day => 1,
hour => 0,
minute => 0,
second => 0,
);
# I wonder if people will still be traveling by train in the year 3000
my $interval_end = $interval_start->clone->add( years => 1000 );
if ( $opt{year} and $opt{month} ) {
$interval_start->set(
year => $opt{year},
month => $opt{month}
);
$interval_end = $interval_start->clone->add( months => 1 );
}
elsif ( $opt{year} ) {
$interval_start->set( year => $opt{year} );
$interval_end = $interval_start->clone->add( years => 1 );
}
my @journeys = $self->get_user_travels(
cancelled => $opt{cancelled} ? 1 : 0,
verbose => 1,
after => $interval_start,
before => $interval_end
);
my $stats = $self->compute_journey_stats(@journeys);
2019-04-17 11:10:49 +00:00
$self->pg->db->insert(
'journey_stats',
{
user_id => $uid,
year => $year,
month => $month,
data => JSON->new->encode($stats),
}
);
2019-04-07 14:55:35 +00:00
return $stats;
}
);
2019-03-22 15:57:24 +00:00
$self->helper(
'get_user_travels' => sub {
my ( $self, %opt ) = @_;
2019-03-27 20:20:59 +00:00
my $uid = $opt{uid} || $self->current_user->{id};
2019-03-22 15:57:24 +00:00
my $query = $self->app->get_all_actions_query;
if ( $opt{limit} ) {
$query = $self->app->get_last_actions_query;
}
2019-03-31 17:32:41 +00:00
if ( $opt{checkout_id} ) {
2019-03-22 15:57:24 +00:00
$query = $self->app->get_journey_actions_query;
$query->execute( $uid, $opt{checkout_id} );
2019-03-22 15:57:24 +00:00
}
2019-03-27 20:20:59 +00:00
elsif ( $opt{after} and $opt{before} ) {
# Each journey consists of exactly two database entries: one for
2019-03-27 20:20:59 +00:00
# checkin, one for checkout. A simple query using e.g.
# after = YYYY-01-01T00:00:00 and before YYYY-02-01T00:00:00
# will miss journeys where checkin and checkout take place in
# different months.
# We therefore add one day to the before timestamp and filter out
# journeys whose checkin lies outside the originally requested
# time range afterwards.
# For an additional twist, get_interval_actions_query filters based
# on the action time, not actual departure, as force
2019-03-27 20:20:59 +00:00
# checkout actions lack sched_time and real_time data. By
# subtracting one day from "after" (i.e., moving it one day into
# the past), we make sure not to miss journeys where the real departure
# time falls into the interval, but the checkin time does not.
# Again, this is addressed in postprocessing at the bottom of this
# helper.
# This works under the assumption that there are no DB trains whose
# journey takes more than 24 hours. If this no longer holds,
# please adjust the intervals accordingly.
$query = $self->app->get_interval_actions_query;
$query->execute(
$uid,
$opt{after}->clone->subtract( days => 1 )->epoch,
$opt{before}->clone->add( days => 1 )->epoch
);
}
2019-03-22 15:57:24 +00:00
else {
$query->execute($uid);
}
my @match_actions = (
$self->app->action_type->{checkout},
$self->app->action_type->{checkin}
);
if ( $opt{cancelled} ) {
@match_actions = (
$self->app->action_type->{cancelled_to},
$self->app->action_type->{cancelled_from}
);
}
2019-03-22 15:57:24 +00:00
my @travels;
my $prev_action = 0;
while ( my @row = $query->fetchrow_array ) {
my (
$action_id, $action, $raw_ts,
$ds100, $name, $train_type,
$train_line, $train_no, $train_id,
$raw_sched_ts, $raw_real_ts, $raw_route,
$raw_messages, $edited
2019-03-22 15:57:24 +00:00
) = @row;
if ( $action == $match_actions[0]
or ( $opt{checkout_id} and not @travels ) )
2019-03-22 15:57:24 +00:00
{
push(
@travels,
{
ids => [ undef, $action_id ],
2019-03-22 15:57:24 +00:00
to_name => $name,
sched_arrival => epoch_to_dt($raw_sched_ts),
rt_arrival => epoch_to_dt($raw_real_ts),
checkout => epoch_to_dt($raw_ts),
type => $train_type,
line => $train_line,
no => $train_no,
messages => $raw_messages
? [ split( qr{[|]}, $raw_messages ) ]
: undef,
route => $raw_route
? [ split( qr{[|]}, $raw_route ) ]
: undef,
completed => 0,
edited => $edited << 8,
2019-03-22 15:57:24 +00:00
}
);
}
elsif (
(
$action == $match_actions[1]
and $prev_action == $match_actions[0]
)
2019-03-31 17:32:41 +00:00
or $opt{checkout_id}
2019-03-22 15:57:24 +00:00
)
{
my $ref = $travels[-1];
$ref->{ids}->[0] = $action_id;
2019-03-22 15:57:24 +00:00
$ref->{from_name} = $name;
$ref->{completed} = 1;
$ref->{sched_departure} = epoch_to_dt($raw_sched_ts);
$ref->{rt_departure} = epoch_to_dt($raw_real_ts);
$ref->{checkin} = epoch_to_dt($raw_ts);
$ref->{type} //= $train_type;
$ref->{line} //= $train_line;
$ref->{no} //= $train_no;
$ref->{messages} //= [ split( qr{[|]}, $raw_messages ) ];
$ref->{route} //= [ split( qr{[|]}, $raw_route ) ];
$ref->{edited} |= $edited;
2019-03-22 15:57:24 +00:00
if ( $opt{verbose} ) {
my @parsed_messages;
for my $message ( @{ $ref->{messages} // [] } ) {
my ( $ts, $msg ) = split( qr{:}, $message );
push( @parsed_messages,
[ epoch_to_dt($ts), $msg ] );
}
$ref->{messages} = [ reverse @parsed_messages ];
$ref->{sched_duration}
= $ref->{sched_arrival}
? $ref->{sched_arrival}->epoch
- $ref->{sched_departure}->epoch
: undef;
$ref->{rt_duration}
= $ref->{rt_arrival}
? $ref->{rt_arrival}->epoch
- $ref->{rt_departure}->epoch
: undef;
my ( $km, $skip )
2019-03-22 15:57:24 +00:00
= $self->get_travel_distance( $ref->{from_name},
$ref->{to_name}, $ref->{route} );
$ref->{km_route} = $km;
$ref->{skip_route} = $skip;
( $km, $skip )
2019-03-22 15:57:24 +00:00
= $self->get_travel_distance( $ref->{from_name},
$ref->{to_name},
[ $ref->{from_name}, $ref->{to_name} ] );
$ref->{km_beeline} = $km;
$ref->{skip_beeline} = $skip;
2019-03-27 20:20:59 +00:00
my $kmh_divisor
= ( $ref->{rt_duration} // $ref->{sched_duration}
// 999999 ) / 3600;
$ref->{kmh_route}
= $kmh_divisor ? $ref->{km_route} / $kmh_divisor : -1;
$ref->{kmh_beeline}
= $kmh_divisor
? $ref->{km_beeline} / $kmh_divisor
: -1;
2019-03-22 15:57:24 +00:00
}
2019-03-31 17:32:41 +00:00
if ( $opt{checkout_id}
2019-03-22 15:57:24 +00:00
and $action
== $self->app->action_type->{cancelled_from} )
{
$ref->{cancelled} = 1;
}
}
$prev_action = $action;
}
2019-03-27 20:20:59 +00:00
if ( $opt{before} and $opt{after} ) {
@travels = grep {
$_->{rt_departure} >= $opt{after}
and $_->{rt_departure} < $opt{before}
} @travels;
}
2019-03-30 20:43:35 +00:00
# user_actions are sorted by action_time. As users are allowed to check
# into trains in arbitrary order, action_time does not always
# correspond to departure/arrival time, so we ensure a proper sort
# order here.
@travels
= sort { $b->{rt_departure} <=> $a->{rt_departure} } @travels;
2019-03-22 15:57:24 +00:00
return @travels;
}
);
$self->helper(
'get_journey' => sub {
my ( $self, %opt ) = @_;
my @journeys = $self->get_user_travels(%opt);
if ( @journeys == 0
or not $journeys[0]{completed}
or $journeys[0]{ids}[1] != $opt{checkout_id} )
{
return undef;
}
return $journeys[0];
}
);
2019-03-22 15:57:24 +00:00
$self->helper(
'get_user_status' => sub {
my ( $self, $uid ) = @_;
2019-03-22 15:57:24 +00:00
$uid //= $self->current_user->{id};
$self->app->get_last_actions_query->execute($uid);
my $rows = $self->app->get_last_actions_query->fetchall_arrayref;
2019-03-22 15:57:24 +00:00
if ( @{$rows} ) {
my $now = DateTime->now( time_zone => 'Europe/Berlin' );
2019-03-22 15:57:24 +00:00
my @cols = @{ $rows->[0] };
my $action_ts = epoch_to_dt( $cols[2] );
my $sched_ts = epoch_to_dt( $cols[9] );
my $real_ts = epoch_to_dt( $cols[10] );
my $checkin_station_name = $cols[4];
my @route = split( qr{[|]}, $cols[11] // q{} );
2019-03-22 15:57:24 +00:00
my @route_after;
my $is_after = 0;
for my $station (@route) {
if ( $station eq $checkin_station_name ) {
$is_after = 1;
}
if ($is_after) {
push( @route_after, $station );
}
}
2019-03-22 15:57:24 +00:00
return {
checked_in =>
( $cols[1] == $self->app->action_type->{checkin} ),
2019-03-22 15:57:24 +00:00
cancelled =>
( $cols[1] == $self->app->action_type->{cancelled_from} ),
2019-03-22 15:57:24 +00:00
timestamp => $action_ts,
timestamp_delta => $now->epoch - $action_ts->epoch,
action_id => $cols[0],
2019-03-22 15:57:24 +00:00
sched_ts => $sched_ts,
real_ts => $real_ts,
station_ds100 => $cols[3],
2019-03-22 15:57:24 +00:00
station_name => $checkin_station_name,
train_type => $cols[5],
train_line => $cols[6],
train_no => $cols[7],
train_id => $cols[8],
2019-03-22 15:57:24 +00:00
route => \@route,
route_after => \@route_after,
};
}
2019-03-22 15:57:24 +00:00
return {
checked_in => 0,
timestamp => epoch_to_dt(0),
sched_ts => epoch_to_dt(0),
real_ts => epoch_to_dt(0),
};
}
2019-03-22 15:57:24 +00:00
);
2019-03-22 15:57:24 +00:00
$self->helper(
'get_travel_distance' => sub {
my ( $self, $from, $to, $route_ref ) = @_;
2019-03-22 15:57:24 +00:00
my $distance = 0;
my $skipped = 0;
2019-03-22 15:57:24 +00:00
my $geo = Geo::Distance->new();
my @route = after_incl { $_ eq $from } @{$route_ref};
@route = before_incl { $_ eq $to } @route;
2019-03-22 15:57:24 +00:00
if ( @route < 2 ) {
2019-03-22 15:57:24 +00:00
# I AM ERROR
return 0;
}
2019-03-22 15:57:24 +00:00
my $prev_station = get_station( shift @route );
if ( not $prev_station ) {
return 0;
}
2019-03-22 15:57:24 +00:00
for my $station_name (@route) {
if ( my $station = get_station($station_name) ) {
if ( $#{$prev_station} >= 4 and $#{$station} >= 4 ) {
$distance
+= $geo->distance( 'kilometer', $prev_station->[3],
$prev_station->[4], $station->[3], $station->[4] );
}
else {
$skipped++;
}
2019-03-22 15:57:24 +00:00
$prev_station = $station;
}
}
2019-03-22 15:57:24 +00:00
return ( $distance, $skipped );
}
2019-03-22 15:57:24 +00:00
);
2019-03-27 20:20:59 +00:00
$self->helper(
'compute_journey_stats' => sub {
my ( $self, @journeys ) = @_;
my $km_route = 0;
my $km_beeline = 0;
my $min_travel_sched = 0;
my $min_travel_real = 0;
my $delay_dep = 0;
my $delay_arr = 0;
my $interchange_real = 0;
2019-03-27 20:20:59 +00:00
my $num_trains = 0;
my $num_journeys = 0;
my $next_departure = 0;
2019-03-27 20:20:59 +00:00
for my $journey (@journeys) {
$num_trains++;
$km_route += $journey->{km_route};
$km_beeline += $journey->{km_beeline};
if ( $journey->{sched_duration} > 0 ) {
$min_travel_sched += $journey->{sched_duration} / 60;
}
if ( $journey->{rt_duration} > 0 ) {
$min_travel_real += $journey->{rt_duration} / 60;
}
if ( $journey->{sched_departure} and $journey->{rt_departure} )
{
$delay_dep
+= ( $journey->{rt_departure}->epoch
- $journey->{sched_departure}->epoch ) / 60;
}
if ( $journey->{sched_arrival} and $journey->{rt_arrival} ) {
$delay_arr
+= ( $journey->{rt_arrival}->epoch
- $journey->{sched_arrival}->epoch ) / 60;
}
# Note that journeys are sorted from recent to older entries
if ( $journey->{rt_arrival}
and $next_departure
and $next_departure - $journey->{rt_arrival}->epoch
< ( 60 * 60 ) )
{
$interchange_real
+= ( $next_departure - $journey->{rt_arrival}->epoch )
/ 60;
}
else {
$num_journeys++;
}
$next_departure = $journey->{rt_departure}->epoch;
2019-03-27 20:20:59 +00:00
}
return {
km_route => $km_route,
km_beeline => $km_beeline,
num_trains => $num_trains,
num_journeys => $num_journeys,
min_travel_sched => $min_travel_sched,
min_travel_real => $min_travel_real,
min_interchange_real => $interchange_real,
delay_dep => $delay_dep,
delay_arr => $delay_arr,
2019-03-27 20:20:59 +00:00
};
}
);
2019-03-22 15:57:24 +00:00
$self->helper(
'navbar_class' => sub {
my ( $self, $path ) = @_;
2019-03-22 15:57:24 +00:00
if ( $self->req->url eq $self->url_for($path) ) {
return 'active';
}
return q{};
}
2019-03-22 15:57:24 +00:00
);
2019-03-22 15:57:24 +00:00
my $r = $self->routes;
$r->get('/')->to('traveling#homepage');
$r->get('/about')->to('static#about');
$r->get('/impressum')->to('static#imprint');
$r->get('/imprint')->to('static#imprint');
$r->get('/api/v0/:user_action/:token')->to('api#get_v0');
$r->get('/login')->to('account#login_form');
$r->get('/register')->to('account#registration_form');
$r->get('/reg/:id/:token')->to('account#verify');
$r->post('/action')->to('traveling#log_action');
$r->post('/geolocation')->to('traveling#geolocation');
$r->post('/list_departures')->to('traveling#redirect_to_station');
$r->post('/login')->to('account#do_login');
$r->post('/register')->to('account#register');
my $authed_r = $r->under(
sub {
my ($self) = @_;
if ( $self->is_user_authenticated ) {
return 1;
}
$self->render( 'login', redirect_to => $self->req->url );
return undef;
}
);
2019-03-22 15:57:24 +00:00
$authed_r->get('/account')->to('account#account');
$authed_r->get('/cancelled')->to('traveling#cancelled');
2019-04-06 09:20:10 +00:00
$authed_r->get('/change_password')->to('account#password_form');
2019-03-22 15:57:24 +00:00
$authed_r->get('/export.json')->to('account#json_export');
$authed_r->get('/history.json')->to('traveling#json_history');
2019-03-22 15:57:24 +00:00
$authed_r->get('/history')->to('traveling#history');
$authed_r->get('/history/:year')->to('traveling#yearly_history');
2019-03-27 20:20:59 +00:00
$authed_r->get('/history/:year/:month')->to('traveling#monthly_history');
$authed_r->get('/journey/add')->to('traveling#add_journey_form');
2019-03-22 15:57:24 +00:00
$authed_r->get('/journey/:id')->to('traveling#journey_details');
$authed_r->get('/s/*station')->to('traveling#station');
$authed_r->post('/journey/add')->to('traveling#add_journey_form');
$authed_r->post('/journey/edit')->to('traveling#edit_journey');
2019-04-06 09:20:10 +00:00
$authed_r->post('/change_password')->to('account#change_password');
2019-03-22 15:57:24 +00:00
$authed_r->post('/delete')->to('account#delete');
$authed_r->post('/logout')->to('account#do_logout');
$authed_r->post('/set_token')->to('api#set_token');
}
1;