net.nemerosa.ontrack.boot.support.DocumentHttpMessageConverter Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of ontrack-ui Show documentation

Ontrack module: ontrack-ui

There is a newer version: 4.4.5

package net.nemerosa.ontrack.boot.support;

import net.nemerosa.ontrack.common.Document;
import org.apache.commons.io.IOUtils;
import org.springframework.http.HttpInputMessage;
import org.springframework.http.HttpOutputMessage;
import org.springframework.http.MediaType;
import org.springframework.http.converter.AbstractHttpMessageConverter;
import org.springframework.http.converter.HttpMessageNotReadableException;
import org.springframework.http.converter.HttpMessageNotWritableException;

import java.io.IOException;

public class DocumentHttpMessageConverter extends AbstractHttpMessageConverter {

    public DocumentHttpMessageConverter() {
        super(MediaType.IMAGE_GIF, MediaType.IMAGE_JPEG, MediaType.IMAGE_PNG);
    }

    @Override
    protected boolean supports(Class clazz) {
        return Document.class.isAssignableFrom(clazz);
    }

    @Override
    protected Document readInternal(Class clazz, HttpInputMessage inputMessage) throws IOException, HttpMessageNotReadableException {
        return new Document(
                inputMessage.getHeaders().getContentType().toString(),
                IOUtils.toByteArray(inputMessage.getBody())
        );
    }

    @Override
    protected void writeInternal(Document document, HttpOutputMessage outputMessage) throws IOException, HttpMessageNotWritableException {
        outputMessage.getHeaders().setContentType(MediaType.parseMediaType(document.getType()));
        byte[] content = document.getContent();
        outputMessage.getHeaders().setContentLength(content.length);
        outputMessage.getBody().write(content);
    }
}