Untitled

require_relative 'route.rb'
require_relative 'train.rb'
require_relative 'cargo_train.rb'
require_relative 'station.rb'
require_relative 'passenger_wagon.rb'
require_relative 'cargo_wagon.rb'
require_relative 'passenger_train.rb'


=begin train=CargoTrain.new '123'

w1=CargoWagon.new '123'
w2=CargoWagon.new '321'


wp=PassengerWagon.new '555'



train.add_wagon(wp)

=end


loop do
  puts 'Enter 1 to create station'
  puts 'Enter 2 to create train'
  puts 'Enter 3 to manage route'
  puts 'Enter 4 to appoint route to the train'
  puts 'Enter 5 to move the train'
  puts 'Enter 6 to see stations'

  x=gets.chomp.to_i

  if x == 1 
    print 'Enter Station name: '
    station_name = gets.chomp
    print 'Enter key word for the station: '
    key = gets.chomp
    key = Station.new(station_name)
    puts "Statin #{station_name} created"
  end


  if x == 2
    print 'What type of train do you want to create? (Cargo/Passenger): '
    type = gets.chomp
    if type == 'cargo'
      print 'Enter number of train: '
      number = gets.chomp.to_i
      print 'Enter key word for this train: '
      key=gets.chomp
      key = CargoTrain.new(number)
      puts "Train number #{number} created"
    end

    if type == 'passenger'
      print 'Enter number of train: '
      number = gets.chomp.to_i
      print 'Enter key word for this train: '
      key=gets.chomp
      key = PassengerTrain.new(number)
      puts "Train number #{number} created"
    end
  end

  if x == 3
    print "To create new route enter 1; to manage existing route enter 2: "
      x = gets.chomp.to_i
      if x == 1
        print "Enter number of route: "
        y = gets.chomp
        print "Enter first station name: "
        first_station = gets.chomp
        print "Enter last station name: "
        last_station = gets.chomp
        print "Enter key word for this route: "
        key = gets.chomp
        key = Route.new(first_station, last_station)
        puts "Route #{y} created"
      end 

      if x == 2
        print "To add station enter: 1; to delete station enter: 2: "
        y = gets.chomp.to_i
        if y == 1
          print "Enter name of new station: "
          new_station = gets.chomp
          print "Enter Route name: "
          route_number = gets.chomp
          route_number.add_station(new_station)
        end

        if y == 2
          print "Enter name of station you want to delete: "
          delete_station = gets.chomp
          print "Enter Route name: "
          route_number = gets.chomp
          route_number.delete_station(delete_station)
        end
    end
  end

  if x == 4
    print "Enter train number: "
    train_number = gets.chomp
    print "Enter route number: "
    route_number = gets.chomp
    train_number.route=(route_number)
  end

  if x == 5
    print "Enter train number: "
    train_number = gets.chomp
    print "Enter 1 to move forward; enter 2 to move back: "
    y = gets.chomp
    if y == 1
      train_number.move_forward
    end
    if y == 2
      train_number.move_back
    end
  end

  if x == 6
    print "Enter route number: "
    route_number = gets.chomp
    route_number.stations
  end
end