org.voovan.http.message.packet.Part Maven / Gradle / Ivy

package org.voovan.http.message.packet;

import org.voovan.network.IoSession;
import org.voovan.tools.buffer.TByteBuffer;
import org.voovan.tools.TFile;
import org.voovan.tools.TString;
import org.voovan.tools.log.Logger;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.nio.ByteBuffer;

/**
 * HTTP 的 part 对象
 * 	改对象仅在 POST 请求,并且Content-Type = multipart/form-data时使用
 * @author helyho
 *
 * Voovan Framework.
 * WebSite: https://github.com/helyho/Voovan
 * Licence: Apache v2 License
 */
public class Part {
	private Header header;
	private Body body;

	/**
	 * Part 类型枚举
	 * @author helyho
	 *
	 */
	public enum PartType{
		BINARY,TEXT
	}

	/**
	 * 构造函数
	 */
	public Part(){
		header = new  Header();
		body = new Body();
	}

	/**
	 * 构造函数
	 * @param name 参数名
	 * @param value 参数值
	 * @param charset 字符集
	 */
	public Part(String name,String value, String charset){
		header = new  Header();
		body = new Body();
		header.put("name", name);
		body.write(value,charset);
	}

	/**
	 * 构造函数
	 * @param name 参数名
	 * @param value 参数值
	 */
	public Part(String name,String value){
		header = new  Header();
		body = new Body();
		header.put("name", name);
		try {
			body.write(value.getBytes("UTF-8"));
		} catch (UnsupportedEncodingException e) {
			Logger.error("This charset is unsupported",e);
		}
	}

	public Part(String name,File file){
		header = new  Header();
		body = new Body();
		header.put("name", name);
		header.put("filename", TFile.getFileName(file.getPath()));
		try {
			body.changeToFile(file);
		} catch (FileNotFoundException e) {
			Logger.error(e);
		}
	}

	/**
	 * Part 的 Header 对象
	 * @return HTTP-Header 对象
	 */
	public Header header(){
		return header;
	}

	/**
	 * Part 的 body 对象
	 * @return Body 对象
	 */
	public Body body(){
		return body;
	}

	/**
	 * 获取 Part 的名称
	 * @return Part 的名称
	 */
	public String getName() {
		return  header.get("name");
	}

	/**
	 * 获取 part 的文件名称
	 * @return part 的文件名称
	 */
	public String getFileName() {
		return header.get("filename");
	}

	/**
	 * 获取 Part 内容的类型
	 * @return Part 内容的类型
	 */
	public PartType getType(){
		if("binary".equals(header.get("Content-Transfer-Encoding"))){
			return PartType.BINARY;
		}else{
			return PartType.TEXT;
		}
	}

	/**
	 * 将 Part 的内容保存为文件
	 * @param file art 的内容保存为文件
	 * @throws IOException IO 异常
	 */
	public void saveAsFile(File file) throws IOException {
		body.saveAsFile(file);
	}

	/**
	 * 将 Part 的内容保存为文件
	 * @param fileName    Part 的内容保存为文件
	 * @throws IOException IO 异常
	 */
	public void saveAsFile(String fileName) throws IOException{
		body.saveAsFile(new File(fileName));
	}

	/**
	 * 清理
	 */
	public void clear(){
		this.header.clear();
		this.body().clear();
	}

	private String readHead(){
		header.put("Content-Disposition", "form-data");
		if(header.get("name")!=null){
			String value = header.get("Content-Disposition")+"; name=\""+header.get("name")+"\"";
			if(header.get("filename")!=null) {
				value = value + "; filename=\""+header.get("filename")+"\"";
			}
			header.put("Content-Disposition", value);
		}

		header.remove("name");
		header.remove("filename");

		return header.toString()+"\r\n";
	}

	public void send(IoSession session, String boundary){
		int readSize = 0;

		//发送缓冲区
		ByteBuffer byteBuffer = TByteBuffer.allocateDirect();

		//发送分段开始
		byteBuffer.put(TString.assembly("--", boundary, "\r\n").getBytes());
		byteBuffer.flip();
		session.send(byteBuffer);
		byteBuffer.clear();

		//发送 part 头
		byteBuffer.put(readHead().getBytes());
		byteBuffer.flip();
		session.send(byteBuffer);
		byteBuffer.clear();

		//发送 part 报文正文
		while(true) {
			readSize = body.read(byteBuffer);
			if (readSize == -1) {
				break;
			}
			session.send(byteBuffer);
			byteBuffer.clear();
		}

		byteBuffer.clear();

		//发送结尾字符
		byteBuffer.put("\r\n".getBytes());
		byteBuffer.flip();
		session.send(byteBuffer);
		byteBuffer.clear();

		TByteBuffer.release(byteBuffer);

	}

	@Override
	public String toString(){
		return readHead();
	}


}