Add prefix to gifs-realtime data (#184)

Author: mattkinnia

src/lib/import-gtfs-realtime.ts

@@ -35,6 +35,7 @@ interface GtfsRealtimeTask {
   gtfsRealtimeExpirationSeconds: number;
   ignoreErrors: boolean;
   sqlitePath: string;
+  prefix: string | undefined;
   currentTimestamp: number;
   log: (message: string, newLine?: boolean) => void;
   logWarning: (message: string) => void;
@@ -150,7 +151,11 @@ function prepareRealtimeFieldValue(
     return task.currentTimestamp + task.gtfsRealtimeExpirationSeconds;
   }
 
-  const value = getNestedProperty(entity, column.default, column.source);
+  let value = getNestedProperty(entity, column.default, column.source);
+
+  if (column.prefix && task.prefix && value !== null) {
+    value = `${task.prefix}${value}`;
+  }
 
   if (column.type === 'json') {
     return sqlString.escape(JSON.stringify(value));
@@ -390,6 +395,7 @@ export async function updateGtfsRealtime(initialConfig: Config) {
           gtfsRealtimeExpirationSeconds: config.gtfsRealtimeExpirationSeconds,
           ignoreErrors: config.ignoreErrors,
           sqlitePath: config.sqlitePath,
+          prefix: agency.prefix,
           currentTimestamp: Math.floor(Date.now() / 1000),
           log: log(config),
           logWarning: logWarning(config),

src/models/gtfs-realtime/service-alert-informed_entities.ts

@@ -8,20 +8,23 @@ export const serviceAlertInformedEntities = {
       required: true,
       primary: true,
       source: 'parent.id',
+      prefix: true,
     },
     {
       name: 'stop_id',
       type: 'text',
       index: true,
       source: 'stopId',
       default: null,
+      prefix: true,
     },
     {
       name: 'route_id',
       type: 'text',
       index: true,
       source: 'routeId',
       default: null,
+      prefix: true,
     },
     {
       name: 'route_type',
@@ -36,6 +39,7 @@ export const serviceAlertInformedEntities = {
       index: true,
       source: 'trip.tripId',
       default: null,
+      prefix: true,
     },
     {
       name: 'direction_id',

src/models/gtfs-realtime/service-alerts.ts

@@ -9,6 +9,7 @@ export const serviceAlerts = {
       primary: true,
       index: true,
       source: 'id',
+      prefix: true,
     },
     {
       name: 'active_period',

src/models/gtfs-realtime/stop-time-updates.ts

@@ -8,6 +8,7 @@ export const stopTimeUpdates = {
       index: true,
       source: 'parent.tripUpdate.trip.tripId',
       default: null,
+      prefix: true,
     },
     {
       name: 'trip_start_time',
@@ -27,13 +28,15 @@ export const stopTimeUpdates = {
       index: true,
       source: 'parent.tripUpdate.trip.routeId',
       default: null,
+      prefix: true,
     },
     {
       name: 'stop_id',
       type: 'text',
       index: true,
       source: 'stopId',
       default: null,
+      prefix: true,
     },
     {
       name: 'stop_sequence',

src/models/gtfs-realtime/trip-updates.ts

@@ -9,20 +9,23 @@ export const tripUpdates = {
       primary: true,
       index: true,
       source: 'id',
+      prefix: true,
     },
     {
       name: 'vehicle_id',
       type: 'text',
       index: true,
       source: 'tripUpdate.vehicle.id',
       default: null,
+      prefix: true,
     },
     {
       name: 'trip_id',
       type: 'text',
       index: true,
       source: 'tripUpdate.trip.tripId',
       default: null,
+      prefix: true,
     },
     {
       name: 'trip_start_time',
@@ -42,6 +45,7 @@ export const tripUpdates = {
       index: true,
       source: 'tripUpdate.trip.routeId',
       default: null,
+      prefix: true,
     },
     {
       name: 'start_date',

src/models/gtfs-realtime/vehicle-positions.ts

@@ -9,6 +9,7 @@ export const vehiclePositions = {
       primary: true,
       index: true,
       source: 'id',
+      prefix: true,
     },
     {
       name: 'bearing',
@@ -51,6 +52,7 @@ export const vehiclePositions = {
       index: true,
       source: 'vehicle.trip.tripId',
       default: null,
+      prefix: true,
     },
     {
       name: 'trip_start_date',
@@ -96,6 +98,7 @@ export const vehiclePositions = {
       index: true,
       source: 'vehicle.vehicle.id',
       default: null,
+      prefix: true,
     },
     {
       name: 'vehicle_label',

src/models/gtfs/stops.ts

@@ -59,6 +59,7 @@ export const stops = {
       name: 'parent_station',
       type: 'text',
       index: true,
+      prefix: true,
     },
     {
       name: 'stop_timezone',