knife-ovh-cloud 1.0.0.pre.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 560ff92c7398e4253a1da4f57c702411e2f34c9c
+  data.tar.gz: 40d70a9004086478882349b37a48a8060c2cf18e
+SHA512:
+  metadata.gz: 22aa6e2eac62d522fb690447a2c50e95b673a400f2c0b22ca5ac85ce6e26b8c6ce32afa2adf1f449ab7130e11a7f0f73a357678d77c8d0efb31e42fb81ed586e
+  data.tar.gz: efc304d1a7af10de6f2b403e1a1ff02a251ba9bef54baab2c8f6e9ee63dd409750d90da52408134108b58b6b1f021529ea95870e623c2a501c27baf28db34e4c

data/lib/chef/knife/base_ovh_cloud_command.rb

@@ -0,0 +1,358 @@
+#
+# Author:: Ezra Pagel (<ezra@cpan.org>)
+# Contributor:: Jesse Campbell (<hikeit@gmail.com>)
+# License:: Apache License, Version 2.0
+#
+
+require 'chef/knife'
+require 'rbvmomi'
+
+# Base class for vsphere knife commands
+class Chef
+  class Knife
+    class BaseOvhCloudCommand < Knife
+
+      deps do
+        require 'chef/knife/bootstrap'
+        Chef::Knife::Bootstrap.load_deps
+        require 'fog'
+        require 'socket'
+        require 'net/ssh/multi'
+        require 'readline'
+        require 'chef/json_compat'
+      end
+
+      def self.get_common_options
+        unless defined? $default
+          $default = Hash.new
+        end
+
+        option :ovh_cloud_user,
+               :short => "-u USERNAME",
+               :long => "--vsuser USERNAME",
+               :description => "The username for vsphere"
+
+        option :ovh_cloud_pass,
+               :short => "-p PASSWORD",
+               :long => "--vspass PASSWORD",
+               :description => "The password for vsphere"
+
+        option :ovh_cloud_host,
+               :long => "--vshost HOST",
+               :description => "The vsphere host"
+
+        option :ovh_cloud_dc,
+               :short => "-D DATACENTER",
+               :long => "--vsdc DATACENTER",
+               :description => "The Datacenter for vsphere"
+
+        option :ovh_cloud_path,
+               :long => "--vspath SOAP_PATH",
+               :description => "The vsphere SOAP endpoint path"
+        $default[:ovh_cloud_path] = "/sdk"
+
+        option :ovh_cloud_port,
+               :long => "--vsport PORT",
+               :description => "The VI SDK port number to use"
+        $default[:ovh_cloud_port] = 443
+
+        option :vshere_nossl,
+               :long => "--vsnossl",
+               :description => "Disable SSL connectivity"
+
+        option :ovh_cloud_insecure,
+               :long => "--vsinsecure",
+               :description => "Disable SSL certificate verification"
+
+        option :folder,
+               :short => "-f FOLDER",
+               :long => "--folder FOLDER",
+               :description => "The folder to get VMs from"
+
+        option :proxy_host,
+               :long => '--proxyhost PROXY_HOSTNAME',
+               :description => 'Proxy hostname'
+
+        option :proxy_port,
+               :long => '--proxyport PROXY_PORT',
+               :description => 'Proxy port'
+
+        $default[:folder] = ''
+      end
+
+      def get_config(key)
+        key = key.to_sym
+        rval = config[key] || Chef::Config[:knife][key] || $default[key]
+        Chef::Log.debug("value for config item #{key}: #{rval}")
+        rval
+      end
+
+      def get_vim_connection
+
+        conn_opts = {
+            :host => get_config(:ovh_cloud_host),
+            :path => get_config(:ovh_cloud_path),
+            :port => get_config(:ovh_cloud_port),
+            :use_ssl => !get_config(:ovh_cloud_nossl),
+            :user => get_config(:ovh_cloud_user),
+            :password => get_config(:ovh_cloud_pass),
+            :insecure => get_config(:ovh_cloud_insecure),
+            :proxyHost => get_config(:proxy_host),
+            :proxyPort => get_config(:proxy_port)
+        }
+
+        # Grab the password from the command line
+        # if tt is not in the config file
+        if not conn_opts[:password]
+          conn_opts[:password] = get_password
+        end
+
+        #    opt :debug, "Log SOAP messages", :short => 'd', :default => (ENV['RBVMOMI_DEBUG'] || false)
+
+        vim = RbVmomi::VIM.connect conn_opts
+        config[:vim] = vim
+        return vim
+      end
+
+      def get_password
+        @password ||= ui.ask("Enter your password: ") { |q| q.echo = false }
+      end
+
+      def get_vm(vmname)
+        vim = get_vim_connection
+        baseFolder = find_folder(get_config(:folder));
+        retval = traverse_folders_for_vm(baseFolder, vmname)
+        return retval
+      end
+
+      def traverse_folders_for_vm(folder, vmname)
+        children = folder.children.find_all
+        children.each do |child|
+          if child.class == RbVmomi::VIM::VirtualMachine && child.name == vmname
+              return child
+          elsif child.class == RbVmomi::VIM::Folder
+            vm = traverse_folders_for_vm(child, vmname)
+            if vm then return vm end
+          end
+        end
+        return false
+      end
+
+      def get_vms(vmname)
+        vim = get_vim_connection
+        baseFolder = find_folder(get_config(:folder));
+        retval = traverse_folders_for_vms(baseFolder, vmname)
+        return retval
+      end
+
+      def traverse_folders_for_vms(folder, vmname)
+	    retval = []
+        children = folder.children.find_all
+        children.each do |child|
+          if child.class == RbVmomi::VIM::VirtualMachine && child.name == vmname
+              retval << child
+          elsif child.class == RbVmomi::VIM::Folder
+            retval.concat(traverse_folders_for_vms(child, vmname))
+          end
+        end
+        return retval
+      end
+
+      def traverse_folders_for_dc(folder, dcname)
+        children = folder.children.find_all
+        children.each do |child|
+          if child.class == RbVmomi::VIM::Datacenter && child.name == dcname
+            return child
+          elsif child.class == RbVmomi::VIM::Folder
+            dc = traverse_folders_for_dc(child, dcname)
+            if dc then return dc end
+          end
+        end
+        return false
+      end
+
+      def get_datacenter
+        dcname = get_config(:ovh_cloud_dc)
+        traverse_folders_for_dc(config[:vim].rootFolder, dcname) or abort "datacenter not found"
+      end
+
+      def find_folder(folderName)
+        dc = get_datacenter
+        baseEntity = dc.vmFolder
+        entityArray = folderName.split('/')
+        entityArray.each do |entityArrItem|
+          if entityArrItem != ''
+            baseEntity = baseEntity.childEntity.grep(RbVmomi::VIM::Folder).find { |f| f.name == entityArrItem } or
+                abort "no such folder #{folderName} while looking for #{entityArrItem}"
+          end
+        end
+        baseEntity
+      end
+
+      def find_network(networkName)
+        dc = get_datacenter
+        baseEntity = dc.network
+        baseEntity.find { |f| f.name == networkName } or abort "no such network #{networkName}"
+      end
+
+      def find_pool(poolName)
+        dc = get_datacenter
+        baseEntity = dc.hostFolder
+        entityArray = poolName.split('/')
+        entityArray.each do |entityArrItem|
+          if entityArrItem != ''
+            if baseEntity.is_a? RbVmomi::VIM::Folder
+              baseEntity = baseEntity.childEntity.find { |f| f.name == entityArrItem } or
+                  abort "no such pool #{poolName} while looking for #{entityArrItem}"
+            elsif baseEntity.is_a? RbVmomi::VIM::ClusterComputeResource or baseEntity.is_a? RbVmomi::VIM::ComputeResource
+              baseEntity = baseEntity.resourcePool.resourcePool.find { |f| f.name == entityArrItem } or
+                  abort "no such pool #{poolName} while looking for #{entityArrItem}"
+            elsif baseEntity.is_a? RbVmomi::VIM::ResourcePool
+              baseEntity = baseEntity.resourcePool.find { |f| f.name == entityArrItem } or
+                  abort "no such pool #{poolName} while looking for #{entityArrItem}"
+            else
+              abort "Unexpected Object type encountered #{baseEntity.type} while finding resourcePool"
+            end
+          end
+        end
+
+        baseEntity = baseEntity.resourcePool if not baseEntity.is_a?(RbVmomi::VIM::ResourcePool) and baseEntity.respond_to?(:resourcePool)
+        baseEntity
+      end
+
+      def choose_datastore(dstores, size)
+        vmdk_size_kb = size.to_i * 1024 * 1024
+        vmdk_size_B = size.to_i * 1024 * 1024 * 1024
+
+        candidates = []
+        dstores.each do |store|
+          avail = number_to_human_size(store.summary[:freeSpace])
+          cap = number_to_human_size(store.summary[:capacity])
+          puts "#{ui.color("Datastore", :cyan)}: #{store.name} (#{avail}(#{store.summary[:freeSpace]}) / #{cap})"
+
+          # vm's can span multiple datastores, so instead of grabbing the first one
+          # let's find the first datastore with the available space on a LUN the vm
+          # is already using, or use a specified LUN (if given)
+
+
+          if (store.summary[:freeSpace] - vmdk_size_B) > 0
+            # also let's not use more than 90% of total space to save room for snapshots.
+            cap_remains = 100 * ((store.summary[:freeSpace].to_f - vmdk_size_B.to_f) / store.summary[:capacity].to_f)
+            if (cap_remains.to_i > 10)
+              candidates.push(store)
+            end
+          end
+        end
+        if candidates.length > 0
+          vmdk_datastore = candidates[0]
+        else
+          puts "Insufficient space on all LUNs designated or assigned to the virtual machine. Please specify a new target."
+          vmdk_datastore = nil
+        end
+        return vmdk_datastore
+      end
+
+
+      def find_datastores_regex(regex)
+        stores = Array.new()
+        puts "Looking for all datastores that match /#{regex}/"
+        dc = get_datacenter
+        baseEntity = dc.datastore
+        baseEntity.each do |ds|
+          if ds.name.match /#{regex}/
+            stores.push ds
+          end
+        end
+        return stores
+      end
+
+      def find_datastore(dsName)
+        dc = get_datacenter
+        baseEntity = dc.datastore
+        baseEntity.find { |f| f.info.name == dsName } or abort "no such datastore #{dsName}"
+      end
+
+      def find_datastorecluster(dsName)
+        dc = get_datacenter
+        baseEntity = dc.datastoreFolder.childEntity
+        baseEntity.find { |f| f.name == dsName and f.instance_of?(RbVmomi::VIM::StoragePod) } or abort "no such datastorecluster #{dsName}"
+      end
+
+      def find_device(vm, deviceName)
+        vm.config.hardware.device.each do |device|
+          return device if device.deviceInfo.label == deviceName
+        end
+        nil
+      end
+
+      def find_all_in_folder(folder, type)
+        if folder.instance_of?(RbVmomi::VIM::ClusterComputeResource) or folder.instance_of?(RbVmomi::VIM::ComputeResource)
+          folder = folder.resourcePool
+        end
+        if folder.instance_of?(RbVmomi::VIM::ResourcePool)
+          folder.resourcePool.grep(type)
+        elsif folder.instance_of?(RbVmomi::VIM::Folder)
+          folder.childEntity.grep(type)
+        else
+          puts "Unknown type #{folder.class}, not enumerating"
+          nil
+        end
+      end
+
+      def get_path_to_object(object)
+        if object.is_a?(RbVmomi::VIM:: ManagedEntity)
+          if object.parent.is_a?(RbVmomi::VIM:: ManagedEntity)
+            return get_path_to_object(object.parent) + "/" + object.parent.name
+          else
+			return ""
+		  end
+        else
+          puts "Unknown type #{object.class}, not enumerating"
+          nil
+        end
+      end
+
+      def find_in_folder(folder, type, name)
+        folder.childEntity.grep(type).find { |o| o.name == name }
+      end
+
+      def fatal_exit(msg)
+        ui.fatal(msg)
+        exit 1
+      end
+
+      def tcp_test_port_vm(vm, port)
+        ip = vm.guest.ipAddress
+        if ip.nil?
+          sleep 2
+          return false
+        end
+        tcp_test_port(ip, port)
+      end
+
+      def tcp_test_port(hostname, port)
+        tcp_socket = TCPSocket.new(hostname, port)
+        readable = IO.select([tcp_socket], nil, nil, 5)
+        if readable
+          Chef::Log.debug("sshd accepting connections on #{hostname}, banner is #{tcp_socket.gets}") if port == 22
+          true
+        else
+          false
+        end
+      rescue Errno::ETIMEDOUT
+        false
+      rescue Errno::EPERM
+        false
+      rescue Errno::ECONNREFUSED
+        sleep 2
+        false
+      rescue Errno::EHOSTUNREACH, Errno::ENETUNREACH
+        sleep 2
+        false
+      ensure
+        tcp_socket && tcp_socket.close
+      end
+
+    end
+  end
+end

data/lib/chef/knife/ovh_cloud_customization_list.rb

@@ -0,0 +1,29 @@
+#
+# Author:: Ezra Pagel (<ezra@cpan.org>)
+# License:: Apache License, Version 2.0
+#
+require 'chef/knife'
+require 'chef/knife/base_ovh_cloud_command'
+
+# Lists all customization specifications in the configured datacenter
+class Chef::Knife::OvhCloudCustomizationList < Chef::Knife::BaseOvhCloudCommand
+
+  banner "knife ovh cloud customization list"
+
+  get_common_options
+
+  def run
+
+    $stdout.sync = true
+
+    vim = get_vim_connection
+
+    csm = vim.serviceContent.customizationSpecManager
+    csm.info.each do |c|
+      puts "#{ui.color("Customization Name", :cyan)}: #{c.name}"
+
+    end
+
+  end
+end
+

data/lib/chef/knife/ovh_cloud_datastore_list.rb

@@ -0,0 +1,58 @@
+# Copyright (C) 2012, SCM Ventures AB
+# Author: Ian Delahorne <ian@scmventures.se>
+#
+# Permission to use, copy, modify, and/or distribute this software for
+# any purpose with or without fee is hereby granted, provided that the
+# above copyright notice and this permission notice appear in all
+# copies.
+#
+# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL
+# WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED
+# WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE
+# AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
+# DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA
+# OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER
+# TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
+# PERFORMANCE OF THIS SOFTWARE
+
+require 'chef/knife'
+require 'chef/knife/base_ovh_cloud_command'
+
+def number_to_human_size(number)
+  number = number.to_f
+  storage_units_fmt = ["byte", "kB", "MB", "GB", "TB"]
+  base = 1024
+  if number.to_i < base
+    unit = storage_units_fmt[0]
+  else
+    max_exp = storage_units_fmt.size - 1
+    exponent = (Math.log(number) / Math.log(base)).to_i # Convert to base
+    exponent = max_exp if exponent > max_exp # we need this to avoid overflow for the highest unit
+    number /= base ** exponent
+    unit = storage_units_fmt[exponent]
+  end
+
+  return sprintf("%0.2f %s", number, unit)
+end
+
+
+# Lists all known data stores in datacenter with sizes
+class Chef::Knife::OvhCloudDatastoreList < Chef::Knife::BaseOvhCloudCommand
+
+  banner "knife ovh cloud datastore list"
+
+  get_common_options
+
+  def run
+    $stdout.sync = true
+
+    vim = get_vim_connection
+    dc = get_datacenter
+    dc.datastore.each do |store|
+      avail = number_to_human_size(store.summary[:freeSpace])
+      cap = number_to_human_size(store.summary[:capacity])
+      puts "#{ui.color("Datastore", :cyan)}: #{store.name} (#{avail} / #{cap})"
+    end
+  end
+end
+