undo: Redirect to departure station when undoing a checkin

1 files changed, 6 insertions, 1 deletions

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