/*
* JBoss, Home of Professional Open Source
* Copyright 2005, JBoss Inc., and individual contributors as indicated
* by the @authors tag. See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* This 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 software 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 software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jbpm.ant;
import java.io.IOException;
import java.io.InputStream;
import java.util.StringTokenizer;
import org.apache.tools.ant.BuildException;
import org.apache.tools.ant.Task;
import org.hibernate.cfg.Configuration;
import org.hibernate.tool.hbm2ddl.SchemaExport;
import org.jbpm.JbpmConfiguration;
import org.jbpm.JbpmContext;
import org.jbpm.db.JbpmSchema;
import org.jbpm.persistence.db.DbPersistenceServiceFactory;
import org.jbpm.svc.Services;
public class JbpmSchemaTask extends Task {
String jbpmCfg = null;
String hibernateCfg = null;
String hibernateProperties = null;
boolean quiet = false;
boolean text = false;
String output = null;
String delimiter = null;
String actions = null;
public void execute() throws BuildException {
if (actions==null) {
// default action is create
actions = "create";
}
// we need a configuration
Configuration configuration = null;
// if there is no jbpm nor hibernate configuration specified
if ( (jbpmCfg==null)
&& (hibernateCfg==null)
) {
// search for the default jbpm.cfg.xml
InputStream defaultJbpmCfgStream = getClass().getClassLoader().getResourceAsStream("jbpm.cfg.xml");
if (defaultJbpmCfgStream!=null) {
jbpmCfg = "jbpm.cfg.xml";
try {
defaultJbpmCfgStream.close();
} catch (IOException e) {
e.printStackTrace();
}
// if still not found, search for the default hibernate.cfg.xml
} else {
InputStream defaultHibernateCfgStream = getClass().getClassLoader().getResourceAsStream("hibernate.cfg.xml");
if (defaultHibernateCfgStream!=null) {
hibernateCfg = "hibernate.cfg.xml";
try {
defaultHibernateCfgStream.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
// first see if the jbpm cfg is specified cause that implies a hibernate configuration
if (jbpmCfg!=null) {
log("using jbpm configuration "+jbpmCfg);
JbpmConfiguration jbpmConfiguration = AntHelper.getJbpmConfiguration(jbpmCfg);
JbpmContext jbpmContext = jbpmConfiguration.createJbpmContext();
try {
DbPersistenceServiceFactory dbPersistenceServiceFactory = (DbPersistenceServiceFactory) jbpmConfiguration.getServiceFactory(Services.SERVICENAME_PERSISTENCE);
configuration = dbPersistenceServiceFactory.getConfiguration();
} finally {
jbpmContext.close();
}
// if there is no jbpm.cfg.xml specified, check if there is a hibernate.cfg.xml specified
} else if (hibernateCfg!=null) {
log("using hibernate configuration "+hibernateCfg);
configuration = AntHelper.getConfiguration(hibernateCfg, hibernateProperties);
// no hibernate configuration specified
} else {
throw new BuildException("couldn't create schema. no jbpm nor hibernate configuration specified.");
}
JbpmSchema jbpmSchema = new JbpmSchema(configuration);
SchemaExport schemaExport = new SchemaExport(configuration);
if (output!=null) schemaExport.setOutputFile(output);
if (delimiter!=null) schemaExport.setDelimiter(delimiter);
StringTokenizer tokenizer = new StringTokenizer(actions, ",");
while (tokenizer.hasMoreTokens()) {
String action = tokenizer.nextToken();
if ("drop".equalsIgnoreCase(action)) {
schemaExport.drop(!quiet, !text);
} else if ("create".equalsIgnoreCase(action)) {
schemaExport.create(!quiet, !text);
} else if ("clean".equalsIgnoreCase(action)) {
jbpmSchema.cleanSchema();
}
}
}
public void setActions(String actions) {
this.actions = actions;
}
public void setJbpmCfg(String jbpmCfg) {
this.jbpmCfg = jbpmCfg;
}
public void setHibernateCfg(String hibernateCfg) {
this.hibernateCfg = hibernateCfg;
}
public void setHibernateProperties(String hibernateProperties) {
this.hibernateProperties = hibernateProperties;
}
public void setDelimiter(String delimiter) {
this.delimiter = delimiter;
}
public void setOutput(String output) {
this.output = output;
}
public void setQuiet(boolean quiet) {
this.quiet = quiet;
}
public void setText(boolean text) {
this.text = text;
}
}