package generator;
import freemarker.template.Configuration;
import freemarker.template.DefaultObjectWrapper;
import freemarker.template.Template;
import freemarker.template.TemplateException;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.hibernate.cfg.JDBCMetaDataConfiguration;
import org.hibernate.mapping.Column;
import org.hibernate.mapping.Component;
import org.hibernate.mapping.MetaAttributable;
import org.hibernate.mapping.MetaAttribute;
import org.hibernate.mapping.PersistentClass;
import org.hibernate.mapping.Property;
import org.hibernate.tool.hbm2x.Cfg2HbmTool;
import org.hibernate.tool.hbm2x.ConfigurationNavigator;
import org.hibernate.tool.hbm2x.pojo.ComponentPOJOClass;
import org.hibernate.tool.hbm2x.pojo.POJOClass;
import org.hibernate.type.CharacterType;
import org.hibernate.type.StringType;
/**
* The local thread hold variable for the whole application scope
*
* @author kenvin
*
*/
public class AppGenerator {
private static Log log = LogFactory.getLog(AppGenerator.class);
GeneratorHelper c2j;
JDBCMetaDataConfiguration cfg;
Connection con;
Configuration freemarker;
String dest;
@SuppressWarnings("unchecked")
Map data = new HashMap();
@SuppressWarnings("unchecked")
public AppGenerator(String templateDir, String dest) {
freemarker = new Configuration();
try {
freemarker.setDirectoryForTemplateLoading(new File(templateDir));
} catch (IOException e) {
log.error("Can't find the template path", e);
}
freemarker.setObjectWrapper(new DefaultObjectWrapper());
this.dest = dest;
c2j = new GeneratorHelper();
cfg = new JDBCMetaDataConfiguration();
cfg.configure();
con = cfg.buildSessionFactory().openSession().connection();
cfg.readFromJDBC();
SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd");
data.put("date", format.format(new Date()));
data.put("version", "1.0");
data.put("jdk5", true);
data.put("ejb3", true);
data.put("generics", true);
data.put("c2j", c2j);
data.put("c2h", new Cfg2HbmTool());
}
public static void main(String[] args) {
AppGenerator generator = new AppGenerator("C:/Program Files/Java/workspace/adempiere/etc/templates",
"C:/Program Files/Java/workspace/adempiere/src");
try {
PreparedStatement clm = generator.con
.prepareStatement("SELECT c.ad_column_id, c.name, c.description, c.help, c.defaultvalue, c.AD_Reference_ID, c.AD_Reference_value_ID "
+ "FROM ad_column c, ad_table t where tablename=? and columnname=? and t.ad_table_id = c.ad_table_id ");
PreparedStatement fld = generator.con
.prepareStatement("SELECT f.seqno, f.isdisplayed, f.isreadonly,f.issameline, f.ad_fieldgroup_id FROM ad_field f where f.ad_column_id=? ");
Iterator<POJOClass> iterator = generator.c2j.getPOJOIterator(generator.cfg.getClassMappings());
ArrayList e = new ArrayList();
String packagename;
while (iterator.hasNext()) {
POJOClass element = (POJOClass) iterator.next();
if ("AdColumn".equals(element.getDeclarationName()) || "AdMenu".equals(element.getDeclarationName())) {
packagename = element.getDeclarationName();
packagename = "domain.ad." + packagename.substring(0, 1).toLowerCase() + packagename.substring(1);
log.info("start set attributes.....");
generator.setApplicationAttributes(element, packagename, clm, fld);
log.info("end set attributes!");
generator.exportPOJOClass(element);
generator.exportJavaFile(element, "struts/ebForm.ftl", "EbForm");
generator.exportJavaFile(element, "struts/Form.ftl", "Form");
generator.exportJavaFile(element, "struts/Action.ftl", "Action");
generator.exportJavaFile(element, "struts/Service.ftl", "Service");
generator.exportJavaFile(element, "struts/ServiceImpl.ftl", "JPAImpl");
generator.exportJSPFile(element, "jsp/Search.ftl", "Search");
generator.exportJSPFile(element, "jsp/ebSearch.ftl", "EbSearch");
generator.exportJSPFile(element, "jsp/Detail.ftl", "Detail");
generator.exportJSPFile(element, "jsp/ebDetail.ftl", "EbDetail");
e.add(element);
}
}
clm.close();
fld.close();
generator.con.close();
generator.exportConfigFile(e, "etc/StrutsConfig.ftl", "struts-config.xml");
generator.exportConfigFile(e, "etc/sun-jaxws.ftl", "sun-jaxws.xml");
} catch (Exception e) {
log.error("Failed to generate file", e);
} finally {
try {
generator.con.close();
generator.cfg = null;
} catch (Exception e1) {
}
}
}
@SuppressWarnings("unchecked")
private void exportConfigFile(ArrayList e, String t, String targetFile) throws TemplateException, IOException {
data.put("pojos", e);
File f = new File(dest);
f.mkdirs();
FileWriter out = new FileWriter(new File(f, targetFile));
log.info("Generating " + targetFile);
Template temp = freemarker.getTemplate(t);
temp.process(data, out);
out.close();
}
private void exportJavaFile(POJOClass element, String t, String suffix) throws TemplateException, IOException {
data.put("pojo", element);
data.put("clazz", element.getDecoratedObject());
File f = new File(dest + "/java/" + element.getPackageName().replace(".", "/"));
f.mkdirs();
FileWriter out = new FileWriter(new File(f, element.getDeclarationName() + suffix + ".java"));
Template temp = freemarker.getTemplate(t);
temp.process(data, out);
out.close();
}
private void exportJSPFile(POJOClass element, String t, String suffix) throws TemplateException, IOException {
data.put("pojo", element);
File f = new File(dest + "/jsp/" + element.getPackageName().replace(".", "/"));
f.mkdirs();
FileWriter out = new FileWriter(new File(f, element.getDeclarationName() + suffix + ".jsp"));
Template temp = freemarker.getTemplate(t);
temp.process(data, out);
out.close();
}
@SuppressWarnings( { "unchecked", "unchecked" })
public void exportPOJOClass(POJOClass element) throws Exception {
exportPOJOFile(element, data);
String packagename = element.getPackageName();
// Expoert the component class(Primary key) of this class
Map keys = new HashMap();
ConfigurationNavigator.collectComponents(keys, element);
Iterator ki = keys.values().iterator();
while (ki.hasNext()) {
Component component = (Component) ki.next();
// change the package name to the same one as the parent
component.setComponentClassName(packagename + "." + component.getComponentClassName());
element = new ComponentPOJOClass(component, c2j);
exportPOJOFile(element, data);
}
}
@SuppressWarnings("unchecked")
private void exportPOJOFile(POJOClass element, Map args) throws TemplateException, IOException {
args.put("pojo", element);
args.put("clazz", element.getDecoratedObject());
File f = new File(dest + "/java/" + element.getPackageName().replace(".", "/"));
f.mkdirs();
FileWriter out = new FileWriter(new File(f, element.getDeclarationName() + ".java"));
Template temp = freemarker.getTemplate("pojo/Pojo.ftl");
temp.process(args, out);
out.close();
}
@SuppressWarnings("unchecked")
public void setApplicationAttributes(POJOClass element, String packagename, PreparedStatement clm,
PreparedStatement fld) throws Exception {
PersistentClass persistent = (PersistentClass) element.getDecoratedObject();
// modify the default package name
setMetaAttribute(persistent, "generated-class", packagename + "." + persistent.getClassName());
clm.setString(1, persistent.getTable().getName());
Iterator iterator = persistent.getPropertyIterator();
Column column;
Property property;
ResultSet rs;
while (iterator.hasNext()) {
property = (Property) iterator.next();
if (property.getColumnSpan() == 1 && property.getColumnIterator().hasNext()) {
column = (Column) property.getColumnIterator().next();
clm.setString(2, column.getName());
rs = clm.executeQuery();
if (rs.next()) {
// Add field description
setMetaAttribute(property, "field-description", rs.getString("description") + ".\r\n"
+ rs.getString("help"));
// add default value
String defaultValue = rs.getString("defaultvalue");
if (column.getDefaultValue() == null && defaultValue != null && !defaultValue.startsWith("@")) {// default-value
if (property.getType() instanceof CharacterType) {
defaultValue = "'" + defaultValue + "'";
} else if (property.getType() instanceof StringType) {
defaultValue = "\"" + defaultValue + "\"";
}
setMetaAttribute(property, "default-value", defaultValue);
}
addMetaAttribute(property, "label", rs, "name");
addMetaAttribute(property, "reference", rs, "AD_Reference_ID");
addMetaAttribute(property, "reference_value", rs, "AD_Reference_value_ID");
addMetaAttribute(property, "column_id", rs, "ad_column_id");
}
rs.close();
}
}
// get other attributes, application specified
iterator = persistent.getPropertyIterator();
while (iterator.hasNext()) {
property = (Property) iterator.next();
fld.setInt(1, c2j.getMetaAsInt(property, "column_id"));
rs = fld.executeQuery();
if (rs.next()) {
addMetaAttribute(property, "isdisplayed", rs, "isdisplayed");
addMetaAttribute(property, "seqno", rs, "seqno");
addMetaAttribute(property, "issameline", rs, "issameline");
addMetaAttribute(property, "isreadonly", rs, "isreadonly");
addMetaAttribute(property, "fieldgroup", rs, "ad_fieldgroup_id");
}
rs.close();
}
}
public void addMetaAttribute(MetaAttributable obj, String name, ResultSet rs, String column) throws SQLException {
setMetaAttribute(obj, name, rs.getString(column));
}
public void setMetaAttribute(MetaAttributable obj, String name, String value) {
if (value == null || "".equals(value))
return;
Map<String, MetaAttribute> metas = obj.getMetaAttributes() == null ? new HashMap() : obj.getMetaAttributes();
// Add a attribute meta
MetaAttribute meta = new MetaAttribute(name);
meta.addValue(value);
metas.put(name, meta);
obj.setMetaAttributes(metas);
}
}
Login
FAQ
Search
