ContextVmware.java

/*******************************************************************************
 * Copyright 2012 Urbancode, Inc
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 ******************************************************************************/
package com.urbancode.terraform.tasks.vmware;

import java.io.IOException;
import java.net.MalformedURLException;
import java.rmi.RemoteException;

import org.apache.log4j.Logger;

import com.urbancode.terraform.credentials.common.Credentials;
import com.urbancode.terraform.credentials.vmware.CredentialsVmware;
import com.urbancode.terraform.tasks.common.EnvironmentTask;
import com.urbancode.terraform.tasks.common.TerraformContext;
import com.urbancode.terraform.tasks.common.exceptions.EnvironmentCreationException;
import com.urbancode.terraform.tasks.common.exceptions.EnvironmentRestorationException;
import com.urbancode.terraform.tasks.vmware.util.GlobalIpAddressPool;
import com.urbancode.terraform.tasks.vmware.util.VirtualHost;
import com.urbancode.x2o.util.PropertyResolver;


public class ContextVmware extends TerraformContext {

    //**********************************************************************************************
    // CLASS
    //**********************************************************************************************
    static private final Logger log = Logger.getLogger(ContextVmware.class);

    //**********************************************************************************************
    // INSTANCE
    //**********************************************************************************************
    private EnvironmentTaskVmware env;
    private CredentialsVmware credentials;

    //----------------------------------------------------------------------------------------------
    public ContextVmware() {

    }

    //----------------------------------------------------------------------------------------------
    public PropertyResolver fetchResolver() {
        return resolver;
    }

    //----------------------------------------------------------------------------------------------
    @Override
    public void setCredentials(Credentials credentials) {
        this.credentials = (CredentialsVmware) credentials;
    }

    //----------------------------------------------------------------------------------------------
    @Override
    public EnvironmentTask getEnvironment() {
        return env;
    }

    //----------------------------------------------------------------------------------------------
    @Override
    public Credentials fetchCredentials() {
        return credentials;
    }

    //----------------------------------------------------------------------------------------------
    public EnvironmentTaskVmware createEnvironment()
    throws RemoteException, MalformedURLException, IOException {
        env = new EnvironmentTaskVmware(this);
        return env;
    }

    //----------------------------------------------------------------------------------------------
    /**
     * A virtual host holds the ServiceInstance object and executes commands at the datacenter level.
     * @return
     * @throws IOException
     */
    private VirtualHost createVirtualHost()
    throws IOException {
        VirtualHost result = null;

        if (credentials != null) {
            String url = credentials.getUrl();
            String user = credentials.getUser();
            String password = credentials.getPassword();
            result = new VirtualHost(url, user, password);
        }
        else {
            log.error("Vmware Credentials null!");
        }

        return result;
    }

    //----------------------------------------------------------------------------------------------
    private void setVcenterPaths() throws EnvironmentCreationException {
        String datacenter = resolver.resolve("${datacenter}");
        String hostName = resolver.resolve("${host.name}");
        String destination = resolver.resolve("${destination}");
        String datastore = resolver.resolve("${datastore}");

        if("null".equals(datacenter)) {
            throw new EnvironmentCreationException("no datacenter property specified; canceling request");
        }
        if("null".equals(hostName)) {
            throw new EnvironmentCreationException("no host.name property specified; canceling request");
        }
        if("null".equals(destination)) {
            throw new EnvironmentCreationException("no destination property specified; canceling request");
        }
        if("null".equals(datastore)) {
            throw new EnvironmentCreationException("no datastore property specified; canceling request");
        }
        env.setAllPaths(datacenter, hostName, destination, datastore);
    }

    //----------------------------------------------------------------------------------------------
    @Override
    public void create() {

        VirtualHost host = null;
        try {
            GlobalIpAddressPool.getInstance().setPropertyResolver(resolver);
            GlobalIpAddressPool.getInstance().createIpPoolFromUserProps();
            setVcenterPaths();
            host = createVirtualHost();

            env.setVirtualHost(host);
            env.setStartTime(System.currentTimeMillis());
            if (host == null) {
                throw new NullPointerException("Host is null!");
            }
            else {
                env.create();
            }
        }
        catch (IOException e) {
            log.fatal("IOException when trying to run create() on context!", e);
        }
        catch(EnvironmentCreationException e) {
            log.error("a required property was not specified", e);
        }

    }

    //----------------------------------------------------------------------------------------------
    @Override
    public void destroy() {
        VirtualHost host = null;
        try {
            GlobalIpAddressPool.getInstance().setPropertyResolver(resolver);
            GlobalIpAddressPool.getInstance().createIpPoolFromUserProps();
            host = createVirtualHost();
        }
        catch (IOException e) {
            log.fatal("IOException when trying to run destroy() on context!", e);
        }
        env.setVirtualHost(host);
        env.destroy();
    }

    //----------------------------------------------------------------------------------------------
    @Override
    public void restore() throws EnvironmentRestorationException {
        VirtualHost host = null;
        try {
            GlobalIpAddressPool.getInstance().setPropertyResolver(resolver);
            GlobalIpAddressPool.getInstance().createIpPoolFromUserProps();
            host = createVirtualHost();
        }
        catch (IOException e) {
            log.fatal("IOException when trying to run destroy() on context!", e);
        }
        env.setVirtualHost(host);
        env.restore();
    }
}