/**
 * 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.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.onebusaway.csv_entities.EntityHandler;
import org.onebusaway.gtfs.model.Agency;
import org.onebusaway.gtfs.model.AgencyAndId;
import org.onebusaway.gtfs.model.Stop;
import org.onebusaway.gtfs.serialization.GtfsReader;
import org.onebusaway.transit_data_federation.bundle.model.GtfsBundle;
import org.onebusaway.transit_data_federation.bundle.tasks.EntityReplacementStrategyFactory;
import org.onebusaway.transit_data_federation.services.AgencyAndIdLibrary;
import org.onebusaway.utility.IOLibrary;

/**
 * Given a stop-consolidation list, verifies that the specified stops still
 * exist in the target GTFS feeds.
 * 
 * @author bdferris
 * @see EntityReplacementStrategyFactory
 */
public class GtfsStopReplacementVerificationMain {

  @SuppressWarnings("unchecked")
  public static void main(String[] args) throws IOException {

    if (args.length == 0) {
      System.err.println("usage: stop-consolidation-file gtfs_path [gtfs_path:defaultAgencyId data-sources.xml ...]");
      System.exit(-1);
    }

    EntityHandlerImpl handler = new EntityHandlerImpl();

    List<String> paths = new ArrayList<String>();
    for (int i = 1; i < args.length; i++) {
      paths.add(args[i]);
    }

    List<GtfsBundle> bundles = UtilityLibrary.getGtfsBundlesForArguments(paths);

    for (GtfsBundle bundle : bundles) {

      GtfsReader reader = new GtfsReader();
      reader.setDefaultAgencyId(bundle.getDefaultAgencyId());
      reader.setInputLocation(bundle.getPath());
      for (Map.Entry<String, String> entry : bundle.getAgencyIdMappings().entrySet())
        reader.addAgencyIdMapping(entry.getKey(), entry.getValue());
      reader.getEntityClasses().retainAll(
          Arrays.asList(Agency.class, Stop.class));
      reader.addEntityHandler(handler);
      reader.run();
    }

    Set<AgencyAndId> ids = handler.getIds();

    InputStream in = IOLibrary.getPathAsInputStream(args[0]);
    BufferedReader reader = new BufferedReader(new InputStreamReader(in));

    String line = null;

    while ((line = reader.readLine()) != null) {
      if (line.startsWith("#") || line.startsWith("{{{")
          || line.startsWith("}}}") || line.length() == 0)
        continue;
      String[] tokens = line.split("\\s+");
      AgencyAndId id = AgencyAndIdLibrary.convertFromString(tokens[0]);
      if (!ids.contains(id))
        System.out.println(id + " <- " + line);
    }

  }

  private static class EntityHandlerImpl implements EntityHandler {

    private Set<AgencyAndId> _ids = new HashSet<AgencyAndId>();

    public Set<AgencyAndId> getIds() {
      return _ids;
    }

    @Override
    public void handleEntity(Object bean) {
      if (!(bean instanceof Stop))
        return;
      Stop stop = (Stop) bean;
      _ids.add(stop.getId());
    }

  }
}