@transportme/load-ptv-gtfs 1.1.6 → 1.1.7

package/lib/loader/TripLoader.mjs

@@ -26,6 +26,7 @@ export default class TripLoader {
   #stopCache = {}
 
   #shapeIDMap = {}
+  #directionIDMap = {}
 
   #tripsToLoad = []
 
@@ -164,7 +165,9 @@ export default class TripLoader {
     if (!processedData) return
 
     this.#shapeIDMap[tripData.shapeID] = tripData.routeGTFSID
-
+    if (!this.#directionIDMap[tripData.routeGTFSID]) this.#directionIDMap[tripData.routeGTFSID] = {}
+    this.#directionIDMap[tripData.routeGTFSID][rawTripData.getHeadsign()] = rawTripData.getGTFSDirection()
+    
     this.#tripsToLoad.push(processedData)
   }
 
@@ -211,4 +214,8 @@ export default class TripLoader {
   getShapeIDMap() {
     return this.#shapeIDMap
   }
+
+  getDirectionIDMap() {
+    return this.#directionIDMap
+  }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@transportme/load-ptv-gtfs",
-  "version": "1.1.6",
+  "version": "1.1.7",
   "description": "",
   "main": "index.mjs",
   "scripts": {

package/test/gtfs-trip-loader.test.mjs

@@ -109,4 +109,31 @@ describe('The TripLoader class', () => {
 
     expect(shapeIDMap['2-ALM-vpt-1.1.R']).to.equal('2-ALM')
   })
+
+  it('Should return a mapping of Route + GTFS Direction ID to PTV Direction Names', async () => {
+    let database = new LokiDatabaseConnection('test-db')
+    let stops = await database.createCollection('stops')
+    let routes = await database.createCollection('routes')
+    let trips = await database.createCollection('gtfs timetables')
+
+    let stopLoader = new StopsLoader(stopsFile, suburbs, TRANSIT_MODES.metroTrain, database)
+    await stopLoader.loadStops()
+
+    let routeLoader = new RouteLoader(routesFile, agencyFile, TRANSIT_MODES.metroTrain, database)
+    await routeLoader.loadRoutes()
+
+    let routeIDMap = routeLoader.getRouteIDMap()
+
+    let tripLoader = new TripLoader({
+      tripsFile, stopTimesFile,
+      calendarFile, calendarDatesFile
+    }, TRANSIT_MODES.metroTrain, database)
+    
+    await tripLoader.loadTrips({ routeIDMap })
+    let directionIDMap = tripLoader.getDirectionIDMap()
+
+    expect(directionIDMap).to.deep.equal({
+      '2-ALM': { 'Alamein': 0, 'Camberwell': 1 }
+    })
+  })
 })