org.testobject.appium.internal.RestClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of testobject-appium-java-api Show documentation
java wrapper for testobject's appium java apis
There is a newer version: 0.2.7
Show newest version
package org.testobject.appium.internal;

import com.google.common.base.Optional;
import com.sun.jersey.api.client.Client;
import com.sun.jersey.api.client.WebResource;
import com.sun.jersey.api.client.filter.HTTPBasicAuthFilter;
import com.sun.jersey.api.client.filter.LoggingFilter;
import com.sun.jersey.client.apache.ApacheHttpClient;
import com.sun.jersey.client.apache.config.ApacheHttpClientConfig;
import com.sun.jersey.client.apache.config.DefaultApacheHttpClientConfig;
import org.apache.commons.httpclient.UsernamePasswordCredentials;
import org.apache.commons.httpclient.auth.AuthScope;

import java.io.Closeable;
import java.net.URI;

public class RestClient implements Closeable {

    public static final String REST_APPIUM_PATH = "/rest/appium/v1/";
    public static final String REST_DEVICES_PATH = "/rest/devices/v1/";

    public static final class Builder {

        private Client client;
        private String baseUrl;
        private String path = "";
        private String token = "";

        public static Builder createClient() {
            return new Builder();
        }

        public Builder withUrl(String baseUrl) {
            this.baseUrl = baseUrl;
            return this;
        }

        public Builder withToken(String token) {
            this.token = token;
            return this;
        }

        public Builder path(String path) {
            this.path = path;
            return this;
        }

        public RestClient build() {

            ApacheHttpClientConfig config = new DefaultApacheHttpClientConfig();
            config.getProperties().put(ApacheHttpClientConfig.PROPERTY_HANDLE_COOKIES, true);

            addProxyConfiguration(config, baseUrl);

            client = ApacheHttpClient.create(config);
            client.addFilter(new LoggingFilter(System.out));
            client.addFilter(new HTTPBasicAuthFilter(token, ""));

            WebResource webResource = client.resource(baseUrl + path);
            return new RestClient(client, webResource);
        }

        private static void addProxyConfiguration(ApacheHttpClientConfig config, String baseUrl) {
            String protocol = URI.create(baseUrl).getScheme().toLowerCase();

            Optional proxyHost = Optional.fromNullable(System.getProperty(protocol + ".proxyHost"));
            if (!proxyHost.isPresent()) {
                return;
            }

            String host = proxyHost.get();
            String port = Optional.fromNullable(System.getProperty(protocol + ".proxyPort")).or("8080");
            String proxyProtocol = Optional.fromNullable(System.getProperty(protocol + ".proxyProtocol")).or("http");
            String url = proxyProtocol + "://" + host + ":" + port;
            config.getProperties().put(DefaultApacheHttpClientConfig.PROPERTY_PROXY_URI, url);

            Optional username = Optional.fromNullable(System.getProperty(protocol + ".proxyUser"));
            Optional password = Optional.fromNullable(System.getProperty(protocol + ".proxyPassword"));
            if (username.isPresent() && password.isPresent()) {
                UsernamePasswordCredentials credentials = new UsernamePasswordCredentials(username.get(), password.get());
                config.getState().getHttpState().setProxyCredentials(AuthScope.ANY, credentials);
            }
        }

    }

    private final Client client;
    private final WebResource baseResource;

    RestClient(Client client, WebResource baseResource){
        this.client = client;
        this.baseResource = baseResource;
    }

    public WebResource path(String path){
        return baseResource.path(path);
    }

    @Override
    public void close() {
        client.destroy();
    }

}