jupiter 0.0.1 → 0.8.0

data/lib/jupiter/configuration.rb

@@ -0,0 +1,18 @@
+module Jupiter
+  class Configuration
+    attr_accessor :nagios, :hosts, :vmusers, :templates
+
+    def self.load_from_yaml!
+      config = YAML.load_file('config.yaml')
+      new(config)
+    end
+
+    def initialize(options = {})
+      @hosts      = options.fetch(:hosts) {[]}
+      @vmusers    = options.fetch(:vmusers) { {} }
+      @templates  = options.fetch(:templates) { {} }
+      @nagios     = options.fetch(:nagios) { {} }
+    end
+
+  end
+end

data/lib/jupiter/host.rb

@@ -0,0 +1,257 @@
+module Jupiter
+  class Host
+
+    attr_accessor :name, :vmpath, :host, :sshport, :sshuser, :sshpass, :dcname, :timestamp
+
+    def initialize(options)
+      self.name       = options.fetch(:name)
+      self.vmpath     = options.fetch(:vmpath)
+      self.host       = options.fetch(:host)
+      self.sshport    = options.fetch(:sshport)
+      self.sshuser    = options.fetch(:sshuser)
+      self.sshpass    = options.fetch(:sshpass)
+      self.dcname     = options.fetch(:dcname)
+      self.timestamp  = Time.new.strftime('%Y-%m-%d-%H%M').freeze
+    end
+
+    def server_uptime
+      puts ssh.exec!("uptime")
+      close_ssh
+    end
+
+    def server_vm_list
+      vm_list = ssh.exec!("ls #{vmpath}").to_s
+      close_ssh
+      vm_list.split("\n")
+    end
+
+    def ssh
+      @ssh ||= Net::SSH.start(host, sshuser, password: sshpass, port: sshport)
+    end
+
+    def close_ssh
+      ssh.close
+      @ssh = nil
+      true
+    end
+
+    def vmhost
+      if /^(?:(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.){3}(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)$/.match(host)
+        vim = RbVmomi::VIM.connect host: host, user: sshuser, password: sshpass, insecure: true
+        dc  = vim.serviceInstance.find_datacenter(dcname) or fail "datacenter not found"
+      else
+        return false
+      end
+    end
+
+    def guestvm(vmname)
+      vm  = vmhost.find_vm(vmname) or fail "\n\nVirtual Machine was not found. Although its files might exist, it may not be registered on the server. You can register the VM with the following command:\n\njupiter register --guest=#{vmname} --host=#{name}\n\n"
+    end
+
+    def guest_ip(vm)
+      result = guestvm(vm).guest_ip
+    end
+
+    def vm_state(vm)
+      result = guestvm(vm).guest.guestState
+    end
+
+    def snapshot_create(vm)
+      result = guestvm(vm).CreateSnapshot_Task(name: "snapshot_#{timestamp}", memory: false, quiesce: true)
+    end
+
+    def snapshots_remove(vm)
+      result = guestvm(vm).RemoveAllSnapshots_Task
+    end
+
+    def consolidate_snapshots(vm)
+      result = guestvm(vm).ConsolidateVMDisks_Task
+    end
+
+    def register_vm(vm)
+      ssh.exec!("vim-cmd solo/registervm #{vmpath}/#{vm}/#{vm}.vmx")
+      snapshots_remove(vm)
+    end
+    
+    def unregister_vm(vm)
+      ssh.exec!("vim-cmd vmsvc/unregister #{vmpath}/#{vm}/#{vm}.vmx")
+    end
+
+    def power_on_vm(vm)
+      result = guestvm(vm).PowerOnVM_Task
+    end
+
+    def power_off_vm(vm)
+      result = guestvm(vm).PowerOffVM_Task
+    end
+
+    def power_cycle_vm(vm)
+      guestvm(vm).PowerOffVM_Task
+      sleep 5
+      guestvm(vm).PowerOnVM_Task
+    end
+
+    def reboot_vm(vm)
+      result = guestvm(vm).RebootGuest
+    end
+
+    def shutdown_vm(vm)
+      result = guestvm(vm).ShutdownGuest
+    end
+
+    def clone_vm(vm, newvm)
+      snapshot_create(vm)
+      sleep 5
+      ssh.exec!("mkdir #{vmpath}/#{newvm}")
+      ssh.exec!("cp -Rfv #{vmpath}/#{vm}/* #{vmpath}/#{newvm}")
+      rename_cp_files(vm, newvm)
+      update_vmx(vm, newvm)
+      update_vmxf(vm, newvm)
+      update_vmdk(vm, newvm)
+      register_vm(newvm)
+      close_ssh
+      snapshots_remove(vm)
+      consolidate_snapshots(vm)
+      sleep 5
+      power_on_vm(newvm)
+      sleep 5
+      answer_vm_question(newvm) # Needed to answer the "Was the VM moved" question that pops up.
+      sleep 5
+      power_on_vm(newvm)
+    end
+
+    def rename_cp_files(vm, newvm)
+      old_files_path = "#{vmpath}/#{newvm}/#{vm}"
+      new_files_path = "#{vmpath}/#{newvm}/#{newvm}"
+      ssh.exec!("rm -f #{vmpath}/#{newvm}/vmware-*.log")
+      ssh.exec!("rm -f #{old_files_path}.vmx~")
+      ssh.exec!("rm -f #{old_files_path}.vmsd")
+      ssh.exec!("rm -f #{old_files_path}-*.vmsn")
+      ssh.exec!("rm -f #{old_files_path}-0000*.vmdk")
+      ssh.exec!("mv #{old_files_path}-aux.xml #{new_files_path}-aux.xml")
+      ssh.exec!("mv #{old_files_path}-flat.vmdk #{new_files_path}-flat.vmdk")
+      ssh.exec!("mv #{old_files_path}.nvram #{new_files_path}.nvram")
+      ssh.exec!("mv #{old_files_path}.vmdk #{new_files_path}.vmdk")
+      ssh.exec!("mv #{old_files_path}.vmx #{new_files_path}.vmx")
+      ssh.exec!("mv #{old_files_path}.vmxf #{new_files_path}.vmxf")
+    end
+
+    def change_config_value(key, value)
+      self[key] = value
+      return self
+    end
+
+    def read_config_to_hash(file)
+      file_contents = ssh.exec!(%Q[cat #{file}])
+      Hash[file_contents.split("\n").map { |line| line.split('=').map(&:strip).map { |value| value.gsub(/\A"|"\z/, '') } }]
+    end
+
+    def uuid(newvm)
+      @uuid ||= Jupiter::UUID.new.generate(newvm)
+    end
+
+    def flatten_hash(hash_to_flatten)
+      hash_to_flatten.map{|k,v| "#{k} = \"#{v}\""}.join("\n")
+    end
+
+    def write_to_file(file_content, full_path_to_file)
+      ssh.exec!(%Q[echo '#{file_content}' > #{full_path_to_file}])
+    end
+   
+    def generate_mac_address
+      (1..6).map{"%0.2X"%rand(256)}.join(":")
+    end 
+
+    def update_vmx(vm, newvm)
+      vmx_hash = read_config_to_hash("#{vmpath}/#{newvm}/#{newvm}.vmx")
+      vmx_hash['ethernet0.generatedAddress'] = generate_mac_address
+      vmx_hash['uuid.bios'] = uuid(newvm)
+      vmx_hash['uuid.location'] = uuid(newvm)
+      vmx_hash['displayName'] = newvm
+      vmx_hash['nvram'] = "#{newvm}.nvram"
+      vmx_hash['extendedConfigFile'] = "#{newvm}.vmxf"
+      vmx_hash['scsi0:0.fileName'] = "#{newvm}.vmdk"
+      swap_derived_name = vmx_hash['sched.swap.derivedName']
+      swap_derived_name["#{vm}/#{vm}"]= "#{newvm}/#{newvm}"
+      vmx_hash['sched.swap.derivedName'] = swap_derived_name
+      write_to_file(flatten_hash(vmx_hash), "#{vmpath}/#{newvm}/#{newvm}.vmx")
+      return true
+    end
+
+    def update_vmxf(vm, newvm)
+      file_with_path = "#{vmpath}/#{newvm}/#{newvm}.vmxf"
+      newuuid = uuid(newvm)
+      file_contents = "<?xml version=\"1.0\"?>\n<Foundry>\n<VM>\n<VMId type=\"string\">#{newuuid}</VMId>\n<ClientMetaData>\n<clientMetaDataAttributes/>\n<HistoryEventList/></ClientMetaData>\n<vmxPathName type=\"string\">#{newvm}.vmx</vmxPathName></VM></Foundry>"
+      write_to_file(file_contents, file_with_path)
+      return true
+    end
+
+    def update_vmdk(vm, newvm)
+      file_with_path = "#{vmpath}/#{newvm}/#{newvm}.vmdk"
+      file_contents = ssh.exec!(%Q[cat #{file_with_path}])
+      file_contents[vm]= newvm
+      write_to_file(file_contents, file_with_path)
+      return true
+    end
+
+    def vmdk_size(vm)
+      file_with_path = "#{vmpath}/#{vm}/#{vm}.vmdk"
+      vmdk_contents = ssh.exec!(%Q[cat #{file_with_path}])
+      close_ssh
+      vmfs_sectors = vmdk_contents[/\RW(.*?)VMFS/,1]
+      gb_size = ((vmfs_sectors.to_f * 512) / (1024 * 1024 * 1024))
+    end
+
+    def resize_vmdk(vm, size_in_GB)
+      size_in_GB.tr!('^.0-9', '')
+      raise InvalidGBSizeError.new("You can only increase the size of the drive, not reduce it.") if vmdk_size(vm).to_f > size_in_GB.to_f
+      raise InvalidVMStateError.new("#{vm} must be shutdown before the drive can be resized.") if vm_state(vm) == 'running'
+      result = ssh.exec!(%Q[vmkfstools -X #{size_in_GB}G #{vmpath}/#{vm}/#{vm}.vmdk])
+      close_ssh
+      return result
+    end
+
+    def answer_vm_question(vm)
+      result = guestvm(vm).AnswerVM(questionId: '_vmx1', answerChoice: '2')
+    end
+
+    def setup_template(original_name, new_name)
+      extract_template(original_name, new_name)
+      rename_copied_template(original_name, new_name)
+      update_vmx(original_name, new_name)
+      update_vmxf(original_name, new_name)
+      update_vmdk(original_name, new_name)
+      register_vm(new_name)
+      close_ssh
+      sleep 5
+      power_on_vm(new_name)
+      sleep 5
+      answer_vm_question(new_name) # Needed to answer the "Was the VM moved" question that pops up.
+      sleep 5
+      power_on_vm(new_name)
+    end
+
+    def extract_template(original_name, new_name)
+      file_exists = ssh.exec!("find #{vmpath}/#{original_name}.tar.gz").chomp
+      if file_exists == "#{vmpath}/#{original_name}.tar.gz"
+        ssh.exec!("cd #{vmpath} && tar -zxvf #{original_name}.tar.gz && rm #{original_name}.tar.gz")
+      else
+        close_ssh
+        return false
+      end
+    end
+
+    def rename_copied_template(original_name, new_name)
+      if !new_name.nil? && !original_name.nil?
+        ssh.exec!("mv #{vmpath}/#{original_name} #{vmpath}/#{new_name}")
+        rename_cp_files(original_name, new_name)
+      else
+        return false
+      end
+    end
+
+    class InvalidVMStateError < StandardError; end
+    class InvalidGBSizeError  < StandardError; end
+
+  end
+end

data/lib/jupiter/nagios.rb

@@ -0,0 +1,188 @@
+module Jupiter
+  class TextTemplate
+
+    def initialize( template )
+      @template = template.clone()
+      @values = {}
+    end
+
+    def set( name, value )
+      @values[ name ] = value
+    end
+
+    def run()
+      @template.gsub( /:::(.*?):::/ ) {
+        raise "Key '#{$1}' found in template but the value has not been set" unless ( @values.has_key?( $1 ) )
+        @values[ $1 ].to_s
+      }
+    end
+
+    def to_s()
+      run()
+    end
+
+  end
+
+  class Nagios
+
+    def initialize
+      @url        = Jupiter.nagios.fetch(:url)
+      @location   = Jupiter.nagios.fetch(:location)
+      @sshuser    = Jupiter.nagios.fetch(:sshuser)
+      @sshpass    = Jupiter.nagios.fetch(:sshpass)
+      @sshport    = Jupiter.nagios.fetch(:sshport)
+      @host       = Jupiter.nagios.fetch(:host)
+    end
+
+    def ask(question)
+      print question + " "
+      answer = STDIN.gets.chomp
+      if answer == "" or answer.nil?
+        return nil
+      else
+        return answer
+      end
+    end
+
+    def host_template(machines_name, machines_alias, machines_ip)
+      template_host = Jupiter::TextTemplate.new(
+        "\ndefine host {"\
+        "\n  use                   linux-server"\
+        "\n  host_name             :::HOSTNAME:::"\
+        "\n  alias                 :::HOSTALIAS:::"\
+        "\n  address               :::IPADDRESS:::"\
+        "\n  statusmap_image       linux40.gd2"\
+        "\n  parents               GATEWAY1"\
+        "\n}\n"
+      )
+      template_host.set( 'HOSTNAME', machines_name)
+      template_host.set( 'HOSTALIAS', machines_alias)
+      template_host.set( 'IPADDRESS', machines_ip)
+
+      return template_host
+    end
+
+    def services_template(machines_name)
+      template_services = Jupiter::TextTemplate.new(
+	      "\ndefine service {"\
+        "\n  use                   local-service"\
+        "\n  host_name             :::HOSTNAME:::"\
+        "\n  service_description   Root Partition"\
+        "\n  check_command         check_local_disk!20%!10%!/"\
+        "\n  notifications_enabled 1"\
+        "\n}\n"\
+        "\ndefine service{"\
+        "\n  use                   local-service"\
+        "\n  host_name             :::HOSTNAME:::"\
+        "\n  service_description   Current Users"\
+        "\n  check_command         check_local_users!20!50"\
+        "\n  notifications_enabled 1"\
+        "\n}\n"\
+        "\ndefine service{"\
+        "\n  use                   local-service"\
+        "\n  host_name             :::HOSTNAME:::"\
+        "\n  service_description   Total Processes"\
+        "\n  check_command         check_local_procs!250!400!RSZDT"\
+        "\n  notifications_enabled 1"\
+        "\n}\n"\
+        "\ndefine service{"\
+        "\n  use                   local-service"\
+        "\n  host_name             :::HOSTNAME:::"\
+        "\n  service_description   Current Load"\
+        "\n  check_command         check_local_load!5.0,4.0,3.0!10.0,6.0,4.0"\
+        "\n  notifications_enabled 1"\
+        "\n}\n"\
+        "\ndefine service{"\
+        "\n  use                   local-service"\
+        "\n  host_name             :::HOSTNAME:::"\
+        "\n  service_description   Swap Usage"\
+        "\n  check_command         check_local_swap!20!10"\
+        "\n  notifications_enabled 1"\
+        "\n}\n"\
+        "\ndefine service{"\
+        "\n  use                   local-service"\
+        "\n  host_name             :::HOSTNAME:::"\
+        "\n  service_description   SSH"\
+        "\n  check_command         check_ssh!-p 2500"\
+        "\n  notifications_enabled 0"\
+        "\n}\n"\
+        "\ndefine service{"\
+        "\n  use                   local-service"\
+        "\n  host_name             :::HOSTNAME:::"\
+        "\n  service_description   HTTP"\
+        "\n  check_command         check_http"\
+        "\n  notifications_enabled 0"\
+        "\n}"\
+      )
+      template_services.set( 'HOSTNAME', machines_name)
+
+      return template_services
+    end
+
+    def action(choice, machines_name, machines_alias, machines_ip)
+      if (choice == 1)
+        include_services = ask("\nWould you like to include default services? (yes / no): ")
+        if (include_services == 'yes')
+          puts "\n\n\n"
+          puts generate_template(machines_name, machines_alias, machines_ip, 1)
+          puts "\n\n\n"
+        end
+        if (include_services == 'no')
+          puts "\n\n\n"
+          puts generate_template(machines_name, machines_alias, machines_ip, 0)
+          puts "\n\n\n"
+        end
+      elsif (choice == 2)
+        include_services = ask("\nWould you like to include default services? (yes / no): ")
+        if (include_services == 'yes')
+          puts "\n\n\n"
+          upload_nagios_config(machines_name, machines_alias, machines_ip, 1)
+          puts "Completed adding nagios config for #{machines_name}. Please restart Nagios to see changes."
+          puts "\n\n\n"
+        end
+        if (include_services == 'no')
+          puts "\n\n\n"
+          upload_nagios_config(machines_name, machines_alias, machines_ip, 0)
+          puts "Added Nagios configuration for #{machines_name}. Please restart Nagios to see changes."
+          puts "\n\n\n"
+        end
+      else
+        raise InvalidSelectionError.new("Not a valid selection.")
+      end
+    end
+
+    def generate_template(machines_name, machines_alias, machines_ip, services)
+      host = host_template(machines_name, machines_alias, machines_ip)
+      serv = services_template(machines_name)
+      if (services == 1)
+        template = "########## GENERATED BY JUPITER ##########\n#{host}#{serv}\n\n########## GENERATED BY JUPITER ##########"
+      else
+        template = "########## GENERATED BY JUPITER ##########\n#{host}\n\n########## GENERATED BY JUPITER ##########"
+      end
+      return template
+    end
+
+    def upload_nagios_config(machines_name, machines_alias, machines_ip, services)
+      template = generate_template(machines_name, machines_alias, machines_ip, services)
+      puts template
+      puts ssh.exec!(%Q[echo "#{template}" > #{@location}/#{machines_name}.cfg])
+      close_ssh
+    end
+    
+    def ssh
+      @ssh ||= Net::SSH.start(@host, @sshuser, password: @sshpass, port: @sshport)
+    end
+
+    def close_ssh
+      ssh.close
+      @ssh = nil
+      true
+    end
+
+  end
+
+  class InvalidSelectionError < StandardError; end
+
+end
+
+