com.chimerapps.niddler.interceptor.okhttp.NiddlerOkHttpResponse Maven / Gradle / Ivy

Go to download
package com.chimerapps.niddler.interceptor.okhttp;

import com.chimerapps.niddler.core.NiddlerRequest;
import com.chimerapps.niddler.core.NiddlerResponse;

import java.io.IOException;
import java.io.OutputStream;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;
import java.util.UUID;

import androidx.annotation.NonNull;
import androidx.annotation.Nullable;
import okhttp3.Response;
import okhttp3.ResponseBody;
import okio.Buffer;
import okio.BufferedSource;

/**
 * @author Maarten Van Giel
 * @author Nicola Verbeeck
 */
final class NiddlerOkHttpResponse implements NiddlerResponse {

	private final Response mResponse;
	private final String mRequestId;
	private final String mMessageId;
	private final long mTimestamp;
	@Nullable
	private final NiddlerRequest mActualNetworkRequest;
	@Nullable
	private final NiddlerResponse mActualNetworkReply;
	private final int mWriteTime;
	private final int mReadTime;
	private final int mWaitTime;
	@Nullable
	private final Map mExtraHeaders;
	private final Map mMetadata;

	NiddlerOkHttpResponse(final Response response,
			final String requestId,
			@Nullable final NiddlerRequest actualNetworkRequest,
			@Nullable final NiddlerResponse actualNetworkReply,
			final int writeTime,
			final int readTime,
			final int waitTime,
			@Nullable final Map extraHeaders,
			@Nullable final Map metadata) {
		mResponse = response;
		mRequestId = requestId;
		mActualNetworkRequest = actualNetworkRequest;
		mActualNetworkReply = actualNetworkReply;
		mWriteTime = writeTime;
		mReadTime = readTime;
		mWaitTime = waitTime;
		mMessageId = UUID.randomUUID().toString();
		mTimestamp = System.currentTimeMillis();
		mExtraHeaders = extraHeaders;
		mMetadata = new TreeMap<>();
		if (metadata != null) {
			mMetadata.putAll(metadata);
		}
	}

	public void addMetadata(@NonNull final String key, @NonNull final String value) {
		mMetadata.put(key, value);
	}

	@NonNull
	@Override
	public String getMessageId() {
		return mMessageId;
	}

	@NonNull
	@Override
	public String getRequestId() {
		return mRequestId;
	}

	@Override
	public long getTimestamp() {
		return mTimestamp;
	}

	@NonNull
	@Override
	public Map> getHeaders() {
		final Map> finalHeaders = mResponse.headers().toMultimap();
		if (mExtraHeaders != null) {
			for (final Map.Entry keyValueEntry : mExtraHeaders.entrySet()) {
				if (!finalHeaders.containsKey(keyValueEntry.getKey())) {
					finalHeaders.put(keyValueEntry.getKey(), Collections.singletonList(keyValueEntry.getValue()));
				}
			}
		}
		return finalHeaders;
	}

	@NonNull
	@Override
	public Map getMetadata() {
		return mMetadata;
	}

	@NonNull
	@Override
	public Integer getStatusCode() {
		return mResponse.code();
	}

	@Nullable
	@Override
	public NiddlerRequest actualNetworkRequest() {
		return mActualNetworkRequest;
	}

	@Nullable
	@Override
	public NiddlerResponse actualNetworkReply() {
		return mActualNetworkReply;
	}

	@NonNull
	@Override
	public String getStatusLine() {
		return mResponse.message();
	}

	@NonNull
	@Override
	public String getHttpVersion() {
		return NiddlerOkHttpRequest.httpVersion(mResponse.protocol());
	}

	@Override
	public int getWriteTime() {
		return mWriteTime;
	}

	@Override
	public int getReadTime() {
		return mReadTime;
	}

	@Override
	public int getWaitTime() {
		return mWaitTime;
	}

	@Override
	public void writeBody(@NonNull final OutputStream stream) {
		final ResponseBody body = mResponse.body();
		try {
			if (body != null) {
				final BufferedSource source = body.source();
				source.request(Long.MAX_VALUE); // Buffer entire body

				final Buffer buffer = source.buffer();
				stream.write(buffer.clone().readByteArray());
				stream.flush();
			}
		} catch (final IOException e) {
			e.printStackTrace();
		}
	}

	@Nullable
	@Override
	public StackTraceElement[] getErrorStackTrace() {
		return null;
	}
}