transportation/lib/import/gtfs/import.calendar.js

Import GTFS into a semantic model

module.exports = importCalendar
module.exports.optional = isOptional

const Service = require('../../service')
const csv = require('./csv')

function importCalendar (filename, transit, callback) {
  csv(filename, function (err, services) {
    if (err) {
      return callback(err)
    }

    services.forEach(function (serviceRow) {
      let service
      if (!transit.services[serviceRow.service_id]) {
        service = new Service(serviceRow.service_id)
        transit.services.add(service)
      } else {
        service = transit.services[serviceRow.service_id]
      }

      service.start = serviceRow.start_date
      service.end = serviceRow.end_date

      const days = [
        'monday',
        'tuesday',
        'wednesday',
        'thursday',
        'friday',
        'saturday',
        'sunday'
      ]
      days.forEach(function (day) {
        if (serviceRow[day] === '1') {
          service.days[day] = Service.Status.Operating
        } else if (serviceRow[day] === '0') {
          service.days[day] = Service.Status.NotOperating
        }
      })
    })

    callback()
  })
}

function isOptional (transit, status) {
  // determine wether the import of 'calendar.txt' is optional
  // only true iff 'calendar_dates.txt' is present

  if (status && status.imported && status.imported.calendar_dates) {
    return true
  }

  return false
}