com.automationrockstars.bmo.event.processor.HttpContent Maven / Gradle / Ivy

Go to download
package com.automationrockstars.bmo.event.processor;

import java.io.File;
import java.io.InputStream;
import java.util.List;
import java.util.Map;

import org.apache.http.NameValuePair;
import org.apache.http.client.fluent.Request;
import org.apache.http.entity.ContentType;
import org.apache.http.message.BasicNameValuePair;

import com.google.common.collect.Lists;
import com.google.common.collect.Maps;

public class HttpContent {

	private Map headers = Maps.newHashMap();
	public Map headers(){
		return headers;
	}
	
	public HttpContent headers(Map headers){
		this.headers.putAll(headers);
		return this;
	}
	
	public HttpContent header(String name, String value){
		this.headers.put(name, value);
		return this;
	}
	
	public HttpContent withContentType(String type){
		this.contentType = ContentType.create(type);
		return this;
	}
	
	public HttpContent withContentType(ContentType type){
		this.contentType = type;
		return this;
	}
	private Object content;
	private Class contentClass;
	private ContentType contentType = ContentType.TEXT_PLAIN;
	public HttpContent content(Object content){
		this.content = content;
		this.contentClass = content.getClass();
		return this;
	}
	public HttpContent setEncoding(String encoding){
		contentType = ContentType.create(encoding);
		return this;
	}
	public Class getContentType(){
		return contentClass;
	}
	
	@SuppressWarnings("unchecked")
	public  T getContent(){
		return (T) content;
	}
	
	public static HttpContent create(){
		return new HttpContent();
	}
	
	public static HttpContent create(Object content){
		return new HttpContent().content(content);
	}
	
	public static HttpContent create(Object content,String encoding){
		return new HttpContent().content(content).setEncoding(encoding);
	}
	@SuppressWarnings("unchecked")
	Request populate(Request toPopulate){
		for (Map.Entry header : headers.entrySet()){
			toPopulate.addHeader(header.getKey(), header.getValue());
		}
		if (content == null) return toPopulate;
		if (contentClass.equals(byte[].class)){
			toPopulate.bodyByteArray((byte[])content);
		} else if (contentClass.equals(String.class)){
			toPopulate.bodyString((String)content, contentType);
		} else if (Map.class.isAssignableFrom(contentClass)){
			List formContent = Lists.newArrayList();
			for (Map.Entry val : ((Map)content).entrySet()){
				formContent.add(new BasicNameValuePair(val.getKey(), val.getValue()));
			}
			toPopulate.bodyForm(formContent);
		} else if (contentClass.equals(File.class)){
			toPopulate.bodyFile((File)content, contentType);
		} else if (InputStream.class.isAssignableFrom(contentClass)){
			toPopulate.bodyStream((InputStream)content);
		} else {
			throw new IllegalArgumentException(String.format("Unknown content class %s. Only byte[], String, Map, File and InputStream are accepted", contentClass));
		}
		return toPopulate;
	}
	
	public String toString(){
		return String.format("HttpContent type %s with headers %s content %s", contentType,headers,content);
	}
	
}