116 lines
3.1 KiB
JavaScript
116 lines
3.1 KiB
JavaScript
const unzip = require('unzip-stream');
|
||
const csv = require('csv-parse');
|
||
const request = require('request');
|
||
const requestp = require('request-promise-native');
|
||
|
||
const overpassEndpoint = 'https://lz4.overpass-api.de/api/interpreter';
|
||
|
||
/**
|
||
* Submit an Overpass query.
|
||
*
|
||
* @async
|
||
* @param query Query in Overpass QL.
|
||
* @return Results as provided by the endpoint.
|
||
*/
|
||
const queryOverpass = query => requestp.post(
|
||
overpassEndpoint,
|
||
{form: 'data=' + query}
|
||
);
|
||
|
||
exports.queryOverpass = queryOverpass;
|
||
|
||
/**
|
||
* Process a CSV stream to extract passings.
|
||
*
|
||
* @private
|
||
* @param csvStream Stream containing CSV data.
|
||
* @param callback See fetchTamRealtime for a description of the callback.
|
||
*/
|
||
const processTamPassingStream = (csvStream, callback) =>
|
||
{
|
||
const parser = csv({
|
||
delimiter: ';',
|
||
});
|
||
|
||
const rowStream = csvStream.pipe(parser);
|
||
|
||
rowStream.on('readable', () =>
|
||
{
|
||
let row;
|
||
|
||
while (row = rowStream.read())
|
||
{
|
||
if (row.length === 0 || row[0] === 'course')
|
||
{
|
||
// Ignore les lignes invalides et l’en-tête
|
||
continue;
|
||
}
|
||
|
||
callback(null, {
|
||
course: row[0],
|
||
stopCode: row[1],
|
||
stopId: row[2],
|
||
stopName: row[3],
|
||
routeShortName: row[4],
|
||
tripHeadsign: row[5],
|
||
directionId: row[6],
|
||
departureTime: row[7],
|
||
isTheorical: row[8],
|
||
delaySec: row[9],
|
||
destArCode: row[10],
|
||
});
|
||
}
|
||
});
|
||
|
||
rowStream.on('end', () => callback(null, null));
|
||
rowStream.on('error', err => callback(err));
|
||
};
|
||
|
||
const tamRealtimeEndpoint = 'http://data.montpellier3m.fr/node/10732/download';
|
||
|
||
/**
|
||
* Fetch realtime passings for the current day across the network.
|
||
*
|
||
* @param callback Called for each passing during parsing. First argument will
|
||
* be non-null only if an error occurred. Second argument will contain passings
|
||
* or be null if the end was reached.
|
||
*/
|
||
const fetchTamRealtime = (callback) =>
|
||
{
|
||
const csvStream = request(tamRealtimeEndpoint);
|
||
processTamPassingStream(csvStream, callback);
|
||
};
|
||
|
||
exports.fetchTamRealtime = fetchTamRealtime;
|
||
|
||
const tamTheoreticalEndpoint =
|
||
'http://data.montpellier3m.fr/node/10731/download';
|
||
const tamTheoreticalFileName = 'offre_du_jour.csv';
|
||
|
||
/**
|
||
* Fetch theoretical passings for the current day across the network.
|
||
*
|
||
* @param callback Called for each passing during parsing. First argument will
|
||
* be non-null only if an error occurred. Second argument will contain passings
|
||
* or be null if the end was reached.
|
||
*/
|
||
const fetchTamTheoretical = (callback) =>
|
||
{
|
||
const fileStream = request(tamTheoreticalEndpoint).pipe(unzip.Parse());
|
||
|
||
fileStream.on('entry', entry =>
|
||
{
|
||
if (entry.type !== 'File' || entry.path !== tamTheoreticalFileName)
|
||
{
|
||
entry.autodrain();
|
||
return;
|
||
}
|
||
|
||
processTamPassingStream(entry, callback);
|
||
});
|
||
|
||
fileStream.on('error', err => callback(err));
|
||
};
|
||
|
||
exports.fetchTamTheoretical = fetchTamTheoretical;
|