@transportme/load-ptv-gtfs 1.4.3 → 1.4.4

package/lib/loader/RouteLoader.mjs

@@ -21,10 +21,12 @@ export default class RouteLoader {
    * 
    * @param {string} routesFile The path for the routes.txt file
    * @param {string} agencyFile The path for the agency.txt file
+   * @param {string} mode The route's mode
    * @param {DatabaseConnection} database A database connection
+   * @param {string} collectionName The collection to store the routes in. Defaults to gtfs-routes
    */
-  constructor(routesFile, agencyFile, mode, database) {
-    this.#routes = database.getCollection('gtfs-routes')
+  constructor(routesFile, agencyFile, mode, database, collectionName = 'gtfs-routes') {
+    this.#routes = database.getCollection(collectionName)
     this.#mode = mode
     this.#routesReader = new GTFSRouteReader(routesFile, mode)
     this.#agencyFile = agencyFile

package/lib/loader/ShapeLoader.mjs

@@ -13,9 +13,10 @@ export default class ShapeLoader {
    * 
    * @param {string} shapeFile The path for the shapes.txt file
    * @param {DatabaseConnection} database A database connection
+   * @param {string} collectionName The collection to store the routes in. Defaults to gtfs-routes
    */
-  constructor(shapeFile, database) {
-    this.#routes = database.getCollection('gtfs-routes')
+  constructor(shapeFile, database, collectionName = 'gtfs-routes') {
+    this.#routes = database.getCollection(collectionName)
     this.#shapeReader = new GTFSShapeReader(shapeFile)
   }
 

package/lib/loader/StopsLoader.mjs

@@ -20,9 +20,11 @@ export default class StopsLoader {
    * @param {string} mode The transit mode being loaded
    * @param {DatabaseConnection} database A database connection
    * @param {function} [suburbHook] A hook that returns the stop's suburb, or null if unavailable. This function is called before processing from the stop's name or location
+   * @param {string} collectionName The collection to store the stops in. Defaults to gtfs-stops
+   * 
    */
-  constructor(stopsFile, suburbs, mode, database, suburbHook) {
-    this.#stops = database.getCollection('gtfs-stops')
+  constructor(stopsFile, suburbs, mode, database, suburbHook, collectionName = 'gtfs-stops') {
+    this.#stops = database.getCollection(collectionName)
     this.#mode = mode
     this.#stopsReader = new GTFSStopsReader(stopsFile, suburbs, suburbHook)
   }

package/lib/loader/TripLoader.mjs

@@ -42,11 +42,19 @@ export default class TripLoader {
    * @param {string} paths.calendarDatesFile The path for the calendar_dates.txt file
    * @param {string} mode The transit mode being loaded
    * @param {DatabaseConnection} database A database connection
+   * @param {Object} colls Used to specify the collection names to write into
+   * @param {string} colls.routeColl The collection to store the routes in. Defaults to gtfs-routes
+   * @param {string} colls.stopColl The collection to store the stops in. Defaults to gtfs-stops
+   * @param {string} colls.timetableColl The collection to store the timetables in. Defaults to gtfs-gtfs timetables
    */
-  constructor({ tripsFile, stopTimesFile, calendarFile, calendarDatesFile }, mode, database) {
-    this.#routes = database.getCollection('gtfs-routes')
-    this.#stops = database.getCollection('gtfs-stops')
-    this.#trips = database.getCollection('gtfs-gtfs timetables')
+  constructor({ tripsFile, stopTimesFile, calendarFile, calendarDatesFile }, mode, database, {
+    routeColl = 'gtfs-routes',
+    stopColl = 'gtfs-stops',
+    timetableColl = 'gtfs-gtfs timetables'
+  } = {}) {
+    this.#routes = database.getCollection(routeColl)
+    this.#stops = database.getCollection(stopColl)
+    this.#trips = database.getCollection(timetableColl)
 
     this.#mode = mode
 

package/lib/post-processing/set-route-stops.mjs

@@ -7,10 +7,19 @@ import { TRANSIT_MODES } from '../constants.mjs'
  * Loads route stop data into the database. This is generated based off the timetable data.
  * 
  * @param {DatabaseConnection} database The database connection to use
+ * @param {Object} directionsMap Used to store PTV API direction data 
+ * @param {function} onMilestone Called for every 25 routes processed
+ * @param {Object} colls Used to specify the collection names to write into
+ * @param {string} colls.routeColl The collection to store the routes in. Defaults to gtfs-routes
+ * @param {string} colls.stopColl The collection to store the stops in. Defaults to gtfs-stops
+ * @param {string} colls.timetableColl The collection to store the timetables in. Defaults to gtfs-gtfs timetables
  */
-export default async function setRouteStops(database, directionsMap, onMilestone = () => {}) {
-  let gtfsTimetables = database.getCollection('gtfs-gtfs timetables')
-  let routes = database.getCollection('gtfs-routes')
+export default async function setRouteStops(database, directionsMap, onMilestone = () => {}, {
+    routeColl = 'gtfs-routes',
+    timetableColl = 'gtfs-gtfs timetables'
+  } = {}) {
+  let gtfsTimetables = database.getCollection(timetableColl)
+  let routes = database.getCollection(routeColl)
 
   let allRoutes = await routes.distinct('routeGTFSID')
 

package/package.json

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