Switch to new carriage formation API

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;
 		}