Directly use DBRIS::Formation for wagon order; enable proxy support thereHEAD4.37.15main

2 files changed, 31 insertions, 99 deletions

diff --git a/lib/DBInfoscreen/Helper/DBRIS.pm b/lib/DBInfoscreen/Helper/DBRIS.pm
index e780213..593dc90 100644
--- a/lib/DBInfoscreen/Helper/DBRIS.pm
+++ b/lib/DBInfoscreen/Helper/DBRIS.pm
@@ -29,8 +29,8 @@ sub new {
 
 }
 
-sub get_journey_p {
-	my ( $self, %opt ) = @_;
+sub get_agent {
+	my ($self) = @_;
 
 	my $agent = $self->{user_agent};
 
@@ -40,11 +40,17 @@ sub get_journey_p {
 		$agent->proxy->https($proxy);
 	}
 
+	return $agent;
+}
+
+sub get_journey_p {
+	my ( $self, %opt ) = @_;
+
 	return Travel::Status::DE::DBRIS->new_p(
 		journey    => $opt{id},
 		cache      => $self->{realtime_cache},
 		promise    => 'Mojo::Promise',
-		user_agent => $agent->request_timeout(10)
+		user_agent => $self->get_agent->request_timeout(10)
 	);
 }
 
@@ -56,20 +62,12 @@ sub get_polyline_p {
 	my $trip_id = $opt{id};
 	my $promise = Mojo::Promise->new;
 
-	my $agent = $self->{user_agent};
-
-	if ( my $proxy = $ENV{DBFAKEDISPLAY_DBRIS_PROXY} ) {
-		$agent = Mojo::UserAgent->new;
-		$agent->proxy->http($proxy);
-		$agent->proxy->https($proxy);
-	}
-
 	Travel::Status::DE::DBRIS->new_p(
 		journey       => $trip_id,
 		with_polyline => 1,
 		cache         => $self->{realtime_cache},
 		promise       => 'Mojo::Promise',
-		user_agent    => $agent->request_timeout(10)
+		user_agent    => $self->get_agent->request_timeout(10)
 	)->then(
 		sub {
 			my ($dbris) = @_;
@@ -90,4 +88,25 @@ sub get_polyline_p {
 	return $promise;
 }
 
+sub get_wagonorder_p {
+	my ( $self, %opt ) = @_;
+
+	$self->{log}
+	  ->debug("get_wagonorder_p($opt{train_type} $opt{train_no} @ $opt{eva})");
+
+	return Travel::Status::DE::DBRIS->new_p(
+		cache         => $self->{main_cache},
+		failure_cache => $self->{realtime_cache},
+		promise       => 'Mojo::Promise',
+		user_agent    => $self->get_agent->request_timeout(10),
+		formation     => {
+			departure    => $opt{datetime},
+			eva          => $opt{eva},
+			train_type   => $opt{train_type},
+			train_number => $opt{train_no}
+		},
+		developer_mode => $self->{log}->is_level('debug') ? 1 : 0,
+	);
+}
+
 1;
diff --git a/lib/DBInfoscreen/Helper/Wagonorder.pm b/lib/DBInfoscreen/Helper/Wagonorder.pm
index 9981244..8ec1a06 100644
--- a/lib/DBInfoscreen/Helper/Wagonorder.pm
+++ b/lib/DBInfoscreen/Helper/Wagonorder.pm
@@ -25,93 +25,6 @@ sub new {
 
 }
 
-sub get_p {
-	my ( $self, %opt ) = @_;
-
-	my %param;
-
-	if ( $opt{param} ) {
-		%param = %{ $opt{param} };
-		delete $param{e};
-	}
-	else {
-		my $datetime = $opt{datetime}->clone->set_time_zone('UTC');
-		%param = (
-			administrationId => 80,
-			category         => $opt{train_type},
-			date             => $datetime->strftime('%Y-%m-%d'),
-			evaNumber        => $opt{eva},
-			number           => $opt{train_number},
-			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;
-
-	if ( my $content = $self->{main_cache}->thaw($url) ) {
-		$self->{log}->debug("wagonorder->get_p($url): cached");
-		if ( $content->{error} ) {
-			return $promise->reject(
-"GET $url: HTTP $content->{error}{code} $content->{error}{message} (cachd)"
-			);
-		}
-		return $promise->resolve( $content, \%param );
-	}
-
-	if ( my $content = $self->{realtime_cache}->thaw($url) ) {
-		$self->{log}->debug("wagonorder->get_p($url): cached");
-		if ( $content->{error} ) {
-			return $promise->reject(
-"GET $url: HTTP $content->{error}{code} $content->{error}{message} (cachd)"
-			);
-		}
-		return $promise->resolve( $content, \%param );
-	}
-
-	$self->{user_agent}->request_timeout(10)->get_p( $url => $self->{header} )
-	  ->then(
-		sub {
-			my ($tx) = @_;
-
-			if ( my $err = $tx->error ) {
-				my $json = {
-					error => {
-						id  => $err->{code},
-						msg => $err->{message}
-					}
-				};
-				$self->{log}->debug(
-					"wagonorder->get_p($url): HTTP $err->{code} $err->{message}"
-				);
-				$self->{realtime_cache}->freeze( $url, $json );
-				$promise->reject("GET $url: HTTP $err->{code} $err->{message}");
-				return;
-			}
-
-			$self->{log}->debug("wagonorder->get_p($url): OK");
-			my $json = $tx->res->json;
-			$json->{ts} = DateTime->now( time_zone => 'Europe/Berlin' )
-			  ->strftime('%d.%m.%Y %H:%M');
-
-			$self->{main_cache}->freeze( $url, $json );
-			$promise->resolve( $json, \%param );
-			return;
-		}
-	)->catch(
-		sub {
-			my ($err) = @_;
-			$self->{log}->warn("wagonorder->get_p($url): $err");
-			$promise->reject("GET $url: $err");
-			return;
-		}
-	)->wait;
-	return $promise;
-}
-
 sub get_stationinfo_p {
 	my ( $self, $eva ) = @_;