panda.net.http.HttpRequest Maven / Gradle / Ivy

Go to download
package panda.net.http;

import java.io.ByteArrayInputStream;
import java.io.DataOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import panda.Panda;
import panda.codec.binary.Base64;
import panda.io.MimeTypes;
import panda.io.Streams;
import panda.io.stream.WriterOutputStream;
import panda.lang.Arrays;
import panda.lang.Charsets;
import panda.lang.Collections;
import panda.lang.Exceptions;
import panda.lang.Iterators;
import panda.lang.Randoms;
import panda.lang.Strings;
import panda.lang.Systems;
import panda.net.Mimes;
import panda.net.URLBuilder;
import panda.vfs.FileItem;

public class HttpRequest {
	// -------------------------------------------------------------
	public static final String DEFAULT_USERAGENT = "Panda/" + Panda.VERSION + " (Java " + Systems.JAVA_VERSION + ")";

	private static final int TOSTRING_BODY_LIMIT = 1024;

	public static HttpRequest get(String url) {
		return new HttpRequest().setUrl(url).setMethod(HttpMethod.GET);
	}

	public static HttpRequest get(String url, HttpHeader header) {
		return new HttpRequest().setUrl(url).setMethod(HttpMethod.GET).setHeader(header);
	}

	public static HttpRequest post(String url) {
		return new HttpRequest().setUrl(url).setMethod(HttpMethod.POST);
	}

	public static HttpRequest post(String url, HttpHeader header) {
		return new HttpRequest().setUrl(url).setMethod(HttpMethod.POST).setHeader(header);
	}

	public static HttpRequest create(String url, String method) {
		return new HttpRequest().setUrl(url).setMethod(method);
	}

	public static HttpRequest create(String url, String method, Map params) {
		return new HttpRequest().setUrl(url).setMethod(method).setParams(params);
	}

	public static HttpRequest create(String url, String method, Map params, HttpHeader header) {
		return new HttpRequest().setUrl(url).setMethod(method).setParams(params).setHeader(header);
	}

	private String url;
	private String method;
	private HttpHeader header;
	private Map params;
	private InputStream body;
	private String encoding = Charsets.UTF_8;
	private String boundary;

	public HttpRequest() {
	}

	public String getMethod() {
		return method;
	}

	public boolean isGet() {
		return HttpMethod.GET.equalsIgnoreCase(method);
	}

	public boolean isPost() {
		return HttpMethod.POST.equalsIgnoreCase(method);
	}

	public HttpRequest setMethod(String method) {
		this.method = Strings.upperCase(method);
		return this;
	}

	public HttpHeader getHeader() {
		if (header == null) {
			header = HttpHeader.create();
		}
		return header;
	}

	public HttpRequest setHeader(HttpHeader header) {
		this.header = header;
		return this;
	}

	public HttpRequest addHeader(String name, Object value) {
		getHeader().add(name, value);
		return this;
	}

	public HttpRequest setHeader(String name, Object value) {
		getHeader().set(name, value);
		return this;
	}
	
	/**
	 * @return the encoding
	 */
	public String getEncoding() {
		return encoding;
	}

	/**
	 * @param encoding the encoding to set
	 * @return this
	 */
	public HttpRequest setEncoding(String encoding) {
		this.encoding = encoding;
		return this;
	}

	public InputStream getBody() {
		return body;
	}

	public HttpRequest setBody(InputStream body) {
		this.body = body;
		return this;
	}

	public HttpRequest setBody(byte[] body) {
		this.body = (body == null ? null : new ByteArrayInputStream(body));
		return this;
	}

	public HttpRequest setBody(String body) {
		this.body = (body == null ? null : new ByteArrayInputStream(Strings.getBytes(body, encoding)));
		return this;
	}

	/**
	 * @return the url
	 */
	public String getUrl() {
		return url;
	}

	public HttpRequest setUrl(String url) {
		if (url != null && url.indexOf("://") < 0) {
			// default http protocol
			this.url = "http://" + url;
		}
		else {
			this.url = url;
		}
		return this;
	}
	
	/**
	 * @return the url with query string
	 */
	public URL getURL() {
		String url = this.url;
		if (!isPost() && Collections.isNotEmpty(params)) {
			url = URLBuilder.buildURL(url, params, encoding);
		}

		try {
			return new URL(url);
		}
		catch (MalformedURLException e) {
			throw new IllegalArgumentException("Invalid URL: " + url, e);
		}
	}

	public Map getParams() {
		if (params == null) {
			params = new HashMap();
		}
		return params;
	}

	public String getURLEncodedParams() {
		return URLBuilder.buildQueryString(params, encoding);
	}

	@SuppressWarnings("unchecked")
	public HttpRequest setParams(Map params) {
		this.params = (Map)params;
		return this;
	}

	public HttpRequest setParam(String name, Object value) {
		getParams().put(name, value);
		return this;
	}

	@SuppressWarnings("unchecked")
	public HttpRequest addParam(String name, Object value) {
		Object o = getParams().get(name);
		if (o == null) {
			getParams().put(name, value);
		}
		else if (o instanceof Collection) {
			((Collection)o).add(o);
		}
		else {
			List