module Fog
  module Compute
    class Vsphere
      class Real
        def list_clusters(filters = { })
          datacenter_name = filters[:datacenter]

          raw_clusters(datacenter_name).map do |cluster|
            cluster_attributes(cluster, datacenter_name)
          end
        end

        def raw_clusters(datacenter)
          find_raw_datacenter(datacenter).hostFolder.childEntity.grep(RbVmomi::VIM::ClusterComputeResource)
        end

        protected

        def cluster_attributes cluster, datacenter_name
          {
            :id             => managed_obj_id(cluster),
            :name           => cluster.name,
            :num_host       => cluster.summary.numHosts,
            :num_cpu_cores  => cluster.summary.numCpuCores,
            :overall_status => cluster.summary.overallStatus,
            :datacenter     => datacenter_name || parent_attribute(cluster.path, :datacenter)[1],
          }
        end
      end
      class Mock
        def list_clusters(filters = { })
        end
      end
    end
  end
end

module Fog
  module Compute
    class Vsphere
      class Real
        def list_clusters(filters = { })
          datacenter_name = filters[:datacenter]

          raw_clusters(datacenter_name).map do |cluster|
            cluster_attributes(cluster, datacenter_name)
          end
        end

        def raw_clusters(datacenter)
          folder ||= find_raw_datacenter(datacenter).hostFolder
          @raw_clusters = get_raw_clusters_from_folder(folder)
        end

        protected

        def get_raw_clusters_from_folder(folder)
          folder.childEntity.map do |child_entity|
            if child_entity.is_a? RbVmomi::VIM::ClusterComputeResource
              child_entity
            elsif child_entity.is_a? RbVmomi::VIM::Folder
              get_raw_clusters_from_folder(child_entity)
            end
          end.flatten
        end

        def cluster_attributes cluster, datacenter_name
          {
            :id             => managed_obj_id(cluster),
            :name           => cluster.name,
            :num_host       => cluster.summary.numHosts,
            :num_cpu_cores  => cluster.summary.numCpuCores,
            :overall_status => cluster.summary.overallStatus,
            :datacenter     => datacenter_name || parent_attribute(cluster.path, :datacenter)[1],
          }
        end
      end
      class Mock
        def list_clusters(filters = { })
        end
      end
    end
  end
end

Support clusters that are located below folders
[vSphere] Support clusters that are located below folders
