panda.net.InternetHeader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of panda-core Show documentation
Panda Core is the core module of Panda Framework, it contains commonly used utility classes similar to apache-commons.
There is a newer version: 1.8.0
package panda.net;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.Date;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;

import panda.lang.Arrays;
import panda.lang.Exceptions;
import panda.lang.Iterators;
import panda.lang.Numbers;
import panda.lang.Strings;
import panda.lang.collection.CaseInsensitiveMap;


public abstract class InternetHeader extends CaseInsensitiveMap {
	@Override
	protected Map init() {
		return new TreeMap();
	}

	//-------------------------------------------------
	protected abstract Date parseDate(String value);
	protected abstract String formatDate(Date value);
	
	//-------------------------------------------------
	public int getInt(String key) {
		String str = getString(key);
		return Numbers.toInt(str, -1);
	}
	
	public Date getDate(String key) {
		String str = getString(key);
		return parseDate(str);
	}
	
	public String getString(String key) {
		return getString(key, null);
	}
	
	public String getString(String key, String defaultValue) {
		Object value = get(key);
		if (value == null)
			return defaultValue;
		
		if (value instanceof String) {
			return (String)value;
		}
		if (value instanceof List) {
			List vs = (List)value;
			return vs.size() > 0 ? vs.get(0).toString() : defaultValue;
		}

		return defaultValue;
	}

	@SuppressWarnings("unchecked")
	public List getStrings(String key) {
		Object value = get(key);
		if (value == null) {
			return null;
		}
		
		if (value instanceof List) {
			return (List)value;
		}
		return Arrays.toList(value.toString());
	}

	public String getValue(String name) {
		Object val = get(name);
		if (val instanceof String) {
			return (String)val;
		}
		if (val instanceof List) {
			List vs = (List)val;
			return Strings.join(vs, ',');
		}
		return "";
	}
	
	//-------------------------------------------------
	public InternetHeader setInt(String key, int value) {
		return set(key, value);
	}

	public InternetHeader setDate(String key, long value) {
		return set(key, new Date(value));
	}

	public InternetHeader setDate(String key, Date value) {
		return set(key, value);
	}
	
	public InternetHeader setString(String key, String value) {
		return set(key, value);
	}
	
	public InternetHeader set(String key, Object value) {
		put(key, value);
		return this;
	}

	public InternetHeader setAll(Map m) {
		putAll(m);
		return this;
	}

	//-------------------------------------------------
	public InternetHeader addInt(String key, int value) {
		return add(key, value);
	}

	public InternetHeader addDate(String key, long value) {
		return add(key, new Date(value));
	}

	public InternetHeader addDate(String key, Date value) {
		return add(key, value);
	}
	
	public InternetHeader addString(String key, String value) {
		add(key, (Object)value);
		return this;
	}
	
	@SuppressWarnings("unchecked")
	public InternetHeader add(String key, Object value) {
		if (key == null || value == null) {
			return this;
		}

		Object object = get(key);
		if (object == null) {
			put(key, value);
			return this;
		}
		
		List vs = null;
		if (object instanceof List) {
			vs = (List)object;
		}
		else {
			vs = new ArrayList();
			vs.add(object);
			super.put(key, vs);
		}
		
		value = convertValue(value);
		if (value instanceof List) {
			vs.addAll((List)value);
		}
		else {
			vs.add(value);
		}

		return this;
	}
	
	public InternetHeader addAll(Map m) {
		for (Entry en : m.entrySet()) {
			add(en.getKey(), en.getValue());
		}
		return this;
	}

	//-------------------------------------------------
	@SuppressWarnings("unchecked")
	private Object convertValue(Object value) {
		if (value == null) {
			return null;
		}
		
		if (value instanceof Date) {
			return formatDate((Date)value);
		}
		if (value instanceof Calendar) {
			return formatDate(((Calendar)value).getTime());
		}

		if (Iterators.isIterable(value)) {
			List vs = new ArrayList();
			for (Object v : Iterators.asIterable(value)) {
				v = convertValue(v);
				if (v instanceof List) {
					for (Object c : (List)v) {
						vs.add(c);
					}
				}
				else {
					vs.add(v);
				}
			}
			return vs;
		}

		return value.toString();
	}

	@Override
	public Object put(String key, Object value) {
		if (key == null) {
			return null;
		}
		
		if (value == null) {
			return remove(key);
		}

		value = convertValue(value);
		return super.put(key, value);
	}

	@Override
	public String toString() {
		StringBuilder sb = new StringBuilder();
		try {
			write(sb);
		}
		catch (IOException e) {
			throw Exceptions.wrapThrow(e);
		}
		return sb.toString();
	}

	public void write(Appendable writer) throws IOException {
		for (Map.Entry en : entrySet()) {
			String key = en.getKey();
			writer.append(key).append(": ");

			Object val = en.getValue();
			if (val != null) {
				Iterator it = Iterators.asIterator(en.getValue());
				while (it.hasNext()) {
					writer.append(it.next().toString());
					if (it.hasNext()) {
						writer.append(',');
					}
				}
			}
			writer.append(Strings.LF);
		}
	}
}