directory.2.rb

require 'csv'
#create an empty array
@students = []
#method for checking if spelling of the month was correct
def spelling(month)
  #making an array for the months
  month_array =[ 
    "January", 
    "February", 
    "March", 
    "April", 
    "May", 
    "June", 
    "July", 
    "August", 
    "September", 
    "October", 
    "November", 
    "December"
  ]
  #checking if the array includes our user unput, if yes returns true
  month_array.include?(month)
end
#method for getting list of names from the user input into array
def input_students
  puts "Please enter the names of the students"
  puts "To finish, just hit return twice"
  #get the first name
  name = STDIN.gets.chomp
  #while the name is not empty, repeat this code
  while !name.empty? do
    puts "Please enter country of birth"
    birth = STDIN.gets.chomp
    puts "Please enter the hobby"
    hobby = STDIN.gets.chomp
    puts "Please enter the cohort"
    cohort = STDIN.gets.chomp.capitalize 
    #sets default month as a november if the user made no input
    if birth.empty?
      birth = "Unknown"
    end
    if hobby.empty?
      hobby = "Unknown"
    end
    if cohort.empty?
      cohort = "November" 
    #if not empty its gonna ask user to correct their spelling until it equals any string from our spelling method
    else
      until spelling(cohort) == true
        puts "Please make sure you have a correct spelling for the cohort!"
        cohort = STDIN.gets.chomp.capitalize
      end
    end  
    #add the student hash to the array
    add_students(name, cohort, birth, hobby)
    #prints appropriative message for the student count
    if @students.count == 1
      puts "Now we have #{@students.count} student. Enter the next student or hit return to exit."
    else
      puts "Now we have #{@students.count} students. Enter the next student or hit return to exit."
    end  
    #get another name from the user
    name = STDIN.gets.chomp.capitalize
  end
end
def add_students(name, cohort, birth, hobby)
  @students << {name: name, cohort: cohort, birth: birth, hobby: hobby}
end 
#method for printing the header
def print_header
  puts "The students of Villains Academy"
  puts "--------------------------------"
end
#method for printing each student name from the array
def print_students_list
  @students.each.with_index(1) do |name, index|
    puts "#{index}.#{name[:name].center(16)}| cohort: #{name[:cohort].center(10)}| Country of birth: #{name[:birth].center(10)}| hobby: #{name[:hobby].center(10)}"
  end
end
#method for printing total number of students
def print_footer
  puts "Overall, we have #{@students.count} great students"
end
#method for printing the menu options
def print_menu
  puts "1. Input the students"
  puts "2. Show the students"
  puts "3. Save the list"
  puts "4. Load the list"
  puts "9. Exit"
end
#method for printing student list
def show_students
  print_header
  print_students_list
  print_footer
end
#method for saving student data
def save_students
  #open the file for writing
  file = File.open("students.csv", "w")
  CSV.open("students.csv", "wb") do |csv|
  #iterate over the array of students
    @students.each do |student|
      csv << [student[:name], student[:cohort], student[:birth], student[:hobby]]
    end
    puts "File is saved successfully!"
  end
end
#method which asks user for input and if the input matches with our csv file it will be saved
def saved_filename
  savedfilename = STDIN.gets.chomp
  if savedfilename == "students.csv"
    #calling method for saving file
    save_students
  else 
    puts "Incorrect file name, Please try again."
  end
end
#method for loading files
def load_students(filename = "students.csv")
  #assigned variable students again because program kept adding loaded array to the existing one, with this command we gonna get only the loaded array
  @students = []
  file = File.open(filename, "r") do |file|
    file.readlines.each do |line|
      name, cohort, birth, hobby = line.chomp.split(',')
      add_students(name, cohort, birth, hobby)
    end
  end
    puts "File is loaded successfully"
end
#method which checks if the filename is correct before loading it
def loaded_filename
  loadedfilename = STDIN.gets.chomp
  if loadedfilename == "students.csv"
    load_students
  else 
    puts "Incoret filename. Please try again."
  end
end
#method for loading the program with argument
def try_load_students
  filename = ARGV.first #first argument from the command line
  return if filename.nil? #get out of the method if it isn't given
  if File.exist?(filename) #if it exists
    load_students(filename)
    puts "Loaded #{@students.count} from #{filename}"
  else #if it doesnt exist
    puts "Sorry, #{filename} doesn't exist."
    exit #quit the program
  end
end
#method for selection
def process(selection)
  case selection
    when "1"
      input_students
    when "2"
      puts "You have chosen to view the list of students."
      show_students
    when "3"
      puts "Trying to save the file...."
      puts "Please type correct filename in the terminal"
      saved_filename
    when "4"
      puts "Trying to load the file..."
      puts "Please type correct filename in the terminal"
      loaded_filename
    when "9"
      puts "Exiting the program...."
      exit #this will cause the program to terminate
    else
      puts "I don't know what you meant, try again"
  end
end
#final method for our interactive menu which will call other methods 
def interactive_menu
  loop do
    print_menu
    process(STDIN.gets.chomp)
  end  
end
#nothing gonna happen until we call our method for the menu
try_load_students
interactive_menu