Allow journey deletion

See issue #3

2 files changed, 88 insertions, 6 deletions

diff --git a/lib/Travelynx.pm b/lib/Travelynx.pm
index 6bdc133..25d45d3 100755
--- a/lib/Travelynx.pm
+++ b/lib/Travelynx.pm
@@ -348,6 +348,19 @@ sub startup {
 		}
 	);
 	$self->attr(
+		drop_journey_query => sub {
+			my ($self) = @_;
+
+			return $self->app->dbh->prepare(
+				qq{
+					delete from user_actions
+					where user_id = ?
+					and (id = ? or id = ?)
+				}
+			);
+		}
+	);
+	$self->attr(
 		get_userid_query => sub {
 			my ($self) = @_;
 
@@ -493,7 +506,9 @@ qq{select * from pending_mails where email = ? and num_tries > 1;}
 		},
 	);
 
-	$self->helper(sendmail => sub { state $sendmail = Travelynx::Helper::Sendmail->new; });
+	$self->helper(
+		sendmail => sub { state $sendmail = Travelynx::Helper::Sendmail->new; }
+	);
 
 	$self->helper(
 		'get_departures' => sub {
@@ -892,6 +907,50 @@ qq{select * from pending_mails where email = ? and num_tries > 1;}
 	);
 
 	$self->helper(
+		'delete_journey' => sub {
+			my ( $self, $checkin_id, $checkout_id, $checkin_epoch,
+				$checkout_epoch )
+			  = @_;
+			my $uid = $self->current_user->{id};
+
+			my @journeys = $self->get_user_travels(
+				uid         => $uid,
+				checkout_id => $checkout_id
+			);
+			if ( @journeys == 0 ) {
+				return 'Journey not found';
+			}
+			my $journey = $journeys[0];
+
+			# Double-check (comparing both ID and action epoch) to make sure we
+			# are really deleting the right journey and the user isn't just
+			# playing around with POST requests.
+			if (   $journey->{ids}[0] != $checkin_id
+				or $journey->{ids}[1] != $checkout_id
+				or $journey->{checkin}->epoch != $checkin_epoch
+				or $journey->{checkout}->epoch != $checkout_epoch )
+			{
+				return 'Invalid journey data';
+			}
+			my $query = $self->app->drop_journey_query;
+			my $success = $query->execute( $uid, $checkin_id, $checkout_id );
+			if ($success) {
+				if ( $query->rows == 2 ) {
+					return undef;
+				}
+				else {
+					return
+					  sprintf( 'Deleted %d rows, expected 2', $query->rows );
+				}
+			}
+			my $err = $self->app->drop_journey_query->errstr;
+			$self->app->log->error(
+				"Delete($uid, $checkin_id, $checkout_id): DELETE failed: $err");
+			return 'DELETE failed: ' . $err;
+		}
+	);
+
+	$self->helper(
 		'get_user_travels' => sub {
 			my ( $self, %opt ) = @_;
 
diff --git a/lib/Travelynx/Controller/Traveling.pm b/lib/Travelynx/Controller/Traveling.pm
index bf61741..a83adb0 100755
--- a/lib/Travelynx/Controller/Traveling.pm
+++ b/lib/Travelynx/Controller/Traveling.pm
@@ -175,6 +175,26 @@ sub log_action {
 			);
 		}
 	}
+	elsif ( $params->{action} eq 'delete' ) {
+		my ( $from, $to ) = split( qr{,}, $params->{ids} );
+		my $error = $self->delete_journey( $from, $to, $params->{checkin},
+			$params->{checkout} );
+		if ($error) {
+			$self->render(
+				json => {
+					success => 0,
+					error   => $error,
+				},
+			);
+		}
+		else {
+			$self->render(
+				json => {
+					success => 1,
+				},
+			);
+		}
+	}
 	else {
 		$self->render(
 			json => {
@@ -312,7 +332,7 @@ sub journey_details {
 	my ($self) = @_;
 	my ( $uid, $checkout_id ) = split( qr{-}, $self->stash('id') );
 
-	if ( not ($uid == $self->current_user->{id} and $checkout_id)) {
+	if ( not( $uid == $self->current_user->{id} and $checkout_id ) ) {
 		$self->render(
 			'journey',
 			error   => 'notfound',
@@ -322,11 +342,14 @@ sub journey_details {
 	}
 
 	my @journeys = $self->get_user_travels(
-		uid            => $uid,
-		checkout_id  => $checkout_id,
-		verbose        => 1,
+		uid         => $uid,
+		checkout_id => $checkout_id,
+		verbose     => 1,
 	);
-	if ( @journeys == 0 or not $journeys[0]{completed} or $journeys[0]{ids}[1] != $checkout_id) {
+	if (   @journeys == 0
+		or not $journeys[0]{completed}
+		or $journeys[0]{ids}[1] != $checkout_id )
+	{
 		$self->render(
 			'journey',
 			error   => 'notfound',