groovy.lang.HttpBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of artemis-core Show documentation
The newest version!
package groovy.lang;

import lombok.Getter;
import lombok.RequiredArgsConstructor;
import lombok.ToString;
import org.apache.hc.client5.http.classic.methods.HttpUriRequestBase;
import org.apache.hc.client5.http.cookie.Cookie;
import org.apache.hc.client5.http.entity.UrlEncodedFormEntity;
import org.apache.hc.client5.http.impl.classic.CloseableHttpClient;
import org.apache.hc.client5.http.impl.classic.CloseableHttpResponse;
import org.apache.hc.client5.http.impl.classic.HttpClients;
import org.apache.hc.client5.http.protocol.HttpClientContext;
import org.apache.hc.core5.http.ContentType;
import org.apache.hc.core5.http.io.entity.StringEntity;
import org.apache.hc.core5.http.message.BasicNameValuePair;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.URI;
import java.nio.charset.StandardCharsets;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

import static groovy.lang.Artemis.jsonify;

public class HttpBuilder {
	private static final HttpBuilder INST = new HttpBuilder();
	private static final CloseableHttpClient HTTP_CLIENT = HttpClients.createDefault();

	private HttpBuilder() {
	}

	public static HttpBuilder get() {
		return INST;
	}

	// ---------------

	public Rq get(String url) {
		return new Rq("GET", url);
	}

	public Rq post(String url) {
		return new Rq("POST", url);
	}

	public Rq put(String url) {
		return new Rq("PUT", url);
	}

	public Rq patch(String url) {
		return new Rq("PATCH", url);
	}

	public Rq delete(String url) {
		return new Rq("DELETE", url);
	}

	// ------------------------------

	@RequiredArgsConstructor
	public static class Rq {
		private final String method;
		private final String url;
		private CharSequence body;

		private final Map headers = new HashMap<>();
		private final Map formParams = new HashMap<>();

		public Rq header(String key, String value) {
			headers.put(key, value);
			return this;
		}

		public Rq body(Object body) {
			return body(jsonify(body));
		}

		public Rq body(CharSequence body) {
			if (!formParams.isEmpty()) {
				throw new RuntimeException("Invalid State: setting both form and body");
			}
			this.body = body;
			return this;
		}

		public Rq form(String key, String value) {
			if (body != null) {
				throw new RuntimeException("Invalid State: setting both form and body");
			}
			formParams.put(key, value);
			return this;
		}

		public Rs send() {
			try {
				final HttpUriRequestBase rq = new HttpUriRequestBase(method, URI.create(url));
				if (body != null) {
					rq.setEntity(new StringEntity(body.toString(), ContentType.APPLICATION_JSON, "UTF-8", false));
				} else {
					final List pairs = formParams
						.entrySet()
						.stream()
						.map(e -> new BasicNameValuePair(e.getKey(), e.getValue()))
						.collect(Collectors.toList());
					rq.setEntity(new UrlEncodedFormEntity(pairs));
				}
				headers.forEach(rq::addHeader);

				final HttpClientContext context = HttpClientContext.create();
				try (final CloseableHttpResponse rs = HTTP_CLIENT.execute(rq, context)) {
					final String rsBody = new BufferedReader(new InputStreamReader(rs.getEntity().getContent(), StandardCharsets.UTF_8))
						.lines()
						.collect(Collectors.joining("\n"));

					final Map cookies;
					if (context.getCookieStore().getCookies().size() > 0) {
						cookies = context.getCookieStore().getCookies().stream()
							.collect(Collectors.toMap(Cookie::getName, Cookie::getValue));
					} else {
						cookies = Collections.emptyMap();
					}
					return new Rs(
						rs.getCode(),
						rs.getReasonPhrase(),
						rs.getEntity().getContentType(),
						rsBody,
						cookies);
				}
			} catch (IOException e) {
				throw new RuntimeException(e);
			}
		}
	}

	// ------------------------------

	@Getter
	@RequiredArgsConstructor
	@ToString
	public static class Rs {
		private final int code;
		private final String codsString;
		private final String contentType;
		private final String body;
		private final Map cookies;

		public Object getBodyAsObject() {
			return Artemis.objectify(body);
		}
	}
}