Send bus notifications if first leg is bus

src/main/java/org/opentripplanner/middleware/triptracker/ManageTripTracking.java

@@ -2,14 +2,19 @@
 
 import org.eclipse.jetty.http.HttpStatus;
 import org.opentripplanner.middleware.models.TrackedJourney;
+import org.opentripplanner.middleware.otp.response.Leg;
 import org.opentripplanner.middleware.persistence.Persistence;
 import org.opentripplanner.middleware.triptracker.instruction.TripInstruction;
 import org.opentripplanner.middleware.triptracker.interactions.busnotifiers.BusOperatorActions;
 import org.opentripplanner.middleware.triptracker.response.EndTrackingResponse;
 import org.opentripplanner.middleware.triptracker.response.TrackingResponse;
 import spark.Request;
 
+import static org.opentripplanner.middleware.triptracker.TravelerLocator.isAtStartOfLeg;
 import static org.opentripplanner.middleware.utils.ConfigUtils.getConfigPropertyAsInt;
+import static org.opentripplanner.middleware.utils.ItineraryUtils.getRouteGtfsIdFromLeg;
+import static org.opentripplanner.middleware.utils.ItineraryUtils.isBusLeg;
+import static org.opentripplanner.middleware.utils.ItineraryUtils.legsMatch;
 import static org.opentripplanner.middleware.utils.JsonUtils.logMessageAndHalt;
 
 public class ManageTripTracking {
@@ -145,19 +150,53 @@ private static EndTrackingResponse completeJourney(TripTrackingData tripData, bo
             tripData.trip.journeyState.matchingItinerary,
             Persistence.otpUsers.getById(tripData.trip.userId)
         );
-        BusOperatorActions
-            .getDefault()
-            .handleCancelNotificationAction(travelerPosition);
+        cancelBusNotification(travelerPosition);
         TrackedJourney trackedJourney = travelerPosition.trackedJourney;
         trackedJourney.end(isForciblyEnded);
         Persistence.trackedJourneys.updateField(trackedJourney.id, TrackedJourney.END_TIME_FIELD_NAME, trackedJourney.endTime);
         Persistence.trackedJourneys.updateField(trackedJourney.id, TrackedJourney.END_CONDITION_FIELD_NAME, trackedJourney.endCondition);
 
-        // Provide response.
         return new EndTrackingResponse(
             TripInstruction.NO_INSTRUCTION,
             TripStatus.ENDED.name()
         );
+    }
+
+    /**
+     * Cancel bus notifications which are no longer needed/relevant.
+     */
+    private static void cancelBusNotification(TravelerPosition travelerPosition) {
+        Leg busLeg = travelerPosition.nextLeg;
+        if (shouldCancelBusNotificationForStartOfTrip(travelerPosition)) {
+            busLeg = travelerPosition.expectedLeg;
+        }
+        BusOperatorActions
+            .getDefault()
+            .handleCancelNotificationAction(travelerPosition, busLeg);
+    }
 
+    /**
+     * Traveler is still waiting to board the bus at the start of a trip and notification has been sent.
+     */
+    public static boolean shouldCancelBusNotificationForStartOfTrip(TravelerPosition travelerPosition) {
+        return hasSentBusNotificationForStartOfTrip(travelerPosition) && isWaitingForBusAtStartOfTrip(travelerPosition);
+    }
+
+    /**
+     * Bus notification has been sent for the start of the trip.
+     */
+    private static boolean hasSentBusNotificationForStartOfTrip(TravelerPosition travelerPosition) {
+        var routeId = getRouteGtfsIdFromLeg(travelerPosition.expectedLeg);
+        return routeId != null && travelerPosition.trackedJourney.busNotificationMessages.containsKey(routeId);
+    }
+
+    /**
+     * Traveler is waiting for a bus at the start of a trip.
+     */
+    private static boolean isWaitingForBusAtStartOfTrip(TravelerPosition travelerPosition) {
+        return
+            legsMatch(travelerPosition.expectedLeg, travelerPosition.firstLegOfTrip) &&
+            isBusLeg(travelerPosition.expectedLeg) &&
+            isAtStartOfLeg(travelerPosition);
     }
-}
+}

src/main/java/org/opentripplanner/middleware/triptracker/TravelerLocator.java

@@ -33,6 +33,7 @@
 import static org.opentripplanner.middleware.utils.GeometryUtils.getDistance;
 import static org.opentripplanner.middleware.utils.GeometryUtils.isPointBetween;
 import static org.opentripplanner.middleware.utils.ItineraryUtils.isBusLeg;
+import static org.opentripplanner.middleware.utils.ItineraryUtils.legsMatch;
 
 /**
  * Locate the traveler in relation to the nearest step or destination and provide the appropriate instructions.
@@ -57,14 +58,14 @@ public static String getInstruction(
     ) {
         if (hasRequiredWalkLeg(travelerPosition)) {
             if (hasRequiredTripStatus(tripStatus)) {
-                TripInstruction tripInstruction = alignTravelerToTrip(travelerPosition, isStartOfTrip, tripStatus);
+                TripInstruction tripInstruction = alignTravelerToTrip(travelerPosition, isStartOfTrip);
                 if (tripInstruction != null) {
                     return tripInstruction.build();
                 }
             }
 
             if (tripStatus.equals(TripStatus.DEVIATED)) {
-                TripInstruction tripInstruction = getBackOnTrack(travelerPosition, isStartOfTrip, tripStatus);
+                TripInstruction tripInstruction = getBackOnTrack(travelerPosition, isStartOfTrip);
                 if (tripInstruction != null) {
                     return tripInstruction.build();
                 }
@@ -111,10 +112,9 @@ private static boolean hasRequiredTripStatus(TripStatus tripStatus) {
     @Nullable
     private static TripInstruction getBackOnTrack(
         TravelerPosition travelerPosition,
-        boolean isStartOfTrip,
-        TripStatus tripStatus
+        boolean isStartOfTrip
     ) {
-        TripInstruction instruction = alignTravelerToTrip(travelerPosition, isStartOfTrip, tripStatus);
+        TripInstruction instruction = alignTravelerToTrip(travelerPosition, isStartOfTrip);
         if (instruction != null && instruction.hasInstruction()) {
             return instruction;
         }
@@ -130,16 +130,12 @@ private static TripInstruction getBackOnTrack(
     @Nullable
     public static TripInstruction alignTravelerToTrip(
         TravelerPosition travelerPosition,
-        boolean isStartOfTrip,
-        TripStatus tripStatus
+        boolean isStartOfTrip
     ) {
         Locale locale = travelerPosition.locale;
 
         if (isApproachingEndOfLeg(travelerPosition)) {
-            if (isBusLeg(travelerPosition.nextLeg) && isWithinOperationalNotifyWindow(travelerPosition)) {
-                BusOperatorActions
-                    .getDefault()
-                    .handleSendNotificationAction(tripStatus, travelerPosition);
+            if (sendBusNotification(travelerPosition)) {
                 // Regardless of whether the notification is sent or qualifies, provide a 'wait for bus' instruction.
                 return new WaitForTransitInstruction(travelerPosition.nextLeg, travelerPosition.currentTime, locale);
             }
@@ -157,6 +153,31 @@ public static TripInstruction alignTravelerToTrip(
         return null;
     }
 
+    /**
+     * Send bus notification if the first leg is a bus leg or approaching a bus leg and within the notify window.
+     */
+    public static boolean sendBusNotification(TravelerPosition travelerPosition) {
+        Leg busLeg = atStartOfTransitTrip(travelerPosition) ? travelerPosition.expectedLeg : travelerPosition.nextLeg;
+        if (isBusLeg(busLeg) && isWithinOperationalNotifyWindow(travelerPosition.currentTime, busLeg)) {
+            BusOperatorActions
+                .getDefault()
+                .handleSendNotificationAction(travelerPosition, busLeg);
+            return true;
+        }
+        return false;
+    }
+
+    /**
+     * A trip which starts with a transit leg and the traveler is on that leg.
+     */
+    private static boolean atStartOfTransitTrip(TravelerPosition travelerPosition) {
+        return
+            travelerPosition.expectedLeg != null &&
+            travelerPosition.firstLegOfTrip != null &&
+            travelerPosition.firstLegOfTrip.transitLeg &&
+            legsMatch(travelerPosition.expectedLeg, travelerPosition.firstLegOfTrip);
+    }
+
     /**
      * Align the traveler's position to the nearest transit stop or destination.
      */
@@ -166,6 +187,11 @@ public static TripInstruction alignTravelerToTransitTrip(TravelerPosition travel
         Leg expectedLeg = travelerPosition.expectedLeg;
         String finalStop = expectedLeg.to.name;
 
+        if (sendBusNotification(travelerPosition)) {
+            // Regardless of whether the notification is sent or qualifies, provide a 'wait for bus' instruction.
+            return new WaitForTransitInstruction(expectedLeg, travelerPosition.currentTime, locale);
+        }
+
         if (isApproachingEndOfLeg(travelerPosition)) {
             return new GetOffHereTransitInstruction(finalStop, locale);
         }
@@ -211,6 +237,14 @@ private static boolean isApproachingEndOfLeg(TravelerPosition travelerPosition)
         return getDistanceToEndOfLeg(travelerPosition) <= TRIP_INSTRUCTION_UPCOMING_RADIUS;
     }
 
+    /**
+     * Is the traveler at the start of a leg.
+     */
+    public static boolean isAtStartOfLeg(TravelerPosition travelerPosition) {
+        Coordinates legDestination = new Coordinates(travelerPosition.expectedLeg.from);
+        return getDistance(travelerPosition.currentPosition, legDestination) <= TRIP_INSTRUCTION_UPCOMING_RADIUS;
+    }
+
     /**
      * Is the traveler at the leg destination.
      */
@@ -222,11 +256,11 @@ public static boolean isAtEndOfLeg(TravelerPosition travelerPosition) {
      * Make sure the traveler is on schedule or ahead of schedule (but not too far) to be within an operational window
      * for the bus service.
      */
-    public static boolean isWithinOperationalNotifyWindow(TravelerPosition travelerPosition) {
-        var busDepartureTime = getBusDepartureTime(travelerPosition.nextLeg);
+    public static boolean isWithinOperationalNotifyWindow(Instant currentTime, Leg busLeg) {
+        var busDepartureTime = getBusDepartureTime(busLeg);
         return
-            (travelerPosition.currentTime.equals(busDepartureTime) || travelerPosition.currentTime.isBefore(busDepartureTime)) &&
-            ACCEPTABLE_AHEAD_OF_SCHEDULE_IN_MINUTES >= getMinutesAheadOfDeparture(travelerPosition.currentTime, busDepartureTime);
+            (currentTime.equals(busDepartureTime) || currentTime.isBefore(busDepartureTime)) &&
+            ACCEPTABLE_AHEAD_OF_SCHEDULE_IN_MINUTES >= getMinutesAheadOfDeparture(currentTime, busDepartureTime);
     }
 
     /**

src/main/java/org/opentripplanner/middleware/triptracker/TravelerPosition.java

@@ -14,6 +14,7 @@
 import static org.opentripplanner.middleware.triptracker.ManageLegTraversal.getNextLeg;
 import static org.opentripplanner.middleware.triptracker.ManageLegTraversal.getSegmentFromPosition;
 import static org.opentripplanner.middleware.utils.GeometryUtils.getDistanceFromLine;
+import static org.opentripplanner.middleware.utils.ItineraryUtils.getFirstLeg;
 
 public class TravelerPosition {
 
@@ -44,6 +45,9 @@ public class TravelerPosition {
     /** The traveler's locale. */
     public Locale locale;
 
+    /** The first leg of the trip. **/
+    public Leg firstLegOfTrip;
+
     public TravelerPosition(TrackedJourney trackedJourney, Itinerary itinerary, OtpUser otpUser) {
         TrackingLocation lastLocation = trackedJourney.locations.get(trackedJourney.locations.size() - 1);
         currentTime = lastLocation.timestamp.toInstant();
@@ -61,6 +65,7 @@ public TravelerPosition(TrackedJourney trackedJourney, Itinerary itinerary, OtpU
             }
             this.locale = I18nUtils.getOtpUserLocale(otpUser);
         }
+        firstLegOfTrip = getFirstLeg(itinerary);
     }
 
     /** Used for unit testing. */
@@ -83,6 +88,14 @@ public TravelerPosition(Leg nextLeg, Instant currentTime) {
         this.currentTime = currentTime;
     }
 
+    /** Used for unit testing. */
+    public TravelerPosition(Leg expectedLeg, TrackedJourney trackedJourney, Leg first, Coordinates currentPosition) {
+        this.expectedLeg = expectedLeg;
+        this.trackedJourney = trackedJourney;
+        this.firstLegOfTrip = first;
+        this.currentPosition = currentPosition;
+    }
+
     /** Computes the current deviation in meters from the expected itinerary. */
     public double getDeviationMeters() {
         return getDistanceFromLine(legSegmentFromPosition.start, legSegmentFromPosition.end, currentPosition);

src/main/java/org/opentripplanner/middleware/triptracker/interactions/busnotifiers/BusOperatorActions.java

@@ -1,8 +1,8 @@
 package org.opentripplanner.middleware.triptracker.interactions.busnotifiers;
 
 import com.fasterxml.jackson.databind.JsonNode;
+import org.opentripplanner.middleware.otp.response.Leg;
 import org.opentripplanner.middleware.triptracker.TravelerPosition;
-import org.opentripplanner.middleware.triptracker.TripStatus;
 import org.opentripplanner.middleware.utils.JsonUtils;
 import org.opentripplanner.middleware.utils.YamlUtils;
 import org.slf4j.Logger;
@@ -48,8 +48,8 @@ public BusOperatorActions(List<AgencyAction> agencyActions) {
     /**
      * Get the action that matches the given agency id.
      */
-    public AgencyAction getAgencyAction(TravelerPosition travelerPosition) {
-        String agencyId = removeAgencyPrefix(getAgencyGtfsIdFromLeg(travelerPosition.nextLeg));
+    public AgencyAction getAgencyAction(Leg busLeg) {
+        String agencyId = removeAgencyPrefix(getAgencyGtfsIdFromLeg(busLeg));
         if (agencyId != null) {
             for (AgencyAction agencyAction : agencyActions) {
                 if (agencyAction.agencyId.equalsIgnoreCase(agencyId)) {
@@ -63,12 +63,12 @@ public AgencyAction getAgencyAction(TravelerPosition travelerPosition) {
     /**
      * Get the correct action for agency and send notification.
      */
-    public void handleSendNotificationAction(TripStatus tripStatus, TravelerPosition travelerPosition) {
-        AgencyAction action = getAgencyAction(travelerPosition);
+    public void handleSendNotificationAction(TravelerPosition travelerPosition, Leg busLeg) {
+        AgencyAction action = getAgencyAction(busLeg);
         if (action != null) {
             BusOperatorInteraction interaction = getBusOperatorInteraction(action);
             try {
-                interaction.sendNotification(tripStatus, travelerPosition);
+                interaction.sendNotification(travelerPosition, busLeg);
             } catch (Exception e) {
                 LOG.error("Could not trigger class {} for agency {}", action.trigger, action.agencyId, e);
                 throw new RuntimeException(e);
@@ -81,12 +81,12 @@ public void handleSendNotificationAction(TripStatus tripStatus, TravelerPosition
     /**
      * Get the correct action for agency and cancel notification.
      */
-    public void handleCancelNotificationAction(TravelerPosition travelerPosition) {
-        AgencyAction action = getAgencyAction(travelerPosition);
+    public void handleCancelNotificationAction(TravelerPosition travelerPosition, Leg busLeg) {
+        AgencyAction action = getAgencyAction(busLeg);
         if (action != null) {
             BusOperatorInteraction interaction = getBusOperatorInteraction(action);
             try {
-                interaction.cancelNotification(travelerPosition);
+                interaction.cancelNotification(travelerPosition, busLeg);
             } catch (Exception e) {
                 LOG.error("Could not trigger class {} for agency {}", action.trigger, action.agencyId, e);
                 throw new RuntimeException(e);

src/main/java/org/opentripplanner/middleware/triptracker/interactions/busnotifiers/BusOperatorInteraction.java

@@ -1,11 +1,12 @@
 package org.opentripplanner.middleware.triptracker.interactions.busnotifiers;
 
+import org.opentripplanner.middleware.otp.response.Leg;
 import org.opentripplanner.middleware.triptracker.TravelerPosition;
 import org.opentripplanner.middleware.triptracker.TripStatus;
 
 public interface BusOperatorInteraction {
 
-    void sendNotification(TripStatus tripStatus, TravelerPosition travelerPosition);
+    void sendNotification(TravelerPosition travelerPosition, Leg busLeg);
 
-    void cancelNotification(TravelerPosition travelerPosition);
+    void cancelNotification(TravelerPosition travelerPosition, Leg busLeg);
 }

src/main/java/org/opentripplanner/middleware/triptracker/interactions/busnotifiers/UsRideGwinnettNotifyBusOperator.java

@@ -4,8 +4,8 @@
 import org.eclipse.jetty.http.HttpMethod;
 import org.eclipse.jetty.http.HttpStatus;
 import org.opentripplanner.middleware.models.TrackedJourney;
+import org.opentripplanner.middleware.otp.response.Leg;
 import org.opentripplanner.middleware.triptracker.TravelerPosition;
-import org.opentripplanner.middleware.triptracker.TripStatus;
 import org.opentripplanner.middleware.utils.HttpUtils;
 import org.opentripplanner.middleware.utils.JsonUtils;
 import org.slf4j.Logger;
@@ -79,8 +79,8 @@ private static List<String> getBusOperatorNotifierQualifyingRoutes() {
     /**
      * Stage notification to bus operator by making sure all required conditions are met.
      */
-    public void sendNotification(TripStatus tripStatus, TravelerPosition travelerPosition) {
-        var routeId = getRouteGtfsIdFromLeg(travelerPosition.nextLeg);
+    public void sendNotification(TravelerPosition travelerPosition, Leg busLeg) {
+        var routeId = getRouteGtfsIdFromLeg(busLeg);
         try {
             if (
                 hasNotSentNotificationForRoute(travelerPosition.trackedJourney, routeId) &&
@@ -98,13 +98,13 @@ public void sendNotification(TripStatus tripStatus, TravelerPosition travelerPos
     }
 
     /**
-     * Cancel a previously sent notification for the next bus leg.
+     * Cancel a previously sent notification for the expected or next leg.
      */
-    public void cancelNotification(TravelerPosition travelerPosition) {
-        var routeId = getRouteGtfsIdFromLeg(travelerPosition.nextLeg);
+    public void cancelNotification(TravelerPosition travelerPosition, Leg busLeg) {
+        var routeId = getRouteGtfsIdFromLeg(busLeg);
         try {
             if (
-                isBusLeg(travelerPosition.nextLeg) && routeId != null &&
+                isBusLeg(busLeg) && routeId != null &&
                 hasNotCanceledNotificationForRoute(travelerPosition.trackedJourney, routeId)
             ) {
                 Map<String, String> busNotificationRequests = travelerPosition.trackedJourney.busNotificationMessages;

src/main/java/org/opentripplanner/middleware/utils/ItineraryUtils.java

@@ -178,7 +178,7 @@ private static boolean isAfterServiceStart(ZonedDateTime time) {
     /**
      * Check whether a new leg of an itinerary matches the previous itinerary leg for the purposes of trip monitoring.
      */
-    private static boolean legsMatch(Leg referenceItineraryLeg, Leg candidateItineraryLeg) {
+    public static boolean legsMatch(Leg referenceItineraryLeg, Leg candidateItineraryLeg) {
         // for now don't analyze non-transit legs
         if (!referenceItineraryLeg.transitLeg) return true;
 
@@ -360,4 +360,15 @@ public static String getStopGtfsIdFromPlace(Place place) {
     public static String getRouteShortNameFromLeg(Leg leg) {
         return (leg != null && leg.route != null) ? leg.route.shortName : null;
     }
+
+    /**
+     * Get the first leg in an itinerary.
+     */
+    public static Leg getFirstLeg(Itinerary itinerary) {
+        if (itinerary != null && itinerary.legs != null && !itinerary.legs.isEmpty()) {
+            return itinerary.legs.get(0);
+        }
+        return null;
+    }
+
 }