rvc 1.0.0

data/lib/rvc/extensions/Network.rb

@@ -0,0 +1,28 @@
+# Copyright (c) 2011 VMware, Inc.  All Rights Reserved.
+# 
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+# 
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+# 
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+class RbVmomi::VIM::Network
+  def display_info
+    summary, = collect(:summary)
+    puts "name: #{summary.name}"
+    puts "accessible: #{summary.accessible}"
+    puts "IP pool name: #{summary.ipPoolName}" unless summary.ipPoolName.empty?
+  end
+end

data/lib/rvc/extensions/ResourcePool.rb

@@ -0,0 +1,52 @@
+# Copyright (c) 2011 VMware, Inc.  All Rights Reserved.
+# 
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+# 
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+# 
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+class RbVmomi::VIM::ResourcePool
+  def self.ls_properties
+    %w(name config.cpuAllocation config.memoryAllocation)
+  end
+
+  def ls_text r
+    cpuAlloc, memAlloc = r['config.cpuAllocation'], r['config.memoryAllocation']
+
+    cpu_shares_text = cpuAlloc.shares.level == 'custom' ? cpuAlloc.shares.shares.to_s : cpuAlloc.shares.level
+    mem_shares_text = memAlloc.shares.level == 'custom' ? memAlloc.shares.shares.to_s : memAlloc.shares.level
+
+    ": cpu %0.2f/%0.2f/%s, mem %0.2f/%0.2f/%s" % [
+      cpuAlloc.reservation/1000.0, cpuAlloc.limit/1000.0, cpu_shares_text,
+      memAlloc.reservation/1000.0, memAlloc.limit/1000.0, mem_shares_text,
+    ]
+  end
+
+  def children
+    {
+      'vms' => RVC::FakeFolder.new(self, :children_vms),
+      'pools' => RVC::FakeFolder.new(self, :children_pools),
+    }
+  end
+
+  def children_vms
+    RVC::Util.collect_children self, :vm
+  end
+
+  def children_pools
+    RVC::Util.collect_children self, :resourcePool
+  end
+end

data/lib/rvc/extensions/VirtualMachine.rb

@@ -0,0 +1,72 @@
+# Copyright (c) 2011 VMware, Inc.  All Rights Reserved.
+# 
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+# 
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+# 
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+class RbVmomi::VIM::VirtualMachine
+  def display_info
+    config, runtime, guest = collect :config, :runtime, :guest
+
+    puts "name: #{config.name}"
+    puts "note: #{config.annotation}" if config.annotation and !config.annotation.empty?
+    puts "host: #{runtime.host.path[1..-1].map { |x| x[1] } * '/'}" if runtime.host
+    puts "tools: #{guest.toolsRunningStatus}"
+    puts "hostname: #{guest.hostName} (#{guest.ipAddress})" if guest.hostName and guest.ipAddress
+    puts "VC UUID: #{config.instanceUuid}" if config.instanceUuid and !config.instanceUuid.empty?
+    puts "power: #{runtime.powerState}"
+    if runtime.question
+      puts "question: #{runtime.question.text.lines.to_a.join("> ")}"
+      puts "choices: #{runtime.question.choice.choiceInfo.map(&:label) * ', '}"
+      if i = runtime.question.choice.defaultIndex
+        puts "default: #{runtime.question.choice.choiceInfo[i].label}"
+      end
+    end
+    puts "cpus: #{config.hardware.numCPU}"
+    puts "memory: #{config.hardware.memoryMB} MB"
+
+    puts "nics:"
+    config.hardware.device.grep RbVmomi::VIM::VirtualEthernetCard do |dev|
+      backing_info = case dev.backing
+      when RbVmomi::VIM::VirtualEthernetCardNetworkBackingInfo
+        dev.backing.deviceName.inspect
+      when RbVmomi::VIM::VirtualEthernetCardDistributedVirtualPortBackingInfo
+        dev.backing.port.portgroupKey.inspect
+      else
+        dev.backing.class.name
+      end
+      guest_net = guest.net.find { |x| x.macAddress == dev.macAddress }
+      puts " #{dev.deviceInfo.label}: #{backing_info} #{dev.connectable.connected ? :connected : :disconnected} #{dev.macAddress} #{guest_net ? (guest_net.ipAddress * ' ') : ''}"
+    end
+  end
+
+  def self.ls_properties
+    %w(name runtime.powerState)
+  end
+  
+  def ls_text r
+    ": #{r['runtime.powerState']}"
+  end
+
+  def children
+    host, resourcePool = collect *%w(runtime.host resourcePool)
+    {
+      'host' => host,
+      'resourcePool' => resourcePool,
+    }
+  end
+end

data/lib/rvc/fs.rb

@@ -0,0 +1,123 @@
+# Copyright (c) 2011 VMware, Inc.  All Rights Reserved.
+# 
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+# 
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+# 
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+module RVC
+
+class Location
+  attr_reader :stack
+
+  def initialize root
+    @stack = [['', root]]
+  end
+
+  def initialize_copy src
+    super
+    @stack = @stack.dup
+  end
+
+  def push name, obj
+    @stack << [name, obj]
+  end
+
+  def pop
+    @stack.pop
+  end
+
+  def obj
+    @stack.empty? ? nil : @stack[-1][1]
+  end
+
+  def path
+    @stack.map { |name,obj| name }
+  end
+end
+
+class FS
+  attr_reader :root, :loc, :marks
+
+  MARK_REGEX = /^~(?:([\d\w]*|~|@))$/
+
+  def initialize root
+    @root = root
+    @loc = Location.new root
+    @marks = {}
+  end
+
+  def cur
+    @loc.obj
+  end
+
+  def display_path
+    @loc.path * '/'
+  end
+
+  def lookup path
+    (lookup_loc(path) || return).obj
+  end
+
+  def cd path
+    new_loc = lookup_loc(path) or return false
+    mark '~', @loc
+    @loc = new_loc
+    true
+  end
+
+  def lookup_loc path
+    els, absolute, trailing_slash = Path.parse path
+    base_loc = absolute ? Location.new(@root) : @loc
+    traverse(base_loc, els)
+  end
+
+  def traverse base_loc, els
+    loc = base_loc.dup
+    els.each_with_index do |el,i|
+      case el
+      when '.'
+      when '..'
+        loc.pop unless loc.obj == @root
+      when '...'
+        loc.push(el, loc.obj.parent) unless loc.obj == @root
+      when MARK_REGEX
+        return unless i == 0
+        loc = @marks[$1] or return
+        loc = loc.dup
+      else
+        # XXX check for ambiguous child
+        if i == 0 and el =~ /^\d+$/ and @marks.member? el
+          loc = @marks[el].dup
+        else
+          x = loc.obj.traverse_one(el) or return
+          loc.push el, x
+        end
+      end
+    end
+    loc
+  end
+
+  def mark key, loc
+    if loc == nil
+      @marks.delete key
+    else
+      @marks[key] = loc
+    end
+  end
+end
+
+end

data/lib/rvc/inventory.rb

@@ -0,0 +1,125 @@
+# Copyright (c) 2011 VMware, Inc.  All Rights Reserved.
+# 
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+# 
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+# 
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+module RVC
+
+module InventoryObject
+  module ClassMethods
+    def ls_properties
+      %w()
+    end
+
+    def folder?
+      false
+    end
+  end
+
+  def self.included m
+    m.extend ClassMethods
+  end
+
+  def display_info
+    puts "class: #{self.class.name}"
+  end
+
+  def ls_text r
+    self.class.folder? ? '/' : ''
+  end
+
+  def traverse_one arc
+    children[arc]
+  end
+
+  def children
+    {}
+  end
+
+  def parent
+    nil
+  end
+end
+
+class FakeFolder
+  include RVC::InventoryObject
+
+  def initialize target, method
+    @target = target
+    @method = method
+  end
+
+  def children
+    @target.send @method
+  end
+
+  def self.folder?
+    true
+  end
+
+  def parent
+    @target
+  end
+
+  def eql? x
+    @target == x.instance_variable_get(:@target) &&
+      @method == x.instance_variable_get(:@method)
+  end
+
+  def hash
+    @target.hash ^ @method.hash
+  end
+end
+
+class RootNode
+  include RVC::InventoryObject
+
+  def children
+    $shell.connections
+  end
+
+  def parent
+    nil
+  end
+
+  def self.folder?
+    true
+  end
+
+  def pretty_print pp
+    pp.text "Root"
+  end
+end
+
+end
+
+class RbVmomi::VIM
+  include RVC::InventoryObject
+
+  def children
+    rootFolder.children
+  end
+
+  def parent
+    $shell.fs.root
+  end
+
+  def self.folder?
+    true
+  end
+end

data/lib/rvc/modules.rb

@@ -0,0 +1,74 @@
+# Copyright (c) 2011 VMware, Inc.  All Rights Reserved.
+# 
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+# 
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+# 
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+module RVC
+
+ALIASES = {}
+MODULES = {}
+
+class CmdModule
+  def initialize module_name
+    @module_name = module_name
+    @opts = {}
+    super()
+  end
+
+  def commands
+    @opts.keys
+  end
+
+  def opts cmd, &b
+    @opts[cmd] = b
+  end
+
+  def opts_for cmd
+    @opts[cmd]
+  end
+
+  def rvc_alias cmd, target=nil
+    target ||= cmd
+    RVC::ALIASES[target.to_s] = "#{@module_name}.#{cmd}"
+  end
+end
+
+BULTIN_MODULE_PATH = [File.expand_path(File.join(File.dirname(__FILE__), 'modules')),
+                      File.join(ENV['HOME'], ".rvc")]
+ENV_MODULE_PATH = (ENV['RVC_MODULE_PATH'] || '').split ':'
+
+def self.reload_modules verbose=true
+  RVC::MODULES.clear
+  RVC::ALIASES.clear
+  RVC::MODULES['custom'] = CmdModule.new 'custom'
+  module_path = (BULTIN_MODULE_PATH+ENV_MODULE_PATH).select { |d| File.directory?(d) }
+  globs = module_path.map { |d| File.join(d, '*.rb') }
+  Dir.glob(globs) do |f|
+    module_name = File.basename(f)[0...-3]
+    puts "loading #{module_name} from #{f}" if verbose
+    code = File.read f
+    unless RVC::MODULES.member? module_name
+      m = CmdModule.new module_name
+      CMD.define_singleton_method(module_name.to_sym) { m }
+      RVC::MODULES[module_name] = m
+    end
+    RVC::MODULES[module_name].instance_eval code, f
+  end
+end
+
+end