* Created on May 13, 2004
*
* To change the template for this generated file go to
* Window>Preferences>Java>Code Generation>Code and Comments
*/
package com.diatem.util.codegenerator;

import java.io.PrintWriter;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.hibernate.tool.hbm2java.AbstractRenderer;
import org.hibernate.tool.hbm2java.ClassMapping;
import org.hibernate.tool.hbm2java.FieldProperty;

/**
* @author luke
*
* To change the template for this generated type comment go to
* Window>Preferences>Java>Code Generation>Code and Comments
*/
public class BusinessObjectIncludesGenerator extends AbstractRenderer {
    protected String savedToPackage;
    protected String savedToClass;
    protected ClassMapping cmap;
    protected Map class2classmap;
    protected PrintWriter writer;

    /**
     *
     */
    public BusinessObjectIncludesGenerator() {
        super();
    }
   
    /* (non-Javadoc)
     * @see org.hibernate.tool.hbm2java.Renderer#render(java.lang.String, java.lang.String, org.hibernate.tool.hbm2java.ClassMapping, java.util.Map, java.io.PrintWriter)
     */
    public void render(
        String savedToPackage,
        String savedToClass,
        ClassMapping classMapping,
        Map class2classmap,
        PrintWriter writer)
        throws Exception {
       
        this.savedToPackage = savedToPackage;
        this.savedToClass = savedToClass;
        this.cmap = classMapping;
        this.class2classmap = class2classmap;
        this.writer = writer;
       
//        writeHelperMethods();
//        writer.println();
//        writer.println();
//        writer.println();
        writeBeanInterfaceDeclarations();
        writer.println();
        writer.println();
        writer.println();
        writeBeanImplementations();

    }
   
    protected void writeHelperMethods() {
        writer.println("/*###  START HELPER METHODS FOR " + cmap.getFullyQualifiedName() + " ###*/");
        List fields = cmap.getFields();
        if (fields != null) {
            for (Iterator it = fields.iterator(); it.hasNext(); ) {
                FieldProperty f = (FieldProperty) it.next();
                if (GeneratorUtil.shouldUsePersistenceHelper(f)) {
                    writeHelperMethod(f);
                    writer.println();
                }
            }
        }
        writer.println("/*###  END HELPER METHODS FOR " + cmap.getFullyQualifiedName() + " ###*/");
    }
   
    protected void writeHelperMethod(FieldProperty f) {
        writer.println("    public " + getGetterSignature(f) + " throws DAOException {");
        writer.println("        " + f.getClassType() + " ret;");
        writer.println();
        writer.println("        try {");
        writer.println("            ret = (" + f.getClassType() + ") utilities." + getGetterCall(f) + ";");
        writer.println("        } catch (RemoteException e) {");
        writer.println("            e.printStackTrace();");
        writer.println("            throw new DAOException(e);");
        writer.println("        }");
        writer.println("        return ret;");
        writer.println("    }");
    }
   
    protected void writeBeanInterfaceDeclarations() {
        writer.println("/*###  START BEAN INTERFACE FOR " + cmap.getFullyQualifiedName() + " ###*/");
        List fields = cmap.getFields();
        if (fields != null) {
            for (Iterator it = fields.iterator(); it.hasNext(); ) {
                FieldProperty f = (FieldProperty) it.next();
                if (GeneratorUtil.shouldUsePersistenceHelper(f)) {
                    writeBeanInterfaceDeclaration(f);
                    writer.println();
                }
            }
        }
        writer.println("/*###  END BEAN INTERFACE FOR " + cmap.getFullyQualifiedName() + " ###*/");
    }
   
    protected void writeBeanInterfaceDeclaration(FieldProperty f) {
        writer.println(getGetterSignature(f) + " throws DAOException, RemoteException;");
    }
   
    protected void writeBeanImplementations() {
        writer.println("/*###  START BEAN METHODS FOR " + cmap.getFullyQualifiedName() + " ###*/");
        List fields = cmap.getFields();
        if (fields != null) {
            for (Iterator it = fields.iterator(); it.hasNext(); ) {
                FieldProperty f = (FieldProperty) it.next();
                if (GeneratorUtil.shouldUsePersistenceHelper(f)) {
                    writeBeanImplementation(f);
                    writer.println();
                }
            }
        }
        writer.println("/*###  END BEAN METHODS FOR " + cmap.getFullyQualifiedName() + " ###*/");
    }

    protected void writeBeanImplementation(FieldProperty f) {
        writer.println("    public " + getGetterSignature(f) + " throws DAOException {");
        writer.println("        PersistentSession sess = null;");
        writer.println("        sess = new PersistentSession();");
        writer.println("        try {");
        writer.println("            if (!Hibernate.isInitialized(owner." + f.getGetterSignature() +")) {");
        writer.println("                sess.getPersistentSession().update(owner);");
        writer.println("                Hibernate.initialize(owner." + f.getGetterSignature() + ");");
        writer.println("            }");
        writer.println("        } catch (HibernateException e) {");
        writer.println("            e.printStackTrace();");
        writer.println("        } finally {");
        writer.println("            sess.close();");
        writer.println("        }");
        writer.println();
        writer.println("        return owner." + f.getGetterSignature() + ";");
        writer.println("    }");
    }

    protected String getGetterSignature(FieldProperty f) {
        return f.getClassType() + " " + f.getGetterSignature().substring(0, f.getGetterSignature().length()-2) + "(" + cmap.getClassName() + " owner)";
    }
   
    protected String getGetterCall(FieldProperty f) {
        return f.getGetterSignature().substring(0, f.getGetterSignature().length()-2) + "(owner)";
    }
   
}