/*
    * Copyright 2004-2006 the Seasar Foundation and the Others.
    *
    * Licensed under the Apache License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    *
    *     http://www.apache.org/licenses/LICENSE-2.0
    *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, 
   * either express or implied. See the License for the specific language
   * governing permissions and limitations under the License.
   */
  package org.seasar.tuigwaa.database.function;
  
  import java.util.ArrayList;
  import java.util.HashMap;
  import java.util.HashSet;
  import java.util.Iterator;
  import java.util.List;
  import java.util.Map;
  import java.util.Set;
  
  import org.hibernate.ObjectNotFoundException;
  import org.seasar.framework.log.Logger;
  import org.seasar.tuigwaa.model.common.EntityUtils;
  import org.seasar.tuigwaa.model.core.TgwAttribute;
  import org.seasar.tuigwaa.model.core.TgwEntity;
  import org.seasar.tuigwaa.model.core.impl.FkAttribute;
  import org.seasar.tuigwaa.model.core.impl.SelfAttribute;
  import org.seasar.tuigwaa.model.core.impl.SetAttribute;
  
  import com.isenshi.util.functor.UnaryFunction;
  import com.isenshi.util.functor.UnaryProcedure;
  
  public class UpdateExeFunction extends AbstractExeFunction implements
  		UnaryFunction, UnaryProcedure {
  	
  	private Logger logger = Logger.getLogger(getClass());
  
  	private Map updateValueMap = new HashMap();
  
  	private List dtoFieldList = new ArrayList();
  
  	private boolean updateOnly;
  	
  	public Map getUpdateValueMap() {
  		return updateValueMap;
  	}
  
  	public boolean isUpdateOnly() {
  		return updateOnly;
  	}
  	
  	public void setUpdateOnly(boolean updateOnly) {
  		this.updateOnly = updateOnly;
  	}
  	
  	public List getDtoFieldList() {
  		return dtoFieldList;
  	}
  
  	public void putUpdateValue(String field, Object value) {
  		updateValueMap.put(field, value);
  	}
  
  	public void addDtoField(String field) {
  		dtoFieldList.add(field);
  	}
  
  	public Object evaluate(Object obj) {
  		if (dtoFieldList.size() == 0) {
  			executeUpdate((Long) obj);
  		} else {
  			executeCustomUpdate(EntityUtils.getId(obj), obj);
  		}
  		return obj;
  	}
  
  	private Object executeUpdate(Long id) { //update only
  		Object obj = getSession().load(getJavaClass(), id);
  		bindUpdateValue(obj);
  		getSession().update(obj);
  //		save(obj);
  		return id;
  	}
  
  	private Object executeCustomUpdate(Long id, Object dto) {
  		if (id == null || id.longValue() <= 0) { //save
  			if(updateOnly){
  				return null;
  			}
  			bindUpdateValue(dto);
  			EntityUtils.bindSelfObject(dto, getEntity());
  			EntityUtils.setUser(dto);
  			EntityUtils.setTimestamp(dto);
  			save(dto);
 		} else { //update
 			Object obj = getSession().load(getJavaClass(), id);
 			bindUpdateValue(obj);
 			bindDto(obj, dto);
 			EntityUtils.bindSelfObject(obj, getEntity());
 			EntityUtils.setUser(obj);
 			EntityUtils.setTimestamp(obj);
 			getSession().update(obj);
 		}
 		return null;
 	}
 
 	public void run(Object obj) {
 		evaluate(obj);
 	}
 
 	private void bindUpdateValue(Object obj) {
 		Iterator itr = updateValueMap.keySet().iterator();
 		while (itr.hasNext()) {
 			String field = (String) itr.next();
 			Object value = updateValueMap.get(field);			
 			Object convertedValue = convertValue(obj,value,field);
 			EntityUtils.setProperty(obj, field, convertedValue);				
 		}
 	}
 
 	private void bindDto(Object obj, Object dto) {
 		Iterator itr = dtoFieldList.iterator();
 		while (itr.hasNext()) {
 			String field = (String) itr.next();
 			EntityUtils.copyPropertyCollectionEscape(dto, obj, field);
 		}
 	}
 	
 	private Object convertValue(Object valueObject, Object updateValue, String field){
 		
 		TgwEntity entity = getEntity();
 		TgwAttribute attr = entity.getField(field);
 		
 		Object ret = updateValue;
 		
 		if(attr instanceof FkAttribute){
 			FkAttribute fkAttr = (FkAttribute)attr;
 			try{
 				Class refClass = fkAttr.getRefClass();
 				Long refId = new Long(Long.parseLong(updateValue.toString()));
 				ret = getSession().load(refClass,refId);
 			}catch(ClassNotFoundException cnfe){
 				
 				return null;
 			}catch(ObjectNotFoundException onfe){
 				logger.log("ETGW1101",new Object[]{getName()});
 				return null;
 			}
 		}else if(attr instanceof SetAttribute){
 			
 			SetAttribute setAttr = (SetAttribute) attr;
 			TgwEntity refEntity = setAttr.getRefEntity();				
 			
 			Object setObj = EntityUtils.getProperty(valueObject,setAttr);
 			Set refSet = null;
 			if(setObj != null && setObj instanceof Set){
 				refSet = (Set) setObj;
 			}else{
 				refSet = new HashSet();
 			}			
 			String strs = updateValue.toString();
 			if(strs != null){
 				String[] refValues = strs.split(",");				
 				for(int i=0;i<refValues.length;i++){
 					Object refObj = refEntity.newInstance();
 					EntityUtils.setProperty(refObj,setAttr.getName(),refValues[i]);
 					refSet.add(refObj);
 				}
 			}
 			ret = refSet;				
 		}else if(attr instanceof SelfAttribute){
 			Class refClass = entity.getJavaClass();
 			Long refId = new Long(Long.parseLong(updateValue.toString()));
 			try{
 				ret = getSession().load(refClass,refId);				
 			}catch(ObjectNotFoundException onfe){
 				logger.log("ETGW1101",new Object[]{getName()});
 				return null;
 			}			
 		}
 		return ret;
 	}
 	
 	
 }