package railo.runtime.type;
import java.util.HashMap;
import java.util.Map;
import railo.commons.lang.CFTypes;
import railo.commons.lang.StringUtil;
import railo.runtime.ComponentImpl;
import railo.runtime.PageContext;
import railo.runtime.component.Property;
import railo.runtime.exp.ExpressionException;
import railo.runtime.exp.PageException;
import railo.runtime.type.Collection.Key;
import railo.runtime.type.util.CollectionUtil;
import railo.runtime.type.util.KeyConstants;
import railo.runtime.type.util.PropertyFactory;
import railo.runtime.type.util.UDFUtil;
public final class UDFAddProperty extends UDFGSProperty {
private final Property prop;
//private ComponentScope scope;
private final Key propName;
private static final Object NULL=new Object();
public UDFAddProperty(ComponentImpl component,Property prop) {
super(component,"add"+StringUtil.ucFirst(PropertyFactory.getSingularName(prop)),getFunctionArgument(prop),CFTypes.TYPE_ANY,"wddx");
this.prop=prop;
this.propName=KeyImpl.getInstance(prop.getName());
}
private static FunctionArgument[] getFunctionArgument(Property prop) {
String t = PropertyFactory.getType(prop);
FunctionArgument value = new FunctionArgumentImpl(KeyImpl.init(PropertyFactory.getSingularName(prop)),"any",CFTypes.TYPE_ANY,true);
if("struct".equalsIgnoreCase(t)){
FunctionArgument key = new FunctionArgumentImpl(KeyConstants._key,"string",CFTypes.TYPE_STRING,true);
return new FunctionArgument[]{key,value};
}
return new FunctionArgument[]{value};
}
public UDF duplicate() {
return new UDFAddProperty(component,prop);
}
@Override
public Object call(PageContext pageContext, Object[] args,boolean doIncludePath) throws PageException {
// struct
if(this.arguments.length==2) {
if(args.length<2)
throw new ExpressionException("The function "+getFunctionName()+" need 2 arguments, only "+args.length+" argment"+(args.length==1?" is":"s are")+" passed in.");
return _call(pageContext, args[0], args[1]);
}
// array
else if(this.arguments.length==1) {
if(args.length<1)
throw new ExpressionException("The parameter "+this.arguments[0].getName()+" to function "+getFunctionName()+" is required but was not passed in.");
return _call(pageContext, null, args[0]);
}
// never reached
return component;
}
@Override
public Object callWithNamedValues(PageContext pageContext, Struct values,boolean doIncludePath) throws PageException {
UDFUtil.argumentCollection(values,getFunctionArguments());
// struct
if(this.arguments.length==2) {
Key keyName = arguments[0].getName();
Key valueName = arguments[1].getName();
Object key = values.get(keyName,null);
Object value = values.get(valueName,null);
if(key==null)
throw new ExpressionException("The parameter "+keyName+" to function "+getFunctionName()+" is required but was not passed in.");
if(value==null)
throw new ExpressionException("The parameter "+valueName+" to function "+getFunctionName()+" is required but was not passed in.");
return _call(pageContext, key, value);
}
// array
else if(this.arguments.length==1) {
Key valueName = arguments[0].getName();
Object value = values.get(valueName,null);
if(value==null){
Key[] keys = CollectionUtil.keys(values);
if(keys.length==1) {
value=values.get(keys[0]);
}
else throw new ExpressionException("The parameter "+valueName+" to function "+getFunctionName()+" is required but was not passed in.");
}
return _call(pageContext, null, value);
}
// never reached
return component;
}
private Object _call(PageContext pageContext, Object key, Object value) throws PageException {
Object propValue = component.getComponentScope().get(propName,null);
// struct
if(this.arguments.length==2) {
key=cast(arguments[0],key,1);
value=cast(arguments[1],value,2);
if(propValue==null){
HashMap map=new HashMap();
component.getComponentScope().setEL(propName,map);
propValue=map;
}
if(propValue instanceof Struct) {
((Struct)propValue).set(KeyImpl.toKey(key), value);
}
else if(propValue instanceof Map) {
((Map)propValue).put(key, value);
}
}
else {
value=cast(arguments[0],value,1);
if(propValue==null){
/* jira2049
PageContext pc = ThreadLocalPageContext.get();
ORMSession sess = ORMUtil.getSession(pc);
SessionImpl s=(SessionImpl) sess.getRawSession();
propValue=new PersistentList(s);
component.getComponentScope().setEL(propName,propValue);*/
Array arr=new ArrayImpl();
component.getComponentScope().setEL(propName,arr);
propValue=arr;
}
if(propValue instanceof Array) {
((Array)propValue).appendEL(value);
}
else if(propValue instanceof java.util.List) {
((java.util.List)propValue).add(value);
}
}
return component;
}
@Override
public Object implementation(PageContext pageContext) throws Throwable {
return null;
}
@Override
public Object getDefaultValue(PageContext pc, int index) throws PageException {
return prop.getDefault();
}
@Override
public Object getDefaultValue(PageContext pc, int index, Object defaultValue) throws PageException {
return prop.getDefault();
}
@Override
public String getReturnTypeAsString() {
return "any";
}
}