vi_api_lib 0.1.0

data/lib/vmware/objects/host_system.rb

@@ -0,0 +1,21 @@
+class VMware::HostSystem < VMware::ManagedEntity
+   
+  def networks
+    self["network"].collect do |network|
+      @session.managed_object_wrapper_factory("Network", network)
+    end
+  end
+  
+  def datastores
+    self["datastore"].collect do |datastore|
+      @session.managed_object_wrapper_factory("Datastore", datastore)
+    end
+  end
+  
+  def vms
+    self["vm"].collect do |vm|
+      @session.managed_object_wrapper_factory("VirtualMachine", vm)
+    end
+  end
+  
+end

data/lib/vmware/objects/managed_entity.rb

@@ -0,0 +1,87 @@
+#
+# Wrapper for managed object references.
+#
+class VMware::ManagedEntity
+  
+  attr_reader :object_reference
+  
+  #
+  # Create a wrapper for managed object based on a ManagedObjectReference
+  #
+  def initialize(session, object_reference)
+    @session, @object_reference = session, object_reference
+    
+    # We cache properties for more efficient access.
+    @cached_properties = nil
+  end
+  
+  #
+  # Retrieve a set of properties from the object as a hash. Specify individual
+  # property names as a list, or leave empty to get all properties.
+  #
+  def properties(prop_list = [])
+    if prop_list.empty? and @cached_properties
+      return @cached_properties
+    end
+    
+    #  Object at which filter starts.
+    object_spec = ObjectSpec.new
+    object_spec.obj = @object_reference
+    object_spec.skip = false
+
+    # Which properties to return.
+    prop_spec = PropertySpec.new
+    prop_spec.type = self.class.to_s.gsub(/^.*::/, "")
+    prop_spec.pathSet = prop_list
+    prop_spec.all = prop_list.empty?
+
+    # Create the filter.
+    prop_filter_spec = PropertyFilterSpec.new
+    prop_filter_spec.objectSet = [object_spec]
+    prop_filter_spec.propSet = prop_spec
+
+    # Create the request.
+    request = RetrievePropertiesRequestType.new(@session.sic.propertyCollector)
+    request.specSet = prop_filter_spec
+
+    prop_hash = {}
+    begin
+      object_list = @session.retrieveProperties(request)
+      if object_list.length > 1
+        raise "Unexpected number of objects returned when fetching properties for #{self.class}:\n#{object_list.inspect}\n"
+      end
+      if object_list.empty?
+        puts "WARNING: #{self.class} has no properties"
+        return {}
+      end
+      
+      # Convert the properties into a hash and wrap data objects.
+      prop_set = object_list[0].propSet
+      prop_set.each do |prop|
+        prop_hash[prop.name] = VMware::DataObject::soap_to_data_object(prop.val)
+      end
+    rescue SOAP::FaultError => e
+      puts e.inspect
+    end
+    
+    # Only cache the properties if we retrieved all of them.
+    if prop_list.empty?
+      @cached_properties = prop_hash
+    end
+    
+    return prop_hash
+  end
+  
+  #
+  # Return a specific property.
+  #
+  def [](key)
+    if @cached_properties
+      p = properties
+    else
+      p = properties(key)
+    end
+    return p[key.to_s]
+  end
+  
+end

data/lib/vmware/objects/network.rb

@@ -0,0 +1,15 @@
+class VMware::Network < VMware::ManagedEntity
+   
+  def hosts
+    self["host"].collect do |host|
+      @session.managed_object_wrapper_factory("HostSystem", host)
+    end
+  end
+  
+  def vms
+    self["vm"].collect do |vm|
+      @session.managed_object_wrapper_factory("VirtualMachine", vm)
+    end
+  end
+  
+end

data/lib/vmware/objects/virtual_machine.rb

@@ -0,0 +1,19 @@
+class VMware::VirtualMachine < VMware::ManagedEntity
+   
+  def networks
+    self["network"].collect do |network|
+      @session.managed_object_wrapper_factory("Network", network)
+    end
+  end
+  
+  def datastores
+    self["datastore"].collect do |datastore|
+      @session.managed_object_wrapper_factory("Datastore", datastore)
+    end
+  end
+  
+  def resource_pool
+    @session.managed_object_wrapper_factory("ResourcePool", self["resourcePool"])
+  end
+  
+end

data/lib/vmware/session.rb

@@ -0,0 +1,67 @@
+
+#
+# A session incorporates the API specific state of the server
+# connection.
+#
+class VMware::Session
+  
+  #
+  # Service content object provides our root to the object hierarchy.
+  #
+  attr_reader :sic
+  
+  #
+  # Create a new session by connecting to the server, getting the 
+  # service instance and logging in.
+  # 
+  def initialize(connection)
+    @connection = connection
+    
+    svcref = ManagedObjectReference.new("ServiceInstance")
+    svcref.xmlattr_type = "ServiceInstance"
+    retrieveServiceContent = RetrieveServiceContentRequestType.new(svcref)
+    @sic = connection.retrieveServiceContent(retrieveServiceContent).returnval
+    
+    login
+  end
+  
+  #
+  # Return the root folder of the inventory hierarchy.
+  #
+  def root_folder
+    managed_object_wrapper_factory("Folder", @sic.rootFolder)
+  end
+  
+  #
+  # Login to the server.
+  #
+  def login
+    loginRequest = LoginRequestType.new(@sic.sessionManager)
+    loginRequest.userName = @connection.username
+    loginRequest.password = @connection.password
+    @connection.login(loginRequest);
+  end
+  
+  #
+  # Based on the type of a ManagedObjectReference, generate the 
+  # appropriate wrapper instance.
+  #
+  def managed_object_wrapper_factory(type, reference)
+    #
+    # If we know about the type try to instantiate it.
+    #
+    begin
+      eval("VMware::#{type}.new(self, reference)")
+    rescue
+      puts "WARNING: Unknown managed object type: #{type}"
+      VMware::ManagedEntity.new(self, reference)
+    end
+  end
+  
+  #
+  # Catch any function call and send it over the soap connection.
+  #
+  def method_missing(methodSymbol, *params)
+    @connection.send(methodSymbol, *params)
+  end
+end