index.js

const path = require("path")
const gtfsDefaultBuilders = require('./src/geojson_to_gtfs/gtfsBuilders')
const writeGtfs = require('./src/geojson_to_gtfs/writeGtfs')
const fs = require('fs')
const {
    osmToGeojson,
    OSMOverpassDownloader,
    OSMPBFReader
} = require('./src/osm_to_geojson')
const geojsonToGtfs = require('./src/geojson_to_gtfs')
const geojsonToTrufiTPData = require('./src/geojson_to_trufi_tp_data')

const defaultGeojsonOptions = {
    osmDataGetter: null,
    transformTypes: ["bus", "share_taxi", "aerialway", "train", "subway", "monorail", "tram", "trolleybus", "ferry", "light_rail"],
    skipRoute: () => true
}
const defatulgtfsOptions = {
    agencyTimezone: "America/La_Paz",
    agencyUrl: "https://www.example.com/",
    defaultCalendar: () => "Mo-Su 06:00-23:00",
    frequencyHeadway: () => 300,
    vehicleSpeed: () => 50,
    fakeStops: () => false,
    skipStopsWithinDistance: 100,
    stopNameBuilder: (stops) => {
        if (!stops) {
            stops = ["unnamed"]
        }
        return stops.join(" and ")
    },
}
const defaultOutFiles = {
    outputDir: null,
    routes: false,
    log: false,
    stops: false,
    readme: true,
    gtfs: false,
    trufiTPData: false
}
async function osmToGtfs(config) {
    const outputFiles = Object.assign({}, defaultOutFiles, config.outputFiles || {})
    const geojsonOptions = Object.assign({}, defaultGeojsonOptions, config.geojsonOptions || {})
    const gtfsOptions = Object.assign({}, defatulgtfsOptions, config.gtfsOptions || {})
    const gtfsBuilders = Object.assign({}, gtfsDefaultBuilders, config.gtfsBuilders || {})
    const { outputDir } = outputFiles
    if (outputDir && !fs.existsSync(path.dirname(outputDir))) {
        throw new Error('Output directory does not exist')
    }

    const geojson = await osmToGeojson(geojsonOptions)
    const gtfs = (outputFiles.gtfs) ? await geojsonToGtfs(geojson.geojsonFeatures, geojson.stops, gtfsOptions, gtfsBuilders) : null
    const trufiTPData = (outputFiles.trufiTPData) ? await geojsonToTrufiTPData(geojson.geojsonFeatures, geojson.stops) : null

    if (outputDir) {
        if (fs.existsSync(path.join(outputDir))) {
            fs.rmdirSync(path.join(outputDir), { recursive: true })
        }
        fs.mkdirSync(path.join(outputDir));
        if (outputFiles.routes) {
            fs.mkdirSync(path.join(outputDir, `routes`));
            for (const key in geojson.geojsonFeatures) {
                const feature = geojson.geojsonFeatures[key]
                fs.writeFileSync(path.join(outputDir, `/routes/${key}.geojson`), JSON.stringify(feature))
            }
        } if (outputFiles.log)
            fs.writeFileSync(path.join(outputDir, 'log.json'), JSON.stringify(geojson.log))
        if (outputFiles.stops)
            fs.writeFileSync(path.join(outputDir, 'stops.json'), JSON.stringify(geojson.stops))
        if (outputFiles.readme)
            fs.writeFileSync(path.join(outputDir, 'README.md'), geojson.readme)
        if (outputFiles.gtfs) {
            fs.mkdirSync(path.join(outputDir, `gtfs`));
            writeGtfs(gtfs, path.join(outputDir, 'gtfs'));
        }
        if (outputFiles.trufiTPData) {
            fs.mkdirSync(path.join(outputDir, `trufiTPData`));
            fs.writeFileSync(path.join(outputDir, 'trufiTPData', 'routes.json'), JSON.stringify(trufiTPData.routes))
            fs.writeFileSync(path.join(outputDir, 'trufiTPData', 'stops.json'), JSON.stringify(trufiTPData.stops))
            fs.writeFileSync(path.join(outputDir, 'trufiTPData', 'search.json'), JSON.stringify(geojson.stops))
        }
    }
}

module.exports = {
    osmToGtfs,
    OSMOverpassDownloader,
    OSMPBFReader
}