io.rocketbase.mail.PostmarkClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of postmark-spring Show documentation
postmark api client in vanilla spring :)
There is a newer version: 1.0.0
package io.rocketbase.mail;

import io.rocketbase.mail.config.PostmarkProperties;
import io.rocketbase.mail.dto.EmailAttachment;
import io.rocketbase.mail.dto.Message;
import io.rocketbase.mail.dto.MessageResponse;
import io.rocketbase.mail.util.MessageJsonWriter;
import lombok.RequiredArgsConstructor;
import org.springframework.core.ParameterizedTypeReference;
import org.springframework.core.io.FileSystemResource;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.ResponseEntity;
import org.springframework.web.client.RestTemplate;
import org.springframework.web.util.UriComponentsBuilder;

import java.io.File;
import java.util.List;

@RequiredArgsConstructor
public class PostmarkClient {

    private final PostmarkProperties postmarkProperties;

    protected RestTemplate restTemplate;

    protected RestTemplate getRestTemplate() {
        if (restTemplate == null) {
            restTemplate = new RestTemplate();
        }
        return restTemplate;
    }

    public MessageResponse deliverMessage(Message msg) {
        ResponseEntity response = getRestTemplate().exchange(createUriBuilder().path("/email/")
                        .toUriString(),
                HttpMethod.POST,
                new HttpEntity<>(msg, buildHeaders()),
                MessageResponse.class);
        return response.getBody();
    }

    public MessageResponse deliverMessage(Message msg, EmailAttachment... emailAttachments) {
        ResponseEntity response = null;
        File json = null;
        try {
            json = new MessageJsonWriter().writeMessageFile(msg, emailAttachments);
            response = getRestTemplate().exchange(createUriBuilder().path("/email/")
                            .toUriString(),
                    HttpMethod.POST,
                    new HttpEntity<>(new FileSystemResource(json), buildHeaders()),
                    MessageResponse.class);
        } finally {
            if (json != null) {
                json.delete();
            }
        }
        return response.getBody();
    }

    public List deliverMessage(List messages) {
        ResponseEntity> response = getRestTemplate().exchange(createUriBuilder().path("/email/batch")
                        .toUriString(),
                HttpMethod.POST,
                new HttpEntity<>(messages, buildHeaders()),
                new ParameterizedTypeReference>() {
                });
        return response.getBody();
    }

    protected UriComponentsBuilder createUriBuilder() {
        return UriComponentsBuilder.fromUriString(postmarkProperties.getApi().getUrl());
    }

    protected HttpHeaders buildHeaders() {
        HttpHeaders headers = new HttpHeaders();
        headers.add(postmarkProperties.getApi().getHeader(), postmarkProperties.getToken());
        return headers;
    }

}