ComponentRESTClient.java

/**
 * DataCleaner (community edition)
 * Copyright (C) 2014 Neopost - Customer Information Management
 *
 * This copyrighted material is made available to anyone wishing to use, modify,
 * copy, or redistribute it subject to the terms and conditions of the GNU
 * Lesser General Public License, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
 * for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this distribution; if not, write to:
 * Free Software Foundation, Inc.
 * 51 Franklin Street, Fifth Floor
 * Boston, MA  02110-1301  USA
 */
package org.datacleaner.restclient;

import java.io.IOException;

import org.datacleaner.api.RestrictedFunctionalityException;

import com.fasterxml.jackson.databind.JsonNode;

/**
 * @since 02. 09. 2015
 */
public class ComponentRESTClient {

    private final RESTClient restClient;
    private final String url;
    private String tenantName;

    public ComponentRESTClient(final String url, final String username, final String password,
            final String dataCleanerVersion) {
        this.url = url;
        restClient = new RESTClientImpl(username, password, dataCleanerVersion);
        getUserTenantName();
    }

    /**
     * Mainly for tests, tenant can be recognized automatically when other
     * constructor is used
     */
    public ComponentRESTClient(final String url, final String username, final String password, final String tenantName,
            final String dataCleanerVersion) {
        this.url = url;
        restClient = new RESTClientImpl(username, password, dataCleanerVersion);
        this.tenantName = tenantName;
    }

    public ComponentList getAllComponents(final boolean iconData) {
        final String response = call(RESTClient.HttpMethod.GET, getURL("?iconData=" + iconData), "");

        return Serializator.componentList(response);
    }

    public ComponentList.ComponentInfo getComponentInfo(String componentName, final boolean iconData) {
        componentName = urlify(componentName);
        final String response = call(RESTClient.HttpMethod.GET, getURL(componentName + "&iconData=" + iconData), "");

        return Serializator.componentInfo(response);
    }

    public OutputColumns getOutputColumns(String componentName, final CreateInput config) {
        componentName = urlify(componentName);
        final String configuration = Serializator.stringCreateInput(config);
        final String response =
                call(RESTClient.HttpMethod.POST, getURL(componentName + "/_outputColumns"), configuration);

        return Serializator.outputColumnsOutput(response);
    }

    public ProcessStatelessOutput processStateless(String componentName,
            final ProcessStatelessInput processStatelessInput) {
        componentName = urlify(componentName);
        final String configurationAndData = Serializator.stringProcessStatelessInput(processStatelessInput);
        final String response = call(RESTClient.HttpMethod.PUT, getURL(componentName), configurationAndData);

        return Serializator.processStatelessOutput(response);
    }

    public String createComponent(String componentName, final String timeout, final CreateInput config) {
        componentName = urlify(componentName);
        final String configuration = Serializator.stringCreateInput(config);

        return call(RESTClient.HttpMethod.POST, getURL(componentName + "?timeout=" + timeout), configuration);
    }

    public ProcessOutput processComponent(final String instanceId, final ProcessInput processInput) {
        final String inputData = Serializator.stringProcessInput(processInput);
        final String response = call(RESTClient.HttpMethod.PUT, getURL("/_instance/" + instanceId), inputData);

        return Serializator.processOutput(response);
    }

    public ProcessResult getFinalResult(final String instanceId) {
        final String response = call(RESTClient.HttpMethod.GET, getURL(instanceId + "/result"), "");

        return Serializator.processResult(response);
    }

    public void deleteComponent(final String instanceId) {
        call(RESTClient.HttpMethod.DELETE, getURL(instanceId), "");
    }

    public String getUserTenantName() {
        final String url = String.format("%s%s", this.url, "/repository/_user");
        final String response = call(RESTClient.HttpMethod.GET, url, "");
        JsonNode userInfo = null;
        try {
            userInfo = Serializator.getJacksonObjectMapper().readTree(response);
            final JsonNode tenantN = userInfo.get("tenant");
            if (tenantN == null) {
                return tenantName = "unknown";
            } else {
                return tenantName = tenantN.asText();
            }
        } catch (final IOException e) {
            throw new RuntimeException(e);
        }
    }

    public DataCloudUser getDataCloudUserInfo() {
        final String userTenantName = getUserTenantName();
        final String response =
                call(RESTClient.HttpMethod.GET, url + "/repository/" + userTenantName + "/userinfo", null);
        return Serializator.processDataCloudUser(response);
    }

    private String urlify(final String string) {
        return ComponentsRestClientUtils.encodeUrlPathSegment(ComponentsRestClientUtils.escapeComponentName(string));
    }

    private String getURL(String suffix) {
        if (suffix != null && !suffix.isEmpty()) {
            suffix = "/" + suffix;
        }

        return String.format("%s/repository/%s/components%s", url, urlify(tenantName), suffix);
    }

    private String call(final RESTClient.HttpMethod httpMethod, final String url, final String requestBody)
            throws RestrictedFunctionalityException {
        return restClient.getResponse(httpMethod, url, requestBody);
    }
}