/*
 * Copyright (c) 2015 EMC Corporation
 * All Rights Reserved
 */
package com.emc.storageos.db.client.model.dao;

import java.net.URI;
import java.util.Iterator;
import java.util.List;

import com.emc.storageos.db.client.ModelClient;
import com.emc.storageos.db.client.constraint.NamedElementQueryResultList.NamedElement;
import com.emc.storageos.db.client.model.Cluster;

public class ClusterFinder extends TenantModelFinder<Cluster> {
    protected static final String DATACENTER_COLUMN_NAME = "vcenterDataCenter";
    protected static final String PROJECT_COLUMN_NAME = "project";

    public ClusterFinder(ModelClient client) {
        super(Cluster.class, client);
    }

    public Iterable<Cluster> findByDatacenter(URI datacenterId, boolean activeOnly) {
        List<NamedElement> clusters = findIdsByDatacenter(datacenterId);
        return findByIds(toURIs(clusters), activeOnly);
    }

    public List<NamedElement> findIdsByDatacenter(URI datacenterId) {
        return client.findBy(Cluster.class, DATACENTER_COLUMN_NAME, datacenterId);
    }

    public Iterable<Cluster> findByProject(URI projectId, boolean activeOnly) {
        List<NamedElement> clusters = findIdsByProject(projectId);
        return findByIds(toURIs(clusters), activeOnly);
    }

    public List<NamedElement> findIdsByProject(URI projectId) {
        return client.findBy(Cluster.class, PROJECT_COLUMN_NAME, projectId);
    }

    /**
     * Find the Cluster in a vCenterDataCenter by name.
     *
     * @param datacenterId vCenterDataCenterId.
     * @param name name of the Cluster.
     * @param activeOnly indicates whether active only or not.
     * @return the Cluster if the name matches, otherwise null.
     */
    public Cluster findClusterByNameAndDatacenter(URI datacenterId, String name, boolean activeOnly) {
        List<NamedElement> clusters = findIdsByDatacenter(datacenterId);
        Iterator<NamedElement> clusterIt = clusters.iterator();
        while(clusterIt.hasNext()) {
            Cluster cluster = findById(clusterIt.next().getId());
            if(cluster != null && cluster.getLabel().equalsIgnoreCase(name)) {
                return cluster;
            }
        }
        return null;
    }
}