/*
 * Copyright (c) 2015 EMC Corporation
 * All Rights Reserved
 */
package com.emc.storageos.volumecontroller.impl.smis.job;

import java.net.URI;

import javax.cim.CIMInstance;
import javax.cim.CIMObjectPath;
import javax.wbem.CloseableIterator;
import javax.wbem.WBEMException;
import javax.wbem.client.WBEMClient;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.emc.storageos.cimadapter.connections.cim.CimObjectPathCreator;
import com.emc.storageos.db.client.model.BlockMirror;
import com.emc.storageos.volumecontroller.TaskCompleter;
import com.emc.storageos.volumecontroller.impl.smis.CIMPropertyFactory;
import com.emc.storageos.volumecontroller.impl.smis.SmisConstants;

public abstract class SmisBlockMirrorJob extends SmisReplicaCreationJobs {

    private static final Logger _log = LoggerFactory.getLogger(SmisBlockMirrorJob.class);

    public SmisBlockMirrorJob(CIMObjectPath cimJob, URI storageSystem, TaskCompleter taskCompleter, String jobName) {
        super(cimJob, storageSystem, taskCompleter, jobName);
    }

    protected void updateSynchronizationAspects(WBEMClient client, BlockMirror mirror) throws WBEMException {
        CIMObjectPath path = CimObjectPathCreator.createInstance(mirror.getSynchronizedInstance());
        _log.info("Updating synchronization aspects of " + path.toString());
        CIMInstance syncInstance = getStorageSyncInstance(client, path);

        if (syncInstance == null) {
            return;
        }

        mirror.setSynchronizedInstance(syncInstance.getObjectPath().toString());
        mirror.setSyncState(CIMPropertyFactory.getPropertyValue(syncInstance, SmisConstants.CP_SYNC_STATE));
        mirror.setSyncType(CIMPropertyFactory.getPropertyValue(syncInstance, SmisConstants.CP_SYNC_TYPE));
    }

    protected CIMInstance getStorageSyncInstance(WBEMClient client, CIMObjectPath syncPath) throws WBEMException {
        return client.getInstance(syncPath, false, false, null);
    }

    protected CIMInstance getStorageSyncInstanceFromVolume(WBEMClient client, CIMObjectPath volumePath) throws WBEMException {
        CloseableIterator<CIMInstance> references = client.referenceInstances(volumePath,
                SmisConstants.CIM_STORAGE_SYNCHRONIZED, null, false, new String[] { SmisConstants.CP_SYNC_TYPE });
        if (!references.hasNext()) {
            _log.error(String.format("No storage synchronized instance was found for %s", volumePath.toString()));
            return null;
        }
        return references.next();
    }

    // Set synchronized instance to GroupSynchronized
    protected CIMInstance getSynchronizedInstance(WBEMClient client, CIMObjectPath repGroupPath) throws Exception {
        CloseableIterator<CIMInstance> syncIter = null;
        try {
            // get reference to the CIM_StorageSynchronized instance
            syncIter = client.referenceInstances(repGroupPath, SmisConstants.CIM_GROUP_SYNCHRONIZED, null, false,
                    new String[] { SmisConstants.CP_SYNC_TYPE });
            return syncIter.next();
        } catch (Exception e) {
            _log.error("Failed to find GroupSynchronized for group {}", repGroupPath.toString(), e);
            throw e;
        } finally {
            if (syncIter != null) {
                syncIter.close();
            }
        }
    }

}