From a830acc69877e204e5136e0862959164691683ca Mon Sep 17 00:00:00 2001
From: Daniel Friesel <derf@finalrewind.org>
Date: Wed, 12 Feb 2020 20:38:24 +0100
Subject: Handle cancelled arrival at checkout when departure at checkin is not
 cancelled

---
 lib/Travelynx/Command/work.pm | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

(limited to 'lib/Travelynx/Command/work.pm')

diff --git a/lib/Travelynx/Command/work.pm b/lib/Travelynx/Command/work.pm
index 405716c..9658ff6 100644
--- a/lib/Travelynx/Command/work.pm
+++ b/lib/Travelynx/Command/work.pm
@@ -134,7 +134,15 @@ sub run {
 						checkout_station_id => $arr
 					}
 				);
-				$self->app->add_route_timestamps( $uid, $train, 0 );
+				if ( $train->arrival_is_cancelled ) {
+
+                  # check out (adds a cancelled journey and resets journey state
+                  # to destination selection)
+					$self->app->checkout( $arr, 0, $uid );
+				}
+				else {
+					$self->app->add_route_timestamps( $uid, $train, 0 );
+				}
 			}
 			elsif ( $entry->{real_arr_ts} ) {
 				my ( undef, $error ) = $self->app->checkout( $arr, 1, $uid );
-- 
cgit v1.2.3