Newer
Older
#!/usr/bin/env perl
use Mojolicious::Lite;
use List::MoreUtils qw();
use Travel::Status::DE::DeutscheBahn;
use Travel::Status::DE::IRIS::Stations;
no if $] >= 5.018, warnings => "experimental::smartmatch";
our $VERSION = qx{git describe --dirty} || '0.05';
my $refresh_interval = 180;
my ( $backend, $station, %opt ) = @_;
Birte Kristina Friesel
committed
cache_root => '/tmp/db-fakedisplay',
default_expires => $refresh_interval . ' sec',
lock_level => Cache::File::LOCK_LOCAL(),
# Cache::File has UTF-8 problems, so strip it (and any other potentially
# problematic chars).
my $cstation = $station;
$cstation =~ tr{[0-9a-zA-Z -]}{}cd;
my $cache_str = "${backend}_${cstation}";
my $data = $cache->thaw($cache_str);
# requests with DS100 codes should be preferred (they avoid
# encoding problems on the IRIS server). However, only use them
# if we have an exact match. Ask the backend otherwise.
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,
serializable => 1,
%opt
$data = {
results => [ $status->results ],
errstr => $status->errstr,
};
$cache->freeze( $cache_str, $data );
my $status = Travel::Status::DE::DeutscheBahn->new(
station => $station,
%opt
);
$data = {
results => [ $status->results ],
errstr => $status->errstr,
};
$cache->freeze( $cache_str, $data );
helper 'is_important' => sub {
my ( $self, $stop ) = @_;
if ( $stop =~ m{ Hbf | Flughafen }ox ) {
return 1;
}
return;
};
Birte Kristina Friesel
committed
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
helper 'json_route_diff' => sub {
my ( $self, $route, $sched_route ) = @_;
my @json_route;
my @route = @{$route};
my @sched_route = @{$sched_route};
my $route_idx = 0;
my $sched_idx = 0;
while ( $route_idx <= $#route and $sched_idx <= $#sched_route ) {
if ( $route[$route_idx] eq $sched_route[$sched_idx] ) {
push( @json_route, { name => $route[$route_idx] } );
$route_idx++;
$sched_idx++;
}
# this branch is inefficient, but won't be taken frequently
elsif ( not( $route[$route_idx] ~~ \@sched_route ) ) {
push(
@json_route,
{
name => $route[$route_idx],
isAdditional => 1
}
);
$route_idx++;
}
else {
push(
@json_route,
{
name => $sched_route[$sched_idx],
isCancelled => 1
}
);
$sched_idx++;
}
}
while ( $route_idx < $#route ) {
Birte Kristina Friesel
committed
push(
@json_route,
{
name => $route[$route_idx],
Birte Kristina Friesel
committed
isAdditional => 1,
isCancelled => 0
}
);
$route_idx++;
Birte Kristina Friesel
committed
}
while ( $sched_idx < $#sched_route ) {
Birte Kristina Friesel
committed
push(
@json_route,
{
name => $sched_route[$sched_idx],
Birte Kristina Friesel
committed
isAdditional => 0,
isCancelled => 1
}
);
$sched_idx++;
Birte Kristina Friesel
committed
}
return @json_route;
};
my $station = $self->stash('station');
my @platforms = split( /,/, $self->param('platforms') // q{} );
my @lines = split( /,/, $self->param('lines') // q{} );
Birte Kristina Friesel
committed
my $template = $self->param('mode') // 'clean';
my $hide_low_delay = $self->param('hidelowdelay') // 0;
my $hide_opts = $self->param('hide_opts') // 0;
my $show_realtime = $self->param('show_realtime') // 0;
Birte Kristina Friesel
committed
my $backend = $self->param('backend') // 'iris';
my $admode = $self->param('admode') // 'deparr';
my $apiver = $self->param('version') // 0;
my $api_version
= $backend eq 'iris'
? $Travel::Status::DE::IRIS::VERSION
: $Travel::Status::DE::DeutscheBahn::VERSION;
$self->stash( departures => [] );
Birte Kristina Friesel
committed
$self->stash( title => 'db-infoscreen' );
$self->stash( version => $VERSION );
if ( not( $template ~~ [qw[clean json marudor_v1 marudor multi single]] ) )
{
Birte Kristina Friesel
committed
$template = 'clean';
Birte Kristina Friesel
committed
}
Birte Kristina Friesel
committed
'landingpage',
hide_opts => 0,
show_intro => 1
);
if ( $template eq 'marudor' and $backend eq 'iris' ) {
$opt{lookahead} = 120;
}
Birte Kristina Friesel
committed
my @departures;
my $data = get_results_for( $backend, $station, %opt );
my $results_ref = $data->{results};
my $errstr = $data->{errstr};
my @results = @{$results_ref};
if ( not @results and $template ~~ [qw[json marudor_v1 marudor]] ) {
$self->res->headers->access_control_allow_origin('*');
api_version => $api_version,
version => $VERSION,
}
);
}
elsif ( $backend eq 'iris' ) {
my @candidates = map { { code => $_->[0], name => $_->[1] } }
Travel::Status::DE::IRIS::Stations::get_station($station);
if ( @candidates > 1
or ( @candidates == 1 and $candidates[0]{code} ne $station ) )
{
$json = $self->render_to_string(
json => {
api_version => $api_version,
version => $VERSION,
error => 'ambiguous station code/name',
candidates => \@candidates,
}
);
}
else {
$json = $self->render_to_string(
json => {
api_version => $api_version,
version => $VERSION,
error => ( $errstr // "Got no results for '$station'" )
}
);
}
}
else {
$json = $self->render_to_string(
json => {
api_version => $api_version,
version => $VERSION,
error => ( $errstr // 'unknown station code/name' )
}
);
}
if ($callback) {
$self->render(
data => "$callback($json);",
format => 'json'
);
}
else {
$self->render(
data => $json,
format => 'json'
);
}
return;
}
if ( $backend eq 'iris' ) {
my @candidates = map { [ "$_->[1] ($_->[0])", $_->[0] ] }
Travel::Status::DE::IRIS::Stations::get_station($station);
if ( @candidates > 1
or ( @candidates == 1 and $candidates[0][1] ne $station ) )
{
$self->render(
Birte Kristina Friesel
committed
'landingpage',
stationlist => \@candidates,
hide_opts => 0
);
}
}
Birte Kristina Friesel
committed
'landingpage',
error => ( $errstr // "Got no results for '$station'" ),
if ( $template eq 'single' ) {
if ( not @platforms ) {
for my $result (@results) {
if ( not( $result->platform ~~ \@platforms ) ) {
push( @platforms, $result->platform );
}
}
@platforms = sort { $a <=> $b } @platforms;
}
my %pcnt;
@results = grep { $pcnt{ $_->platform }++ < 1 } @results;
@results = sort { $a->platform <=> $b->platform } @results;
}
if ( $backend eq 'iris' and $show_realtime ) {
if ( $admode eq 'arr' ) {
@results = sort {
( $a->arrival // $a->departure )
<=> ( $b->arrival // $b->departure )
} @results;
}
else {
@results = sort {
( $a->departure // $a->arrival )
<=> ( $b->departure // $b->arrival )
} @results;
}
}
my $platform = ( split( / /, $result->platform ) )[0];
my $line = $result->line;
if ( $result->isa('Travel::Status::DE::IRIS::Result') ) {
@route = $result->route_post;
}
if ( not( List::MoreUtils::any { $_ =~ m{$via}io } @route ) ) {
if ( @platforms
and not( List::MoreUtils::any { $_ eq $platform } @platforms ) )
{
if ( @lines and not( List::MoreUtils::any { $line =~ m{^$_} } @lines ) )
{
if ( $backend eq 'iris' and $admode eq 'arr' and not $result->arrival )
{
next;
}
if ( $backend eq 'iris'
and $admode eq 'dep'
and not $result->departure )
{
next;
}
Birte Kristina Friesel
committed
my ( $info, $moreinfo );
my $delaymsg
= join( ', ', map { $_->[1] } $result->delay_messages );
my $qosmsg = join( ' +++ ', map { $_->[1] } $result->qos_messages );
$info = "Fahrt fällt aus: ${delaymsg}";
elsif ( $result->delay and $result->delay > 0 ) {
if ( $template eq 'clean' ) {
$info = $delaymsg;
$delay = $result->delay;
}
else {
$info = sprintf( 'Verspätung ca. %d Min.%s%s',
$result->delay, $delaymsg ? q{: } : q{}, $delaymsg );
if ( $result->replacement_for and $template ne 'clean' ) {
for my $rep ( $result->replacement_for ) {
$info = sprintf(
'Ersatzzug für %s %s %s%s',
$rep->type, $rep->train_no,
$info ? '+++ ' : q{}, $info // q{}
Birte Kristina Friesel
committed
);
}
}
if ( $info and $qosmsg ) {
$info .= ' +++ ';
}
$info .= $qosmsg;
Birte Kristina Friesel
committed
if ( $result->additional_stops and not $result->is_cancelled ) {
my $additional_line = join( q{, }, $result->additional_stops );
$info
= 'Zusätzliche Halte: '
. $additional_line
. ( $info ? ' +++ ' : q{} )
. $info;
if ( $template ne 'marudor_v1' and $template ne 'marudor' ) {
push(
@{$moreinfo},
[ 'Zusätzliche Halte', $additional_line ]
);
}
if ( $result->canceled_stops and not $result->is_cancelled ) {
my $cancel_line = join( q{, }, $result->canceled_stops );
$info
= 'Ohne Halt in: '
. $cancel_line
. ( $info ? ' +++ ' : q{} )
. $info;
if ( $template ne 'marudor_v1' and $template ne 'marudor' ) {
push( @{$moreinfo}, [ 'Ohne Halt in', $cancel_line ] );
}
}
push( @{$moreinfo}, $result->messages );
Birte Kristina Friesel
committed
if ($info) {
$moreinfo = [ [ 'RIS', $info ] ];
}
my $time = $result->time;
if ( $backend eq 'iris' ) {
# ->time defaults to dep, so we only need to overwrite $time
# if we want arrival times
if ( $admode eq 'arr' ) {
$time = $result->sched_arrival->strftime('%H:%M');
}
if ($show_realtime) {
if ( ( $admode eq 'arr' and $result->arrival )
or not $result->departure )
{
$time = $result->arrival->strftime('%H:%M');
}
else {
$time = $result->departure->strftime('%H:%M');
}
}
if ( $info eq '+0' ) {
if ( $template eq 'clean'
and $info
and $info =~ s{ (?: ca \. \s* )? \+ (\d+) :? \s* }{}x )
{
$delay = $1;
}
if ( $hide_low_delay and $info ) {
Birte Kristina Friesel
committed
$info =~ s{ (?: ca\. \s* )? \+ [ 1 2 3 4 ] $ }{}x;
Birte Kristina Friesel
committed
$info =~ s{ (?: ca\. \s* )? \+ (\d+) }{Verspätung ca $1 Min.}x;
}
if ( $template eq 'marudor' ) {
Birte Kristina Friesel
committed
my @json_route = $self->json_route_diff( [ $result->route ],
[ $result->sched_route ] );
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
if ( $apiver == 1 ) {
push(
@departures,
{
delay => $delay,
destination => $result->destination,
isCancelled => $result->can('is_cancelled')
? $result->is_cancelled
: undef,
messages => {
delay => [
map {
{
timestamp => $_->[0],
text => $_->[1]
}
} $result->delay_messages
],
qos => [
map {
{
timestamp => $_->[0],
text => $_->[1]
}
} $result->qos_messages
],
},
platform => $result->platform,
route => \@json_route,
scheduledPlatform => $result->sched_platform,
time => $time,
train => $result->train,
via => [ $result->route_interesting(3) ],
}
);
}
else { # apiver == 2
my ( $delay_arr, $delay_dep, $sched_arr, $sched_dep );
if ( $result->arrival ) {
$delay_arr = $result->arrival->subtract_datetime(
$result->sched_arrival )->in_units('minutes');
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
if ( $result->departure ) {
$delay_dep = $result->departure->subtract_datetime(
$result->sched_departure )->in_units('minutes');
}
if ( $result->sched_arrival ) {
$sched_arr = $result->sched_arrival->strftime('%H:%M');
}
if ( $result->sched_departure ) {
$sched_dep = $result->sched_departure->strftime('%H:%M');
}
push(
@departures,
{
delayArrival => $delay_arr,
delayDeparture => $delay_dep,
destination => $result->destination,
isCancelled => $result->can('is_cancelled')
? $result->is_cancelled
: undef,
messages => {
delay => [
map {
{
timestamp => $_->[0],
text => $_->[1]
}
} $result->delay_messages
],
qos => [
map {
{
timestamp => $_->[0],
text => $_->[1]
}
} $result->qos_messages
],
},
platform => $result->platform,
route => \@json_route,
scheduledPlatform => $result->sched_platform,
scheduledArrival => $sched_arr,
scheduledDeparture => $sched_dep,
train => $result->train,
via => [ $result->route_interesting(3) ],
}
);
}
}
elsif ( $backend eq 'iris' ) {
time => $time,
sched_arrival => $result->sched_arrival
? $result->sched_arrival->strftime('%H:%M')
: undef,
sched_departure => $result->sched_departure
? $result->sched_departure->strftime('%H:%M')
: undef,
arrival => $result->arrival
? $result->arrival->strftime('%H:%M')
: undef,
departure => $result->departure
? $result->departure->strftime('%H:%M')
: undef,
train_type => $result->type,
train_line => $result->line_no,
train_no => $result->train_no,
scheduled_route => [ $result->sched_route ],
route_post => [ $result->route_post ],
Birte Kristina Friesel
committed
route_post_diff => [
$self->json_route_diff(
[ $result->route_post ],
[ $result->sched_route_post ]
)
],
destination => $result->destination,
origin => $result->origin,
Birte Kristina Friesel
committed
platform => $result->platform,
scheduled_platform => $result->sched_platform,
info => $info,
is_cancelled => $result->is_cancelled,
messages => {
delay => [
map { { timestamp => $_->[0], text => $_->[1] } }
],
qos => [
map { { timestamp => $_->[0], text => $_->[1] } }
],
},
moreinfo => $moreinfo,
delay => $delay,
additional_stops => [ $result->additional_stops ],
canceled_stops => [ $result->canceled_stops ],
map { $_->type . q{ } . $_->train_no }
$result->replaced_by
],
replacement_for => [
map { $_->type . q{ } . $_->train_no }
$result->replacement_for
}
);
}
else {
push(
@departures,
{
time => $time,
train => $result->train,
train_type => $result->type,
via => [ $result->route_interesting(3) ],
destination => $result->destination,
platform => $platform,
info => $info,
is_cancelled => $result->can('is_cancelled')
? $result->is_cancelled
: undef,
messages => {
},
moreinfo => $moreinfo,
delay => $delay,
replaced_by => [],
replacement_for => [],
route_timetable => [ $result->route_timetable ],
route_info => $result->route_info,
$self->res->headers->access_control_allow_origin('*');
preformatted => \@departures,
version => $VERSION,
raw => \@results,
}
);
if ($callback) {
$self->render(
data => "$callback($json);",
format => 'json'
);
}
else {
$self->render(
data => $json,
format => 'json'
);
}
elsif ( $template eq 'marudor' ) {
$self->res->headers->access_control_allow_origin('*');
my $json = $self->render_to_string(
json => {
departures => \@departures,
}
);
if ($callback) {
$self->render(
data => "$callback($json);",
format => 'json'
);
}
else {
$self->render(
data => $json,
format => 'json'
);
}
}
my $json = $self->render_to_string(
json => {
api_version => $api_version,
preformatted => \@departures,
version => $VERSION,
}
);
if ($callback) {
$self->render(
data => "$callback($json);",
format => 'json'
);
}
else {
$self->render(
data => $json,
format => 'json'
);
}
else {
$self->render(
$template,
departures => \@departures,
version => $VERSION,
title => "departures for ${station}",
refresh_interval => $refresh_interval + 3,
hide_opts => $hide_opts,
show_realtime => $show_realtime,
get '/_redirect' => sub {
my $station = $self->param('station');
my $params = $self->req->params;
$params->remove('station');
$params->remove('via');
for my $param (qw(platforms)) {
$params->remove($param);
}
}
$params = $params->to_string;
$self->redirect_to("/${station}/${via}?${params}");
$self->redirect_to("/${station}?${params}");
app->defaults( layout => 'default' );
get '/' => \&handle_request;
get '/:station' => \&handle_request;
get '/multi/:station' => \&handle_request;
accepts => 10,
listen => ['http://*:8092'],
Birte Kristina Friesel
committed
pid_file => '/tmp/db-fakedisplay.pid',
workers => $ENV{VRRFAKEDISPLAY_WORKERS} // 2,
app->types->type( json => 'application/json; charset=utf-8' );
app->plugin('browser_detect');