@transportme/load-ptv-gtfs 1.3.2 → 1.3.3

package/lib/gtfs-parser/GTFSTrip.mjs

@@ -48,7 +48,7 @@ export default class GTFSTrip {
    * @returns {GTFSTrip} A trip instance representing the data being passed in
    */
   static create(data, mode) {
-    let types = [ SmartrakTrip, MetroRRBTrip, MetroTrip, GTFSTrip ]
+    let types = [ SmartrakTrip, MetroTrip, GTFSTrip ]
     for (let type of types) {
       if (type.canProcess(data, mode)) return new type(data)
     }
@@ -148,12 +148,12 @@ export class MetroTrip extends GTFSTrip {
   constructor(data) {
     super(data)
 
-    let runID = data.id.match(/^02-\w{3}-.*-(\w{1,2}\d{3})$/)
+    let runID = data.id.match(/^02-\w{3}-.*-(\w{1,2}\d{2,3})$/)
     this.#runID = runID[1]
   }
 
   static canProcess(data, mode) {
-    return mode === TRANSIT_MODES.metroTrain && data.id.match(/^02-\w{3}-.*-(\w{1,2}\d{3})$/)
+    return mode === TRANSIT_MODES.metroTrain && data.id.match(/^02-\w{3}-.*-(\w{1,2}\d{2,3})$/)
   }
 
   getTrainDirection() {
@@ -163,11 +163,12 @@ export class MetroTrip extends GTFSTrip {
 
   getTripData() {
     let runID = this.getRunID()
+    let tripID = this.getTripID()
 
     return {
       ...super.getTripData(),
       runID,
-      isRailReplacementBus: runID.length === 5 && runID[0] === 'B',
+      isRailReplacementBus: (runID.length === 5 && runID[0] === 'B') || !!tripID.match(/^02-\w{3}-R-/),
       direction: this.getTrainDirection()
     }
   }

package/package.json

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

package/test/gtfs-trips.test.mjs

@@ -11,21 +11,20 @@ const __dirname = path.dirname(__filename)
 
 const tripsFile = path.join(__dirname, 'sample-data', 'trips', 'trips.txt')
 const busTripsFile = path.join(__dirname, 'sample-data', 'trips', 'bus_trips.txt')
+const calendar = new GTFSCalendar({
+  id: 'T1',
+  startDay: '20241122',
+  endDay: '20241129',
+  daysOfWeek: ["0","0","0","0","1","1","0"]
+})
 
 describe('The GTFSTrip class', () => {
   it('Should process the skeleton trip data and populate the operation days', () => {
-    let calendar = new GTFSCalendar({
-      id: 'T1',
-      startDay: '20241122',
-      endDay: '20241129',
-      daysOfWeek: ["0","0","0","0","1","1","0"]
-    })
-
     // route_id,service_id,trip_id,shape_id,trip_headsign,direction_id,block_id
     // "6-452-mjp-1","T0_16","1.T0.6-452-mjp-1.1.H","6-452-mjp-1.1.H","Eynesbury","0",""
     let trip = new GTFSTrip({
       routeGTFSID: '6-452',
-      calendar: calendar,
+      calendar,
       id: '1.T0.6-452-mjp-1.1.H',
       shapeID: '6-452-mjp-1.1.H',
       headsign: 'Eynesbury',

package/test/new-metro-loader-test.test.mjs

@@ -7,6 +7,8 @@ import { expect } from 'chai'
 import suburbs from './sample-data/suburbs.json' with { type: 'json' }
 import RouteLoader from '../lib/loader/RouteLoader.mjs'
 import TripLoader from '../lib/loader/TripLoader.mjs'
+import GTFSCalendar from '../lib/gtfs-parser/GTFSCalendar.mjs'
+import { MetroTrip } from '../lib/gtfs-parser/GTFSTrip.mjs'
 
 const __filename = url.fileURLToPath(import.meta.url)
 const __dirname = path.dirname(__filename)
@@ -198,4 +200,45 @@ describe('The GTFS Loaders with the new Metro data', () => {
       expect(trip.stopTimings[2].arrivalTime).to.equal('00:10')
     })
   })
+
+  describe('The MetroTrip subclass', () => {
+    it('Should detect rail buses', () => {
+      let calendar = new GTFSCalendar({
+        id: 'T1',
+        startDay: '20241122',
+        endDay: '20241129',
+        daysOfWeek: ["0","0","0","0","1","1","0"]
+      })
+
+      expect(new MetroTrip({
+        routeGTFSID: '2-PKM',
+        calendar,
+        id: '02-PKM--54-T6-C962',
+        shapeID: '2-PKM-vpt-54.70.R',
+        headsign: 'South Yarra',
+        direction: '0',
+        block: ''
+      }).getTripData().isRailReplacementBus).to.be.false
+
+      expect(new MetroTrip({
+        routeGTFSID: '2-PKM',
+        calendar,
+        id: '02-PKM-R-6-T5_bp-BF400',
+        shapeID: '2-PKM-R-vpt-6.7.R',
+        headsign: 'South Yarra',
+        direction: '0',
+        block: ''
+      }).getTripData().isRailReplacementBus).to.be.true
+
+      expect(new MetroTrip({
+        routeGTFSID: '2-PKM',
+        calendar,
+        id: '02-PKM-R-6-T5_bp-BF22',
+        shapeID: '2-PKM-R-vpt-6.7.R',
+        headsign: 'South Yarra',
+        direction: '0',
+        block: ''
+      }).getTripData().isRailReplacementBus).to.be.true
+    })
+  })
 })