fog--fog/lib/fog/vcloud/models/compute/servers.rb

require 'fog/vcloud/models/compute/server'

module Fog
  module Vcloud
    class Compute

      class Servers < Fog::Vcloud::Collection

        undef_method :create

        model Fog::Vcloud::Compute::Server

        attribute :href, :aliases => :Href

        def all
          if self.href =~ /\/vdc\//
            check_href!("Vdc")
            load(_vapps)
          else
            check_href!("Vapp")
            attributes[:vapp].load_unless_loaded!
            load(attributes[:vapp].children)
          end
        end

        def get(uri)
          if data = connection.get_vapp(uri)
            # If no tasks returned, set a mock entry to flush on reload
            data.body[:Tasks] = {} unless data.body[:Tasks]
            new(data.body)
          end
        rescue Fog::Errors::NotFound
          nil
        end

        def create options
          check_href!
          options[:vdc_uri] = href
          data = connection.instantiate_vapp_template(options).body
          object = new(data)
          object
        end

        private

        def _resource_entities
          if Hash === resource_entities = connection.get_vdc(href).body[:ResourceEntities]
            resource_entities[:ResourceEntity]
          end
        end

        def _vapps
          resource_entities = _resource_entities
          if resource_entities.nil?
            []
          else
            resource_entities.select {|re| re[:type] == 'application/vnd.vmware.vcloud.vApp+xml' }
          end
        end

      end
    end
  end
end