com.github.dreamhead.moco.model.MessageContent Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of moco-core Show documentation
Moco is an easy setup stub framework, mainly focusing on testing and integration.
There is a newer version: 1.5.0
package com.github.dreamhead.moco.model;

import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
import com.fasterxml.jackson.databind.annotation.JsonSerialize;
import com.github.dreamhead.moco.MocoException;
import com.github.dreamhead.moco.dumper.MessageContentDeserializer;
import com.github.dreamhead.moco.dumper.MessageContentSerializer;
import com.google.common.base.Objects;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.Charset;
import java.util.Arrays;

import static com.google.common.io.ByteStreams.toByteArray;

@JsonSerialize(using = MessageContentSerializer.class)
@JsonDeserialize(using = MessageContentDeserializer.class)
public class MessageContent {
    private byte[] content;
    private Charset charset;

    public final byte[] getContent() {
        return content;
    }

    public final Charset getCharset() {
        if (hasCharset()) {
            return charset;
        }

        return Charset.defaultCharset();
    }

    public final boolean hasCharset() {
        return charset != null;
    }

    public final boolean hasContent() {
        return content.length > 0;
    }

    @Override
    public final String toString() {
        return new String(content, getCharset());
    }

    @Override
    public final boolean equals(final Object obj) {
        if (obj == null) {
            return false;
        }

        if (!(obj instanceof MessageContent)) {
            return false;
        }

        MessageContent that = (MessageContent) obj;
        return Objects.equal(this.charset, that.charset) && Arrays.equals(content, that.content);
    }

    @Override
    public final int hashCode() {
        return Objects.hashCode(Arrays.hashCode(this.content), this.charset);
    }

    public static Builder content() {
        return new Builder();
    }

    public static MessageContent content(final String content) {
        return content().withContent(content).build();
    }

    public final InputStream toInputStream() {
        return new ByteArrayInputStream(this.content);
    }

    public static class Builder {
        private byte[] content;
        private Charset charset;

        public final Builder withContent(final String content) {
            this.content = content.getBytes();
            return this;
        }

        public final Builder withContent(final InputStream is) {
            try {
                this.content = toByteArray(is);
                return this;
            } catch (IOException e) {
                throw new MocoException(e);
            }
        }

        public final Builder withContent(final byte[] content) {
            this.content = content;
            return this;
        }

        public final Builder withCharset(final Charset charset) {
            this.charset = charset;
            return this;
        }

        public final MessageContent build() {
            MessageContent messageContent = new MessageContent();
            messageContent.charset = charset;
            messageContent.content = targetContent(content);
            return messageContent;
        }

        private byte[] targetContent(final byte[] content) {
            if (content == null) {
                return new byte[0];
            }

            return this.content;
        }
    }
}