/*
 * Copyright (c) 2012-2015 iWave Software LLC
 * All Rights Reserved
 */
package com.emc.sa.service.vipr.block.tasks;

import com.emc.sa.service.vipr.block.BlockStorageUtils;
import com.emc.sa.service.vipr.tasks.ViPRExecutionTask;
import com.emc.storageos.model.block.export.ExportGroupRestRep;

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

/**
 */
public class FindExportByCluster extends ViPRExecutionTask<ExportGroupRestRep> {
    private URI cluster;
    private URI project;
    private URI varray;
    private URI volume;

    public FindExportByCluster(String cluster, String project, String varray, String volume) {
        this(uri(cluster), uri(project), uri(varray), uri(volume));
    }

    public FindExportByCluster(URI cluster, URI project, URI varray, URI volume) {
        this.cluster = cluster;
        this.project = project;
        this.varray = varray;
        this.volume = volume;
        provideDetailArgs(cluster, project, varray);
    }

    @Override
    public ExportGroupRestRep executeTask() throws Exception {
        List<ExportGroupRestRep> exports = getClient().blockExports().findByCluster(cluster, project, varray);
        if (volume != null) {
            for (ExportGroupRestRep export : exports) {
                if (BlockStorageUtils.isVolumeInExportGroup(export, volume)) {
                    return export;
                }
            }
        }

        return exports.isEmpty() ? null : exports.get(0);
    }
}