/**
 *
 * Copyright (c) 2014, the Railo Company Ltd. All rights reserved.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either 
 * version 2.1 of the License, or (at your option) any later version.
 * 
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public 
 * License along with this library.  If not, see <http://www.gnu.org/licenses/>.
 * 
 **/
package lucee.runtime.functions.struct;

import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import lucee.runtime.PageContext;
import lucee.runtime.exp.PageException;
import lucee.runtime.ext.function.BIF;
import lucee.runtime.op.Caster;
import lucee.runtime.op.Decision;
import lucee.runtime.type.Collection;
import lucee.runtime.type.Collection.Key;
import lucee.runtime.type.KeyImpl;
import lucee.runtime.type.Struct;
import lucee.runtime.type.StructImpl;

public class StructKeyTranslate extends BIF {
	
	private static final long serialVersionUID = -7978129950865681102L;

	public static double call(PageContext pc ,Struct sct) throws PageException {
		return call(pc, sct,false,false);
    }
	public static double call(PageContext pc ,Struct sct,boolean deepTranslation) throws PageException {
		return call(pc, sct,deepTranslation,false);
    }
	public static double call(PageContext pc ,Struct sct,boolean deepTranslation,boolean leaveOriginalKey) throws PageException {
		return translate(sct, deepTranslation,leaveOriginalKey);
    }
	
	private static int translate(Collection coll,boolean deep,boolean leaveOrg) throws PageException {
		Key[] keys = coll.keys(); //we do not entry to avoid ConcurrentModificationException
		
		boolean isStruct=coll instanceof Struct;
		String key;
		Object value;
		int index;
		int count=0;
		for(Key k:keys) {
			key=k.getString();
			value=coll.get(k);
			if(deep)count+=translate(value,leaveOrg);
			if(isStruct && (index=key.indexOf('.'))!=-1){
				count++;
				translate(index,k,key,coll,leaveOrg);
			}
		}
		return count;
    }

	private static int translate(Object value,boolean leaveOrg) throws PageException {
		if(value instanceof Collection)
			return translate((Collection)value, true,leaveOrg);
		if(value instanceof List)
			return translate((List<?>)value, leaveOrg);
		if(value instanceof Map)
			return translate((Map<?,?>)value, leaveOrg);
		if(Decision.isArray(value))
			return translate(Caster.toNativeArray(value), leaveOrg);
		return 0;
	}

	private static int translate(List<?> list,boolean leaveOrg) throws PageException {
		Iterator<?> it = list.iterator();
		int count=0;
		while(it.hasNext()){
			count+=translate(it.next(),leaveOrg);
		}
		return count;
	}

	private static int translate(Map<?,?> map,boolean leaveOrg) throws PageException {
		Iterator<?> it = map.entrySet().iterator();
		int count=0;
		while(it.hasNext()){
			count+=translate(((Map.Entry<?,?>)it.next()).getValue(),leaveOrg);
		}
		return count;
	}

	private static int translate(Object[] arr,boolean leaveOrg) throws PageException {
		int count=0;
		for(int i=0;i<arr.length;i++){
			count+=translate(arr[i],leaveOrg);
		}
		return count;
	}

	private static void translate(int index, Key key, String strKey, Collection coll,boolean leaveOrg) throws PageException {
		String left;
		Object value=leaveOrg?coll.get(key):coll.remove(key);
		do{
			left=strKey.substring(0,index);
			strKey=strKey.substring(index+1);
			coll=touch(coll,KeyImpl.init(left));
			
		}
		while((index=strKey.indexOf('.'))!=-1);
		coll.set(KeyImpl.init(strKey), value);
	}

	private static Collection touch(Collection coll, Key key) throws PageException {
		Object obj = coll.get(key,null);
		if(obj instanceof Collection) return (Collection) obj;
		if(Decision.isCastableToStruct(obj))
			return Caster.toStruct(obj);
		coll.set(key, coll=new StructImpl());
		return coll;
		
	}
	@Override
	public Object invoke(PageContext pc, Object[] args) throws PageException {
		if(args.length==3) return call(pc,Caster.toStruct(args[0]),Caster.toBooleanValue(args[1]),Caster.toBooleanValue(args[2]));
		if(args.length==2) return call(pc,Caster.toStruct(args[0]),Caster.toBooleanValue(args[1]));
		return call(pc,Caster.toStruct(args[0]));
	}
	
}