package com.cynergysystems.servlet;

public class HttpServletResponseWrapper extends javax.servlet.http.HttpServletResponseWrapper
{
	private boolean isFlashPlayer = false;


	public HttpServletResponseWrapper(javax.servlet.http.HttpServletResponse response)
	{
		super(response);
	}

	public void sendError(int errorCode) throws java.io.IOException
	{
		this.setStatus(handleStatusCode(errorCode));
		super.sendError(handleStatusCode(errorCode));
	}

	public void sendError(int errorCode, String errorMessage) throws java.io.IOException
	{
		this.setStatus(handleStatusCode(errorCode));
		super.sendError(handleStatusCode(errorCode), errorMessage);
	}

	public void setStatus(int statusCode)
	{
		super.setStatus(handleStatusCode(statusCode));
	}

	public void setFlashPlayer( boolean val )
	{
		this.isFlashPlayer = val;
	}

	public boolean getFlashPlayer()
	{
		return this.isFlashPlayer;
	}

	public int handleStatusCode( int code )
	{
		if ( isFlashPlayer )
		{
			if ( code == 500 )
			{
				return 200;
			}
		}

		return code;
	}
}