org.arquillian.cube.impl.client.CubeConfigurator Maven / Gradle / Ivy

Go to download
package org.arquillian.cube.impl.client;

import java.io.File;
import java.net.URI;
import java.util.Map;

import org.arquillian.cube.impl.util.Boot2Docker;
import org.arquillian.cube.impl.util.HomeResolverUtil;
import org.arquillian.cube.impl.util.OperatingSystemResolver;
import org.jboss.arquillian.config.descriptor.api.ArquillianDescriptor;
import org.jboss.arquillian.core.api.Instance;
import org.jboss.arquillian.core.api.InstanceProducer;
import org.jboss.arquillian.core.api.annotation.ApplicationScoped;
import org.jboss.arquillian.core.api.annotation.Inject;
import org.jboss.arquillian.core.api.annotation.Observes;

public class CubeConfigurator {

    private static final String EXTENSION_NAME = "docker";
    private static final String UNIX_SOCKET_SCHEME = "unix";

    @Inject
    @ApplicationScoped
    private InstanceProducer configurationProducer;

    @Inject
    private Instance boot2DockerInstance;

    public void configure(@Observes ArquillianDescriptor arquillianDescriptor) {
        Map config = arquillianDescriptor.extension(EXTENSION_NAME).getExtensionProperties();
        config = resolveServerUriByOperativeSystem(config);
        config = resolveServerIp(config);
        CubeConfiguration cubeConfiguration = CubeConfiguration.fromMap(config);
        configurationProducer.set(cubeConfiguration);
    }

    private Map resolveServerIp(Map config) {
        String dockerServerUri = config.get(CubeConfiguration.DOCKER_URI);
        if(dockerServerUri.contains(Boot2Docker.BOOT2DOCKER_TAG)) {
            dockerServerUri = resolveBoot2Docker(dockerServerUri, config.get(CubeConfiguration.BOOT2DOCKER_PATH));
            config.put(CubeConfiguration.DOCKER_URI, dockerServerUri);
            if(!config.containsKey(CubeConfiguration.CERT_PATH)) {
                config.put(CubeConfiguration.CERT_PATH, HomeResolverUtil.resolveHomeDirectoryChar(getDefaultTlsDirectory()));
            }
        }

        URI serverUri = URI.create(dockerServerUri);
        String serverIp = UNIX_SOCKET_SCHEME.equalsIgnoreCase(serverUri.getScheme()) ? "localhost" : serverUri.getHost();
        config.put(CubeConfiguration.DOCKER_SERVER_IP, serverIp);
        return config;
    }

    private String resolveBoot2Docker(String tag,
            String boot2DockerPath) {
        return tag.replaceAll(Boot2Docker.BOOT2DOCKER_TAG, boot2DockerInstance.get().ip(boot2DockerPath, false));
    }

    private String getDefaultTlsDirectory() {
        return "~" + File.separator + ".boot2docker" + File.separator + "certs" + File.separator + "boot2docker-vm";
    }

    private Map resolveServerUriByOperativeSystem(Map cubeConfiguration) {
        if(!cubeConfiguration.containsKey(CubeConfiguration.DOCKER_URI)) {
            String serverUri = new OperatingSystemResolver().currentOperatingSystem().getFamily().getServerUri();
            cubeConfiguration.put(CubeConfiguration.DOCKER_URI, serverUri);
        }
        return cubeConfiguration;
    }
}