Travel API: comment support

1 files changed, 9 insertions, 0 deletions

diff --git a/lib/Travelynx/Controller/Api.pm b/lib/Travelynx/Controller/Api.pm
index f420a9e..2e1fa49 100755
--- a/lib/Travelynx/Controller/Api.pm
+++ b/lib/Travelynx/Controller/Api.pm
@@ -260,6 +260,10 @@ sub travel_v1 {
 
 		my ( $train, $error )
 		  = $self->checkin( $from_station, $train_id, $uid );
+		if ( $payload->{comment} ) {
+			$self->update_in_transit_comment(
+				sanitize( q{}, $payload->{comment} ), $uid );
+		}
 		if ( $to_station and not $error ) {
 			( $train, $error ) = $self->checkout( $to_station, 0, $uid );
 		}
@@ -284,6 +288,11 @@ sub travel_v1 {
 	elsif ( $payload->{action} eq 'checkout' ) {
 		my $to_station = sanitize( q{}, $payload->{toStation} );
 
+		if ( $payload->{comment} ) {
+			$self->update_in_transit_comment(
+				sanitize( q{}, $payload->{comment} ), $uid );
+		}
+
 		my ( $train, $error )
 		  = $self->checkout( $to_station, $payload->{force} ? 1 : 0, $uid );
 		if ($error) {