project_wizard.rb in bebox-0.1.0

- old
+ new

@@ -1,12 +1,10 @@
-require 'bebox/project'
-require 'net/http'
-require 'uri'
 
 module Bebox
   class ProjectWizard
     include Bebox::Logger
+    include Bebox::WizardsHelper
     # Bebox boxes directory
     BEBOX_BOXES_PATH = '~/.bebox/boxes'
 
     # Asks for the project parameters and create the project skeleton
     def create_new_project(project_name)
@@ -14,103 +12,87 @@
       return error('Project not created. There is already a project with that name in the current directory.') if project_exists?(Dir.pwd, project_name)
       # Setup the bebox boxes directory
       bebox_boxes_setup
       # Asks to choose an existing box
       current_box = choose_box(get_existing_boxes)
-      # If choose to download/select new box
-      if current_box.nil?
-        # Keep asking for valid uri or overwriting until confirmation
-        confirm = false
-        begin
-          # Asks vagrant box location to user if not choose an existing box
-          valid_box_uri = ask_uri
-          # Confirm if the box already exist
-          confirm = box_exists?(valid_box_uri) ? confirm_overwrite? : true
-        end while !confirm
-        # Setup the box with the valid uri
-        set_box(valid_box_uri)
-      else
-        valid_box_uri = current_box
-      end
-      vagrant_box_base = "#{BEBOX_BOXES_PATH}/#{valid_box_uri}"
+      vagrant_box_base = "#{BEBOX_BOXES_PATH}/#{get_valid_box_uri(current_box)}"
       # Asks user to choose vagrant box provider
-      vagrant_box_provider = ask_box_provider
+      vagrant_box_provider = choose_option(%w{virtualbox vmware}, 'Choose the vagrant box provider')
       # Set default environments
       default_environments = %w{vagrant staging production}
       # Project creation
       project = Bebox::Project.new(project_name, vagrant_box_base, Dir.pwd, vagrant_box_provider, default_environments)
       project.create
       ok "Project '#{project_name}' created!.\nMake: cd #{project_name}\nNow you can add new environments or new nodes to your project.\nSee bebox help."
     end
 
+    # If choose to download/select new box get a valid uri
+    def get_valid_box_uri(current_box)
+      return (valid_box_uri = current_box) unless current_box.nil?
+      # Keep asking for valid uri or overwriting until confirmation
+      confirm = false
+      begin
+        # Asks vagrant box location to user if not choose an existing box
+        valid_box_uri = ask_uri
+        # Confirm if the box already exist
+        confirm = box_exists?(valid_box_uri) ? confirm_action?('There is already a box with that name, do you want to overwrite it?') : true
+      end while !confirm
+      # Setup the box with the valid uri
+      set_box(valid_box_uri)
+    end
+
     # Check if there's an existing project in that dir
     def project_exists?(parent_path, project_name)
       Dir.exists?("#{parent_path}/#{project_name}")
     end
 
-    # Menu to choose vagrant box provider
-    def ask_box_provider
-      choose do |menu|
-        menu.header = title('Choose the vagrant box provider')
-        menu.choices('virtualbox', 'vmware')
-      end
-    end
-
     # Setup the bebox boxes directory
     def bebox_boxes_setup
       # Create user project directories
       `mkdir -p #{BEBOX_BOXES_PATH}/tmp`
       # Clear partial downloaded boxes
       `rm -f #{BEBOX_BOXES_PATH}/tmp/*`
     end
 
     # Asks vagrant box location to user until is valid
     def ask_uri
-      vbox_uri =  ask(highline_quest('Write the URI (http, local_path) for the vagrant box to be used in the project:')) do |q|
-        q.default = 'http://puppet-vagrant-boxes.puppetlabs.com/ubuntu-server-12042-x64-vbox4210-nocm.box'
-      end
+      vbox_uri = write_input('Write the URI (http, local_path) for the vagrant box to be used in the project:', 'http://puppet-vagrant-boxes.puppetlabs.com/ubuntu-server-12042-x64-vbox4210-nocm.box')
       # If valid return uri if not keep asking for uri
       uri_valid?(vbox_uri) ? (return vbox_uri) : ask_uri
     end
 
     # Setup the box in the bebox boxes directory
     def set_box(box_uri)
+      require 'uri'
       uri = URI.parse(box_uri)
       if uri.scheme == ('http' || 'https')
+        info 'Downloading box ...'
         download_box(uri)
       else
-        file_name = uri.path.split('/').last
-        `ln -fs #{uri.path} #{BEBOX_BOXES_PATH}/#{file_name}`
+        `ln -fs #{uri.path} #{BEBOX_BOXES_PATH}/#{uri.path.split('/').last}`
       end
     end
 
     # Validate uri download or local box existence
     def uri_valid?(vbox_uri)
+      require 'uri'
       uri = URI.parse(vbox_uri)
-      if uri.scheme == ('http' || 'https')
-        request = Net::HTTP.new uri.host
-        response = request.request_head uri.path
-        if response.code.to_i == 302
-          uri = URI.parse(response['location'])
-          request = Net::HTTP.new uri.host
-          response = request.request_head uri.path
-        end
-        ( response.code.to_i == 200) ? (return true) : error('Download link not valid!.')
-      else
-        File.file?(uri.path) ? (return true) : error('File path not exist!.')
-      end
+      %w{http https}.include?(uri.scheme) ? http_uri_valid?(uri) : file_uri_valid?(uri)
     end
 
-    # Ask for confirmation of overwrite a box
-    def confirm_overwrite?
-      quest 'There is already a box with that name, do you want to overwrite it?'
-      response =  ask(highline_quest('(y/n)')) do |q|
-        q.default = 'n'
-      end
-      return response == 'y' ? true : false
+    def http_uri_valid?(uri)
+      require 'net/http'
+      request = Net::HTTP.new uri.host
+      response = request.request_head uri.path
+      error('Redirections not supported.') if response.code.to_i == 302
+      ( response.code.to_i == 200) ? (return true) : error('Download link not valid!.')
     end
 
+    def file_uri_valid?(uri)
+      File.file?(uri.path) ? (return true) : error('File path not exist!.')
+    end
+
     # Check if a box with the same name already exist
     def box_exists?(valid_box_uri)
       box_name = valid_box_uri.split('/').last
       boxes = get_existing_boxes
       boxes.any? { |val| /#{box_name}/ =~ val }
@@ -119,44 +101,32 @@
     # Obtain the current boxes downloaded or linked in the bebox user home
     def get_existing_boxes
       # Converts the bebox boxes directory to an absolute pathname
       expanded_directory = File.expand_path("#{BEBOX_BOXES_PATH}")
       # Get an array of bebox boxes paths
-      Dir["#{expanded_directory}/*"].reject {|f| File.directory? f}
+      boxes = Dir["#{expanded_directory}/*"].reject {|f| File.directory? f}
+      boxes.map{|box| box.split('/').last}
     end
 
     # Asks to choose an existing box in the bebox boxes directory
     def choose_box(boxes)
       # Menu to choose vagrant box provider
       other_box_message = 'Download/Select a new box'
-      current_box = choose do |menu|
-        menu.header = title('Choose an existing box or download/select a new box')
-        boxes.each do |box|
-          menu.choice(box.split('/').last)
-        end
-        menu.choice(other_box_message)
-      end
+      boxes << other_box_message
+      current_box = choose_option(boxes, 'Choose an existing box or download/select a new box')
       current_box = (current_box == other_box_message) ? nil : current_box
     end
 
     # Download a box by the specified uri
     def download_box(uri)
-      require 'progressbar'
-      info 'Downloading box ...'
       @counter = 0
-      # Manage redirections
-      request = Net::HTTP.new uri.host
-      response = request.request_head uri.path
-      uri = URI.parse(response['location']) if response.code.to_i == 302
-
-      # Set url variables
       url = uri.path
-      url_base = uri.host
       file_name = uri.path.split('/').last
-      expanded_directory = File.expand_path("#{BEBOX_BOXES_PATH}")
-
+      expanded_directory = File.expand_path(BEBOX_BOXES_PATH)
       # Download file to bebox boxes tmp
-      Net::HTTP.start(url_base) do |http|
+      require 'net/http'
+      require 'uri'
+      Net::HTTP.start(uri.host) do |http|
         response = http.request_head(URI.escape(url))
         ProgressBar
         pbar = ProgressBar.new('file name:', response['content-length'].to_i)
         File.open("#{expanded_directory}/tmp/#{file_name}", 'w') {|f|
           http.get(URI.escape(url)) do |str|
\ No newline at end of file