package railo.runtime.net.http;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.util.Arrays;

import org.apache.commons.fileupload.MultipartStream;
import org.apache.commons.lang.StringUtils;

import railo.commons.io.IOUtil;
import railo.commons.lang.StringUtil;
import railo.runtime.exp.PageException;
import railo.runtime.type.Array;
import railo.runtime.type.ArrayImpl;
import railo.runtime.type.Struct;
import railo.runtime.type.StructImpl;
import railo.runtime.type.util.KeyConstants;
import railo.runtime.type.util.ListUtil;


public class MultiPartResponseUtils {

	public static boolean isMultipart(String mimetype) {
		return !StringUtil.isEmpty(extractBoundary(mimetype,null)) 
			&& StringUtil.startsWithIgnoreCase(mimetype, "multipart/");
	}

	public static Array getParts(byte[] barr,String contentTypeHeader) throws IOException, PageException {
		String boundary = extractBoundary(contentTypeHeader,"");
		ByteArrayInputStream bis = new ByteArrayInputStream(barr);
		MultipartStream stream;
		Array result = new ArrayImpl();
		stream = new MultipartStream(bis,getBytes(boundary,"UTF-8"));// 
		
		boolean hasNextPart = stream.skipPreamble();
		while (hasNextPart) {
			result.append(getPartData(stream));
			hasNextPart = stream.readBoundary();
		}
		return result;
	}

	private static String extractBoundary(String contentTypeHeader, String defaultValue) {
		if (contentTypeHeader == null) return defaultValue;
		String[] headerSections = ListUtil.listToStringArray(contentTypeHeader, ';');
		for (String section: headerSections) {
			String[] subHeaderSections = ListUtil.listToStringArray(section,'=');
			String headerName = subHeaderSections[0].trim();
			if (headerName.toLowerCase().equals("boundary")) {
				return subHeaderSections[1];
			}
		
		}
		return defaultValue;
	}

	private static Struct getPartData(MultipartStream stream) throws IOException, PageException {
		Struct headers = extractHeaders(stream.readHeaders());
		ByteArrayOutputStream baos = new ByteArrayOutputStream();
		stream.readBodyData(baos);
		Struct fileStruct = new StructImpl();
		fileStruct.set(KeyConstants._content, baos.toByteArray());
		fileStruct.set(KeyConstants._headers, headers);
		IOUtil.closeEL(baos);
		return fileStruct;
	}

	private static Struct extractHeaders(String rawHeaders) throws PageException {
		Struct result = new StructImpl();
		String[] headers = ListUtil.listToStringArray(rawHeaders,'\n');
		for(String rawHeader :headers) {
			String[] headerArray = ListUtil.listToStringArray(rawHeader,':');
			String headerName = headerArray[0];
			if (!StringUtil.isEmpty(headerName,true)) {
				String value = StringUtils.join(Arrays.copyOfRange(headerArray, 1, headerArray.length),":").trim();
				result.set(headerName, value);
			}
		}
		return result;
	}

	private static byte[] getBytes(String string, String charset) {
		byte[] bytes;
		try {
			bytes = string.getBytes(charset);
		} catch (UnsupportedEncodingException e) {
			bytes = string.getBytes();
		}
		return bytes;
	}


}