/*
    * 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.
   */
  /*
   * Created on 2005/08/10
   *
   * TODO To change the template for this generated file go to
   * Window - Preferences - Java - Code Style - Code Templates
   */
  package org.seasar.tuigwaa.model;
  
  import java.util.Collection;
  import java.util.Iterator;
  
  import javassist.CannotCompileException;
  import javassist.ClassPool;
  
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  import org.seasar.framework.util.ClassPoolUtil;
  import org.seasar.tuigwaa.model.common.DomainUtils;
  import org.seasar.tuigwaa.model.common.EntityUtils;
  import org.seasar.tuigwaa.model.common.PojoClassGenerator;
  import org.seasar.tuigwaa.model.common.TgwJavaClassUtils;
  import org.seasar.tuigwaa.model.core.TgwAttribute;
  import org.seasar.tuigwaa.model.core.TgwDomain;
  import org.seasar.tuigwaa.model.core.TgwEntity;
  import org.seasar.tuigwaa.model.core.impl.DateAttribute;
  import org.seasar.tuigwaa.system.Constants;
  import org.seasar.tuigwaa.util.TgwNameUtils;
  
  import com.isenshi.util.UniqueNameGenerator;
  
  /***
   * @author nishioka
   */
  public class JavaClassServiceImpl implements JavaClassService {
  
  	private UniqueNameGenerator nameGenerator = new UniqueNameGenerator();
  
  	private Log log = LogFactory.getLog(getClass());
  
  	public void createPackage(TgwDomain domain) {
  		alterClass(domain.getEntityList());
  	}
  
  	public void createClass(TgwEntity entity) {
  		String resolvedClassName = resolveJavaClassName(entity);
  		entity.setJavaClassName(resolvedClassName);
  		doCreateClass(entity);
  	}
  
  	public void alterClass(TgwEntity entity) {
  		Collection refEntitySet = DomainUtils.getReferencedEntities(entity);
  		refEntitySet.add(entity);
  		alterClass(refEntitySet);
  	}
  
  	public void alterClass(Collection entitySet) {
  		Iterator itr = entitySet.iterator();
  
  		// update class name
  		while (itr.hasNext()) {
  			TgwEntity anEntity = (TgwEntity) itr.next();
  			String resolvedClassName = resolveJavaClassName(anEntity);
  			anEntity.setJavaClassName(resolvedClassName);
  		}
  
  		// create each class
  		itr = entitySet.iterator();
  		while (itr.hasNext()) {
  			doCreateClass((TgwEntity) itr.next());
  		}
  	}
  
  	// [Start] ----- Private Method ----
  
  	private void doCreateClass(TgwEntity entity) {
  		String importedClassName = entity.getImportedClassName();
  		if (importedClassName != null) {
  			log.info("Import ..." + entity.getImportedClassName());
  
  			try {
  				Class checkedClass = TgwJavaClassUtils
  						.checkClass(importedClassName);
  				entity.setJavaClass(checkedClass);
  				entity.setJavaClassName(checkedClass.getName());
 
 				// entity.setJavaClassName(importedClassName);
 				// entity.setJavaClass(Class.forName(importedClassName));
 			} catch (ClassNotFoundException e) {
 				e.printStackTrace();
 			}
 			return;
 		}
 
 		log.info("Creating ..." + entity.getJavaClassName());
 		// ClassPool cp = ClassPoolUtil.getClassPool();
 		ClassPool cp = ClassPoolUtil.getClassPool(null);
 
 		PojoClassGenerator generator = new PojoClassGenerator(cp, entity
 				.getJavaClassName());
 
 		Iterator itr = entity.getFieldIterator();
 
 		try {
 			generator
 					.addProperty(Constants.ENTITY_BUILTIN_ID, "java.lang.Long");
 			while (itr.hasNext()) {
 				TgwAttribute attr = (TgwAttribute) itr.next();
 				String name = attr.getName();
 				String javaType = EntityUtils.toJavaType(attr);
 				generator.addProperty(name, javaType);
 
 				if (attr instanceof DateAttribute) {
 					String[] parts = TgwNameUtils.toDateParts(name);
 					for (int i = 0; i < parts.length; i++) {
 						generator.addProperty(parts[i], "java.lang.Integer");
 					}
 				}
 			}
 		} catch (CannotCompileException e) {
 			e.printStackTrace();
 			throw new RuntimeException(e);
 		} catch (ClassNotFoundException e) {
 			e.printStackTrace();
 			throw new RuntimeException(e);
 		}
 		Class generatedClass = generator.toClass();
 		entity.setJavaClass(generatedClass);
 	}
 
 	private String resolveJavaClassName(TgwEntity entity) {
 		String pureClassName = TgwNameUtils.toPureClassName(entity);
 		String versionedClassName = nameGenerator.nextUniqueName(pureClassName);
 		// entity.setJavaClassName(versionedClassName);
 		return versionedClassName;
 	}
 }