SchemaExportAction.java

/*
 * 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.hibernate.eclipse.console.views;

import java.util.Iterator;

import org.eclipse.jface.dialogs.IDialogConstants;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.viewers.StructuredViewer;
import org.eclipse.osgi.util.NLS;
import org.eclipse.swt.widgets.Shell;
import org.hibernate.console.ConsoleConfiguration;
import org.hibernate.console.execution.ExecutionContext.Command;
import org.hibernate.eclipse.console.HibernateConsoleMessages;
import org.hibernate.eclipse.console.HibernateConsolePlugin;
import org.hibernate.eclipse.console.actions.ConsoleConfigReadyUseBaseAction;
import org.jboss.tools.hibernate.runtime.spi.IConfiguration;
import org.jboss.tools.hibernate.runtime.spi.ISchemaExport;
import org.jboss.tools.hibernate.runtime.spi.IService;

/**
 * @author max
 *
 */
public class SchemaExportAction extends ConsoleConfigReadyUseBaseAction {

	public static final String SCHEMAEXPORT_ACTIONID = "actionid.schemaexport"; //$NON-NLS-1$

	/**
	 * @param text
	 */
	protected SchemaExportAction(String text) {
		super(text);
		setId(SCHEMAEXPORT_ACTIONID);
		init(null);
	}

	/**
	 * @param selectionProvider
	 */
	public SchemaExportAction(StructuredViewer selectionProvider) {
		super(HibernateConsoleMessages.SchemaExportAction_run_schemaexport);
		setId(SCHEMAEXPORT_ACTIONID);
		init(selectionProvider);
	}

    /**
     * Convenience method to open a standard warning Yes/No dialog.
     * 
     * @param parent the parent shell of the dialog, or <code>null</code> if none
     * @param title the dialog's title, or <code>null</code> if none
     * @param msg the message
     * @return <code>true</code> if the user presses the Yes button,
     *         <code>false</code> otherwise
     */
	public static boolean openWarningYesNoDlg(Shell parent, String title, String msg) {
		String[] dialogButtonLabels = new String[] { IDialogConstants.YES_LABEL,
			IDialogConstants.NO_LABEL };
		MessageDialog dialog = new MessageDialog(parent, title, null, msg,
			MessageDialog.WARNING, dialogButtonLabels, 0);
		return dialog.open() == 0;
	}
	
	public void doRun() {
		for (Iterator<?> i = getSelectedNonResources().iterator(); i.hasNext();) {
			Object node = i.next();
			if (!(node instanceof ConsoleConfiguration)) {
				continue;
			}
			final ConsoleConfiguration config = (ConsoleConfiguration) node;
			try {
				config.execute( new Command() {
					public Object execute() {
						final IConfiguration cfg = config.getConfiguration();
						if (cfg == null) {
							return null;
						}
						String out = NLS.bind(HibernateConsoleMessages.SchemaExportAction_sure_run_schemaexport, config.getName());
						boolean res = openWarningYesNoDlg(viewer.getControl().getShell(),
							HibernateConsoleMessages.SchemaExportAction_run_schemaexport, out);
						if (!res) {
							return null;
						}
						IService service = config.getHibernateExtension().getHibernateService();
						ISchemaExport export = service.newSchemaExport(cfg);
						export.create(false, true);
						if (!export.getExceptions().isEmpty()) {
							Iterator<Throwable> iterator = export.getExceptions().iterator();
							int cnt = 1;
							while (iterator.hasNext()) {
								Throwable element = iterator.next();
								String outStr = NLS.bind(HibernateConsoleMessages.SchemaExportAction_errornum_while_performing_schemaexport, cnt++);
								HibernateConsolePlugin.getDefault().logErrorMessage(outStr, element);
							}
							HibernateConsolePlugin.getDefault().showError(viewer.getControl().getShell(),
								NLS.bind(HibernateConsoleMessages.SchemaExportAction_error_while_performing_schemaexport, cnt - 1),
								(Throwable)null );
						}
						return null;
					}
				} );
				viewer.refresh( node ); // todo: should we do it here or should
				// the view just react to config being
				// build ?
			}
			catch (Exception he) {
				HibernateConsolePlugin.getDefault().showError(
						viewer.getControl().getShell(),
						HibernateConsoleMessages.SchemaExportAction_exception_running_schemaexport, he );
			}
		}
	}
}