com.xqbase.util.http.HttpForm Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of xqbase-util-jdk17 Show documentation
Reusable Java components for www.xqbase.com
There is a newer version: 0.2.18
package com.xqbase.util.http;

import java.io.IOException;
import java.net.URLEncoder;
import java.nio.charset.Charset;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import com.xqbase.util.ByteArrayQueue;
import com.xqbase.util.Bytes;

public class HttpForm {
	private ByteArrayQueue body = new ByteArrayQueue();
	private Map> headers = new LinkedHashMap<>();
	private boolean multipart, appended = false;
	private byte[] boundaryBytes = null;
	private Charset charset;

	public HttpForm(boolean multipart, Charset charset) {
		this.multipart = multipart;
		this.charset = charset;
		String contentType;
		if (multipart) {
			String boundary = Bytes.toHexLower(Bytes.random(16));
			boundaryBytes = boundary.getBytes();
			contentType = "multipart/form-data; boundary=" + boundary;
		} else {
			contentType = "application/x-www-form-urlencoded; charset=" + charset;
		}
		headers.put("Content-Type", Collections.singletonList(contentType));
	}

	public ByteArrayQueue getBody() {
		return body;
	}

	public Map> getHeaders() {
		return headers;
	}

	private static final byte[] EQ = {'='};
	private static final byte[] AMP = {'&'};
	private static final byte[] QUOT = {'"'};
	private static final byte[] DASH = {'-', '-'};
	private static final byte[] CRLF = {'\r', '\n'};
	private static final byte[] CONTENT_DISPOSITION_NAME =
			"Content-Disposition: form-data; name=\"".getBytes();
	private static final byte[] FILENAME =
			"; filename=\"".getBytes();
	private static final byte[] CONTENT_TYPE =
			"Content-Type: ".getBytes();

	private byte[] encode(String s) {
		try {
			return URLEncoder.encode(s, charset.name()).getBytes();
		} catch (IOException e) {
			throw new RuntimeException(e);
		}
	}

	public void add(String name, String value) {
		if (multipart) {
			next(name, null, null);
			body.add(value.getBytes(charset));
			return;
		}
		if (appended) {
			body.add(AMP);
		}
		appended = true;
		body.add(encode(name)).add(EQ).add(encode(value));
	}

	public void next(String name, String fileName, String contentType) {
		if (appended) {
			body.add(CRLF);
		}
		appended = true;
		body.add(DASH).add(boundaryBytes).add(CRLF);
		body.add(CONTENT_DISPOSITION_NAME).add(name.getBytes(charset)).add(QUOT);
		if (fileName != null) {
			body.add(FILENAME).add(fileName.getBytes(charset)).add(QUOT);
		}
		body.add(CRLF);
		if (contentType != null) {
			body.add(CONTENT_TYPE).add(contentType.getBytes()).add(CRLF);
		}
		body.add(CRLF);
	}

	public void finish() {
		if (appended) {
			body.add(CRLF);
		}
		body.add(DASH).add(boundaryBytes).add(DASH).add(CRLF);
	}
}