com.tascape.reactor.ws.comm.RestCommunication Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2015 - 2016 Nebula Bay.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.tascape.reactor.ws.comm;

import com.jayway.restassured.RestAssured;
import com.jayway.restassured.response.Response;
import com.jayway.restassured.specification.RequestSpecification;
import com.tascape.reactor.comm.EntityCommunication;
import java.io.IOException;
import javax.naming.OperationNotSupportedException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * A wrap-up of rest-assured.
 *
 * @author linsong wang
 */
public class RestCommunication extends EntityCommunication {
    private static final Logger LOG = LoggerFactory.getLogger(RestCommunication.class);

    /**
     * REST service host name or IP address
     */
    public static final String SYSPROP_HOST = "reactor.comm.rest.HOST";

    /**
     * REST service port
     */
    public static final String SYSPROP_PORT = "reactor.comm.rest.PORT";

    /**
     * Certificate used to authenticate REST service client
     */
    public static final String SYSPROP_CLIENT_CERT = "reactor.comm.rest.CLIENT_CERT";

    /**
     * Passcode of client certificate
     */
    public static final String SYSPROP_CLIENT_CERT_PASS = "reactor.comm.rest.CLIENT_CERT_PASS";

    /**
     * REST service user name for basic authentication
     */
    public static final String SYSPROP_USER = "reactor.comm.rest.USER";

    /**
     * REST service user password for basic authentication
     */
    public static final String SYSPROP_PASS = "reactor.comm.rest.PASS";

    private RequestSpecification reqSpec;

    /**
     * Create a RequestSpecification object as template, with host/port, user/pass, and certificate info.
     *
     * @throws Exception in case of any issue
     */
    @Override
    public void connect() throws Exception {
        String host = sysConfig.getProperty(RestCommunication.SYSPROP_HOST, "localhost");
        int port = sysConfig.getIntProperty(RestCommunication.SYSPROP_PORT, 443);
        RestAssured.useRelaxedHTTPSValidation();
        this.reqSpec = RestAssured.given();

        if (port % 1000 == 443) {
            this.reqSpec = this.reqSpec.baseUri("https://" + host + ":" + port);
        } else {
            this.reqSpec = this.reqSpec.baseUri("http://" + host + ":" + port);
        }

        String user = sysConfig.getProperty(RestCommunication.SYSPROP_USER);
        String pass = sysConfig.getProperty(RestCommunication.SYSPROP_PASS);
        if (null != user && null != pass) {
            this.reqSpec = this.reqSpec.auth().preemptive().basic(user, pass);
        }

        String clientCert = sysConfig.getProperty(RestCommunication.SYSPROP_CLIENT_CERT);
        String clientCertPass = sysConfig.getProperty(RestCommunication.SYSPROP_CLIENT_CERT_PASS);
        if (null != clientCert && null != clientCertPass) {
            this.reqSpec = this.reqSpec.auth().certificate(clientCert, clientCertPass);
        }
    }

    /**
     * Throws OperationNotSupportedException.
     *
     * @throws Exception OperationNotSupportedException
     */
    @Override
    public void disconnect() throws Exception {
        throw new OperationNotSupportedException();
    }

    /**
     * Gets a new copy of defined RequestSpecification.
     *
     * @return a new copy of defined RequestSpecification
     */
    public RequestSpecification given() {
        return RestAssured.given(reqSpec);
    }

    /**
     *
     * @param response HTTP response
     *
     * @return response body
     *
     * @throws IOException in case of any issue
     */
    public static String checkResponse(Response response) throws IOException {
        String res = "";
        if (response.body() != null) {
            res = response.body().asString();
        }
        int code = response.getStatusCode();
        if (code < 200 || code >= 300) {
            LOG.warn("{}", response.getStatusLine());
            throw new RestException(code, res);
        }
        return res;
    }
}