1 files changed, 60 insertions, 35 deletions

diff --git a/lib/Travelynx/Helper/DBDB.pm b/lib/Travelynx/Helper/DBDB.pm
index da3bfb1..79e9c0b 100644
--- a/lib/Travelynx/Helper/DBDB.pm
+++ b/lib/Travelynx/Helper/DBDB.pm
@@ -27,48 +27,65 @@ 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 $cache   = $self->{realtime_cache};
+	my ( $self, %opt ) = @_;
+
+	$opt{train_type} //= q{};
+	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{$_} } sort keys %param ) );
+
 	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 ( my $content = $self->{main_cache}->get("HEAD $url")
+		// $self->{realtime_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}: ${content} (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");
-				$cache->set( "HEAD $url", 'a' );
+				$self->{log}->debug("${debug_prefix}: a");
+				$self->{main_cache}->set( "HEAD $url", 'a' );
+				my $body = decode( 'utf-8', $tx->res->body );
+				my $json = JSON->new->decode($body);
+				$self->{main_cache}->freeze( $url, $json );
 				$promise->resolve('a');
 			}
 			else {
-				$self->{log}
-				  ->debug("has_wagonorder_p(${train_no}/${api_ts}): n");
-				$cache->set( "HEAD $url", 'n' );
+				my $code = $tx->res->code;
+				$self->{log}->debug("${debug_prefix}: n (HTTP $code)");
+				$self->{realtime_cache}->set( "HEAD $url", 'n' );
 				$promise->reject;
 			}
 			return;
 		}
 	)->catch(
 		sub {
-			$self->{log}->debug("has_wagonorder_p(${train_no}/${api_ts}): n");
-			$cache->set( "HEAD $url", 'n' );
+			my ($err) = @_;
+			$self->{log}->debug("${debug_prefix}: n ($err)");
+			$self->{realtime_cache}->set( "HEAD $url", 'n' );
 			$promise->reject;
 			return;
 		}
@@ -77,17 +94,28 @@ 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{$_} } sort 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)");
+	if ( my $content = $self->{main_cache}->thaw($url) ) {
+		$self->{log}->debug("${debug_prefix}: (cached)");
 		$promise->resolve($content);
 		return $promise;
 	}
@@ -100,15 +128,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");
-				$cache->freeze( $url, $json );
+				$self->{log}->debug("${debug_prefix}: success");
+				$self->{main_cache}->freeze( $url, $json );
 				$promise->resolve($json);
 			}
 			else {
-				my $code = $tx->code;
-				$self->{log}->debug(
-					"get_wagonorder_p(${train_no}/${api_ts}): HTTP ${code}");
+				my $code = $tx->res->code;
+				$self->{log}->debug("${debug_prefix}: HTTP ${code}");
 				$promise->reject("HTTP ${code}");
 			}
 			return;
@@ -116,8 +142,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;
 		}