DB: support "best price search"

index.js

@@ -1,12 +1,14 @@
 import isObj from 'lodash/isObject.js';
 import sortBy from 'lodash/sortBy.js';
 import omit from 'lodash/omit.js';
+import {DateTime} from 'luxon';
 
 import {defaultProfile} from './lib/default-profile.js';
 import {validateProfile} from './lib/validate-profile.js';
 import {INVALID_REQUEST} from './lib/errors.js';
 import {sliceLeg} from './lib/slice-leg.js';
 import {HafasError} from './lib/errors.js';
+import {profile as dbProfile} from './p/db/index.js';
 
 // background info: https://github.com/public-transport/hafas-client/issues/286
 const FORBIDDEN_USER_AGENTS = [
@@ -268,6 +270,87 @@ const createClient = (profile, userAgent, opt = {}) => {
 		};
 	};
 
+	const bestPrices = async (from, to, opt = {}) => {
+		from = profile.formatLocation(profile, from, 'from');
+		to = profile.formatLocation(profile, to, 'to');
+
+		opt = Object.assign({
+			via: null, // let journeys pass this station?
+			transfers: -1, // maximum nr of transfers
+			transferTime: 0, // minimum time for a single transfer in minutes
+			bike: false, // only bike-friendly journeys
+			tickets: false, // return tickets?
+			polylines: false, // return leg shapes?
+			subStops: false, // parse & expose sub-stops of stations?
+			entrances: false, // parse & expose entrances of stops/stations?
+			remarks: true, // parse & expose hints & warnings?
+			scheduledDays: false, // parse & expose dates each journey is valid on?
+		}, opt);
+		if (opt.via) {
+			opt.via = profile.formatLocation(profile, opt.via, 'opt.via');
+		}
+
+		let when = new Date();
+		if (opt.departure !== undefined && opt.departure !== null) {
+			when = new Date(opt.departure);
+			if (Number.isNaN(Number(when))) {
+				throw new TypeError('opt.departure is invalid');
+			}
+			const now = new Date();
+			let today = DateTime.fromObject({
+				year: now.year, month: now.month, day: now.day,
+				hour: 0, minute: 0, second: 0, millisecond: 0,
+			}, {
+				zone: profile.timezone,
+				locale: profile.locale,
+			});
+			if (today > when) {
+				throw new TypeError('opt.departure date older than current date.');
+			}
+		}
+
+		const filters = [
+			profile.formatProductsFilter({profile}, opt.products || {}),
+		];
+
+		const query = {
+			maxChg: opt.transfers,
+			depLocL: [from],
+			viaLocL: opt.via ? [{loc: opt.via}] : [],
+			arrLocL: [to],
+			jnyFltrL: filters,
+			getTariff: Boolean(opt.tickets),
+
+			getPolyline: Boolean(opt.polylines),
+		};
+		query.outDate = profile.formatDate(profile, when);
+
+		if (profile.endpoint !== dbProfile.endpoint) {
+			throw new Error('bestPrices() only works with the DB profile.');
+		}
+
+		const {res, common} = await profile.request({profile, opt}, userAgent, {
+			cfg: {polyEnc: 'GPA'},
+			meth: 'BestPriceSearch',
+			req: profile.transformJourneysQuery({profile, opt}, query),
+		});
+		if (!Array.isArray(res.outConL)) {
+			return null;
+		}
+		// todo: outConGrpL
+
+		const ctx = {profile, opt, common, res};
+		const journeys = res.outConL.map(j => profile.parseJourney(ctx, j));
+		const bestPrices = res.outDaySegL.map(j => profile.parseBestPrice(ctx, j, journeys));
+
+		return {
+			bestPrices,
+			realtimeDataUpdatedAt: res.planrtTS && res.planrtTS !== '0'
+				? parseInt(res.planrtTS)
+				: null,
+		};
+	};
+
 	const refreshJourney = async (refreshToken, opt = {}) => {
 		if ('string' !== typeof refreshToken || !refreshToken) {
 			throw new TypeError('refreshToken must be a non-empty string.');
@@ -889,6 +972,9 @@ const createClient = (profile, userAgent, opt = {}) => {
 	if (profile.lines !== false) {
 		client.lines = lines;
 	}
+	if (profile.bestPrices !== false) {
+		client.bestPrices = bestPrices;
+	}
 	Object.defineProperty(client, 'profile', {value: profile});
 	return client;
 };

lib/default-profile.js

@@ -33,6 +33,7 @@ import {parseOperator} from '../parse/operator.js';
 import {parseHint} from '../parse/hint.js';
 import {parseWarning} from '../parse/warning.js';
 import {parseStopover} from '../parse/stopover.js';
+import {parseBestPrice} from '../parse/bestprice.js';
 
 import {formatAddress} from '../format/address.js';
 import {formatCoord} from '../format/coord.js';
@@ -91,6 +92,7 @@ const defaultProfile = {
 	parseTrip,
 	parseJourneyLeg,
 	parseJourney,
+	parseBestPrice,
 	parseLine,
 	parseStationName: (_, name) => name,
 	parseLocation,
@@ -123,6 +125,7 @@ const defaultProfile = {
 	// `departures()` method: support for `stbFltrEquiv` field?
 	departuresStbFltrEquiv: false,
 
+	bestPrices: false,
 	trip: false,
 	radar: false,
 	refreshJourney: true,

p/db/index.js

@@ -692,6 +692,7 @@ const profile = {
 
 	generateUnreliableTicketUrls: false,
 	refreshJourneyUseOutReconL: true,
+	bestPrices: true,
 	trip: true,
 	journeysFromTrip: true,
 	radar: true,

parse/bestprice.js

@@ -0,0 +1,26 @@
+
+const parseBestPrice = (ctx, outDaySeg, journeys) => {
+	const {profile, res} = ctx;
+
+	const bpjourneys = outDaySeg.conRefL
+		? outDaySeg.conRefL
+			.map(outConLIdx => journeys.find(j => j.refreshToken == res.outConL[outConLIdx].ctxRecon))
+			.filter(j => Boolean(j))
+		: [];
+
+	const amount = outDaySeg.bestPrice.amount / 100;
+	const currency = bpjourneys?.[0]?.price?.currency;
+
+	const result = {
+		journeys: bpjourneys,
+		from: profile.parseDateTime(ctx, outDaySeg.fromDate, outDaySeg.fromTime),
+		to: profile.parseDateTime(ctx, outDaySeg.toDate, outDaySeg.toTime),
+		bestPrice: amount > 0 && currency ? {amount, currency} : null,
+	};
+
+	return result;
+};
+
+export {
+	parseBestPrice,
+};

test/db-bestprice.js

@@ -0,0 +1,39 @@
+// todo: use import assertions once they're supported by Node.js & ESLint
+// https://github.com/tc39/proposal-import-assertions
+import {createRequire} from 'module';
+const require = createRequire(import.meta.url);
+
+import tap from 'tap';
+
+import {createClient} from '../index.js';
+import {profile as rawProfile} from '../p/db/index.js';
+const response = require('./fixtures/db-bestprice.json');
+import {dbBestPrices as expected} from './fixtures/db-bestprice.js';
+
+const client = createClient(rawProfile, 'public-transport/hafas-client:test');
+const {profile} = client;
+
+const opt = {
+	via: null,
+	transfers: -1,
+	transferTime: 0,
+	bike: false,
+	tickets: true,
+	polylines: true,
+	remarks: true,
+	departure: '2023-06-15',
+	products: {},
+	firstClass: false,
+	ageGroup: 'E',
+};
+
+tap.test('parses a bestprice with a DEVI leg correctly (DB)', (t) => {
+	const res = response.svcResL[0].res;
+	const common = profile.parseCommon({profile, opt, res});
+	const ctx = {profile, opt, common, res};
+	const journeys = res.outConL.map(j => profile.parseJourney(ctx, j));
+	const bestPrices = res.outDaySegL.map(j => profile.parseBestPrice(ctx, j, journeys));
+
+	t.same(bestPrices, expected.bestPrices);
+	t.end();
+});