From 2cae5a51d2f2a8002864860f452fa9474dc2ecb3 Mon Sep 17 00:00:00 2001
From: Daniel Friesel <derf@finalrewind.org>
Date: Thu, 25 Apr 2019 09:22:54 +0200
Subject: [PATCH] undo: Redirect to departure station when undoing a checkin

---
 lib/Travelynx/Controller/Traveling.pm | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/lib/Travelynx/Controller/Traveling.pm b/lib/Travelynx/Controller/Traveling.pm
index 8c15782..5282d9a 100755
--- a/lib/Travelynx/Controller/Traveling.pm
+++ b/lib/Travelynx/Controller/Traveling.pm
@@ -140,6 +140,7 @@ sub log_action {
 		}
 	}
 	elsif ( $params->{action} eq 'undo' ) {
+		my $status = $self->get_user_status;
 		my $error = $self->undo( $params->{undo_id} );
 		if ($error) {
 			$self->render(
@@ -150,10 +151,14 @@ sub log_action {
 			);
 		}
 		else {
+			my $redir = '/';
+			if ($status->{checked_in} or $status->{cancelled}) {
+				$redir = '/s/' . $status->{dep_ds100};
+			}
 			$self->render(
 				json => {
 					success     => 1,
-					redirect_to => '/',
+					redirect_to => $redir,
 				},
 			);
 		}