bebox 0.0.1 → 0.1.0

data/lib/bebox/project.rb

@@ -1,12 +1,9 @@
-require 'tilt'
-require 'bebox/environment'
-require 'bebox/provision'
-require 'bebox/logger'
 
 module Bebox
   class Project
 
     include Bebox::Logger
+    include Bebox::FilesHelper
 
     attr_accessor :name, :vagrant_box_base, :parent_path, :vagrant_box_provider, :environments, :path, :created_at
 
@@ -53,7 +50,7 @@ module Bebox
       generate_ruby_version
       # Generate Capfile and deploy files
       create_capfile
-      generate_deploy_file
+      generate_deploy_files
       # Generate Gemfile
       create_gemfile
       # Create the default environments
@@ -96,18 +93,12 @@ module Bebox
       # Set the creation time for the project
       self.created_at = DateTime.now.to_s
       # Create the .bebox file from template
-      dotbebox_template = Tilt::ERBTemplate.new("#{Bebox::Project.templates_path}/project/dot_bebox.erb")
-      File.open("#{self.path}/.bebox", 'w') do |f|
-        f.write dotbebox_template.render(nil, project: self)
-      end
+      generate_file_from_template("#{Bebox::Project.templates_path}/project/dot_bebox.erb", "#{self.path}/.bebox", {project: self})
     end
 
     # Generate .gitignore file
     def generate_gitignore_file
-      gitignore_template = Tilt::ERBTemplate.new("#{Bebox::Project.templates_path}/project/gitignore.erb")
-      File.open("#{self.path}/.gitignore", 'w') do |f|
-        f.write gitignore_template.render(nil, steps: Bebox::PROVISION_STEP_NAMES)
-      end
+      generate_file_from_template("#{Bebox::Project.templates_path}/project/gitignore.erb", "#{self.path}/.gitignore", {steps: Bebox::PROVISION_STEP_NAMES})
     end
 
     # Create templates directories
@@ -127,7 +118,7 @@ module Bebox
 
     # Create config deploy and keys directories
     def create_config_deploy_directories
-      `cd #{self.path} && mkdir -p config/{deploy,keys/environments}`
+      `cd #{self.path} && mkdir -p config/{deploy/steps,keys/environments}`
     end
 
     # Create the default environments
@@ -137,18 +128,12 @@ module Bebox
 
     # Create Capfile for the project
     def create_capfile
-      capfile_content = File.read("#{Bebox::Project.templates_path}/project/Capfile.erb")
-      File::open("#{self.path}/Capfile", "w")do |f|
-        f.write(capfile_content)
-      end
+      write_content_to_file("#{path}/Capfile", File.read("#{Bebox::Project.templates_path}/project/Capfile.erb"))
     end
 
     # Create Gemfile for the project
     def create_gemfile
-      gemfile_content = File.read("#{Bebox::Project.templates_path}/project/Gemfile.erb")
-      File::open("#{self.path}/Gemfile", "w")do |f|
-        f.write(gemfile_content)
-      end
+      write_content_to_file("#{self.path}/Gemfile", File.read("#{Bebox::Project.templates_path}/project/Gemfile.erb"))
     end
 
     # Create puppet base directories and files
@@ -180,20 +165,11 @@ module Bebox
         step_dir = Bebox::Provision.step_name(step)
         templates_path = Bebox::Project::templates_path
         # Generate site.pp template
-        manifest_template = Tilt::ERBTemplate.new("#{templates_path}/puppet/#{step}/manifests/site.pp.erb")
-        File.open("#{self.path}/puppet/steps/#{step_dir}/manifests/site.pp", 'w') do |f|
-          f.write manifest_template.render(nil, :nodes => [])
-        end
+        generate_file_from_template("#{templates_path}/puppet/#{step}/manifests/site.pp.erb", "#{self.path}/puppet/steps/#{step_dir}/manifests/site.pp", {nodes: []})
         # Generate hiera.yaml template
-        hiera_template = Tilt::ERBTemplate.new("#{templates_path}/puppet/#{step}/hiera/hiera.yaml.erb")
-        File.open("#{self.path}/puppet/steps/#{step_dir}/hiera/hiera.yaml", 'w') do |f|
-          f.write hiera_template.render(nil, :step_dir => step_dir)
-        end
+        generate_file_from_template("#{templates_path}/puppet/#{step}/hiera/hiera.yaml.erb", "#{self.path}/puppet/steps/#{step_dir}/hiera/hiera.yaml", {step_dir: step_dir})
         # Generate common.yaml template
-        hiera_template = Tilt::ERBTemplate.new("#{templates_path}/puppet/#{step}/hiera/data/common.yaml.erb")
-        File.open("#{self.path}/puppet/steps/#{step_dir}/hiera/data/common.yaml", 'w') do |f|
-          f.write hiera_template.render(nil, :step_dir => step_dir, :ssh_key => ssh_key, :project_name => self.shortname)
-        end
+        generate_file_from_template("#{templates_path}/puppet/#{step}/hiera/data/common.yaml.erb", "#{self.path}/puppet/steps/#{step_dir}/hiera/data/common.yaml", {step_dir: step_dir, ssh_key: ssh_key, project_name: self.shortname})
       end
     end
 
@@ -227,10 +203,10 @@ module Bebox
     end
 
     # Generate the deploy file for the project
-    def generate_deploy_file
-      config_deploy_template = Tilt::ERBTemplate.new("#{Bebox::Project.templates_path}/project/config/deploy.erb")
-      File.open("#{self.path}/config/deploy.rb", 'w') do |f|
-        f.write config_deploy_template.render(nil, project: self)
+    def generate_deploy_files
+      generate_file_from_template("#{Bebox::Project.templates_path}/project/config/deploy.erb", "#{self.path}/config/deploy.rb", {project: self})
+      Bebox::PROVISION_STEPS.each do |step|
+        generate_file_from_template("#{Bebox::Project.templates_path}/project/config/deploy/steps/#{step}.erb", "#{self.path}/config/deploy/steps/#{step}.rb", {})
       end
     end
 

data/lib/bebox/provision.rb

@@ -1,4 +1,3 @@
-require 'tilt'
 
 module Bebox
 
@@ -8,6 +7,8 @@ module Bebox
 
   class Provision
 
+    include Bebox::FilesHelper
+
     attr_accessor :environment, :project_root, :node, :step, :started_at, :finished_at
 
     def initialize(project_root, environment, node, step)
@@ -32,45 +33,41 @@ module Bebox
 
     # Check if it's necessary a Puppetfile accord to it's content
     def check_puppetfile_content
-      puppetfile_content = File.read("#{project_root}/puppet/steps/#{Bebox::Provision.step_name(step)}/Puppetfile").strip
-      `rm "#{project_root}/puppet/steps/#{Bebox::Provision.step_name(step)}/Puppetfile"` if puppetfile_content.scan(/^\s*(mod\s*.+?)$/).flatten.empty?
+      puppetfile_content = File.read("#{project_root}/puppet/steps/#{step_name}/Puppetfile").strip
+      `rm "#{project_root}/puppet/steps/#{step_name}/Puppetfile"` if puppetfile_content.scan(/^\s*(mod\s*.+?)$/).flatten.empty?
     end
 
     # Copy the static modules to the step-N modules path
     def copy_static_modules
-      `cp -R #{Bebox::Provision::templates_path}/puppet/#{self.step}/modules/* #{self.project_root}/puppet/steps/#{Bebox::Provision.step_name(self.step)}/modules/`
+      `cp -R #{Bebox::FilesHelper::templates_path}/puppet/#{self.step}/modules/* #{self.project_root}/puppet/steps/#{step_name}/modules/`
     end
 
     # Generate the hiera data for step from the template
     def generate_hiera
+      generate_hiera_yaml
+      generate_hiera_data_common
+    end
+
+    def generate_hiera_yaml
+      generate_file_from_template("#{Bebox::FilesHelper::templates_path}/puppet/#{self.step}/hiera/hiera.yaml.erb", "#{self.project_root}/puppet/steps/#{step_name}/hiera/hiera.yaml", {step_dir: step_name})
+    end
+
+    def generate_hiera_data_common
       ssh_key = Bebox::Project.public_ssh_key_from_file(self.project_root, self.environment)
       project_name = Bebox::Project.shortname_from_file(self.project_root)
-      hiera_template = Tilt::ERBTemplate.new("#{Bebox::Provision::templates_path}/puppet/#{self.step}/hiera/hiera.yaml.erb")
-      File.open("#{self.project_root}/puppet/steps/#{Bebox::Provision.step_name(self.step)}/hiera/hiera.yaml", 'w') do |f|
-        f.write hiera_template.render(nil, :step_dir => Bebox::Provision.step_name(self.step))
-      end
-      common_hiera_template = Tilt::ERBTemplate.new("#{Bebox::Provision::templates_path}/puppet/#{self.step}/hiera/data/common_apply.yaml.erb")
-      File.open("#{self.project_root}/puppet/steps/#{Bebox::Provision.step_name(self.step)}/hiera/data/common.yaml", 'w') do |f|
-        f.write common_hiera_template.render(nil, :ssh_key => ssh_key, :project_name => project_name)
-      end
+      generate_file_from_template("#{Bebox::FilesHelper::templates_path}/puppet/#{self.step}/hiera/data/common_apply.yaml.erb", "#{self.project_root}/puppet/steps/#{step_name}/hiera/data/common.yaml", {ssh_key: ssh_key, project_name: project_name})
     end
 
     # Generate the site.pp manifests file for step
     def self.generate_manifests(project_root, step, nodes)
-      manifest_template = Tilt::ERBTemplate.new("#{Bebox::Provision::templates_path}/puppet/#{step}/manifests/site_apply.pp.erb", :trim => true)
-      File.open("#{project_root}/puppet/steps/#{Bebox::Provision.step_name(step)}/manifests/site.pp", 'w') do |f|
-        f.write manifest_template.render(nil, :nodes => nodes)
-      end
+      generate_file_from_template("#{Bebox::FilesHelper::templates_path}/puppet/#{step}/manifests/site_apply.pp.erb", "#{project_root}/puppet/steps/#{Bebox::Provision.step_name(step)}/manifests/site.pp", {nodes: nodes})
     end
 
     # Generate the hiera templates for each step
     def self.generate_hiera_for_steps(project_root, template_file, filename, options)
       Bebox::PROVISION_STEPS.each do |step|
         step_dir = Bebox::Provision.step_name(step)
-        hiera_template = Tilt::ERBTemplate.new("#{Bebox::Provision::templates_path}/puppet/#{step}/hiera/data/#{template_file}")
-        File.open("#{project_root}/puppet/steps/#{step_dir}/hiera/data/#{filename}.yaml", 'w') do |f|
-          f.write hiera_template.render(nil, options)
-        end
+        generate_file_from_template("#{Bebox::FilesHelper::templates_path}/puppet/#{step}/hiera/data/#{template_file}", "#{project_root}/puppet/steps/#{step_dir}/hiera/data/#{filename}.yaml", options)
       end
     end
 
@@ -99,10 +96,7 @@ module Bebox
         puppetfile_content = File.read(profile_puppetfile_path)
         modules << puppetfile_content.scan(/^\s*(mod\s*.+?)$/).uniq
       end
-      puppetfile_template = Tilt::ERBTemplate.new("#{Bebox::Provision::templates_path}/puppet/#{step}/Puppetfile.erb", :trim => true)
-      File.open("#{project_root}/puppet/steps/#{Bebox::Provision.step_name(step)}/Puppetfile", 'w') do |f|
-        f.write puppetfile_template.render(nil, :profile_modules => modules.flatten)
-      end
+      generate_file_from_template("#{Bebox::FilesHelper::templates_path}/puppet/#{step}/Puppetfile.erb", "#{project_root}/puppet/steps/#{Bebox::Provision.step_name(step)}/Puppetfile", {profile_modules: modules.flatten})
     end
 
     # Get the role name associated with a node
@@ -142,13 +136,12 @@ module Bebox
     # Add a node to site.pp
     def self.add_node_to_step_manifests(project_root, node)
       Bebox::PROVISION_STEPS.each do |step|
-        manifest_node_template = Tilt::ERBTemplate.new("#{Bebox::Provision::templates_path}/puppet/#{step}/manifests/node.erb", :trim => true)
-        manifest_node = manifest_node_template.render(nil, :node => node)
+        manifest_node = render_erb_template("#{Bebox::FilesHelper::templates_path}/puppet/#{step}/manifests/node.erb", {node: node})
         Bebox::Provision.remove_node(project_root, node.hostname, step)
         manifest_path = "#{project_root}/puppet/steps/#{Bebox::Provision.step_name(step)}/manifests/site.pp"
         content = File.read(manifest_path)
         content += "\n#{manifest_node}\n"
-        File.open(manifest_path, 'wb') { |file| file.write(content) }
+        write_content_to_file(manifest_path, content)
       end
     end
 
@@ -224,15 +217,7 @@ module Bebox
     def create_step_checkpoint(started_at)
       self.node.started_at = started_at
       self.node.finished_at = DateTime.now.to_s
-      checkpoint_template = Tilt::ERBTemplate.new("#{Bebox::Provision::templates_path}/node/provisioned_node.yml.erb")
-      File.open("#{self.project_root}/.checkpoints/environments/#{self.environment}/steps/#{self.step}/#{self.node.hostname}.yml", 'w') do |f|
-        f.write checkpoint_template.render(nil, :node => self.node)
-      end
-    end
-
-    # Get the templates path inside the gem
-    def self.templates_path
-      File.join((File.expand_path '..', File.dirname(__FILE__)), 'templates')
+      generate_file_from_template("#{Bebox::FilesHelper::templates_path}/node/provisioned_node.yml.erb", "#{self.project_root}/.checkpoints/environments/#{self.environment}/steps/#{self.step}/#{self.node.hostname}.yml", {node: self.node})
     end
 
     # Translate step name to directory name

data/lib/bebox/role.rb

@@ -1,8 +1,7 @@
-require 'tilt'
 
 module Bebox
   class Role
-
+    include Bebox::FilesHelper
     attr_accessor :project_root, :name
 
     def initialize(name, project_root)
@@ -33,10 +32,7 @@ module Bebox
 
     # Generate the manifests init.pp file
     def generate_manifests_file
-      manifests_template = Tilt::ERBTemplate.new("#{templates_path}/puppet/roles/manifests/init.pp.erb")
-      File.open("#{self.path}/manifests/init.pp", 'w') do |f|
-        f.write manifests_template.render(nil, :role => self)
-      end
+      generate_file_from_template("#{templates_path}/puppet/roles/manifests/init.pp.erb", "#{self.path}/manifests/init.pp", {role: self})
     end
 
     # Path to the templates directory in the gem
@@ -93,11 +89,5 @@ module Bebox
       role_profiles = Bebox::Role.list_profiles(project_root, role)
       role_profiles.include?(profile) ? true : false
     end
-
-    # Check if the role has a valid name
-    def self.valid_name?(name)
-      valid_name = (name =~ /\A[a-z][a-z0-9_]*\Z/).nil? ? false : true
-      valid_name && !Bebox::RESERVED_WORDS.include?(name)
-    end
   end
 end

data/lib/bebox/vagrant_helper.rb

@@ -0,0 +1,96 @@
+
+module Bebox
+  module VagrantHelper
+
+    include Bebox::Logger
+    include Bebox::FilesHelper
+
+    # Return the existence status of vagrant node
+    def vagrant_box_exist?(node)
+      vagrant_boxes = `cd #{node.project_root} && vagrant box list`
+      project_name = Bebox::Project.name_from_file(node.project_root)
+      vagrant_box_provider = Bebox::Project.vagrant_box_provider_from_file(node.project_root)
+      (vagrant_boxes =~ /#{project_name}-#{node.hostname}\s+\(#{vagrant_box_provider}/) ? true : false
+    end
+
+    # Return the running status of vagrant node
+    def vagrant_box_running?(node)
+      status = `cd #{node.project_root} && vagrant status`
+      (status =~ /#{node.hostname}\s+running/) ? true : false
+    end
+
+    # Remove the specified boxes from vagrant
+    def remove_vagrant_box(node)
+      return unless (node.environment == 'vagrant' && node.prepared_nodes_count > 0)
+      project_name = Bebox::Project.name_from_file(node.project_root)
+      vagrant_box_provider = Bebox::Project.vagrant_box_provider_from_file(node.project_root)
+      `cd #{node.project_root} && vagrant destroy -f #{node.hostname}`
+      `cd #{node.project_root} && vagrant box remove #{project_name}-#{node.hostname} --provider #{vagrant_box_provider}`
+    end
+
+    # Backup and add the vagrant hosts to local hosts file
+    def configure_local_hosts(project_name, node)
+      info "\nPlease provide your local password, if asked, to configure the local hosts file."
+      backup_local_hosts(project_name)
+      add_to_local_hosts(node)
+    end
+
+    # Backup the local hosts file
+    def backup_local_hosts(project_name)
+      hosts_backup_file = "#{local_hosts_path}/hosts_before_#{project_name}"
+      `sudo cp #{local_hosts_path}/hosts #{hosts_backup_file}` unless File.exist?(hosts_backup_file)
+    end
+
+    # Add the vagrant hosts to the local hosts file
+    def add_to_local_hosts(node)
+      host_command = `echo '#{node.ip} #{node.hostname}     # Added by bebox' >> #{local_hosts_path}/hosts`
+      host_command if (file_content_trimmed("#{local_hosts_path}/hosts") =~ /#{node.ip}\s+#{node.hostname}/)
+    end
+
+    # Obtain the local hosts file for the OS
+    def local_hosts_path
+      RUBY_PLATFORM =~ /darwin/ ? '/private/etc' : '/etc'
+    end
+
+    # Prepare the vagrant nodes
+    def prepare_vagrant(node)
+      project_name = Bebox::Project.name_from_file(node.project_root)
+      vagrant_box_base = Bebox::Project.vagrant_box_base_from_file(node.project_root)
+      configure_local_hosts(project_name, node)
+      add_vagrant_node(project_name, vagrant_box_base, node)
+    end
+
+    # Add the boxes to vagrant for each node
+    def add_vagrant_node(project_name, vagrant_box_base, node)
+      already_installed_boxes = installed_vagrant_box_names(node)
+      box_name = "#{project_name}-#{node.hostname}"
+      info "Adding server to vagrant: #{node.hostname}..."
+      `cd #{node.project_root} && vagrant box add #{box_name} #{vagrant_box_base}` unless already_installed_boxes.include? box_name
+    end
+
+    # Up the vagrant boxes in Vagrantfile
+    def self.up_vagrant_nodes(project_root)
+      `cd #{project_root} && vagrant up --provision`
+    end
+
+    # Halt the vagrant boxes running
+    def self.halt_vagrant_nodes(project_root)
+      `cd #{project_root} && vagrant halt`
+    end
+
+    # Generate the Vagrantfile
+    def self.generate_vagrantfile(nodes)
+      project_root = nodes.first.project_root
+      network_interface = RUBY_PLATFORM =~ /darwin/ ? 'en0' : 'eth0'
+      project_name = Bebox::Project.name_from_file(project_root)
+      vagrant_box_provider = Bebox::Project.vagrant_box_provider_from_file(project_root)
+      generate_file_from_template("#{Bebox::FilesHelper::templates_path}/node/Vagrantfile.erb", "#{project_root}/Vagrantfile", {nodes: nodes, project_name: project_name, vagrant_box_provider: vagrant_box_provider, network_interface: network_interface})
+    end
+
+    # return an Array with the names of the currently installed vagrant boxes
+    # @returns Array
+    def installed_vagrant_box_names(node)
+      (`cd #{node.project_root} && vagrant box list`).split("\n").map{|vagrant_box| vagrant_box.split(' ').first}
+    end
+  end
+end

data/lib/bebox/version.rb

@@ -1,3 +1,3 @@
 module Bebox
-  VERSION = '0.0.1'
+  VERSION = '0.1.0'
 end

data/lib/bebox/wizards/environment_wizard.rb

@@ -1,6 +1,9 @@
+
 module Bebox
   class EnvironmentWizard
     include Bebox::Logger
+    include Bebox::WizardsHelper
+
     # Create a new environment
     def create_new_environment(project_root, environment_name)
       # Check if the environment exist
@@ -16,30 +19,11 @@ module Bebox
       # Check if the environment exist
       return error("The '#{environment_name}' environment do not exist!.") unless Bebox::Environment.environment_exists?(project_root, environment_name)
       # Confirm deletion
-      return warn('No changes were made.') unless confirm_environment_deletion?
+      return warn('No changes were made.') unless confirm_action?('Are you sure that you want to delete the environment?')
       # Environment deletion
       environment = Bebox::Environment.new(environment_name, project_root)
       environment.remove
       ok 'Environment removed!.'
     end
-
-    # Ask for confirmation of environment deletion
-    def confirm_environment_deletion?
-      quest 'Are you sure that you want to delete the environment?'
-      response =  ask(highline_quest('(y/n)')) do |q|
-        q.default = "n"
-      end
-      return response == 'y' ? true : false
-    end
-
-    # Asks to choose an existing environment
-    def choose_environment(environments)
-      choose do |menu|
-        menu.header = title('Choose an existing environment:')
-        environments.each do |box|
-          menu.choice(box.split('/').last)
-        end
-      end
-    end
   end
 end

data/lib/bebox/wizards/node_wizard.rb

@@ -1,6 +1,10 @@
+
 module Bebox
   class NodeWizard
     include Bebox::Logger
+    include Bebox::WizardsHelper
+    include Bebox::VagrantHelper
+
     # Create a new node
     def create_new_node(project_root, environment)
       # Ask the hostname for node
@@ -18,12 +22,12 @@ module Bebox
       # Ask for a node to remove
       nodes = Bebox::Node.list(project_root, environment, 'nodes')
       if nodes.count > 0
-        hostname = choose_node(nodes, 'Choose the node to remove:')
+        hostname = choose_option(nodes, 'Choose the node to remove:')
       else
         return error "There are no nodes in the '#{environment}' environment to remove. No changes were made."
       end
       # Ask for deletion confirmation
-      return warn('No changes were made.') unless confirm_node_deletion?
+      return warn('No changes were made.') unless confirm_action?('Are you sure that you want to delete the node?')
       # Node deletion
       node = Bebox::Node.new(environment, project_root, hostname, nil)
       node.remove
@@ -34,9 +38,8 @@ module Bebox
     def set_role(project_root, environment)
       roles = Bebox::Role.list(project_root)
       nodes = Bebox::Node.list(project_root, environment, 'nodes')
-      node = choose_node(nodes, 'Choose an existing node:')
-      require 'bebox/wizards/role_wizard'
-      role = Bebox::RoleWizard.new.choose_role(roles, 'Choose an existing role:')
+      node = choose_option(nodes, 'Choose an existing node:')
+      role = choose_option(roles, 'Choose an existing role:')
       Bebox::Provision.associate_node_role(project_root, environment, node, role)
       ok 'Role associated to node!.'
     end
@@ -54,9 +57,9 @@ module Bebox
         Bebox::Node.regenerate_deploy_file(project_root, environment, nodes_to_prepare)
         # If environment is 'vagrant' Prepare and Up the machines
         if environment == 'vagrant'
-          Bebox::Node.generate_vagrantfile(project_root, nodes_to_prepare)
-          nodes_to_prepare.each{|node| node.prepare_vagrant}
-          Bebox::Node.up_vagrant_nodes(project_root)
+          Bebox::VagrantHelper.generate_vagrantfile(nodes_to_prepare)
+          nodes_to_prepare.each{|node| prepare_vagrant(node)}
+          Bebox::VagrantHelper.up_vagrant_nodes(project_root)
         end
         # For all the environments do the preparation
         nodes_to_prepare.each do |node|
@@ -75,7 +78,9 @@ module Bebox
       prepared_nodes = Bebox::Node.list(project_root, environment, 'prepared_nodes')
       nodes.each do |node|
         if prepared_nodes.include?(node.hostname)
-          nodes_to_prepare << node if confirm_node_preparation?(node)
+          checkpoint_status = "(start: #{node.checkpoint_parameter_from_file('prepared_nodes', 'started_at')} - end: #{node.checkpoint_parameter_from_file('prepared_nodes', 'finished_at')})"
+          message = "The node '#{node.hostname}' was already prepared #{checkpoint_status}.\nDo you want to re-prepare it?"
+          nodes_to_prepare << node if confirm_action?(message)
         else
           nodes_to_prepare << node
         end
@@ -88,35 +93,6 @@ module Bebox
       File.exists?("#{project_root}/.checkpoints/environments/#{environment}/nodes/#{node_name}.yml")
     end
 
-    # Ask for confirmation of node preparation
-    def confirm_node_preparation?(node)
-      checkpoint_status = "(start: #{node.checkpoint_parameter_from_file('prepared_nodes', 'started_at')} - end: #{node.checkpoint_parameter_from_file('prepared_nodes', 'finished_at')})"
-      quest "The node '#{node.hostname}' was already prepared #{checkpoint_status}.\nDo you want to re-prepare it?"
-      response =  ask(highline_quest('(y/n)')) do |q|
-        q.default = "n"
-      end
-      return response == 'y' ? true : false
-    end
-
-    # Ask for confirmation of node deletion
-    def confirm_node_deletion?
-      quest 'Are you sure that you want to delete the node?'
-      response =  ask(highline_quest('(y/n)')) do |q|
-        q.default = "n"
-      end
-      return response == 'y' ? true : false
-    end
-
-    # Asks to choose an existing node
-    def choose_node(nodes, question)
-      choose do |menu|
-        menu.header = title(question)
-        nodes.each do |box|
-          menu.choice(box.split('/').last)
-        end
-      end
-    end
-
     # Keep asking for a hostname that not exist
     def ask_not_existing_hostname(project_root, environment)
       hostname = ask_hostname(project_root, environment)
@@ -129,20 +105,14 @@ module Bebox
       end
     end
 
-    # Ask for the hostname until is valid
+    # Ask for the hostname
     def ask_hostname(project_root, environment)
-      ask(highline_quest('Write the hostname for the node:')) do |q|
-        q.validate = /\.(.*)/
-        q.responses[:not_valid] = highline_warn('Enter valid hostname. Ex. host.server1.com')
-      end
+      write_input('Write the hostname for the node:', nil, /\.(.*)/, 'Enter valid hostname. Ex. host.server1.com')
     end
 
     # Ask for the ip until is valid
     def ask_ip(environment)
-      ip = ask(highline_quest('Write the IP address for the node:')) do |q|
-        q.validate = /\.(.*)/
-        q.responses[:not_valid] = highline_warn('Enter a valid IP address. Ex. 192.168.0.50')
-      end
+      ip = write_input('Write the IP address for the node:', nil, /\.(.*)/, 'Enter a valid IP address. Ex. 192.168.0.50')
       # If the environment is not vagrant don't check ip free
       return ip if environment != 'vagrant'
       # Check if the ip address is free

data/lib/bebox/wizards/profile_wizard.rb

@@ -1,8 +1,9 @@
-require 'bebox/profile'
 
 module Bebox
   class ProfileWizard
     include Bebox::Logger
+    include Bebox::WizardsHelper
+
     # Create a new profile
     def create_new_profile(project_root, profile_name, profile_base_path)
       # Check if the profile name is valid
@@ -12,10 +13,10 @@ module Bebox
       \n* Underscores
       \n* Must begin with an Lowercase letter
       \n* Can not be any of: #{Bebox::RESERVED_WORDS.join(', ')}
-      \n\nNo changes were made." unless Bebox::Profile.valid_name?(profile_name)
-      # Clean the profile_path to make it a valid path
-      profile_base_path = Bebox::Profile.cleanpath(profile_base_path)
+      \n\nNo changes were made." unless valid_puppet_class_name?(profile_name)
       unless profile_base_path.empty?
+        # Clean the profile_path to make it a valid path
+        profile_base_path = Bebox::Profile.cleanpath(profile_base_path)
         # Check if the path name is valid
         return error "Each part of the path can only contain:\n
           \n* Lowercase letters
@@ -26,7 +27,7 @@ module Bebox
           \n\nNo changes were made." unless Bebox::Profile.valid_pathname?(profile_base_path)
       end
       # Check if the profile exist
-      profile_path = profile_complete_path(profile_base_path, profile_name)
+      profile_path = profile_base_path.empty? ? profile_name : profile_complete_path(profile_base_path, profile_name)
       return error("The profile '#{profile_path}' already exist. No changes were made.") if profile_exists?(project_root, profile_path)
       # Profile creation
       profile = Bebox::Profile.new(profile_name, project_root, profile_base_path)
@@ -40,12 +41,12 @@ module Bebox
       profiles = Bebox::Profile.list(project_root)
       # Get a profile if exist
       if profiles.count > 0
-        profile = choose_profile(profiles, 'Choose the profile to remove:')
+        profile = choose_option(profiles, 'Choose the profile to remove:')
       else
         return error "There are no profiles to remove. No changes were made."
       end
       # Ask for deletion confirmation
-      return warn('No changes were made.') unless confirm_profile_deletion?
+      return warn('No changes were made.') unless confirm_action?('Are you sure that you want to delete the profile?')
       # Profile deletion
       profile_name = profile.split('/').last
       profile_base_path = profile.split('/')[0...-1].join('/')
@@ -68,24 +69,5 @@ module Bebox
     def profile_exists?(project_root, profile_path)
       Dir.exists?( File.join("#{project_root}/puppet/profiles", "#{profile_path}") )
     end
-
-    # Ask for confirmation of profile deletion
-    def confirm_profile_deletion?
-      quest 'Are you sure that you want to delete the profile?'
-      response =  ask(highline_quest('(y/n)')) do |q|
-        q.default = "n"
-      end
-      return response == 'y' ? true : false
-    end
-
-    # Asks to choose a profile
-    def choose_profile(profiles, question)
-      choose do |menu|
-        menu.header = title(question)
-        profiles.each do |box|
-          menu.choice(box)
-        end
-      end
-    end
   end
 end