Calculator - Reboot batch #100

calculator.rb

def calculate(first, second, operation)
  if operation == "+"
    result = first + second
  elsif operation == "-"
    result = first - second
  elsif operation == "*"
    result = first * second
  elsif operation == "/"
    result = first / second
  else
    result = nil
  end
  return result
end

interface.rb

require_relative "calculator"

choice = "y"
while choice == "y"

  print "Entrez le premier chiffre de votre opération > "
  first_number = gets.chomp.to_f

  print "Entrez le second chiffre de votre opération > "
  second_number = gets.chomp.to_f

  print "Entrez le type d'opération souhaité (+, -, /, *) > "
  operation = gets.chomp

  result = calculate(first_number, second_number, operation)

  if result
    puts "Votre resultat est #{result}"
  else
    puts "resultat invalide."
  end
  puts "Veux tu continuer? (y/n)"
  choice = gets.chomp
end