class RbVmomi::VIM::Folder

# File lib/rbvmomi/vim/Folder.rb, line 112
def children
  childEntity
end

# File lib/rbvmomi/vim/Folder.rb, line 9
def find name, type=Object
  x = _connection.searchIndex.FindChild(:entity => self, :name => name)
  x if x.is_a? type
end

# File lib/rbvmomi/vim/Folder.rb, line 19
def findByDnsName name, type=RbVmomi::VIM::VirtualMachine, dc=nil
  propSpecs = {
    :entity => self, :dnsName => name,
    :vmSearch => type == RbVmomi::VIM::VirtualMachine
  }
  propSpecs[:datacenter] = dc if dc
  x = _connection.searchIndex.FindByDnsName(propSpecs)
  x if x.is_a? type
end

# File lib/rbvmomi/vim/Folder.rb, line 66
def findByInventoryPath path
  propSpecs = {
    :entity => self, :inventoryPath => path
  }
  _connection.searchIndex.FindByInventoryPath(propSpecs)
end

# File lib/rbvmomi/vim/Folder.rb, line 34
def findByIp ip, type=RbVmomi::VIM::VirtualMachine, dc=nil
  propSpecs = {
    :entity => self, :ip => ip,
    :vmSearch => type == RbVmomi::VIM::VirtualMachine
  }
  propSpecs[:datacenter] = dc if dc
  x = _connection.searchIndex.FindByIp(propSpecs)
  x if x.is_a? type
end

# File lib/rbvmomi/vim/Folder.rb, line 51
def findByUuid(uuid, type = RbVmomi::VIM::VirtualMachine, dc = nil, instance_uuid = false)
  prop_specs = {
    :entity       => self,
    :instanceUuid => instance_uuid,
    :uuid         => uuid,
    :vmSearch     => type == RbVmomi::VIM::VirtualMachine
  }
  prop_specs[:datacenter] = dc if dc
  x = _connection.searchIndex.FindByUuid(prop_specs)
  x if x.is_a?(type)
end

# File lib/rbvmomi/vim/Folder.rb, line 203
def inventory propSpecs={}
  inv = inventory_flat propSpecs
  tree = { self => {} }
  inv.each do |obj,x|
    next if obj == self
    h = Hash[x.propSet.map { |y| [y.name, y.val] }]
    tree[h['parent']][h['name']] = [obj, h]
    tree[obj] = {} if obj.is_a? RbVmomi::VIM::Folder
  end
  tree
end

# File lib/rbvmomi/vim/Folder.rb, line 125
def inventory_flat propSpecs={}
  propSet = [{ :type => 'Folder', :pathSet => ['name', 'parent', 'childEntity'] }]
  propSpecs.each do |k,v|
    case k
    when Class
      fail "key must be a subclass of ManagedEntity" unless k < RbVmomi::VIM::ManagedEntity
      k = k.wsdl_name
    when Symbol, String
      k = k.to_s
    else
      fail "invalid key"
    end

    h = { :type => k }
    if v == :all
      h[:all] = true
    elsif v.is_a? Array
      h[:pathSet] = v + %w(parent)
    else
      fail "value must be an array of property paths or :all"
    end
    propSet << h
  end

  filterSpec = RbVmomi::VIM.PropertyFilterSpec(
    :objectSet => [
      :obj => self,
      :selectSet => [
        RbVmomi::VIM.TraversalSpec(
          :name => 'tsFolder',
          :type => 'Folder',
          :path => 'childEntity',
          :skip => false,
          :selectSet => [
            RbVmomi::VIM.SelectionSpec(:name => 'tsFolder')
          ]
        )
      ]
    ],
    :propSet => propSet
  )

  result = _connection.propertyCollector.RetrieveProperties(:specSet => [filterSpec])
  {}.tap do |h|
    result.each { |r| h[r.obj] = r }
  end
end

# File lib/rbvmomi/vim/Folder.rb, line 183
def inventory_tree propSpecs={}
  inv = inventory_flat propSpecs
  children = inv.values.group_by { |v| v['parent'] }
  rec = lambda { |parent| Hash[(children[parent]||[]).map { |x| [x, rec[x.obj]] }] }
  rec[self]
end

# File lib/rbvmomi/vim/Folder.rb, line 85
def traverse path, type=Object, create=false
  if path.is_a? String
    es = path.split('/').reject(&:empty?)
  elsif path.is_a? Enumerable
    es = path
  else
    fail "unexpected path class #{path.class}"
  end
  return self if es.empty?
  final = es.pop

  p = es.inject(self) do |f,e|
    f.find(e, RbVmomi::VIM::Folder) || (create && f.CreateFolder(:name => e)) || return
  end

  if x = p.find(final, type)
    x
  elsif create and type == RbVmomi::VIM::Folder
    p.CreateFolder(:name => final)
  elsif create and type == RbVmomi::VIM::Datacenter
    p.CreateDatacenter(:name => final)
  else
    nil
  end
end

# File lib/rbvmomi/vim/Folder.rb, line 75
def traverse! path, type=Object
  traverse path, type, true
end