Fix a crash when a train has unscheduled additional stops starting at the selected station

1 files changed, 15 insertions, 1 deletions

diff --git a/lib/Travel/Status/DE/IRIS/Result.pm b/lib/Travel/Status/DE/IRIS/Result.pm
index d4d994e..e1cb513 100644
--- a/lib/Travel/Status/DE/IRIS/Result.pm
+++ b/lib/Travel/Status/DE/IRIS/Result.pm
@@ -102,6 +102,13 @@ sub add_ar {
 		time_zone => 'Europe/Berlin',
 	);
 
+	# unscheduled arrivals may not appear in the plan, but we do need to
+	# know their planned arrival time
+	if ( $attrib{plan_arrival_ts} ) {
+		$self->{sched_arrival}
+		  = $strp->parse_datetime( $attrib{plan_arrival_ts} );
+	}
+
 	if ( $attrib{arrival_ts} ) {
 		$self->{arrival} = $strp->parse_datetime( $attrib{arrival_ts} );
 		$self->{delay}
@@ -139,6 +146,13 @@ sub add_dp {
 		time_zone => 'Europe/Berlin',
 	);
 
+	# unscheduled arrivals may not appear in the plan, but we do need to
+	# know their planned arrival time
+	if ( $attrib{plan_departure_ts} ) {
+		$self->{sched_departure}
+		  = $strp->parse_datetime( $attrib{plan_departure_ts} );
+	}
+
 	if ( $attrib{departure_ts} ) {
 		$self->{departure} = $strp->parse_datetime( $attrib{departure_ts} );
 		$self->{delay}
@@ -535,7 +549,7 @@ sub translate_msg {
 		64 => 'Weichenstörung',
 		55 => 'Technische Störung an einem anderen Zug',        # ?
 		57 => 'Zusätzlicher Halt',                              # ?
-		58 => 'Umleitung',                                    # ?
+		58 => 'Umleitung',                                       # ?
 		61 => 'Türstörung',
 		62 => 'Behobene technische Störung am Zug',
 		63 => 'Technische Untersuchung am Zug',