Improved menu.

Rodrigo Siqueira de Melo
1 parent 5e7a172c
Showing 1 changed file with 13 additions and 7 deletions Show diff stats
Vagrantfile
@@ -9,22 +9,28 @@
 default_box = "precise64"
 if $stdin.isatty
   if Dir.glob(File.join(File.dirname("__FILE__"), '.vagrant/**/id')).empty?
+    options = ["precise64", "trusty64", "chef/centos-7.0"]
+
     puts "Bases boxes available locally:"
     puts '------------------------------'
     system('vagrant', 'box', 'list')
     puts
     puts 'Base boxes we can provide you:'
     puts '------------------------------'
-    puts 'precise64 (virtualbox)'
-    puts 'trusty64  (virtualbox)'
-    puts 'chef/centos-7.0 (virtualbox)'
-    puts
+    options.each_with_index do |value, i|
+      puts "[#{i + 1}] #{value}"
+    end
+
     print "Which box to use [#{default_box}]: "
     choice = $stdin.gets.strip
-    if !choice.empty?
-      default_box = choice
+    options.each_with_index do |option, i|
+      if option == choice || choice.to_i == (i + 1)
+        default_box = option
+        puts default_box
+        break
+      end
     end
-  end
+   end
 end
 # Vagrantfile API/syntax version. Don't touch unless you know what you're doing!