@opentripplanner/core-utils 9.0.0-alpha.30 → 9.0.0-alpha.32

package/src/__tests__/itinerary.js

@@ -2,6 +2,7 @@ import {
   calculateTncFares,
   getCompanyFromLeg,
   getDisplayedStopId,
+  getLegCost,
   isTransit
 } from "../itinerary";
 
@@ -86,4 +87,55 @@ describe("util > itinerary", () => {
       expect(getDisplayedStopId(basePlace)).toBeFalsy();
     });
   });
+
+  describe("getLegCost", () => {
+    it("should return the total cost for a leg", () => {
+      const leg = {
+        fareProducts: [
+          {
+            product: {
+              medium: { id: "cash" },
+              riderCategory: { id: "regular" },
+              name: "rideCost",
+              price: { amount: 200, currency: "USD" }
+            }
+          }
+        ]
+      };
+      const result = getLegCost(leg, "cash", "regular");
+      expect(result.price).toEqual({ amount: 200, currency: "USD" });
+    });
+
+    it("should return the transfer discount amount if a transfer was used", () => {
+      const leg = {
+        fareProducts: [
+          {
+            product: {
+              medium: { id: "cash" },
+              riderCategory: { id: "regular" },
+              name: "rideCost",
+              price: { amount: 200, currency: "USD" }
+            }
+          },
+          {
+            product: {
+              name: "transfer",
+              price: { amount: 50, currency: "USD" },
+              medium: { id: "cash" },
+              riderCategory: { id: "regular" }
+            }
+          }
+        ]
+      };
+      const result = getLegCost(leg, "cash", "regular");
+      expect(result.price).toEqual({ amount: 200, currency: "USD" });
+      expect(result.transferAmount).toEqual({ amount: 50, currency: "USD" });
+    });
+
+    it("should return undefined if no fare products exist on the leg", () => {
+      const leg = {};
+      const result = getLegCost(leg, "cash", "regular");
+      expect(result.price).toBeUndefined();
+    });
+  });
 });

package/src/itinerary.ts

@@ -614,3 +614,48 @@ export function getItineraryCost(
       { amount: 0, currency: null }
     );
 }
+
+const pickupDropoffTypeToOtp1 = otp2Type => {
+  switch (otp2Type) {
+    case "COORDINATE_WITH_DRIVER":
+      return "coordinateWithDriver";
+    case "CALL_AGENCY":
+      return "mustPhone";
+    case "SCHEDULED":
+      return "scheduled";
+    case "NONE":
+      return "none";
+    default:
+      return null;
+  }
+};
+
+export const convertGraphQLResponseToLegacy = (leg: any): any => ({
+  ...leg,
+  agencyBrandingUrl: leg.agency?.url,
+  agencyName: leg.agency?.name,
+  agencyUrl: leg.agency?.url,
+  alightRule: pickupDropoffTypeToOtp1(leg.dropoffType),
+  boardRule: pickupDropoffTypeToOtp1(leg.pickupType),
+  dropOffBookingInfo: {
+    latestBookingTime: leg.dropOffBookingInfo
+  },
+  from: {
+    ...leg.from,
+    stopCode: leg.from.stop?.code,
+    stopId: leg.from.stop?.gtfsId
+  },
+  route: leg.route?.shortName,
+  routeColor: leg.route?.color,
+  routeId: leg.route?.id,
+  routeLongName: leg.route?.longName,
+  routeShortName: leg.route?.shortName,
+  routeTextColor: leg.route?.textColor,
+  to: {
+    ...leg.to,
+    stopCode: leg.to.stop?.code,
+    stopId: leg.to.stop?.gtfsId
+  },
+  tripHeadsign: leg.trip?.tripHeadsign,
+  tripId: leg.trip?.gtfsId
+});