package com.x.base.core.project.jaxrs;

import java.net.URI;
import java.util.Objects;
import java.util.zip.CRC32;

import javax.servlet.http.HttpServletRequest;
import javax.ws.rs.core.CacheControl;
import javax.ws.rs.core.EntityTag;
import javax.ws.rs.core.Response;

import com.x.base.core.project.exception.CallbackPromptException;
import com.x.base.core.project.gson.XGsonBuilder;
import com.x.base.core.project.http.ActionResult;
import com.x.base.core.project.http.HttpMediaType;
import com.x.base.core.project.tools.DefaultCharset;

import org.apache.commons.lang3.StringUtils;
import org.eclipse.jetty.http.HttpHeader;

public class ResponseFactory {

	private static CacheControl defaultCacheControl = CacheControlFactory.getDefault();

	public static final String Content_Disposition = "Content-Disposition";
	public static final String Accept_Ranges = "Accept-Ranges";
	public static final String Content_Type = "Content-Type";
	public static final String Content_Length = "Content-Length";

	public static <T> Response getDefaultActionResultResponse(ActionResult<T> result) {
		if (result.getType().equals(ActionResult.Type.error)) {
			if ((result.throwable instanceof CallbackPromptException)) {
				return Response.ok(callbackError(result)).cacheControl(defaultCacheControl).build();
			} else {
				return Response.serverError().entity(result.toJson()).cacheControl(defaultCacheControl).build();
			}
		} else {
			if ((null != result.getData()) && (result.getData() instanceof WoFile)) {
				WoFile wo = (WoFile) result.getData();
				return Response.ok(wo.getBytes()).header(Content_Disposition, wo.getContentDisposition())
						.header(Content_Type, wo.getContentType()).header(Content_Length, wo.getBytes().length)
						.header(Accept_Ranges, "bytes").build();
			} else if ((null != result.getData()) && (result.getData() instanceof WoText)) {
				WoText wo = (WoText) result.getData();
				return Response.ok(wo.getText()).cacheControl(defaultCacheControl).type(HttpMediaType.TEXT_PLAIN_UTF_8)
						.build();
			} else if ((null != result.getData()) && (result.getData() instanceof WoContentType)) {
				WoContentType wo = (WoContentType) result.getData();
				return Response.ok(wo.getBody()).type(wo.getContentType()).build();
			} else if ((null != result.getData()) && (result.getData() instanceof WoCallback)) {
				return Response.ok(callback((WoCallback) result.getData())).cacheControl(defaultCacheControl).build();
			} else if ((null != result.getData()) && (result.getData() instanceof WoSeeOther)) {
				WoSeeOther wo = (WoSeeOther) result.getData();
				try {
					return Response.seeOther(new URI(wo.getUrl())).build();
				} catch (Exception e) {
					return Response.serverError().entity(Objects.toString(wo.getUrl(), ""))
							.cacheControl(defaultCacheControl).build();
				}
			} else if ((null != result.getData()) && (result.getData() instanceof WoTemporaryRedirect)) {
				WoTemporaryRedirect wo = (WoTemporaryRedirect) result.getData();
				try {
					return Response.temporaryRedirect(new URI(wo.getUrl())).build();
				} catch (Exception e) {
					return Response.serverError().entity(Objects.toString(wo.getUrl(), ""))
							.cacheControl(defaultCacheControl).build();
				}
			} else {
				return Response.ok(result.toJson()).cacheControl(defaultCacheControl).build();
			}
		}
	}

	public static <T> Response getEntityTagActionResultResponse(HttpServletRequest request, ActionResult<T> result) {
		if (result.getType().equals(ActionResult.Type.error)) {
			if ((result.throwable instanceof CallbackPromptException)) {
				return Response.ok(callbackError(result)).build();
			} else {
				return Response.serverError().entity(result.toJson()).build();
			}
		} else {
			if ((null != result.getData()) && (result.getData() instanceof WoFile)) {
				/* 附件,二进制流文件 */
				WoFile wo = (WoFile) result.getData();
				EntityTag tag = new EntityTag(etagWoFile(wo));
				if (notModified(request, tag)) {
					return Response.notModified().tag(tag).build();
				}
				return Response.ok(wo.getBytes()).header(Content_Disposition, wo.getContentDisposition())
						.header(Content_Type, wo.getContentType()).header(Content_Length, wo.getBytes().length)
						.header(Accept_Ranges, "bytes").tag(tag).build();
			} else if ((null != result.getData()) && (result.getData() instanceof WoText)) {
				/* 纯文本text */
				WoText wo = (WoText) result.getData();
				EntityTag tag = new EntityTag(etagWoText(wo));
				if (notModified(request, tag)) {
					return Response.notModified().tag(tag).build();
				}
				return Response.ok(wo.getText()).type(HttpMediaType.TEXT_PLAIN_UTF_8).tag(tag).build();
			} else if ((null != result.getData()) && (result.getData() instanceof WoContentType)) {
				WoContentType wo = (WoContentType) result.getData();
				EntityTag tag = new EntityTag(etagWoContentType(wo));
				if (notModified(request, tag)) {
					return Response.notModified().tag(tag).build();
				}
				return Response.ok(wo.getBody()).type(wo.getContentType()).tag(tag).build();
			} else if ((null != result.getData()) && (result.getData() instanceof WoCallback)) {
				/* jsonp callback */
				return Response.ok(callback((WoCallback) result.getData())).build();
			} else if ((null != result.getData()) && (result.getData() instanceof WoSeeOther)) {
				/* 303 */
				WoSeeOther wo = (WoSeeOther) result.getData();
				try {
					return Response.seeOther(new URI(wo.getUrl())).build();
				} catch (Exception e) {
					return Response.serverError().entity(Objects.toString(wo.getUrl(), "")).build();
				}
			} else if ((null != result.getData()) && (result.getData() instanceof WoTemporaryRedirect)) {
				/* 304 */
				WoTemporaryRedirect wo = (WoTemporaryRedirect) result.getData();
				try {
					return Response.temporaryRedirect(new URI(wo.getUrl())).build();
				} catch (Exception e) {
					return Response.serverError().entity(Objects.toString(wo.getUrl(), "")).build();
				}
			} else {
				/* default */
				EntityTag tag = new EntityTag(etagDefault(result.getData()));
				if (notModified(request, tag)) {
					return Response.notModified().tag(tag).build();
				}
				return Response.ok(result.toJson()).tag(tag).build();
			}
		}
	}

	private static boolean notModified(HttpServletRequest request, EntityTag tag) {
		String If_None_Match = request.getHeader(HttpHeader.IF_NONE_MATCH.toString());
		if (StringUtils.isNotEmpty(If_None_Match)) {
			if (StringUtils.equals(If_None_Match, "\"" + tag.getValue() + "\"")) {
				return true;
			}
		}
		return false;
	}

	private static String etagWoFile(WoFile wo) {
		CRC32 crc = new CRC32();
		crc.update(wo.getBytes());
		return crc.getValue() + "";
	}

	private static String etagWoContentType(WoContentType wo) {
		CRC32 crc = new CRC32();
		crc.update((wo.getBody().toString() + wo.getContentType()).getBytes());
		return crc.getValue() + "";
	}

	private static String etagWoText(WoText wo) {
		CRC32 crc = new CRC32();
		crc.update(wo.getText().getBytes(DefaultCharset.charset_utf_8));
		return crc.getValue() + "";
	}

	private static String etagDefault(Object o) {
		CRC32 crc = new CRC32();
		crc.update(XGsonBuilder.toJson(o).getBytes(DefaultCharset.charset_utf_8));
		return crc.getValue() + "";
	}

	private static String callback(WoCallback woCallback) {
		ActionResult<Object> result = new ActionResult<>();
		result.setData(woCallback.getObject());
		StringBuffer sb = new StringBuffer();
		sb.append("<html><head><meta http-equiv=\"charset\" content=\"" + DefaultCharset.name + "\" /><script>");
		sb.append(woCallback.getCallbackName() + "('" + result.toCompactJson() + "')");
		sb.append("</script></head></html>");
		return sb.toString();
	}

	private static String callbackError(ActionResult result) {
		StringBuffer sb = new StringBuffer();
		CallbackPromptException ex = (CallbackPromptException) result.throwable;
		sb.append("<html><head><meta http-equiv=\"charset\" content=\"" + DefaultCharset.name + "\" /><script>");
		sb.append(ex.getCallbackName() + "('" + result.toCompactJson() + "')");
		sb.append("</script></head></html>");
		return sb.toString();
	}
}