UserPortalExistingVmModelBehavior.java

package org.ovirt.engine.ui.uicommonweb.models.vms;

import java.util.Arrays;
import java.util.Collections;
import java.util.List;

import org.ovirt.engine.core.common.businessentities.ActionGroup;
import org.ovirt.engine.core.common.businessentities.Cluster;
import org.ovirt.engine.core.common.businessentities.StoragePool;
import org.ovirt.engine.core.common.businessentities.VDS;
import org.ovirt.engine.core.common.businessentities.VM;
import org.ovirt.engine.core.common.businessentities.comparators.NameableComparator;
import org.ovirt.engine.core.compat.Guid;
import org.ovirt.engine.ui.frontend.AsyncQuery;
import org.ovirt.engine.ui.uicommonweb.dataprovider.AsyncDataProvider;
import org.ovirt.engine.ui.uicommonweb.models.SystemTreeItemModel;

public class UserPortalExistingVmModelBehavior extends ExistingVmModelBehavior {
    public UserPortalExistingVmModelBehavior(VM vm) {
        super(vm);
    }

    @Override
    public void initialize(SystemTreeItemModel systemTreeSelectedItem) {
        super.initialize(systemTreeSelectedItem);

        // The custom properties tab should be hidden on the User Portal
        getModel().setIsCustomPropertiesTabAvailable(false);

        // Affinity labels are only available in WebAdmin at this time
        getModel().getLabelList().setIsAvailable(false);
    }

    @Override
    protected void initClusters(final List<StoragePool> dataCenters) {
        // Get clusters with permitted edit action
        ActionGroup actionGroup = getModel().isCreateInstanceOnly() ? ActionGroup.CREATE_INSTANCE : ActionGroup.CREATE_VM;
        AsyncDataProvider.getInstance().getClustersWithPermittedAction(new AsyncQuery<>(
                clusters -> {
                    // filter clusters by architecture
                    clusters = AsyncDataProvider.getInstance().filterByArchitecture(clusters, vm.getClusterArch());

                    if (containsVmCluster(clusters)) {
                        Collections.sort(clusters, new NameableComparator());
                        getModel().setDataCentersAndClusters(getModel(), dataCenters, clusters, vm.getClusterId());
                    } else {
                        // Add VM's cluster if not contained in the cluster list
                        addVmCluster(dataCenters, clusters);
                    }

                    initTemplate();
                    initCdImage();

                }), actionGroup, true, false);

    }


    private boolean containsVmCluster(List<Cluster> clusters) {

        for (Cluster cluster : clusters) {
            if (cluster.getStoragePoolId() != null) {
                if (vm.getClusterId().equals(cluster.getId())) {
                    return true;
                }
            }
        }

        return false;
    }

    @Override
    protected void updateCdImage() {
        updateUserCdImage(getVm().getStoragePoolId());
    }

    private void addVmCluster(final List<StoragePool> dataCenters, final List<Cluster> clusters) {
        AsyncDataProvider.getInstance().getClusterById(new AsyncQuery<>(
                cluster -> {
                    if (cluster != null) {
                        clusters.add(cluster);
                    }
                    Collections.sort(clusters, new NameableComparator());
                    getModel().setDataCentersAndClusters(getModel(), dataCenters, clusters, vm.getClusterId());
                }), vm.getClusterId());
    }

    /**
     * Fills the default host according to the selected host set in webadmin. Since this value can be set only in
     * webadmin and can be set also to host, which is not visible to the user in userportal, this fakes the VDS value in
     * a way, that the rest of the code can use it normally and send it back to the server as-is (like Null Object
     * Pattern).
     */
    @Override
    protected void doChangeDefaultHost(List<Guid> hostGuids) {
        if (hostGuids != null && hostGuids.size() == 1) {
            VDS vds = new VDS();
            vds.setId(hostGuids.get(0));
            getModel().getDefaultHost().setItems(Arrays.asList(vds));
        }

        super.doChangeDefaultHost(hostGuids);
    }

    @Override
    protected void getHostListByCluster(Cluster cluster, AsyncQuery<List<VDS>> query) {
        AsyncDataProvider.getInstance().getHostListByClusterId(query, cluster.getId());
    }

    @Override
    protected void updateNumaEnabled() {
    }
}