bus-detective/bus-detective

namespace :metro do
  desc "Import metro data for a new agency"
  task :import, [:gtfs_endpoint] => [:environment] do |t, args|
    gtfs_endpoint = args[:gtfs_endpoint]

    if gtfs_endpoint.blank?
      puts "GTFS endpoint is required for import.\n\n"
      puts "rake metro:import['http://www.go-metro.com/uploads/GTFS/google_transit_info.zip']"
      exit 1
    end

    ActiveRecord::Base.logger.level = 2
    agency = Agency.find_or_create_by!(gtfs_endpoint: gtfs_endpoint)
    Metro::Importer.new(agency, logger: Logger.new(STDOUT)).import!
  end

  desc "Import metro data for all existing agencies"
  task import_existing: :environment do
    ActiveRecord::Base.logger.level = 2
    Agency.find_each do |agency|
      Metro::Importer.new(agency, logger: Logger.new(STDOUT)).import!
    end
  end

  desc "Update route_stop cache"
  task update_route_stop_cache: :environment do
    ActiveRecord::Base.logger.level = 2
    RouteStop.update_cache
  end

  desc "Titleize existing metro data"
  task titleize: :environment do
    Stop.find_each do |s|
      s.update(
        name: Metro::StringHelper.titleize(s.name),
        description: Metro::StringHelper.titleize(s.description),
      )
    end
    Route.find_each do |r|
      r.update(
        long_name: Metro::StringHelper.titleize(r.long_name),
      )
    end
    Trip.find_each do |t|
      t.update(
        headsign: Metro::StringHelper.titleize_headsign(t.headsign),
      )
    end
  end

  desc "Recalculate route text colors"
  task update_route_text_colors: :environment do
    Route.find_each do |r|
      r.update(
        text_color: Metro::ColorHelper.text_color_for_bg_color(r.color),
      )
    end
  end
end