Gruntfile.js

var config = require('./package.json').config;

var request = require('superagent');
var Promise = require('es6-promise').Promise;

var fs = require('fs');
var _ = require('lodash');

var settings = require('./miccolis.json');
var GtfsMaker = require('gtfs-maker');

var gtfsMaker = new GtfsMaker({
  settings:settings,
  data:{
    miccolis:{
      format:'csv',
      ext:'.csv',
      dir:'./cache/'
    },
    timetables:{
      isDirectory:true,
      format:'csv',
      ext:'.csv',
      dir:'./extracted/timetables/',
      transform:function(item){
        function lookup(lineName){
          return _.result(
            _.find( settings.lines, function(line){
              return line.number == lineName;
            }), 'osmId' );
        }
        var matches = /MT(.*)\.csv/.exec(item.name);
        if ( !matches ){
          throw new Error('Timetables file not in correct format.');
        }
        var name = matches[1];
        var osmId = lookup( name );
        return {
          osmId:osmId,
          stopTimes:item.content
        };
      }
    }
  }
});

module.exports = function(grunt){

  // utility function to parse command line options
  function fetchOptions(){
    var include = grunt.option( "include" );
    if (!include){
      throw new Error('No line included!');
    }
    return {
      include:include.toString().split(',')
    };
  }

  grunt.registerTask('cache', function(){

   var done = this.async();

   function fetchCSV(){
    return new Promise(function(resolve, reject){
      request.get( config.miccolis.url )
          .end(function(err, res){
            if(err){
              reject(err);
            } else {
              resolve(res.text);
            }
          });
      });
    }

    function saveData(data){
      fs.writeFileSync('./cache/miccolis.csv', data);
    }

    gtfsMaker.cache()
      .then(
        fetchCSV().then(saveData)
      ).catch(function(err){
        console.log(err);
    }).then(done);

  });

  grunt.registerTask('trips', function(){
    var done = this.async();
    var tripsBuilder = require('./builders/trips');
    var trips = tripsBuilder( gtfsMaker.loadData(['masters', 'miccolis']), fetchOptions() );

    gtfsMaker.saveDataAsCsv(trips, './gtfs/trips.txt')
      .catch(function(err){
        console.log(err);
      }).then(done);

  });

  grunt.registerTask('calendar', function(){

    var done = this.async();
    var calendarBuilder = require('./builders/calendar');
    var calendar = calendarBuilder( gtfsMaker.loadData(['miccolis']), fetchOptions() );

    gtfsMaker.saveDataAsCsv(calendar, './gtfs/calendar.txt')
      .catch(function(err){
        console.log(err);
      })
      .then(function(){
        console.log('created ' + calendar.length + ' services.');
        console.log('please edit service properties in calendar.txt file.');
      }).then(done);

  });

  grunt.registerTask('frequencies', function(){
    var done = this.async();
    var frequenciesBuilder = require('./builders/frequencies');
    var frequencies = frequenciesBuilder( gtfsMaker.loadData(['miccolis']), fetchOptions() );
    gtfsMaker.saveDataAsCsv( frequencies, './gtfs/frequencies.txt' )
      .catch(function(err){
        console.log(err);
      }).then(done);

  });

  grunt.registerTask('routes', function(){
    var done = this.async();
    var routes = gtfsMaker.builders.routes( gtfsMaker.loadData(['masters']), fetchOptions() );
    gtfsMaker.saveDataAsCsv( routes, './gtfs/routes.txt' )
      .catch(function(err){
        console.log(err);
      }).then(done);
  });

  grunt.registerTask('stop_times', function(){
    var done = this.async();
    var stopTimes = gtfsMaker.builders.stopTimes(
      gtfsMaker.loadData([ 'masters', 'routes', 'stops', 'timetables', 'trips']),
      fetchOptions()
    );
    gtfsMaker.saveDataAsCsv( stopTimes, './gtfs/stop_times.txt' )
      .catch(function(err){
        console.log(err);
      }).then(done);
  });

  grunt.registerTask('shapes', function(){
    var done = this.async();
    var shapes = gtfsMaker.builders.shapes(
      gtfsMaker.loadData(['masters', 'routes', 'ways', 'nodes']),
      fetchOptions() );
    gtfsMaker.saveDataAsCsv( shapes, './gtfs/shapes.txt' )
      .catch(function(err){
        console.log(err);
      }).then(done);
  });

  grunt.registerTask('stops', function(){
    var done = this.async();
    var stops = gtfsMaker.builders.stops( gtfsMaker.loadData(['stops']) );
    gtfsMaker.saveDataAsCsv( stops, './gtfs/stops.txt' )
      .catch(function(err){
        console.log(err);
      }).then(done);
  });

  grunt.registerTask('compile', ['routes', 'shapes', 'stops', 'trips', 'frequencies', 'calendar', 'stop_times']);

  /**
   * validate GTFS
   * FIXME: move to GtfsMaker?
   */
  grunt.loadNpmTasks('grunt-exec');

  grunt.initConfig({
    exec: {
      validate: {
        command: 'python2.7 ./utils/transitfeed/feedvalidator.py gtfs/'
      }
    }
  });

  grunt.registerTask('validate', ['exec:validate']);

};