jupiter 0.0.1 → 0.8.0

data/lib/jupiter/cli.rb

@@ -0,0 +1,100 @@
+module Jupiter
+  class Cli < Thor
+
+    desc "state", "Shows power state for supplied virtual machine (VMware Tools must be running on VM)"
+    method_option :host,      type: :string, required: true, desc: 'Host server the VM resides on'
+    method_option :guest,     type: :string, required: true, desc: 'The virtual machine'
+
+    def state
+      vmserver = Jupiter.host_by_name(options[:host])
+      puts vmserver.vm_state(options[:guest])
+      vmserver.close_ssh
+    end
+
+    desc "ip", "Shows IP address for supplied virtual machine (VMware Tools must be running on VM)"
+    method_option :host,      type: :string, required: true, desc: 'Host server the VM resides on'
+    method_option :guest,     type: :string, required: true, desc: 'The virtual machine'
+
+    def ip
+      vmserver = Jupiter.host_by_name(options[:host])
+      puts vmserver.guest_ip(options[:guest])
+      vmserver.close_ssh
+    end
+
+    desc "manage", "Provision a new or manage an existing virtual machine"
+    method_option :pretend,   default: false, desc: 'Connect to the vmware host but do nothing'
+    method_option :host,      type: :string, required: true, desc: 'Host server the VM resides on'
+    method_option :guest,     type: :string, required: false, desc: 'The virtual machine to manage'
+    
+    def manage
+      Screen.new.clear
+      output = Jupiter::Cli::Output.new(self)
+      output.draw_menu
+      output.draw_new_line
+
+      if options[:guest].nil?
+        task = output.ask_which_task
+      else
+        task = '2'
+      end
+
+      if task == '1'
+        targetvm = ''
+        vmserver = Jupiter.host_by_name(options[:host])
+        output.manage_host(vmserver, options)
+      end
+
+      if task == '2'
+        vmserver = Jupiter.host_by_name(options[:host])
+        output.manage_vm(vmserver, options)
+      end
+    end
+
+    desc "register", "Registers the named VM with the supplied ESXi server"
+    method_option :host,      type: :string, required: true, desc: 'Host server the VM resides on'
+    method_option :guest,     type: :string, required: true, desc: 'The virtual machine'
+
+    def register
+      vmserver = Jupiter.host_by_name(options[:host])
+      puts vmserver.register_vm(options[:guest])
+      vmserver.close_ssh
+    end
+
+    desc "unregister", "Unregisters the named VM from the supplied ESXi server"
+    method_option :host,      type: :string, required: true, desc: 'Host server the VM resides on'
+    method_option :guest,     type: :string, required: true, desc: 'The virtual machine'
+
+    def unregister
+      vmserver = Jupiter.host_by_name(options[:host])
+      puts vmserver.unregister_vm(options[:guest])
+      vmserver.close_ssh
+    end
+
+    desc "runtime", "Start an Pry session with Jupiter loaded"
+    method_option :config
+    def runtime
+      # puts config
+      require 'pry'
+      pry
+    end
+
+  end
+
+  class Input
+    def clean_filename(filename)
+      filename.gsub(/[^a-zA-Z 0-9\-_]/, "").gsub(/\s/,'-')
+      # filename.gsub(/[^\w\s_-]+/, '')
+      # filename.gsub(/(^|\b\s)\s+($|\s?\b)/, '\\1\\2')
+      # filename.gsub(/\s+/, '_')
+    end
+  end
+
+  class Screen
+    def clear
+      puts "\e[H\e[2J" # ANSI escape sequence to clear screen
+    end
+  end
+
+
+end
+

data/lib/jupiter/cli/colorizer.rb

@@ -0,0 +1,44 @@
+module Jupiter
+  class Cli
+    module Colorizer
+      extend self
+
+      def color_map
+        {
+          red:      31,
+          white:    37,
+          yellow:   33,
+          blue:     34,
+          green:    32,
+          cyan:     36,
+          magenta:  35,
+        }
+      end
+
+      color_map.each do |color, code|
+        define_method color do |text|
+          colorize text, code
+        end
+      end
+
+      def color(color, text)
+        public_send(color, text)
+      end
+
+      def colorize(text, color_code)
+        "\e[#{color_code}m#{text}\e[0m"
+      end
+
+      # ANSI control codes
+      # 30	Black text
+      # 31	Red text
+      # 32	Green text
+      # 33	Yellow text
+      # 34	Blue text
+      # 35	Magenta text
+      # 36	Cyan text
+      # 37	White text
+      # 39	Default text color
+    end
+  end
+end

data/lib/jupiter/cli/output.rb

@@ -0,0 +1,355 @@
+module Jupiter
+  class Cli
+    class Output 
+
+      attr_accessor :cli
+
+      def initialize(cli)
+        self.cli = cli
+      end
+
+      def clearscreen
+        puts "\e[H\e[2J" # ANSI escape sequence to clear screen
+      end
+
+      def draw_menu
+        draw_line('#', 80, '', 'white', '')
+        draw_line('#', 80, 'WELCOME TO JUPITER', 'white', 'green')
+        draw_line('#', 80, '', 'white', '')
+        draw_line('#', 80, "We hope the god of sky and thunder will make your day most good and stuff.", 'white', 'green')
+        draw_line('#', 80, '', 'white', '')
+      end
+
+      def ask_which_task
+        valid_answers = ['1','2']
+        task = nil
+        until valid_answers.include? task do
+          draw_line('', 80, 'Would you like to create a new virtual machine or work with an existing one?', 'white', 'cyan')
+          draw_new_line
+          draw_line('', 80, '1. Clone / Create a Virtual Machine     ', 'white', 'cyan')
+          draw_line('', 80, '2. Work with an existing Virtual Machine', 'white', 'cyan')
+          draw_new_line
+          task = cli.ask("Please select a number:")
+          if valid_answers.include?(task) == false
+            puts "\n\n* INVALID ANSWER *\n\n"
+          end
+        end
+        return task
+      end
+
+      def ask_how_to_clone
+        valid_answers = ['1','2']
+        task = nil
+        until valid_answers.include? task do
+          draw_line('', 80, 'Would you like to clone one of these virtual machines or use a template?', 'white', 'cyan')
+          draw_new_line
+          draw_line('', 80, '1. Clone Virtual Machine', 'white', 'cyan')
+          draw_line('', 80, '2. Use a Template', 'white', 'cyan')
+          draw_new_line
+          task = cli.ask("Please select a number:")
+          if valid_answers.include?(task) == false
+            puts "\n\n* INVALID ANSWER *\n\n"
+          end
+        end
+        return task
+      end
+
+      def ask_vm_to_clone(vmserver)
+        valid_answers = vmserver.server_vm_list
+        list_virtual_machines(vmserver)
+        draw_line('#', 80, "Note that these VMs might not all be registered on the server.", 'white', 'red')
+        draw_line('#', 80, "You can only clone VMs that are registered.", 'white', 'red')
+        draw_line('#', 80, '', 'white', '')
+        targetvm = nil
+        until valid_answers.include? targetvm do
+          targetvm = cli.ask("\nPlease enter the valid name of the virtual machine you wish to clone:")
+          if valid_answers.include?(targetvm) == false
+            puts "\n\n* INVALID VIRTUAL MACHINE NAME *\n\n"
+          end
+        end
+        return targetvm
+      end
+
+      def select_vm_to_manage(vmserver)
+        vm_list = vmserver.server_vm_list
+        draw_line('#', 80, '', 'white', '')
+        draw_line('#', 80, "These are the currect virtual machines found on #{vmserver.name}", 'white', 'green')
+        draw_line('#', 80, '', 'white', '')
+        draw_new_line
+        vm_list.each_index { |x| puts "#{x + 1}. #{vm_list[x]}" }
+        draw_new_line
+        draw_line('#', 80, '', 'white', '')
+        targetvm  = cli.ask("\nPlease select the number of the Virtual Machine you wish to manage:")
+        target    = targetvm.to_i - 1
+        if vm_list[target].nil? || target < 0
+          raise InvalidSelectionError.new("Not a valid selection.")
+        else
+          targetvm = vmserver.server_vm_list[target].to_s
+          draw_new_line
+          draw_line('#', 80, '', 'white', '')
+          draw_line('#', 80, "#{targetvm} is selected for managment", 'white', 'green')
+          draw_line('#', 80, '', 'white', '')
+          return targetvm
+        end
+      end
+
+      def ask_new_vm_name(vmserver)
+        draw_line('#', 80, '', 'white', '')
+        draw_line('#', 80, "Naming the new virtual machine", 'white', 'green')
+        draw_line('#', 80, '', 'white', '')
+        draw_new_line
+        draw_line('', 80, 'When naming the new virtual machine please use only alpha-numeric', 'white', 'cyan')
+        draw_line('', 80, 'characters, dashes and underscores.', 'white', 'cyan')
+        draw_new_line
+        new_vm_name = cli.ask("\nPlease enter the name of the new virtual machine without spaces:")
+        new_vm = Jupiter::Input.new.clean_filename(new_vm_name)
+        draw_line('#', 80, '', 'white', '')
+        draw_line('#', 80, "The new virtual machine's name will be", 'white', 'cyan')
+        draw_line('#', 80, '', 'white', '')
+        draw_line(' ', 80, "#{new_vm}", 'white', 'green')
+        draw_line('#', 80, '', 'white', '')
+        return new_vm
+      end
+
+      def vm_current_state(vmserver, targetvm)
+        puts %Q[\nConnecting to VMware host server #{vmserver.name}...]
+        puts %Q[\n#{targetvm}'s current state is "#{vmserver.vm_state(targetvm)}"\nNote that this is only valid if VMware Tools is running on the virtual machine.\nIf the kernel has been upgraded recently, you many need to run vmware-config-tools.pl again.]
+        return vmserver.vm_state(targetvm)
+      end
+
+      def show_vm_ip(vmserver, vm)
+        draw_line('#', 80, '', 'white', '')
+        draw_line('#', 80, "#{vm}'s current IP is: #{vmserver.guest_ip(vm)}", 'white', 'green')
+        draw_line('#', 80, '', 'white', '')
+        draw_new_line
+      end
+
+      def list_virtual_machines(vmserver)
+        draw_line('#', 80, '', 'white', '')
+        draw_line('#', 80, "These are the currect virtual machines found on #{vmserver.name}", 'white', 'green')
+        draw_line('#', 80, '', 'white', '')
+        draw_new_line
+        vmserver.server_vm_list.each { |x| puts "* #{x}" }
+        draw_new_line
+        draw_line('#', 80, '', 'white', '')
+      end
+
+      def show_templates_available(templatehost)
+        templatevm = ''
+        templatelist = templatehost.templates_list
+        draw_line('#', 80, '', 'white', '')
+        draw_line('#', 80, "These are the currect templates found on #{templatehost.host}", 'white', 'green')
+        draw_line('#', 80, '', 'white', '')
+        draw_new_line
+        templatelist.each { |x| puts "* #{x.gsub('.tar.gz','')}" }
+        while templatelist.include?(templatevm + '.tar.gz') == false do
+          templatevm = cli.ask("\nPlease enter the valid name of the template you wish to use:")
+        end
+        return templatevm
+      end
+
+      def draw_new_line
+        puts "\n"
+      end
+
+      def draw_line(character, length, text, color1, color2)
+        line_minus_text   = length - text.length
+        half_line         = line_minus_text / 2
+        line_segment1     = half_line - 1
+        line_segment2     = half_line - 1
+        line_to_messure   = %Q[#{character * line_segment1} #{text} #{character * line_segment2}]
+
+        line_segment2     += 1 if line_to_messure.length < length 
+
+        if color1.empty? && text.empty?
+          output_line = character * length
+        elsif color1.empty? && text.length > 0
+          output_line = %Q[#{character * line_segment1} #{text} #{character * line_segment2}]
+        elsif text.empty?
+          output_line     = character * length
+          output_line     = Colorizer.color(color1, output_line)
+        else
+          line_segment1     = character * line_segment1
+          line_segment2     = character * line_segment2
+          line_segment1     = Colorizer.color(color1, line_segment1)
+          line_segment2     = Colorizer.color(color1, line_segment2)
+          text              = Colorizer.color(color2, text)
+          output_line       = %Q[#{line_segment1} #{text} #{line_segment2}]
+        end
+
+        puts output_line
+      end
+
+      def manage_vm(vmserver, options)
+        if options[:guest].nil?
+          vmname = select_vm_to_manage(vmserver) 
+        else
+          vmname = options[:guest]
+        end
+        vmstate = vm_current_state(vmserver, vmname)
+        return if options[:pretend]
+
+        if vmstate != 'running'
+          power_on = cli.ask("\nWould you like to power on #{vmname}? (yes/no)\n")
+          if power_on == 'yes'
+            vmserver.power_on_vm(vmname)
+          elsif power_on == 'no'
+            puts %Q[\nSince #{vmname} is not powered on some options will not be available.\n]
+          else
+            puts %Q[\nNot a valid response. Exiting...\n]
+            return
+          end
+        end
+        draw_new_line
+        draw_line('#', 80, '', 'white', '')
+        draw_line('#', 80, "Please select a task to perform on #{vmname}", 'white', 'green')
+        draw_line('#', 80, '', 'white', '')
+        draw_new_line
+        puts %Q[  1. Change hostname\n]
+        puts %Q[  2. Reboot (graceful restart)\n]
+        puts %Q[  3. Shutdown (graceful stop)\n]
+        puts %Q[  4. Cycle Power (non-graceful restart)\n]
+        puts %Q[  5. Kill Power (non-graceful)\n]
+        puts %Q[  6. Remove deploy user from sudoers\n]
+        puts %Q[  7. Display current IP address of #{vmname}\n]
+        puts %Q[  8. Increase the alocated HD space for #{vmname} (VM must be off)\n]
+        puts %Q[  9. Grow the LVM partition to fill the HD on #{vmname}\n]
+        puts %Q[ 10. Generate Nagios config file for #{vmname}\n]
+        choice = cli.ask("\n  Enter Choice:").to_i
+
+        case choice
+        when 1
+          newname = cli.ask("\nPlease enter a valid hostname for the new virtual machine (lowercase with no spaces or special characters): ")
+          new_hostname = Jupiter::Input.new.clean_filename(newname)
+          draw_line('#', 80, '', 'white', '')
+          draw_line('#', 80, "The new hostname will be", 'white', 'cyan')
+          draw_line('#', 80, '', 'white', '')
+          draw_line(' ', 80, "#{new_hostname}", 'white', 'green')
+          draw_line('#', 80, '', 'white', '')
+          ip = vmserver.guest_ip(vmname)
+          vm = Jupiter::VirtualMachine.new(ip, 2500)
+          puts vm.rename_linux_guest(new_hostname)
+          manage_vm(vmserver, options)
+        when 2
+          puts %Q[\nSending reboot command to #{vmname}...\n]
+          vmserver.reboot_vm(vmname)
+        when 3
+          puts %Q[\nSending shutdown command to #{vmname}...\n]
+          vmserver.shutdown_vm(vmname)
+        when 4
+          cycle = cli.ask("\nAre you sure? Only use this option if the system is locked up. Otherwise use the standard restart options for the guest OS. (yes / no): ")
+          if cycle == 'yes'
+            puts %Q[\nCycling the power for #{vmname}...\n]
+            vmserver.power_cycle_vm(vmname)
+          else
+            puts %[\nExiting...\n]
+            return
+          end
+        when 5
+          kill = cli.ask("\nAre you sure? (yes / no): ")
+          if kill == 'yes'
+            puts %Q[\nKilling power for #{vmname}...\n]
+            vmserver.power_off_vm(vmname)
+          else
+            puts %[\nExiting...\n]
+            return
+          end
+        when 6
+          ip = vmserver.guest_ip(vmname)
+          vm = Jupiter::VirtualMachine.new(ip, 2500)
+          puts vm.remove_from_sudo_group('deploy')
+          manage_vm(vmserver, options)
+        when 7
+          puts vmserver.guest_ip(vmname)
+          manage_vm(vmserver, options)
+        when 8
+          current_size = vmserver.vmdk_size(vmname)
+          puts %Q[\nThe current HD size is #{current_size} GB.\n]
+          size_in_gb = cli.ask(%Q[\nPlease enter the new size for the virtual HD in gigabytes: ])
+          vmserver.resize_vmdk(vmname, size_in_gb)
+          new_size = vmserver.vmdk_size(vmname)
+          puts %Q[\nThe new HD size is #{new_size} GB.\n]
+          manage_vm(vmserver, options)
+        when 9
+          ip = vmserver.guest_ip(vmname)
+          vm = Jupiter::VirtualMachine.new(ip, 2500)
+          puts vm.resize_lvm_volume_step_1
+          vmserver.reboot_vm(vmname)
+          sleep 45
+          puts vm.resize_lvm_volume_step_2
+          manage_vm(vmserver, options)
+        when 10
+          draw_new_line
+          draw_line('#', 80, '', 'white', '')
+          draw_line('#', 80, "Nagios File Generation", 'white', 'cyan')
+          draw_line('#', 80, '', 'white', '')
+          draw_line(' ', 80, "This will generate a Nagios object entry that can be pasted into", 'white', 'green')
+          draw_line(' ', 80, "your nagios config or uploaded to your nagios server if a proper", 'white', 'green')
+          draw_line(' ', 80, "nagios server entry is in your Jupiter yaml file.", 'white', 'green')
+          draw_new_line
+          draw_line(' ', 80, "This will also only be valid if your system has a static IP address and", 'white', 'green')
+          draw_line(' ', 80, "the services will only be functional if the nagios client is installed.", 'white', 'green')
+          draw_line('#', 80, '', 'white', '')
+          nagios_choice = cli.ask("\n  Enter 1 to display Nagios entry or 2 to upload to Nagios Server:").to_i
+          nagios = Jupiter::Nagios.new
+          nagios.action(nagios_choice, vmname, vmname, vmserver.guest_ip(vmname))
+          manage_vm(vmserver, options)
+        else
+          puts %[\nInvalid choice. Exiting...\n]
+          return
+        end
+      end
+
+      def manage_host(vmserver, options)
+
+        list_virtual_machines(vmserver)
+        clone_from = ask_how_to_clone
+
+        if clone_from == '1'
+          targetvm      = ask_vm_to_clone(vmserver)
+          new_vm_name   = ask_new_vm_name(vmserver)
+          vm_current_state(vmserver, targetvm)
+          return if options[:pretend]
+
+          puts %Q[\nStarting cloning of #{targetvm} to a new virtual machine called #{new_vm_name}. This may take some time. Please wait...]
+          time = Benchmark.realtime {
+            vmserver.clone_vm(targetvm, new_vm_name)
+          }
+          puts %Q[\nFinished cloning.]
+          puts %Q[\nElapses time: #{time}\n]
+          puts %Q[\nNow booting the new VM...\n]
+          sleep 10
+
+          show_vm_ip(vmserver, new_vm_name)
+
+          newname = cli.ask("\n#{new_vm_name} is the name used by the ESX server to reference the VM but we need to also change the VM's actually hostname.\nPlease enter a valid hostname for the new virtual machine: (lowercase with no spaces or special characters)")
+          ip = vmserver.guest_ip(new_vm_name)
+          vm = Jupiter::VirtualMachine.new(ip, 2500)
+          puts vm.rename_linux_guest(newname)
+          draw_line('#', 80, '', 'white', '')
+
+        elsif clone_from == '2'
+          templatehost = Jupiter::Template.new(vmserver)
+          templatevm = show_templates_available(templatehost)
+          new_vm_name = ask_new_vm_name(vmserver)
+          templatelist = templatehost.templates_list
+          raise InvalidNameError.new("#{new_vm_name} is already taken as a name.") if templatelist.include?(new_vm_name) 
+          puts %Q[\nConnecting to template share on #{templatehost.host} to transfer template.]
+          puts %Q[The transfer may take some time depending on the file size and bandwidth...]
+          return if options[:pretend]
+          puts templatehost.transfer_template(templatevm, new_vm_name)
+          sleep 10 # Give time for new VM to spinup before offering managment menu
+          manage_vm(vmserver, options)
+        else
+          puts %Q[\n"#{clone_from}" is not a valid option. Exiting...\n]
+          return
+        end
+      end
+
+      class InvalidNameError      < StandardError; end
+      class InvalidSelectionError < StandardError; end
+
+    end
+  end
+  
+end