[jbosstools-commits] JBoss Tools SVN: r7180 - in trunk/hibernatetools/plugins/org.hibernate.eclipse.console: src/org/hibernate/eclipse/console and 1 other directory.
jbosstools-commits at lists.jboss.org
jbosstools-commits at lists.jboss.org
Fri Mar 28 09:03:00 EDT 2008
Author: dgeraskov
Date: 2008-03-28 09:03:00 -0400 (Fri, 28 Mar 2008)
New Revision: 7180
Added:
trunk/hibernatetools/plugins/org.hibernate.eclipse.console/src/org/hibernate/eclipse/console/GenerateSeamEntities.java
Modified:
trunk/hibernatetools/plugins/org.hibernate.eclipse.console/plugin.xml
trunk/hibernatetools/plugins/org.hibernate.eclipse.console/src/org/hibernate/eclipse/console/Messages.java
trunk/hibernatetools/plugins/org.hibernate.eclipse.console/src/org/hibernate/eclipse/console/messages.properties
Log:
http://jira.jboss.com/jira/browse/JBIDE-1178
Modified: trunk/hibernatetools/plugins/org.hibernate.eclipse.console/plugin.xml
===================================================================
--- trunk/hibernatetools/plugins/org.hibernate.eclipse.console/plugin.xml 2008-03-28 12:18:57 UTC (rev 7179)
+++ trunk/hibernatetools/plugins/org.hibernate.eclipse.console/plugin.xml 2008-03-28 13:03:00 UTC (rev 7180)
@@ -654,6 +654,36 @@
</moveParticipant-->
</extension>
+ <extension
+ point="org.eclipse.ui.popupMenus">
+ <!--objectContribution
+ id="org.hibernate.eclipse.console.objectContribution1"
+ objectClass="org.eclipse.datatools.connectivity.sqm.core.rte.jdbc.JDBCTable">
+ <action
+ class="org.hibernate.eclipse.console.GenerateSeamEntities3"
+ id="org.hibernate.eclipse.console.action2"
+ label="Generate Seam Entities">
+ </action>
+ </objectContribution-->
+ <objectContribution
+ id="org.hibernate.eclipse.console.objectContribution2"
+ objectClass="org.hibernate.mapping.Table">
+ <action
+ class="org.hibernate.eclipse.console.GenerateSeamEntities"
+ id="org.hibernate.eclipse.console.action1"
+ label="Generate Seam Entities">
+ </action>
+ </objectContribution>
+ <!--objectContribution
+ id="org.hibernate.eclipse.console.objectContribution2"
+ objectClass="org.hibernate.eclipse.console.workbench.TableContainer">
+ <action
+ class="org.hibernate.eclipse.console.GenerateSeamEntities"
+ id="org.hibernate.eclipse.console.action1"
+ label="Generate Seam Entities">
+ </action>
+ </objectContribution-->
+ </extension>
<!-- <extension
point="org.eclipse.ui.themes">
<themeElementCategory
Added: trunk/hibernatetools/plugins/org.hibernate.eclipse.console/src/org/hibernate/eclipse/console/GenerateSeamEntities.java
===================================================================
--- trunk/hibernatetools/plugins/org.hibernate.eclipse.console/src/org/hibernate/eclipse/console/GenerateSeamEntities.java (rev 0)
+++ trunk/hibernatetools/plugins/org.hibernate.eclipse.console/src/org/hibernate/eclipse/console/GenerateSeamEntities.java 2008-03-28 13:03:00 UTC (rev 7180)
@@ -0,0 +1,271 @@
+package org.hibernate.eclipse.console;
+
+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.Properties;
+import java.util.Set;
+
+import org.eclipse.core.resources.IResource;
+import org.eclipse.core.resources.ResourcesPlugin;
+import org.eclipse.core.runtime.CoreException;
+import org.eclipse.core.runtime.IPath;
+import org.eclipse.core.runtime.IProgressMonitor;
+import org.eclipse.core.runtime.NullProgressMonitor;
+import org.eclipse.jface.action.IAction;
+import org.eclipse.jface.viewers.ISelection;
+import org.eclipse.jface.viewers.TreePath;
+import org.eclipse.jface.viewers.TreeSelection;
+import org.eclipse.ui.IObjectActionDelegate;
+import org.eclipse.ui.IWorkbenchPart;
+import org.hibernate.cfg.Configuration;
+import org.hibernate.cfg.JDBCMetaDataConfiguration;
+import org.hibernate.cfg.reveng.DefaultReverseEngineeringStrategy;
+import org.hibernate.cfg.reveng.OverrideRepository;
+import org.hibernate.cfg.reveng.ReverseEngineeringSettings;
+import org.hibernate.cfg.reveng.ReverseEngineeringStrategy;
+import org.hibernate.cfg.reveng.TableFilter;
+import org.hibernate.console.ConsoleConfiguration;
+import org.hibernate.console.HibernateConsoleRuntimeException;
+import org.hibernate.console.execution.ExecutionContext.Command;
+import org.hibernate.eclipse.console.model.impl.ExporterDefinition;
+import org.hibernate.eclipse.console.model.impl.ExporterFactory;
+import org.hibernate.eclipse.console.utils.DialogSelectionHelper;
+import org.hibernate.eclipse.launch.PathHelper;
+import org.hibernate.mapping.Table;
+import org.hibernate.tool.hbm2x.ArtifactCollector;
+import org.hibernate.tool.hbm2x.Exporter;
+
+public class GenerateSeamEntities implements IObjectActionDelegate {
+
+ private TreeSelection selection;
+
+ private String outputPath;
+
+ private Map<ConsoleConfiguration, List<Table>> tableGroups;
+
+ public void run(IAction action) {
+
+ IPath[] paths = DialogSelectionHelper.chooseFolderEntries(null, null,
+ Messages.GenerateSeamEntities_dialog_title,
+ Messages.GenerateSeamEntities_dialog_message,
+ false);
+
+ if (paths.length == 0) return;
+
+ outputPath = paths[0].toOSString();
+
+ splitOnGroupsByConsoleCfg();
+
+ ExporterFactory[] exporters = getExporterFactories();
+
+ Set<ConsoleConfiguration> ccs = tableGroups.keySet();
+
+ for (Iterator<ConsoleConfiguration> iterator = ccs.iterator(); iterator
+ .hasNext();) {
+ ConsoleConfiguration consoleConfiguration = iterator.next();
+ runExporters(exporters, consoleConfiguration);
+ }
+
+ refreshOutputDir(getOutputPath());
+ }
+
+ private ArtifactCollector runExporters(
+ final ExporterFactory[] exporterFactories, ConsoleConfiguration cc) {
+ final IProgressMonitor monitor = new NullProgressMonitor();
+
+ monitor.beginTask("Generating code for " + cc.getName(), exporterFactories.length + 1); //$NON-NLS-1$
+
+ final Configuration cfg = buildConfiguration(cc, tableGroups.get(cc));
+
+ monitor.worked(1);
+
+ if (monitor.isCanceled())
+ return null;
+
+ return (ArtifactCollector) cc.execute(new Command() {
+
+ public Object execute() {
+ ArtifactCollector artifactCollector = new ArtifactCollector();
+
+ // Global properties
+ Properties props = new Properties();
+ props.put("ejb3", "true"); //$NON-NLS-1$ //$NON-NLS-2$
+ props.put("jdk5", "true"); //$NON-NLS-1$ //$NON-NLS-2$
+
+ for (int i = 0; i < exporterFactories.length; i++) {
+ monitor.subTask(exporterFactories[i]
+ .getExporterDefinition().getDescription());
+
+ Properties globalProperties = new Properties();
+ globalProperties.putAll(props);
+
+ Set outputDirectories = new HashSet();
+ Exporter exporter;
+ try {
+ exporter = exporterFactories[i]
+ .createConfiguredExporter(cfg, getOutputPath(),
+ null, globalProperties,
+ outputDirectories, artifactCollector);
+ } catch (CoreException e) {
+ throw new HibernateConsoleRuntimeException(
+ "Error while setting up " //$NON-NLS-1$
+ + exporterFactories[i]
+ .getExporterDefinition(), e);
+ }
+
+ exporter.start();
+ monitor.worked(1);
+ }
+ return artifactCollector;
+ }
+
+ });
+
+ }
+
+ private ExporterFactory[] getExporterFactories() {
+ List<String> exporterNames = new ArrayList<String>();
+ exporterNames.add("org.hibernate.tools.hbm2java"); //$NON-NLS-1$
+ //exporterNames.add("org.hibernate.tools.hbm2dao"); //$NON-NLS-1$
+
+ Map exDefinitions = ExtensionManager.findExporterDefinitionsAsMap();
+ List<ExporterFactory> factories = new ArrayList<ExporterFactory>();
+
+ for (Iterator<String> iterator = exporterNames.iterator(); iterator
+ .hasNext();) {
+ String exporterId = iterator.next();
+
+ ExporterDefinition expDef = (ExporterDefinition) exDefinitions
+ .get(exporterId);
+ if (expDef == null) {
+ throw new HibernateConsoleRuntimeException(
+ "Could not locate exporter for '" + exporterId + "'"); //$NON-NLS-1$ //$NON-NLS-2$
+ } else {
+ ExporterFactory exporterFactory = new ExporterFactory(expDef,
+ exporterId);
+ factories.add(exporterFactory);
+ }
+ }
+ return factories.toArray(new ExporterFactory[factories.size()]);
+ }
+
+ private Configuration buildConfiguration(ConsoleConfiguration cc,
+ final List<Table> tables) {
+ Configuration configuration = null;
+ if (cc.hasConfiguration()) {
+ configuration = cc.getConfiguration();
+ } else {
+ configuration = cc.buildWith(null, false);
+ }
+
+ final JDBCMetaDataConfiguration cfg = new JDBCMetaDataConfiguration();
+ Properties properties = configuration.getProperties();
+ cfg.setProperties(properties);
+ cc.buildWith(cfg, false);
+
+ cfg.setPreferBasicCompositeIds(true);
+
+ cc.execute(new Command() {
+ /* need to execute in the console configuration to let it handle classpath stuff!*/
+ public Object execute() {
+
+ OverrideRepository repository = new OverrideRepository();
+
+ TableFilter filter = null;
+ for (int i = 0; i < tables.size(); i++) {
+ Table table = tables.get(i);
+
+ filter = new TableFilter();
+ filter.setExclude(false);
+ if (table.getCatalog() != null) {
+ filter.setMatchCatalog(table.getCatalog());
+ }
+ if (table.getSchema() != null) {
+ filter.setMatchSchema(table.getSchema());
+ }
+ filter.setMatchName(tables.get(i).getName());
+
+ repository.addTableFilter(filter);
+ }
+
+ ReverseEngineeringStrategy res = repository
+ .getReverseEngineeringStrategy(new DefaultReverseEngineeringStrategy());
+
+ ReverseEngineeringSettings qqsettings = new ReverseEngineeringSettings(res)
+ .setDetectManyToMany(true)
+ .setDetectOptimisticLock(true);
+
+ res.setSettings(qqsettings);
+
+ cfg.setReverseEngineeringStrategy(res);
+
+ cfg.readFromJDBC();
+ cfg.buildMappings();
+ return null;
+ }
+ });
+
+ return cfg;
+ }
+
+ private String getOutputPath() {
+ return outputPath;
+ }
+
+ private void splitOnGroupsByConsoleCfg() {
+ TreePath[] paths = selection.getPaths();
+
+ tableGroups = new HashMap<ConsoleConfiguration, List<Table>>();
+
+ for (int i = 0; i < paths.length; i++) {
+ if (paths[i].getLastSegment() instanceof Table) {
+ ConsoleConfiguration cc = getConsoleConfiguration(paths[i]);
+ if (cc == null) continue;
+
+ if (tableGroups.containsKey(cc)) {
+ List<Table> tables = tableGroups.get(cc);
+ tables.add((Table) paths[i].getLastSegment());
+ } else {
+ List<Table> tables = new ArrayList<Table>();
+ tables.add((Table) paths[i].getLastSegment());
+ tableGroups.put(cc, tables);
+ }
+ }
+ }
+ }
+
+ private ConsoleConfiguration getConsoleConfiguration(TreePath path) {
+ for (int i = 0; i < path.getSegmentCount(); i++) {
+ if (path.getSegment(i) instanceof ConsoleConfiguration)
+ return (ConsoleConfiguration) path.getSegment(i);
+ }
+ return null;
+ }
+
+ public void setActivePart(IAction action, IWorkbenchPart targetPart) {
+ }
+
+ public void selectionChanged(IAction action, ISelection selection) {
+ if (selection instanceof TreeSelection) {
+ this.selection = (TreeSelection) selection;
+ }
+ }
+
+ private void refreshOutputDir(String outputdir) {
+ IResource bufferRes = PathHelper.findMember(ResourcesPlugin
+ .getWorkspace().getRoot(), outputdir);
+
+ if (bufferRes != null && bufferRes.isAccessible()) {
+ try {
+ bufferRes.refreshLocal(IResource.DEPTH_INFINITE, null);
+ } catch (CoreException e) {
+ // ignore, maybe merge into possible existing status.
+ }
+ }
+ }
+
+}
Property changes on: trunk/hibernatetools/plugins/org.hibernate.eclipse.console/src/org/hibernate/eclipse/console/GenerateSeamEntities.java
___________________________________________________________________
Name: svn:mime-type
+ text/plain
Name: svn:keywords
+ Author Id Revision Date
Name: svn:eol-style
+ native
Modified: trunk/hibernatetools/plugins/org.hibernate.eclipse.console/src/org/hibernate/eclipse/console/Messages.java
===================================================================
--- trunk/hibernatetools/plugins/org.hibernate.eclipse.console/src/org/hibernate/eclipse/console/Messages.java 2008-03-28 12:18:57 UTC (rev 7179)
+++ trunk/hibernatetools/plugins/org.hibernate.eclipse.console/src/org/hibernate/eclipse/console/Messages.java 2008-03-28 13:03:00 UTC (rev 7180)
@@ -34,6 +34,8 @@
public static String find_replace_action_tooltip;
public static String find_replace_action_image;
public static String find_replace_action_description;
+ public static String GenerateSeamEntities_dialog_title;
+ public static String GenerateSeamEntities_dialog_message;
private Messages() {
// noop
Modified: trunk/hibernatetools/plugins/org.hibernate.eclipse.console/src/org/hibernate/eclipse/console/messages.properties
===================================================================
--- trunk/hibernatetools/plugins/org.hibernate.eclipse.console/src/org/hibernate/eclipse/console/messages.properties 2008-03-28 12:18:57 UTC (rev 7179)
+++ trunk/hibernatetools/plugins/org.hibernate.eclipse.console/src/org/hibernate/eclipse/console/messages.properties 2008-03-28 13:03:00 UTC (rev 7180)
@@ -20,3 +20,6 @@
ContentFormat.tooltip =Format HQL
ContentFormat.image =
ContentFormat.description =Format HQL source
+
+GenerateSeamEntities_dialog_title = Select output directory
+GenerateSeamEntities_dialog_message = Choose directory in which the generated files will be stored
More information about the jbosstools-commits
mailing list