com.github.dreamhead.moco.dumper.HttpDumpers 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.dumper;

import com.github.dreamhead.moco.HttpMessage;
import com.github.dreamhead.moco.model.MessageContent;
import com.google.common.base.Function;
import com.google.common.base.Joiner;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.Maps;
import com.google.common.net.HttpHeaders;
import com.google.common.net.MediaType;
import io.netty.util.internal.StringUtil;

import java.util.Map;

import static com.google.common.collect.FluentIterable.from;

public final class HttpDumpers {
    public static String asContent(final HttpMessage message) {
        if (hasContent(message)) {
            return StringUtil.NEWLINE + StringUtil.NEWLINE + contentForDump(message);
        }

        return "";
    }

    private static String contentForDump(final HttpMessage message) {
        String type = message.getHeader(HttpHeaders.CONTENT_TYPE);
        if (isText(type)) {
            return message.getContent().toString();
        }

        return "";
    }

    private static boolean isText(final String type) {
        try {
            MediaType mediaType = MediaType.parse(type);
            return mediaType.is(MediaType.ANY_TEXT_TYPE)
                    || mediaType.subtype().endsWith("javascript")
                    || mediaType.subtype().endsWith("json")
                    || mediaType.subtype().endsWith("xml")
                    || mediaType.is(MediaType.FORM_DATA)
                    || mediaType.subtype().endsWith("form-data");
        } catch (Exception e) {
            return false;
        }
    }

    private static boolean hasContent(final HttpMessage message) {
        String lengthText = message.getHeader(HttpHeaders.CONTENT_LENGTH);
        if (lengthText != null) {
            return true;
        }

        MessageContent content = message.getContent();
        return content != null && content.hasContent();

    }

    private static final Joiner.MapJoiner HEAD_JOINER = Joiner.on(StringUtil.NEWLINE).withKeyValueSeparator(": ");

    public static String asHeaders(final HttpMessage message) {
        return HEAD_JOINER.join(from(message.getHeaders().entrySet())
                .transformAndConcat(toMapEntries()));
    }

    private static Function, Iterable>> toMapEntries() {
        return new Function, Iterable>>() {
            @Override
            public Iterable> apply(final Map.Entry input) {
                String key = input.getKey();
                ImmutableList.Builder> builder = ImmutableList.builder();
                for (String value : input.getValue()) {
                    builder.add(Maps.immutableEntry(key, value));
                }
                return builder.build();
            }
        };
    }

    private HttpDumpers() {
    }
}