package com.kodcu.rapid.config;

import com.kodcu.rapid.network.UnixFactory;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.config.RegistryBuilder;
import org.apache.http.conn.socket.ConnectionSocketFactory;
import org.apache.http.conn.socket.PlainConnectionSocketFactory;
import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
import org.apache.http.impl.conn.PoolingHttpClientConnectionManager;
import org.glassfish.jersey.apache.connector.ApacheClientProperties;
import org.glassfish.jersey.apache.connector.ApacheConnectorProvider;
import org.glassfish.jersey.client.ClientConfig;

import javax.ws.rs.client.Client;
import javax.ws.rs.client.ClientBuilder;
import javax.ws.rs.client.WebTarget;
import java.net.URI;

import static java.util.concurrent.TimeUnit.SECONDS;

/**
 * Created by Hakan on 2/11/2016.
 */
public abstract class DockerClient {

    private static final String DEFAULT_UNIX_ENDPOINT = "unix:///var/run/docker.sock";
    private Client client;
    private URI sanitizeUri;

    public DockerClient() {
        this.init();
    }

    private void init() {
        final URI originalUri = URI.create(DEFAULT_UNIX_ENDPOINT);
        sanitizeUri = UnixFactory.sanitizeUri(originalUri);

        final RegistryBuilder<ConnectionSocketFactory> registryBuilder =
                RegistryBuilder.<ConnectionSocketFactory>create()
                        .register("https", SSLConnectionSocketFactory.getSocketFactory())
                        .register("http", PlainConnectionSocketFactory.getSocketFactory())
                        .register("unix", new UnixFactory(originalUri));

        final PoolingHttpClientConnectionManager cm =
                new PoolingHttpClientConnectionManager(registryBuilder.build());

        final RequestConfig requestConfig = RequestConfig.custom()
                .setConnectionRequestTimeout((int) SECONDS.toMillis(5))
                .setConnectTimeout((int) SECONDS.toMillis(5))
                .setSocketTimeout((int) SECONDS.toMillis(30))
                .build();

        final ClientConfig config = new ClientConfig()
                .connectorProvider(new ApacheConnectorProvider())
                .property(ApacheClientProperties.CONNECTION_MANAGER, cm)
                .property(ApacheClientProperties.REQUEST_CONFIG, requestConfig);

        client = ClientBuilder.newBuilder().withConfig(config).build();
    }

    protected WebTarget resource() {
        return client.target(sanitizeUri);
    }


}