PlatformVirtualMachineTypesToJsonConverter.java

package com.sequenceiq.cloudbreak.converter;

import java.util.Collection;
import java.util.Map;

import org.springframework.stereotype.Component;

import com.google.api.client.util.Lists;
import com.google.api.client.util.Maps;
import com.sequenceiq.cloudbreak.api.model.PlatformVirtualMachinesJson;
import com.sequenceiq.cloudbreak.api.model.VmTypeJson;
import com.sequenceiq.cloudbreak.api.model.VmTypeMetaJson;
import com.sequenceiq.cloudbreak.api.model.VolumeParameterConfigJson;
import com.sequenceiq.cloudbreak.cloud.model.AvailabilityZone;
import com.sequenceiq.cloudbreak.cloud.model.Platform;
import com.sequenceiq.cloudbreak.cloud.model.PlatformVirtualMachines;
import com.sequenceiq.cloudbreak.cloud.model.VmType;
import com.sequenceiq.cloudbreak.cloud.model.VolumeParameterConfig;
import com.sequenceiq.cloudbreak.cloud.model.generic.StringType;
import com.sequenceiq.cloudbreak.converter.util.PlatformConverterUtil;

@Component
public class PlatformVirtualMachineTypesToJsonConverter extends AbstractConversionServiceAwareConverter<PlatformVirtualMachines,
        PlatformVirtualMachinesJson> {

    @Override
    public PlatformVirtualMachinesJson convert(PlatformVirtualMachines source) {
        PlatformVirtualMachinesJson json = new PlatformVirtualMachinesJson();
        json.setDefaultVirtualMachines(PlatformConverterUtil.convertDefaults(source.getDefaultVirtualMachines()));
        json.setVirtualMachines(convertVmMap(source.getVirtualMachines()));
        json.setVmTypesPerZones(convertVmsPerZoneMap(source.getVmTypesPerZones()));
        json.setDefaultVmTypePerZones(convertDefaultVmsPerZoneMap(source.getDefaultVmTypePerZones()));
        return json;
    }

    public Map<String, Map<String, String>> convertDefaultVmsPerZoneMap(Map<Platform, Map<AvailabilityZone, VmType>> defaultVmsPerZone) {
        Map<String, Map<String, String>> result = Maps.newHashMap();
        for (Map.Entry<Platform, Map<AvailabilityZone, VmType>> entry : defaultVmsPerZone.entrySet()) {
            Map<String, String> zoneResult = Maps.newHashMap();
            result.put(entry.getKey().value(), zoneResult);
            for (Map.Entry<AvailabilityZone, VmType> zoneEntry : entry.getValue().entrySet()) {
                zoneResult.put(zoneEntry.getKey().value(), zoneEntry.getValue().value());
            }
        }
        return result;
    }

    public Map<String, Map<String, Collection<VmTypeJson>>> convertVmsPerZoneMap(Map<Platform, Map<AvailabilityZone, Collection<VmType>>> zoneVms) {
        Map<String, Map<String, Collection<VmTypeJson>>> result = Maps.newHashMap();
        for (Map.Entry<Platform, Map<AvailabilityZone, Collection<VmType>>> entry : zoneVms.entrySet()) {
            result.put(entry.getKey().value(), convertVmMap(entry.getValue()));
        }
        return result;
    }

    public Map<String, Collection<VmTypeJson>> convertVmMap(Map<? extends StringType, Collection<VmType>> vms) {
        Map<String, Collection<VmTypeJson>> result = Maps.newHashMap();
        for (Map.Entry<? extends StringType, Collection<VmType>> entry : vms.entrySet()) {
            result.put(entry.getKey().value(), convertVmList(entry.getValue()));
        }
        return result;
    }

    public Collection<VmTypeJson> convertVmList(Collection<VmType> vmlist) {
        Collection<VmTypeJson> result = Lists.newArrayList();
        for (VmType item : vmlist) {
            result.add(convertVmType(item));
        }
        return result;
    }

    private VmTypeJson convertVmType(VmType item) {
        VmTypeMetaJson vmTypeMetaJson = new VmTypeMetaJson();
        vmTypeMetaJson.setProperties(item.getMetaData().getProperties());
        VolumeParameterConfig autoAttachedConfig = item.getMetaData().getAutoAttachedConfig();
        if (autoAttachedConfig != null) {
            VolumeParameterConfigJson volumeParameterConfigJson = new VolumeParameterConfigJson();
            volumeParameterConfigJson.setVolumeParameterType(autoAttachedConfig.volumeParameterType().name());
            volumeParameterConfigJson.setMaximumNumber(autoAttachedConfig.maximumNumber());
            volumeParameterConfigJson.setMinimumNumber(autoAttachedConfig.minimumNumber());
            volumeParameterConfigJson.setMaximumSize(autoAttachedConfig.maximumSize());
            volumeParameterConfigJson.setMinimumSize(autoAttachedConfig.minimumSize());
            vmTypeMetaJson.getConfigs().add(volumeParameterConfigJson);
        }
        VolumeParameterConfig ephemeralConfig = item.getMetaData().getEphemeralConfig();
        if (ephemeralConfig != null) {
            VolumeParameterConfigJson volumeParameterConfigJson = new VolumeParameterConfigJson();
            volumeParameterConfigJson.setVolumeParameterType(ephemeralConfig.volumeParameterType().name());
            volumeParameterConfigJson.setMaximumNumber(ephemeralConfig.maximumNumber());
            volumeParameterConfigJson.setMinimumNumber(ephemeralConfig.minimumNumber());
            volumeParameterConfigJson.setMaximumSize(ephemeralConfig.maximumSize());
            volumeParameterConfigJson.setMinimumSize(ephemeralConfig.minimumSize());
            vmTypeMetaJson.getConfigs().add(volumeParameterConfigJson);
        }
        VolumeParameterConfig magneticConfig = item.getMetaData().getMagneticConfig();
        if (magneticConfig != null) {
            VolumeParameterConfigJson volumeParameterConfigJson = new VolumeParameterConfigJson();
            volumeParameterConfigJson.setVolumeParameterType(magneticConfig.volumeParameterType().name());
            volumeParameterConfigJson.setMaximumNumber(magneticConfig.maximumNumber());
            volumeParameterConfigJson.setMinimumNumber(magneticConfig.minimumNumber());
            volumeParameterConfigJson.setMaximumSize(magneticConfig.maximumSize());
            volumeParameterConfigJson.setMinimumSize(magneticConfig.minimumSize());
            vmTypeMetaJson.getConfigs().add(volumeParameterConfigJson);
        }
        VolumeParameterConfig ssdConfig = item.getMetaData().getSsdConfig();
        if (ssdConfig != null) {
            VolumeParameterConfigJson volumeParameterConfigJson = new VolumeParameterConfigJson();
            volumeParameterConfigJson.setVolumeParameterType(ssdConfig.volumeParameterType().name());
            volumeParameterConfigJson.setMaximumNumber(ssdConfig.maximumNumber());
            volumeParameterConfigJson.setMinimumNumber(ssdConfig.minimumNumber());
            volumeParameterConfigJson.setMaximumSize(ssdConfig.maximumSize());
            volumeParameterConfigJson.setMinimumSize(ssdConfig.minimumSize());
            vmTypeMetaJson.getConfigs().add(volumeParameterConfigJson);
        }
        VolumeParameterConfig st1Config = item.getMetaData().getSt1Config();
        if (st1Config != null) {
            VolumeParameterConfigJson volumeParameterConfigJson = new VolumeParameterConfigJson();
            volumeParameterConfigJson.setVolumeParameterType(st1Config.volumeParameterType().name());
            volumeParameterConfigJson.setMaximumNumber(st1Config.maximumNumber());
            volumeParameterConfigJson.setMinimumNumber(st1Config.minimumNumber());
            volumeParameterConfigJson.setMaximumSize(st1Config.maximumSize());
            volumeParameterConfigJson.setMinimumSize(st1Config.minimumSize());
            vmTypeMetaJson.getConfigs().add(volumeParameterConfigJson);
        }
        return new VmTypeJson(item.value(), vmTypeMetaJson);
    }
}