knife-vsphere 1.0.1 → 1.2.0

data/lib/chef/knife/vsphere_datastorecluster_list.rb

@@ -1,71 +1,66 @@
-# 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_vsphere_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
-
-def traverse_folders_for_dsclusters(folder)
-  print_dsclusters_in_folder(folder)
-  folder.childEntity.each do |child|
-    if child.class.to_s == 'Folder'
-     traverse_folders_for_dsclusters(child)
-    end
-  end
-end
-
-def print_dsclusters_in_folder (folder)
-  folder.childEntity.each do |child|
-    if child.class.to_s == "StoragePod"
-      avail = number_to_human_size(child.summary[:freeSpace])
-      cap = number_to_human_size(child.summary[:capacity])
-      puts "#{ui.color("DatastoreCluster", :cyan)}: #{child.name} (#{avail} / #{cap})"
-    end
-  end
-end
-
-# Lists all known data store cluster in datacenter with sizes
-class Chef::Knife::VsphereDatastoreclusterList < Chef::Knife::BaseVsphereCommand
-
-  banner "knife vsphere datastorecluster list"
-
-  get_common_options
-
-  def run
-    $stdout.sync = true
-    vim = get_vim_connection
-    dc = get_datacenter
-    traverse_folders_for_dsclusters(dc.datastoreFolder)
-  end
-end
-
+# 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_vsphere_command'
+
+def number_to_human_size(number)
+  number = number.to_f
+  storage_units_fmt = %w(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
+
+  format('%0.2f %s', number, unit)
+end
+
+def traverse_folders_for_dsclusters(folder)
+  print_dsclusters_in_folder(folder)
+  folder.childEntity.each do |child|
+    traverse_folders_for_dsclusters(child) if child.class.to_s == 'Folder'
+  end
+end
+
+def print_dsclusters_in_folder(folder)
+  folder.childEntity.each do |child|
+    next unless child.class.to_s == 'StoragePod'
+    avail = number_to_human_size(child.summary[:freeSpace])
+    cap = number_to_human_size(child.summary[:capacity])
+    puts "#{ui.color('DatastoreCluster', :cyan)}: #{child.name} (#{avail} / #{cap})"
+  end
+end
+
+# Lists all known data store cluster in datacenter with sizes
+class Chef::Knife::VsphereDatastoreclusterList < Chef::Knife::BaseVsphereCommand
+  banner 'knife vsphere datastorecluster list'
+
+  common_options
+
+  def run
+    $stdout.sync = true
+    vim_connection
+    dc = datacenter
+    traverse_folders_for_dsclusters(dc.datastoreFolder)
+  end
+end

data/lib/chef/knife/vsphere_datastorecluster_maxfree.rb

@@ -1,84 +1,75 @@
-# 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_vsphere_command'
-
-def is_max_dscluster(dscluster, max_dscluster)
-  if ! max_dscluster 
-    return true
-  end
-
-  if dscluster.summary[:freeSpace] > max_dscluster.summary[:freeSpace]
-    return true
-  end
-
-  return false
-end
-
-def find_max_dscluster(folder, max_dscluster, regex)
-  folder.childEntity.each do |child|
-    if child.class.to_s == 'Folder'
-      sub_max = find_max_dscluster(child, max_dscluster, regex)
-      if is_max_dscluster(sub_max, max_dscluster)
-        max_dscluster = sub_max
-      end
-    elsif child.class.to_s == 'StoragePod'
-      
-      if is_max_dscluster(child, max_dscluster) && regex.match(child.name)
-        max_dscluster = child
-      end
-    end
-  end
-
-  return max_dscluster
-end
-
-# Gets the data store cluster with the most free space in datacenter
-class Chef::Knife::VsphereDatastoreclusterMaxfree < Chef::Knife::BaseVsphereCommand
-
-  banner "knife vsphere datastorecluster maxfree"
-
-  option :regex,
-         :short => "-r REGEX",
-         :long => "--regex REGEX",
-         :description => "Regex to match the datastore cluster name"
-  get_common_options
-  $default[:regex] = ''
-
-  def run
-    $stdout.sync = true
-
-    vim = get_vim_connection
-    dcname = get_config(:vsphere_dc)
-    regex = /#{Regexp.escape( get_config(:regex))}/
-    max_dscluster = nil
-    
-    vim = get_vim_connection
-    dc = get_datacenter
-
-    max_dscluster = find_max_dscluster(dc.datastoreFolder, max_dscluster, regex)
-  
-    if max_dscluster 
-      puts max_dscluster.name
-    else
-      puts "No datastore clusters found"
-      exit 1
-    end
-
-  end
-end
+# 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_vsphere_command'
+
+def max_dscluster(dscluster, max_dscluster)
+  return true unless max_dscluster
+
+  if dscluster.summary[:freeSpace] > max_dscluster.summary[:freeSpace]
+    return true
+  end
+
+  false
+end
+
+def find_max_dscluster(folder, max_dscluster, regex)
+  folder.childEntity.each do |child|
+    if child.class.to_s == 'Folder'
+      sub_max = find_max_dscluster(child, max_dscluster, regex)
+      max_dscluster = sub_max if max_dscluster(sub_max, max_dscluster)
+    elsif child.class.to_s == 'StoragePod'
+      if max_dscluster(child, max_dscluster) && regex.match(child.name)
+        max_dscluster = child
+      end
+    end
+  end
+
+  max_dscluster
+end
+
+# Gets the data store cluster with the most free space in datacenter
+class Chef::Knife::VsphereDatastoreclusterMaxfree < Chef::Knife::BaseVsphereCommand
+  banner 'knife vsphere datastorecluster maxfree'
+
+  option :regex,
+         short: '-r REGEX',
+         long: '--regex REGEX',
+         description: 'Regex to match the datastore cluster name',
+         default: ''
+  common_options
+
+  def run
+    $stdout.sync = true
+
+    regex = /#{Regexp.escape(get_config(:regex))}/
+    max_dscluster = nil
+
+    vim_connection
+    dc = datacenter
+
+    max_dscluster = find_max_dscluster(dc.datastoreFolder, max_dscluster, regex)
+
+    if max_dscluster
+      puts max_dscluster.name
+    else
+      puts 'No datastore clusters found'
+      exit 1
+    end
+  end
+end

data/lib/chef/knife/vsphere_folder_list.rb

@@ -1,30 +1,28 @@
-#
-# Author:: Ezra Pagel (<ezra@cpan.org>)
-# License:: Apache License, Version 2.0
-#
-require 'chef/knife'
-require 'chef/knife/base_vsphere_command'
-
-# Lists all vm folders
-class Chef::Knife::VsphereFolderList < Chef::Knife::BaseVsphereCommand
-
-  banner "knife vsphere folder list"
-
-  get_common_options
-
-  def traverse_folders(folder, indent_level)
-
-    puts "#{"  " * indent_level} #{ui.color("Folder", :cyan)}: " + folder.name
-
-    folders = find_all_in_folder(folder, RbVmomi::VIM::Folder)
-    folders.each do |child|
-      traverse_folders(child, indent_level  + 1)
-    end
-  end
-
-  def run
-    vim = get_vim_connection
-    baseFolder = find_folder(get_config(:folder));
-    traverse_folders(baseFolder, 0)
-  end
-end
+#
+# Author:: Ezra Pagel (<ezra@cpan.org>)
+# License:: Apache License, Version 2.0
+#
+require 'chef/knife'
+require 'chef/knife/base_vsphere_command'
+
+# Lists all vm folders
+class Chef::Knife::VsphereFolderList < Chef::Knife::BaseVsphereCommand
+  banner 'knife vsphere folder list'
+
+  common_options
+
+  def traverse_folders(folder, indent_level)
+    puts "#{'  ' * indent_level} #{ui.color('Folder', :cyan)}: " + folder.name
+
+    folders = find_all_in_folder(folder, RbVmomi::VIM::Folder)
+    folders.each do |child|
+      traverse_folders(child, indent_level  + 1)
+    end
+  end
+
+  def run
+    vim_connection
+    base_folder = find_folder(get_config(:folder))
+    traverse_folders(base_folder, 0)
+  end
+end

data/lib/chef/knife/vsphere_hosts_list.rb

@@ -1,42 +1,42 @@
-require 'chef/knife'
-require 'chef/knife/base_vsphere_command'
-require 'rbvmomi'
-require 'netaddr'
-
-#list hosts belonging to pool
-class Chef::Knife::VsphereHostsList < Chef::Knife::BaseVsphereCommand
-  banner "knife vsphere hosts list"
-
-  get_common_options
-  option :pool,
-         :long => "--pool pool",
-         :short => "-h",
-         :description => "Target pool"
-
-  def find_pools(folder, poolname = nil)
-    pools = folder.children.find_all.select {|p| p.is_a?(RbVmomi::VIM::ComputeResource) || p.is_a?(RbVmomi::VIM::ResourcePool) }
-    poolname.nil? ? pools : pools.select {|p| p.name == poolname }
-  end
-
-  def run
-    vim = get_vim_connection
-    dc = get_datacenter
-    folder = dc.hostFolder
-
-    target_pool = config[:pool]
-
-    pools = find_pools(folder, target_pool)
-    if target_pool && pools.empty? 
-      puts "Pool #{target_pool} not found"
-      return
-    end
-
-    pools.each do |pool|
-      puts "#{ui.color("Pool", :cyan)}: #{pool.name}"
-      hosts = pool.host || []
-      hosts.each do |hostc|
-        puts "  #{ui.color("Host", :cyan)}: #{hostc.name}"
-      end
-    end
-  end
-end
+require 'chef/knife'
+require 'chef/knife/base_vsphere_command'
+require 'rbvmomi'
+require 'netaddr'
+
+# list hosts belonging to pool
+class Chef::Knife::VsphereHostsList < Chef::Knife::BaseVsphereCommand
+  banner 'knife vsphere hosts list'
+
+  common_options
+  option :pool,
+         long: '--pool pool',
+         short: '-h',
+         description: 'Target pool'
+
+  def find_pools(folder, poolname = nil)
+    pools = folder.children.find_all.select { |p| p.is_a?(RbVmomi::VIM::ComputeResource) || p.is_a?(RbVmomi::VIM::ResourcePool) }
+    poolname.nil? ? pools : pools.select { |p| p.name == poolname }
+  end
+
+  def run
+    vim_connection
+    dc = datacenter
+    folder = dc.hostFolder
+
+    target_pool = config[:pool]
+
+    pools = find_pools(folder, target_pool)
+    if target_pool && pools.empty?
+      puts "Pool #{target_pool} not found"
+      return
+    end
+
+    pools.each do |pool|
+      puts "#{ui.color('Pool', :cyan)}: #{pool.name}"
+      hosts = pool.host || []
+      hosts.each do |hostc|
+        puts "  #{ui.color('Host', :cyan)}: #{hostc.name}"
+      end
+    end
+  end
+end

data/lib/chef/knife/vsphere_pool_list.rb

@@ -1,48 +1,46 @@
-#
-# Author:: Jesse Campbell (<hikeit@gmail.com>)
-# License:: Apache License, Version 2.0
-#
-require 'chef/knife'
-require 'chef/knife/base_vsphere_command'
-
-# Lists all known pools in the configured datacenter
-class Chef::Knife::VspherePoolList < Chef::Knife::BaseVsphereCommand
-
-  banner "knife vsphere pool list"
-
-  get_common_options
-
-  def traverse_folders(folder)
-    return if folder.is_a? RbVmomi::VIM::VirtualApp
-
-    if folder.is_a? RbVmomi::VIM::ResourcePool
-      pools = folder.path[3..-1].reject { |p| p.last == "Resources" }
-      puts "#{ui.color("Pool", :cyan)}: " + pools.map(&:last).join('/')
-    end
-
-    folders = find_all_in_folder(folder, RbVmomi::VIM::ManagedObject) || []
-    folders.each do |child|
-      traverse_folders(child)
-    end
-
-  end
-
-  def find_pool_folder(folderName)
-    dc = get_datacenter
-    baseEntity = dc.hostFolder
-    entityArray = folderName.split('/')
-    entityArray.each do |entityArrItem|
-      if entityArrItem != ''
-        baseEntity = baseEntity.childEntity.grep(RbVmomi::VIM::ManagedObject).find { |f| f.name == entityArrItem } or
-            abort "no such folder #{folderName} while looking for #{entityArrItem}"
-      end
-    end
-    baseEntity
-  end
-
-  def run
-    vim = get_vim_connection
-    baseFolder = find_pool_folder(get_config(:folder));
-    traverse_folders(baseFolder)
-  end
-end
+#
+# Author:: Jesse Campbell (<hikeit@gmail.com>)
+# License:: Apache License, Version 2.0
+#
+require 'chef/knife'
+require 'chef/knife/base_vsphere_command'
+
+# Lists all known pools in the configured datacenter
+class Chef::Knife::VspherePoolList < Chef::Knife::BaseVsphereCommand
+  banner 'knife vsphere pool list'
+
+  common_options
+
+  def traverse_folders(folder)
+    return if folder.is_a? RbVmomi::VIM::VirtualApp
+
+    if folder.is_a? RbVmomi::VIM::ResourcePool
+      pools = folder.path[3..-1].reject { |p| p.last == 'Resources' }
+      puts "#{ui.color('Pool', :cyan)}: " + pools.map(&:last).join('/')
+    end
+
+    folders = find_all_in_folder(folder, RbVmomi::VIM::ManagedObject) || []
+    folders.each do |child|
+      traverse_folders(child)
+    end
+  end
+
+  def find_pool_folder(folderName)
+    dc = datacenter
+    base_entity = dc.hostFolder
+    entity_array = folderName.split('/')
+    entity_array.each do |entityArrItem|
+      if entityArrItem != ''
+        base_entity = base_entity.childEntity.grep(RbVmomi::VIM::ManagedObject).find { |f| f.name == entityArrItem } ||
+                      abort("no such folder #{folderName} while looking for #{entityArrItem}")
+      end
+    end
+    base_entity
+  end
+
+  def run
+    vim_connection
+    base_folder = find_pool_folder(get_config(:folder))
+    traverse_folders(base_folder)
+  end
+end