The Next Train assignment.

next_train.rb

EASTEREGG = "journey.txt"

class TrainSchedule
  FILENAME = 'schedule.txt'
  @@schedule = {}

  def initialize
    if @@schedule.empty?
      File.open(FILENAME, 'r').each do |line|
        train, time = line.split(",")
        @@schedule[train] = time.to_f
      end
    end
  end

  def next(now)
    @@schedule.select { |train, time| time > now }.first
  end

end

trains = TrainSchedule.new

puts "What time are you leaving? "
print "> "
leaving_time = gets.chomp.to_f

train, time = trains.next(leaving_time)
puts "You should catch #{train} leaving at #{time}\n"

# journey easter egg
if time == 24
  File.open(EASTEREGG, 'r').each { |line| puts line }
end