AbstractPolicyRawAction.java

/**
 * Copyright (c) 2004-2011 Wang Jinbao(Julian Wong), http://www.ralasafe.com
 * Licensed under the MIT license: http://www.opensource.org/licenses/mit-license.php
 */
package org.ralasafe.servlet;

import java.io.IOException;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.ralasafe.EntityExistException;
import org.ralasafe.RalasafeException;
import org.ralasafe.SystemConstant;
import org.ralasafe.db.sql.xml.ContextValue;
import org.ralasafe.db.sql.xml.DefineVariable;
import org.ralasafe.db.sql.xml.Formula;
import org.ralasafe.db.sql.xml.HintValue;
import org.ralasafe.db.sql.xml.QueryRef;
import org.ralasafe.db.sql.xml.SimpleValue;
import org.ralasafe.db.sql.xml.UserValue;
import org.ralasafe.db.sql.xml.Variable;
import org.ralasafe.db.sql.xml.types.FormulaTypeOperatorType;
import org.ralasafe.db.sql.xml.types.FormulaTypeTypeType;
import org.ralasafe.db.sql.xml.types.SimpleValueTypeTypeType;
import org.ralasafe.entitle.QueryManager;

public abstract class AbstractPolicyRawAction extends Action {
	private static final Log log=LogFactory.getLog( AbstractPolicyRawAction.class );
	
	public abstract AbstractPolicyDesignHandler createPolicyHandler( HttpServletRequest req );
	public abstract String getPolicyHandlerAttributeKey( HttpServletRequest req );
	
	private AbstractPolicyDesignHandler getPolicyHandler( HttpServletRequest req ) {
		String oper=req.getParameter( "oper" );
		String key=getPolicyHandlerAttributeKey( req );
		AbstractPolicyDesignHandler handler=(AbstractPolicyDesignHandler) req.getSession().getAttribute( key );
		
		if( handler==null||"loadFresh".equals(oper) ) {
			handler=createPolicyHandler( req );
		
			// save into session, will be removed when policy is saved
			req.getSession().setAttribute( key, handler );
		} 
		
		handler.setRawMode();
		
		return handler;
	}
	
	protected void doGet( HttpServletRequest req, HttpServletResponse resp )
			throws ServletException, IOException {
		String oper=req.getParameter( "oper" );
		if( log.isDebugEnabled() ) {
			log.debug( "oper=" + oper );
		}
		
		AbstractPolicyDesignHandler handler=getPolicyHandler( req );
		req.setAttribute( "handler", handler );
		
		if( "designVariables".equals( oper ) ) {
			DefineVariable[] variables=handler.getVariables();
			req.setAttribute( "variables", variables );
			
			WebUtil.forward( req, resp, handler.getVariablesPage() );
		} else if( "getVariable".equals( oper ) ) {
			DefineVariable[] variables=handler.getVariables();
			int index=WebUtil.getIntParameter( req, "index", 0 );
			
			DefineVariable var=variables[index];
			String gotoPage=getVariableDefinePage( var );
			
			String[] userFields=WebUtil.getUserType( req ).getUserMetadata().getUserFields();
			req.setAttribute( "userFields", userFields );
			req.setAttribute( "variable", var );
			
			WebUtil.forward( req, resp, gotoPage );
		} else if( "newVariable".equals( oper ) ) {
			String gotoPage=getVariableDefinePage( req );
			String[] userFields=WebUtil.getUserType( req ).getUserMetadata().getUserFields();
			req.setAttribute( "userFields", userFields );
			
			WebUtil.forward( req, resp, gotoPage );
		} else if( "return".equals( oper ) ) {
			String gotoPage=handler.getManagePage();
			
			// remove handler from session
			req.getSession().removeAttribute( getPolicyHandlerAttributeKey( req ) );
			
			// goto manage page
			resp.sendRedirect( gotoPage );
			return;
		} else {
			// default: goto design page
			WebUtil.forward( req, resp, handler.getRawMainPage() );
		}
	}

	private String getVariableDefinePage( HttpServletRequest req ) {
		String varType=req.getParameter( "type" );
		
		if( "contextValue".equals( varType ) ) {
			return "/ralasafe/common/contextValue.jsp";
		} else if( "formula".equals( varType ) ) {
			return "/ralasafe/common/formula.jsp";
		} else if( "hintValue".equals( varType ) ) {
			return "/ralasafe/common/businessData.jsp";
		} else if( "query".equals( varType ) ) {
			return "/ralasafe/common/queryRef.jsp";
		} else if( "simpleValue".equals( varType ) ) {
			return "/ralasafe/common/simpleValue.jsp";
		} else if( "userValue".equals( varType ) ) {
			return "/ralasafe/common/userValue.jsp";
		} else if( "hintValue".equals( varType ) ) {
			return "/ralasafe/common/businessData.jsp";
		} else {
			throw new RalasafeException( "Unknown variable type" );
		}
	}
	
	private String getVariableDefinePage( DefineVariable var ) {
		if( var.getContextValue()!=null ) {
			return "/ralasafe/common/contextValue.jsp";
		} else if( var.getFormula()!=null ) {
			return "/ralasafe/common/formula.jsp";
		} else if( var.getHintValue()!=null ) {
			return "/ralasafe/common/businessData.jsp";
		} else if( var.getQueryRef()!=null ) {
			return "/ralasafe/common/queryRef.jsp";
		} else if( var.getSimpleValue()!=null ) {
			return "/ralasafe/common/simpleValue.jsp";
		} else if( var.getUserValue()!=null ) {
			return "/ralasafe/common/userValue.jsp";
		} else {
			throw new RalasafeException( "Unknown variable type" );
		}
	}
	
	protected void doPost( HttpServletRequest req, HttpServletResponse resp )
			throws ServletException, IOException {
		String oper=req.getParameter( "oper" );
		if( log.isDebugEnabled() ) {
			log.debug( "oper=" + oper );
		}
		
		AbstractPolicyDesignHandler handler=getPolicyHandler( req );
		req.setAttribute( "handler", handler );
		
		if( "deleteVariable".equals( oper ) ) {
			int id=WebUtil.getIntParameter( req, "index", 0 );
			handler.deleteVariable( id );
		} else if( "addVariable".equals( oper ) ) {
			DefineVariable var=getDefineVariable( req );
			
			handler.addVariable( var );
		} else if( "updateVariable".equals( oper ) ) {
			DefineVariable var=getDefineVariable( req );
			
			int varIndex=WebUtil.getIntParameter( req, "index", -1 );
			handler.updateVariable( varIndex, var );
		} else if( "save".equals( oper ) ) {
			// remove handler from session
			req.getSession().removeAttribute( getPolicyHandlerAttributeKey( req ) );
			
			int id=WebUtil.getIntParameter( req, "id", -23 );
			try {
				handler.save( id );
			} catch( EntityExistException e ) {
				log.error( "", e );
				throw new ServletException( e );
			}
			
			return;
		} else if( "setRawScript".equals( oper ) ) {
			String script=req.getParameter( "rawScript" );
			handler.setRawScript( script );
			return;
		}
	}
	
	private DefineVariable getDefineVariable( HttpServletRequest req ) {
		String type=req.getParameter( "type" );
		String name=req.getParameter( "name" );		
		
		DefineVariable var=new DefineVariable();
		var.setName( name );
		
		if( "userValue".equals( type ) ) {
			String key=req.getParameter( "key" );
			
			UserValue userValue=new UserValue();
			userValue.setKey( key );
			
			var.setUserValue( userValue );
		} else if( "simpleValue".equals( type ) ) {
			String content=req.getParameter( "content" );
			String stype=req.getParameter( "stype" );
			SimpleValue value=new SimpleValue();
			value.setContent( content );
			value.setType( SimpleValueTypeTypeType.valueOf( stype ) );
			
			var.setSimpleValue( value );
		} else if( "contextValue".equals( type ) ) {
			String key=req.getParameter( "key" );
			
			ContextValue value=new ContextValue();
			value.setKey( key );
			
			var.setContextValue( value );
		} else if( "hintValue".equals( type ) ) {
			String key=req.getParameter( "key" );
			
			HintValue value=new HintValue();
			value.setKey( key );
			value.setHint( SystemConstant.BUSINESS_DATA );
			
			var.setHintValue( value );
		} else if( "formula".equals( type ) ) {
			String operName1=req.getParameter( "operName1" );
			String operName2=req.getParameter( "operName2" );
			String operator=req.getParameter( "operator" );
			String returnType=req.getParameter( "returnType" );
			
			Variable var0=new Variable();
			var0.setName( operName1 );
			Variable var1=new Variable();
			var1.setName( operName2 );
			
			Formula value=new Formula();
			
			value.addVariable( var0 );
			value.addVariable( var1 );
			value.setOperator( FormulaTypeOperatorType.valueOf( operator ) );
			value.setType( FormulaTypeTypeType.valueOf( returnType ) );
			var.setFormula( value );
		} else if( "queryRef".equals( type ) ) {
			int id=WebUtil.getIntParameter( req, "queryId", -999 );
			QueryManager mng=WebUtil.getQueryManager( req );
			String queryName=mng.getQuery( id ).getName();
			
			QueryRef value=new QueryRef();
			value.setId( id );
			value.setName( queryName );
			
			var.setQueryRef( value );
		} 
		
		return var;
	}
}