io.payrun.helpers.RequestHelper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sdk Show documentation
Java SDK for accessing the UK Payroll API - PayRun.io
There is a newer version: 0.13.0
package io.payrun.helpers;

import io.payrun.models.Link;
import io.payrun.oauth1.OAuth1;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.http.*;
import org.springframework.lang.Nullable;
import org.springframework.web.client.HttpClientErrorException;
import org.springframework.web.client.RestTemplate;

import java.net.URI;
import java.net.URISyntaxException;

import static java.util.Collections.singletonList;

public class RequestHelper {

    private final Logger log = LoggerFactory.getLogger(getClass());

    private final String apiHost;
    private final String consumerKey;
    private final String consumerSecret;
    private final String apiVersion;
    private final SerializerHelper serializerHelper;

    public RequestHelper(String apiHost, String clientKey, String clientSecret, String apiVersion)
    {
        this.apiHost = apiHost;
        this.consumerKey = clientKey;
        this.consumerSecret = clientSecret;
        this.apiVersion = apiVersion;
        this.serializerHelper = new SerializerHelper();
    }

    public RequestHelper(String apiHost, String clientKey, String clientSecret){
        this.apiHost = apiHost;
        this.consumerKey = clientKey;
        this.consumerSecret = clientSecret;
        this.apiVersion = "default";
        this.serializerHelper = new SerializerHelper();
    }

    public String getRaw(String path){
        URI uri = getUri(path);

        if (uri == null) return null;

        ResponseEntity response = this.getResponse(uri, HttpMethod.GET);

        return response.getBody();
    }

    public  T get(String path, Class clss){
        URI uri = getUri(path);

        if (uri == null) return null;

        ResponseEntity response = this.getResponse(uri, HttpMethod.GET);

        T output = serializerHelper.fromJson(response.getBody(), clss);

        return output;
    }

    public Link post(String path, Object objectToPost){
        URI uri = getUri(path);

        if (uri == null) return null;

        ResponseEntity response = this.getResponse(uri, HttpMethod.POST, objectToPost);

        Link output = serializerHelper.fromJson(response.getBody(), Link.class);

        return output;
    }

    public  T put(String path, T objectToPut){
        URI uri = getUri(path);

        if (uri == null) return null;

        ResponseEntity response = this.getResponse(uri, HttpMethod.PUT, objectToPut);

        T output = (T)serializerHelper.fromJson(response.getBody(), objectToPut.getClass());

        return output;
    }

    public  T patch(String path, String patch, Class clss){
        URI uri = getUri(path);

        if (uri == null) return null;

        ResponseEntity response = this.getResponse(uri, HttpMethod.PATCH, patch);

        T output = serializerHelper.fromJson(response.getBody(), clss);

        return output;
    }

    public void delete(String path){
        URI uri = getUri(path);

        if (uri == null) return;

        this.getResponse(uri, HttpMethod.DELETE);
    }

    private URI getUri(String path) {
        try {
            return new URI(this.apiHost + path);
        } catch (URISyntaxException e) {
            throw new RuntimeException(e);
        }
    }

    private ResponseEntity getResponse(URI uri, HttpMethod verb) {
        return getResponse(uri, verb, null);
    }

    private ResponseEntity getResponse(URI uri, HttpMethod verb, @Nullable Object objectToTransmit)
    {
        OAuth1 oAuth1 = new OAuth1(this.consumerKey, this.consumerSecret);
        String signature = oAuth1.generateSignature(uri, verb.name());
        String headerValue = oAuth1.buildHeader(signature);

        HttpHeaders headers = new HttpHeaders();
        headers.add("Authorization", headerValue);
        headers.add("Api-Version", this.apiVersion);
        headers.setAccept(singletonList(MediaType.APPLICATION_JSON));

        RestTemplate restTemplate = new RestTemplate();

        String payload = null;

        if (objectToTransmit != null){
            if (objectToTransmit instanceof String){
                payload = (String)objectToTransmit;
            }
            else {
                payload = serializerHelper.toJson(objectToTransmit);
            }

            headers.setContentType(MediaType.APPLICATION_JSON);
            headers.setContentLength(payload.length());
        }

        final HttpEntity request = new HttpEntity<>(payload, headers);

        try {
            ResponseEntity response = restTemplate.exchange(uri, verb, request, String.class);

            return response;
        } catch (HttpClientErrorException e) {
            if (log.isDebugEnabled()) {
                log.debug("VERB: {}\nURI: {}\nPAYLOAD:\n{}\nSTATUS CODE: {}\nRESPONSE: {}", verb, uri, payload, e.getStatusCode(), e.getResponseBodyAsString());
            }

            throw e;
        }
    }
}