io.orangebeard.client.AbstractClient Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of java-client Show documentation

Orangebeard Java Client, to be used in Orangebeard Java Listeners

There is a newer version: 3.0.1

package io.orangebeard.client;

import io.orangebeard.client.entity.Attachment;
import io.orangebeard.client.entity.Log;

import java.util.Collections;
import java.util.Set;
import java.util.UUID;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.util.LinkedMultiValueMap;

import static java.lang.String.format;
import static org.springframework.http.HttpHeaders.CONTENT_DISPOSITION;
import static org.springframework.http.HttpHeaders.CONTENT_TYPE;
import static org.springframework.http.MediaType.APPLICATION_JSON;
import static org.springframework.http.MediaType.MULTIPART_FORM_DATA;

public abstract class AbstractClient implements OrangebeardClient {
    protected final UUID uuid;

    public AbstractClient(UUID uuid){
        this.uuid = uuid;
    }

    protected HttpHeaders getAuthorizationHeaders(String accessToken) {
        HttpHeaders headers = new HttpHeaders();
        headers.setBearerAuth((accessToken));
        headers.setContentType(APPLICATION_JSON);
        return headers;
    }

    protected HttpEntity> getMultipartLogRequest(Attachment attachmentLogItem) {
        LinkedMultiValueMap filePartHeaders = new LinkedMultiValueMap<>();
        filePartHeaders.add(CONTENT_DISPOSITION, format("form-data; name=\"file\"; filename=\"%s\"", attachmentLogItem.getFile().getName()));
        filePartHeaders.add(CONTENT_TYPE, attachmentLogItem.getFile().getContentType());

        byte[] fileContents = attachmentLogItem.getFile().getContent();
        HttpEntity filePart = new HttpEntity<>(fileContents, filePartHeaders);

        LinkedMultiValueMap parts = new LinkedMultiValueMap<>();
        parts.add("json_request_part", Collections.singletonList(attachmentLogItem));
        parts.add("file", filePart);

        HttpHeaders headers = getAuthorizationHeaders(uuid.toString());
        headers.setContentType(MULTIPART_FORM_DATA);

        return new HttpEntity<>(parts, headers);
    }

    protected HttpEntity> getMultipartLogRequest(Set logs) {
        LinkedMultiValueMap parts = new LinkedMultiValueMap<>();
        parts.add("json_request_part", logs);

        HttpHeaders headers = getAuthorizationHeaders(uuid.toString());
        headers.setContentType(MULTIPART_FORM_DATA);

        return new HttpEntity<>(parts, headers);
    }
}