CloudSlang/cs-actions

View on GitHub
cs-nutanix-prism/src/main/java/io/cloudslang/content/nutanix/prism/utils/HttpUtils.java

Summary

Maintainability
C
1 day
Test Coverage


package io.cloudslang.content.nutanix.prism.utils;

import com.jayway.jsonpath.JsonPath;
import io.cloudslang.content.httpclient.entities.HttpClientInputs;
import io.cloudslang.content.nutanix.prism.entities.NutanixCommonInputs;
import io.cloudslang.content.utils.StringUtilities;
import org.apache.http.client.utils.URIBuilder;
import org.jetbrains.annotations.NotNull;

import java.net.Authenticator;
import java.net.InetSocketAddress;
import java.net.PasswordAuthentication;
import java.net.Proxy;
import java.util.HashMap;
import java.util.Map;

import static io.cloudslang.content.constants.OutputNames.*;
import static io.cloudslang.content.nutanix.prism.utils.Constants.Common.*;
import static io.cloudslang.content.nutanix.prism.utils.Constants.DeleteNICConstants.VM_LOGICAL_TIMESTAMP_QUERY_PARAM;
import static io.cloudslang.content.nutanix.prism.utils.Constants.GetTaskDetailsConstants.FAILED;
import static io.cloudslang.content.nutanix.prism.utils.Constants.GetTaskDetailsConstants.TASK_FAILURE_PATH;
import static io.cloudslang.content.nutanix.prism.utils.Outputs.CommonOutputs.DOCUMENT;
import static io.cloudslang.content.nutanix.prism.utils.Outputs.GetTaskDetailsOutputs.TASK_STATUS;
import static io.cloudslang.content.utils.OutputUtilities.getFailureResultsMap;
import static io.cloudslang.content.utils.OutputUtilities.getSuccessResultsMap;
import static java.net.Proxy.Type.HTTP;
import static org.apache.commons.lang3.StringUtils.isEmpty;

public class HttpUtils {


    @NotNull
    public static Proxy getProxy(@NotNull final String proxyHost, final int proxyPort, @NotNull final String proxyUser, @NotNull final String proxyPassword) {
        if (StringUtilities.isBlank(proxyHost)) {
            return Proxy.NO_PROXY;
        }
        if (StringUtilities.isNotEmpty(proxyUser)) {
            Authenticator.setDefault(new Authenticator() {
                @Override
                public PasswordAuthentication getPasswordAuthentication() {
                    return new PasswordAuthentication(proxyUser, proxyPassword.toCharArray());
                }
            });
        }
        return new Proxy(HTTP, InetSocketAddress.createUnresolved(proxyHost, proxyPort));
    }

    public static void setProxy(@NotNull final HttpClientInputs httpClientInputs,
                                @NotNull final String proxyHost,
                                @NotNull final String proxyPort,
                                @NotNull final String proxyUsername,
                                @NotNull final String proxyPassword) {
        httpClientInputs.setProxyHost(proxyHost);
        httpClientInputs.setProxyPort(proxyPort);
        httpClientInputs.setProxyUsername(proxyUsername);
        httpClientInputs.setProxyPassword(proxyPassword);
    }

    @NotNull
    public static URIBuilder getUriBuilder(String url) {
        final URIBuilder uriBuilder = new URIBuilder();
        uriBuilder.setHost(url);
        uriBuilder.setScheme(HTTPS);
        return uriBuilder;
    }

    @NotNull
    public static String getAuthHeaders(@NotNull final String username, @NotNull final String password) {
        final StringBuilder headerBuilder = new StringBuilder();
        headerBuilder.append(AUTHORIZATION).append(username).append(":").append(password);
        return headerBuilder.toString();
    }

    @NotNull
    public static Map<String, String> getOperationResults(@NotNull final Map<String, String> result,
                                                          @NotNull final String successMessage,
                                                          final String failureMessage,
                                                          final String document) {
        final Map<String, String> results;
        final String statusCode = result.get(STATUS_CODE);
        if (Integer.parseInt(statusCode) >= 200 && Integer.parseInt(statusCode) < 300) {
            results = getSuccessResultsMap(successMessage);
            if (!isEmpty(document))
                results.put(DOCUMENT, document);
        } else {
            results = getFailureResultsMap(failureMessage);
        }
        results.put(STATUS_CODE, statusCode);
        return results;
    }

    public static void setSecurityInputs(@NotNull final HttpClientInputs httpClientInputs,
                                         @NotNull final String trustAllRoots,
                                         @NotNull final String x509HostnameVerifier,
                                         @NotNull final String trustKeystore,
                                         @NotNull final String trustPassword,
                                         @NotNull final String keystore,
                                         @NotNull final String keystorePassword) {
        httpClientInputs.setTrustAllRoots(trustAllRoots);
        httpClientInputs.setX509HostnameVerifier(x509HostnameVerifier);
        httpClientInputs.setTrustKeystore(trustKeystore);
        httpClientInputs.setTrustPassword(trustPassword);
        httpClientInputs.setKeystore(keystore);
        httpClientInputs.setKeystorePassword(keystorePassword);
    }

    public static void setConnectionParameters(HttpClientInputs httpClientInputs,
                                               @NotNull final String connectTimeout,
                                               @NotNull final String socketTimeout,
                                               @NotNull final String keepAlive,
                                               @NotNull final String connectionsMaxPerRoot,
                                               @NotNull final String connectionsMaxTotal,
                                               @NotNull final String preemptiveAuth) {
        httpClientInputs.setConnectTimeout(connectTimeout);
        httpClientInputs.setSocketTimeout(socketTimeout);
        httpClientInputs.setKeepAlive(keepAlive);
        httpClientInputs.setConnectionsMaxPerRoute(connectionsMaxPerRoot);
        httpClientInputs.setConnectionsMaxTotal(connectionsMaxTotal);
        httpClientInputs.setPreemptiveAuth(String.valueOf(true));
    }

    public static void setTLSParameters(HttpClientInputs httpClientInputs) {
        httpClientInputs.setTlsVersion(io.cloudslang.content.httpclient.entities.Constants.TLSv12);
        httpClientInputs.setAllowedCyphers(ALLOWED_CYPHERS);
    }

    @NotNull
    public static URIBuilder getUriBuilder(NutanixCommonInputs nutanixCommonInputs) {
        final URIBuilder uriBuilder = new URIBuilder();
        uriBuilder.setHost(nutanixCommonInputs.getHostname());
        uriBuilder.setPort(Integer.parseInt(nutanixCommonInputs.getPort()));
        uriBuilder.setScheme(HTTPS);
        return uriBuilder;
    }

    @NotNull
    public static String getQueryParams(String includeVMDiskConfigInfo,
                                        final String includeVMNicConfigInfo) {
        final StringBuilder queryParams = new StringBuilder()
                .append(Constants.GetVMDetailsConstants.INCLUDE_VM_DISK_CONFIG_INFO)
                .append(includeVMDiskConfigInfo)
                .append(AND)
                .append(Constants.GetVMDetailsConstants.INCLUDE_VM_NIC_CONFIG_INFO)
                .append(includeVMNicConfigInfo);
        return queryParams.toString();
    }

    @NotNull
    public static String getQueryParams(String vmLogicalTimestamp) {
        final StringBuilder queryParams = new StringBuilder()
                .append(QUERY)
                .append(VM_LOGICAL_TIMESTAMP_QUERY_PARAM)
                .append(vmLogicalTimestamp);
        return queryParams.toString();
    }


    @NotNull
    public static String getTaskDetailsQueryParams(@NotNull String includeSubtasksInfo) {

        final StringBuilder queryParams = new StringBuilder()
                .append(Constants.GetTaskDetailsConstants.INCLUDE_SUBTASKS_INFO)
                .append(includeSubtasksInfo);
        return queryParams.toString();
    }

    @NotNull
    public static String getDeleteVMQueryParams(@NotNull String deleteSnapshots,
                                                @NotNull String logicalTimestamp) {
        final StringBuilder queryParams = new StringBuilder()
                .append(Constants.DeleteVMConstants.DELETE_SNAPSHOTS)
                .append(deleteSnapshots)
                .append(Constants.Common.AND)
                .append(Constants.DeleteVMConstants.LOGICAL_TIMESTAMP)
                .append(logicalTimestamp);
        return queryParams.toString();
    }

    @NotNull
    public static String getQueryParams(String filter, String offset, String length,
                                        String sortOrder, String sortAttribute, String includeVMDiskConfigInfo, final String includeVMNicConfigInfo) {
        final StringBuilder queryParams = new StringBuilder()
                .append(Constants.Common.QUERY)
                .append(Constants.ListVMsConstants.FILTER)
                .append(filter)
                .append(Constants.Common.AND)
                .append(Constants.ListVMsConstants.OFFSET)
                .append(offset)
                .append(Constants.Common.AND)
                .append(Constants.ListVMsConstants.LENGTH)
                .append(length)
                .append(Constants.Common.AND)
                .append(Constants.ListVMsConstants.SORT_ORDER)
                .append(sortOrder)
                .append(Constants.Common.AND)
                .append(Constants.ListVMsConstants.SORT_ATTRIBUTE)
                .append(sortAttribute)
                .append(Constants.Common.AND)
                .append(Constants.GetVMDetailsConstants.INCLUDE_VM_DISK_CONFIG_INFO)
                .append(includeVMDiskConfigInfo)
                .append(Constants.Common.AND)
                .append(Constants.GetVMDetailsConstants.INCLUDE_VM_NIC_CONFIG_INFO)
                .append(includeVMNicConfigInfo);
        if (length.equals(ZERO)) {
            queryParams.delete(17, 26);
        }
        return queryParams.toString();

    }


    @NotNull
    public static Map<String, String> getTaskFailureResults(@NotNull String inputName, @NotNull Integer statusCode,
                                                            @NotNull String taskStatus, @NotNull String returnMessage,
                                                            @NotNull String throwable) {
        Map<String, String> results = new HashMap();
        results.put(RETURN_CODE, "-1");
        results.put(STATUS_CODE, statusCode.toString());
        if (statusCode.equals(401)) {
            results.put(RETURN_RESULT, inputName + " not found, or user unauthorized to perform action");
            results.put(EXCEPTION, "status : " + statusCode + ", Title :  " + inputName + " not found, or user unauthorized to perform action");
        } else if (statusCode.equals(200) || statusCode.equals(201) && taskStatus.equals(FAILED)) {
            final String errorDetail = JsonPath.read(returnMessage, TASK_FAILURE_PATH);
            results.put(RETURN_RESULT, errorDetail);
            results.put(TASK_STATUS, taskStatus);
            results.put(EXCEPTION, " status : " + statusCode + ", Title :  " + errorDetail);
        } else if (statusCode.equals(500)) {
            final String errorDetail = JsonPath.read(returnMessage, ERROR_MESSAGE_PATH);
            results.put(RETURN_RESULT, "  error Message : " + errorDetail);
            results.put(EXCEPTION, " statusCode : " + statusCode + ", Title : message " + errorDetail);
        } else {
            results.put(RETURN_RESULT, throwable);
            results.put(EXCEPTION, throwable);
        }
        return results;
    }

    @NotNull
    public static Map<String, String> getFailureResults(@NotNull String inputName, @NotNull Integer statusCode,
                                                        @NotNull String returnMessage,
                                                        @NotNull String throwable) {
        Map<String, String> results = new HashMap();
        results.put(RETURN_CODE, "-1");
        results.put(STATUS_CODE, statusCode.toString());
        if (statusCode.equals(401)) {
            results.put(RETURN_RESULT, inputName + " not found, or user unauthorized to perform action");
            results.put(EXCEPTION, "status : " + statusCode + ", Title :  " + inputName + " not found, or user unauthorized to perform action");
        } else if (statusCode.equals(500)) {
            final String errorDetail = JsonPath.read(returnMessage, ERROR_MESSAGE_PATH);
            results.put(RETURN_RESULT, "  error Message : " + errorDetail);
            results.put(EXCEPTION, " statusCode : " + statusCode + ", Title : message " + errorDetail);
        } else {
            results.put(RETURN_RESULT, throwable);
            results.put(EXCEPTION, throwable);
        }
        return results;
    }
}