com.icegreen.greenmail.standalone.JacksonObjectMapperProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of greenmail-standalone Show documentation
Run GreenMail standalone
There is a newer version: 2.1.0-rc-1
package com.icegreen.greenmail.standalone;

import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.SerializerProvider;
import com.fasterxml.jackson.databind.module.SimpleModule;
import com.fasterxml.jackson.databind.ser.std.StdSerializer;
import com.icegreen.greenmail.store.StoredMessage;
import com.icegreen.greenmail.user.GreenMailUser;
import com.icegreen.greenmail.util.GreenMailUtil;
import com.icegreen.greenmail.util.ServerSetup;
import jakarta.mail.MessagingException;
import jakarta.ws.rs.ext.ContextResolver;
import jakarta.ws.rs.ext.Provider;

import java.io.IOException;

import static com.fasterxml.jackson.databind.SerializationFeature.FAIL_ON_EMPTY_BEANS;

@Provider
public class JacksonObjectMapperProvider implements ContextResolver {
    final ObjectMapper defaultObjectMapper;

    public JacksonObjectMapperProvider() {
        defaultObjectMapper = createDefaultMapper();
    }

    @Override
    public ObjectMapper getContext(Class type) {
        return defaultObjectMapper;
    }

    /**
     * Serializes GreenMail internal StoredMessage using minimal-/essential attributes only.
     * Note: Ignores errors when serializing (best effort)
     */
    static class StoredMessageSerializer extends StdSerializer {
        public StoredMessageSerializer() {
            super(StoredMessage.class);
        }

        @Override
        public void serialize(StoredMessage value, JsonGenerator generator, SerializerProvider provider) throws IOException {
            generator.writeStartObject();
            generator.writeStringField("uid", Long.toString(value.getUid()));
            try {
                generator.writeStringField("Message-ID", value.getMimeMessage().getMessageID());
            } catch (MessagingException e) {
                // Ignore
            }
            try {
                generator.writeStringField("subject", value.getMimeMessage().getSubject());
            } catch (MessagingException e) {
                // Ignore
            }
            try {
                generator.writeStringField("contentType", value.getMimeMessage().getContentType());
            } catch (MessagingException e) {
                // Ignore
            }
            generator.writeStringField("mimeMessage", GreenMailUtil.toString(value.getMimeMessage()));
            generator.writeEndObject();
        }
    }

    static class GreenMailUserSerializer extends StdSerializer {
        private static final long serialVersionUID = 1L;

        public GreenMailUserSerializer() {
            super(GreenMailUser.class);
        }

        @Override
        public void serialize(GreenMailUser value, JsonGenerator generator, SerializerProvider provider) throws IOException {
            generator.writeStartObject();
            generator.writeStringField("login", value.getLogin());
            generator.writeStringField("email", value.getEmail());
            generator.writeEndObject();
        }
    }

    static class GreenMailServerSetupSerializer extends StdSerializer {
        private static final long serialVersionUID = 1L;

        public GreenMailServerSetupSerializer() {
            super(ServerSetup.class);
        }

        @Override
        public void serialize(ServerSetup value, JsonGenerator generator, SerializerProvider provider) throws IOException {
            generator.writeStartObject();
            generator.writeNumberField("port", value.getPort());
            generator.writeStringField("address", value.getBindAddress());
            generator.writeStringField("protocol", value.getProtocol());
            generator.writeBooleanField("isSecure", value.isSecure());
            generator.writeNumberField("readTimeout", value.getReadTimeout());
            generator.writeNumberField("writeTimeout", value.getWriteTimeout());
            generator.writeNumberField("connectionTimeout", value.getConnectionTimeout());
            generator.writeNumberField("serverStartupTimeout", value.getServerStartupTimeout());
            generator.writeBooleanField("isDynamicPort", value.isDynamicPort());
            generator.writeEndObject();
        }
    }

    private static ObjectMapper createDefaultMapper() {
        final ObjectMapper mapper = new ObjectMapper();
        mapper.configure(FAIL_ON_EMPTY_BEANS, false);

        SimpleModule module = new SimpleModule();
        module.addSerializer(StoredMessage.class, new StoredMessageSerializer());
        module.addSerializer(GreenMailUser.class, new GreenMailUserSerializer());
        module.addSerializer(ServerSetup.class, new GreenMailServerSetupSerializer());
        mapper.registerModule(module);

        return mapper;
    }
}