/**
 * Copyright (C) 2011 Brian Ferris <bdferris@onebusaway.org>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.onebusaway.transit_data_federation.bundle.utilities;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import org.onebusaway.container.ContainerLibrary;
import org.onebusaway.csv_entities.schema.DefaultEntitySchemaFactory;
import org.onebusaway.gtfs.impl.GtfsDaoImpl;
import org.onebusaway.gtfs.model.Stop;
import org.onebusaway.gtfs.model.Trip;
import org.onebusaway.gtfs.serialization.GtfsEntitySchemaFactory;
import org.onebusaway.gtfs.serialization.GtfsWriter;
import org.onebusaway.transit_data_federation.bundle.tasks.GtfsReadingSupport;
import org.springframework.context.ConfigurableApplicationContext;

/**
 * Utility for combining multiple GTFS feeds into one unified feed
 * 
 * @author bdferris
 */
public class GtfsCombinerMain {

  public static void main(String[] args) throws IOException {

    if (args.length != 2) {
      System.err.println("usage: bundle.xml output_path");
      System.exit(-1);
    }

    List<String> paths = new ArrayList<String>();

    paths.add(args[0]);

    ConfigurableApplicationContext context = ContainerLibrary.createContext(paths);
    GtfsDaoImpl store = new GtfsDaoImpl();

    DefaultEntitySchemaFactory schema = GtfsEntitySchemaFactory.createEntitySchemaFactory();
    GenericAdditionalFieldMapping.addGenericFieldMapping(schema, Stop.class,
        "stop_direction", "stopDirection");
    GenericAdditionalFieldMapping.addGenericFieldMapping(schema, Trip.class,
        "block_sequence_id", "blockSequenceId");

    GtfsReadingSupport.readGtfsIntoStore(context, store,schema);

    writeGtfs(new File(args[1]), store, schema);
  }

  private static void writeGtfs(File outputDirectory, GtfsDaoImpl store,
      DefaultEntitySchemaFactory schema) throws IOException {
    
    GtfsWriter writer = new GtfsWriter();
    writer.setOutputLocation(outputDirectory);
    writer.setEntitySchemaFactory(schema);
    writer.run(store);
  }
}