diff options
Diffstat (limited to 'lib/Travelynx')
-rw-r--r-- | lib/Travelynx/Command/work.pm | 44 | ||||
-rw-r--r-- | lib/Travelynx/Helper/DBDB.pm | 76 |
2 files changed, 86 insertions, 34 deletions
diff --git a/lib/Travelynx/Command/work.pm b/lib/Travelynx/Command/work.pm index 44d780a..42b2bc8 100644 --- a/lib/Travelynx/Command/work.pm +++ b/lib/Travelynx/Command/work.pm @@ -84,8 +84,15 @@ sub run { if ( $journey->class <= 16 and $found_dep->rt_dep->epoch > $now->epoch ) { - $self->app->add_wagonorder( $uid, 1, $train_id, - $found_dep->sched_dep, $journey->number ); + $self->app->add_wagonorder( + uid => $uid, + train_id => $journey->id, + is_departure => 1, + eva => $dep, + datetime => $found_dep->sched_dep, + train_type => $journey->type, + train_no => $journey->number, + ); $self->app->add_stationinfo( $uid, 1, $journey->id, $found_dep->loc->eva ); } @@ -102,8 +109,15 @@ sub run { if ( $journey->class <= 16 and $found_arr->rt_arr->epoch - $now->epoch < 600 ) { - $self->app->add_wagonorder( $uid, 0, $train_id, - $found_dep->sched_dep, $journey->number ); + $self->app->add_wagonorder( + uid => $uid, + train_id => $journey->id, + is_arrival => 1, + eva => $arr, + datetime => $found_arr->sched_dep, + train_type => $journey->type, + train_no => $journey->number, + ); $self->app->add_stationinfo( $uid, 0, $journey->id, $found_dep->loc->eva, $found_arr->loc->eva ); } @@ -203,8 +217,15 @@ sub run { } else { $self->app->add_route_timestamps( $uid, $train, 1 ); - $self->app->add_wagonorder( $uid, 1, $train->train_id, - $train->sched_departure, $train->train_no ); + $self->app->add_wagonorder( + uid => $uid, + train_id => $train->train_id, + is_departure => 1, + eva => $dep, + datetime => $train->sched_departure, + train_type => $train->type, + train_no => $train->train_no + ); $self->app->add_stationinfo( $uid, 1, $train->train_id, $dep, $arr ); } @@ -279,8 +300,15 @@ sub run { and $now->epoch > $entry->{real_arr_ts} ) ? 1 : 0 ); - $self->app->add_wagonorder( $uid, 0, $train->train_id, - $train->sched_departure, $train->train_no ); + $self->app->add_wagonorder( + uid => $uid, + train_id => $train->train_id, + is_arrival => 1, + eva => $arr, + datetime => $train->sched_departure, + train_type => $train->type, + train_no => $train->train_no + ); $self->app->add_stationinfo( $uid, 0, $train->train_id, $dep, $arr ); } diff --git a/lib/Travelynx/Helper/DBDB.pm b/lib/Travelynx/Helper/DBDB.pm index da3bfb1..c7c9d4c 100644 --- a/lib/Travelynx/Helper/DBDB.pm +++ b/lib/Travelynx/Helper/DBDB.pm @@ -27,39 +27,53 @@ sub new { } sub has_wagonorder_p { - my ( $self, $ts, $train_no ) = @_; - my $api_ts = $ts->strftime('%Y%m%d%H%M'); - my $url - = "https://ist-wr.noncd.db.de/wagenreihung/1.0/${train_no}/${api_ts}"; + my ( $self, %opt ) = @_; + + my $datetime = $opt{datetime}->clone->set_time_zone('UTC'); + my %param = ( + administrationId => 80, + category => $opt{train_type}, + date => $datetime->strftime('%Y-%m-%d'), + evaNumber => $opt{eva}, + number => $opt{train_no}, + time => $datetime->rfc3339 =~ s{(?=Z)}{.000}r + ); + + my $url = sprintf( '%s?%s', +'https://www.bahn.de/web/api/reisebegleitung/wagenreihung/vehicle-sequence', + join( '&', map { $_ . '=' . $param{$_} } keys %param ) ); + my $cache = $self->{realtime_cache}; my $promise = Mojo::Promise->new; + my $debug_prefix + = "has_wagonorder_p($opt{train_type} $opt{train_no} @ $opt{eva})"; if ( my $content = $cache->get("HEAD $url") ) { if ( $content eq 'n' ) { - $self->{log} - ->debug("has_wagonorder_p(${train_no}/${api_ts}): n (cached)"); + $self->{log}->debug("${debug_prefix}: n (cached)"); return $promise->reject; } else { - $self->{log} - ->debug("has_wagonorder_p(${train_no}/${api_ts}): y (cached)"); + $self->{log}->debug("${debug_prefix}: y (cached)"); return $promise->resolve($content); } } - $self->{user_agent}->request_timeout(5)->head_p( $url => $self->{header} ) + $self->{user_agent}->request_timeout(5)->get_p( $url => $self->{header} ) ->then( sub { my ($tx) = @_; if ( $tx->result->is_success ) { - $self->{log} - ->debug("has_wagonorder_p(${train_no}/${api_ts}): a"); + $self->{log}->debug("${debug_prefix}: a"); $cache->set( "HEAD $url", 'a' ); + my $body = decode( 'utf-8', $tx->res->body ); + my $json = JSON->new->decode($body); + $cache->freeze( $url, $json ); $promise->resolve('a'); } else { - $self->{log} - ->debug("has_wagonorder_p(${train_no}/${api_ts}): n"); + my $code = $tx->code; + $self->{log}->debug("${debug_prefix}: n (HTTP $code)"); $cache->set( "HEAD $url", 'n' ); $promise->reject; } @@ -67,7 +81,8 @@ sub has_wagonorder_p { } )->catch( sub { - $self->{log}->debug("has_wagonorder_p(${train_no}/${api_ts}): n"); + my ($err) = @_; + $self->{log}->debug("${debug_prefix}: n ($err)"); $cache->set( "HEAD $url", 'n' ); $promise->reject; return; @@ -77,17 +92,29 @@ sub has_wagonorder_p { } sub get_wagonorder_p { - my ( $self, $api, $ts, $train_no ) = @_; - my $api_ts = $ts->strftime('%Y%m%d%H%M'); - my $url - = "https://ist-wr.noncd.db.de/wagenreihung/1.0/${train_no}/${api_ts}"; + my ( $self, %opt ) = @_; + + my $datetime = $opt{datetime}->clone->set_time_zone('UTC'); + my %param = ( + administrationId => 80, + category => $opt{train_type}, + date => $datetime->strftime('%Y-%m-%d'), + evaNumber => $opt{eva}, + number => $opt{train_no}, + time => $datetime->rfc3339 =~ s{(?=Z)}{.000}r + ); + + my $url = sprintf( '%s?%s', +'https://www.bahn.de/web/api/reisebegleitung/wagenreihung/vehicle-sequence', + join( '&', map { $_ . '=' . $param{$_} } keys %param ) ); + my $debug_prefix + = "get_wagonorder_p($opt{train_type} $opt{train_no} @ $opt{eva})"; my $cache = $self->{realtime_cache}; my $promise = Mojo::Promise->new; if ( my $content = $cache->thaw($url) ) { - $self->{log} - ->debug("get_wagonorder_p(${train_no}/${api_ts}): (cached)"); + $self->{log}->debug("${debug_prefix}: (cached)"); $promise->resolve($content); return $promise; } @@ -100,15 +127,13 @@ sub get_wagonorder_p { if ( $tx->result->is_success ) { my $body = decode( 'utf-8', $tx->res->body ); my $json = JSON->new->decode($body); - $self->{log} - ->debug("get_wagonorder_p(${train_no}/${api_ts}): success"); + $self->{log}->debug("${debug_prefix}: success"); $cache->freeze( $url, $json ); $promise->resolve($json); } else { my $code = $tx->code; - $self->{log}->debug( - "get_wagonorder_p(${train_no}/${api_ts}): HTTP ${code}"); + $self->{log}->debug("${debug_prefix}: HTTP ${code}"); $promise->reject("HTTP ${code}"); } return; @@ -116,8 +141,7 @@ sub get_wagonorder_p { )->catch( sub { my ($err) = @_; - $self->{log} - ->debug("get_wagonorder_p(${train_no}/${api_ts}): error ${err}"); + $self->{log}->debug("${debug_prefix}: error ${err}"); $promise->reject($err); return; } |